Added export of augmentation schemas to Binding Context

Change-Id: I881fcf83ec76639fa0c37c272e84a9866fe03ce9
Signed-off-by: Tony Tkacik <ttkacik@cisco.com>

diff --git a/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/BindingGeneratorImpl.xtend b/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/BindingGeneratorImpl.xtend
index a2836e2041c860c338d3e78b939fb996504842b2..71f360726b840381307b1e2b6c3d4bd400e34990 100644 (file)
--- a/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/BindingGeneratorImpl.xtend
+++ b/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/BindingGeneratorImpl.xtend
@@ -750,7 +750,8 @@ public class BindingGeneratorImpl implements BindingGenerator {
                 }
                 val augTypeBuilder = addRawAugmentGenTypeDefinition(module, packageName, augmentPackageName,
                     targetTypeBuilder.toInstance, augSchema);
-                genCtx.get(module).addAugmentType(augTypeBuilder)
+                genCtx.get(module).addAugmentType(augTypeBuilder)\r
+                genCtx.get(module).addTypeToAugmentation(augTypeBuilder,augSchema);
             } else {
                 generateTypesFromAugmentedChoiceCases(module, augmentPackageName, targetTypeBuilder.toInstance,
                     targetSchemaNode as ChoiceNode, augSchema.childNodes);

diff --git a/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/ModuleContext.java b/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/ModuleContext.java
index beb310511827c81e43c6c44a5ecd9e560edafa2b..cf4885faa5b1fa3718182f2e5f9f5c42542056e1 100644 (file)
--- a/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/ModuleContext.java
+++ b/code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/ModuleContext.java
@@ -17,6 +17,7 @@ import java.util.Set;
 import org.opendaylight.yangtools.sal.binding.model.api.Type;
 import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTOBuilder;
 import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTypeBuilder;
+import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
 
 public final class ModuleContext {
@@ -29,6 +30,8 @@ public final class ModuleContext {
     private final Set<GeneratedTOBuilder> identities = new HashSet<GeneratedTOBuilder>();
     private final Set<GeneratedTypeBuilder> topLevelNodes = new HashSet<GeneratedTypeBuilder>();
     private final List<GeneratedTypeBuilder> augmentations = new ArrayList<GeneratedTypeBuilder>();
+    private final Map<Type,AugmentationSchema> typeToAugmentation = new HashMap<>();
+
 
 
     List<Type> getGeneratedTypes() {
@@ -147,4 +150,12 @@ public final class ModuleContext {
         return augmentations;
     }
 
+    public Map<Type, AugmentationSchema> getTypeToAugmentation() {
+        return typeToAugmentation;
+    }
+
+    public void addTypeToAugmentation(GeneratedTypeBuilder builder, AugmentationSchema schema) {
+        typeToAugmentation.put(builder, schema);
+    }
+
 }