Adjust to yangtools-2.0.0 changes

[mdsal.git] / binding / mdsal-binding-dom-codec / src / main / java / org / opendaylight / mdsal / binding / dom / codec / gen / impl / AbstractStreamWriterGenerator.java

diff --git a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/gen/impl/AbstractStreamWriterGenerator.java b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/gen/impl/AbstractStreamWriterGenerator.java
index b54256c1a50059907df43b70548fa729c22b273b..a54a83513b3dc1ede2c45edf97ec6cce02894430 100644 (file)
--- a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/gen/impl/AbstractStreamWriterGenerator.java
+++ b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/gen/impl/AbstractStreamWriterGenerator.java
@@ -34,8 +34,8 @@ import org.opendaylight.yangtools.yang.binding.DataObject;
 import org.opendaylight.yangtools.yang.binding.DataObjectSerializerImplementation;
 import org.opendaylight.yangtools.yang.binding.DataObjectSerializerRegistry;
 import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
-import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
-import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
+import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
@@ -177,10 +177,10 @@ abstract class AbstractStreamWriterGenerator extends AbstractGenerator implement
             } else {
                 source = generateMapEntrySerializer(generatedType, casted);
             }
-        } else if (schema instanceof AugmentationSchema) {
-            source = generateSerializer(generatedType,(AugmentationSchema) schema);
-        } else if (schema instanceof ChoiceCaseNode) {
-            source = generateCaseSerializer(generatedType,(ChoiceCaseNode) schema);
+        } else if (schema instanceof AugmentationSchemaNode) {
+            source = generateSerializer(generatedType,(AugmentationSchemaNode) schema);
+        } else if (schema instanceof CaseSchemaNode) {
+            source = generateCaseSerializer(generatedType,(CaseSchemaNode) schema);
         } else if (schema instanceof NotificationDefinition) {
             source = generateNotificationSerializer(generatedType,(NotificationDefinition) schema);
         } else {
@@ -253,7 +253,7 @@ abstract class AbstractStreamWriterGenerator extends AbstractGenerator implement
      * @param node Schema of case
      * @return Source for case node writer
      */
-    protected abstract DataObjectSerializerSource generateCaseSerializer(GeneratedType type, ChoiceCaseNode node);
+    protected abstract DataObjectSerializerSource generateCaseSerializer(GeneratedType type, CaseSchemaNode node);
 
     /**
      * Generates serializer source for supplied list node, which will read supplied binding type and invoke proper
@@ -293,7 +293,7 @@ abstract class AbstractStreamWriterGenerator extends AbstractGenerator implement
      * @param schema Schema of augmentation
      * @return Source for augmentation node writer
      */
-    protected abstract DataObjectSerializerSource generateSerializer(GeneratedType type, AugmentationSchema schema);
+    protected abstract DataObjectSerializerSource generateSerializer(GeneratedType type, AugmentationSchemaNode schema);
 
     /**
      * Generates serializer source for notification node, which will read supplied binding type and invoke proper