From: Tony Tkacik <ttkacik@cisco.com>
Date: Tue, 19 Nov 2013 13:47:05 +0000 (+0100)
Subject: Added export of augmentation schemas to Binding Context
X-Git-Tag: yangtools-0.6.0~177^2
X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=commitdiff_plain;h=bd6fee58322b627f82f3f8ad3913774f23b45c94;p=yangtools.git

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 a2836e2041..71f360726b 100644
--- 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)
+                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 beb3105118..cf4885faa5 100644
--- 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);
+    }
+
 }