Added getParent() method to DataSchemaNode and DataNodeContainer. Fixed Bugs.
[yangtools.git] / code-generator / binding-generator-impl / src / main / java / org / opendaylight / yangtools / sal / binding / generator / impl / BindingGeneratorImpl.xtend
index 5b3bbb28e25fb971575aeddee9e52f593a291d20..c0802aecc5be43e0d8d794bc13be66a219966724 100644 (file)
@@ -1,4 +1,3 @@
-
 /*\r
  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.\r
  *\r
@@ -54,18 +53,25 @@ import org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition;
 import org.opendaylight.yangtools.yang.model.api.type.EnumTypeDefinition;\r
 import org.opendaylight.yangtools.yang.model.api.type.UnionTypeDefinition;\r
 import org.opendaylight.yangtools.yang.model.util.DataNodeIterator;\r
-import org.opendaylight.yangtools.yang.model.util.ExtendedType;\r
 import org.opendaylight.yangtools.yang.model.util.SchemaContextUtil;\r
 import org.opendaylight.yangtools.yang.model.util.UnionType;\r
 import static com.google.common.base.Preconditions.*;\r
 import static extension org.opendaylight.yangtools.binding.generator.util.Types.*;\r
-import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.*;
-import static org.opendaylight.yangtools.binding.generator.util.BindingTypes.*;
-import static org.opendaylight.yangtools.yang.model.util.SchemaContextUtil.*;
+import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.*;\r
+import static org.opendaylight.yangtools.binding.generator.util.BindingTypes.*;\r
+import static org.opendaylight.yangtools.yang.model.util.SchemaContextUtil.*;\r
+import org.opendaylight.yangtools.yang.parser.util.ModuleDependencySort\r
+import org.opendaylight.yangtools.yang.model.util.ExtendedType;\r
+import org.opendaylight.yangtools.yang.model.api.UsesNode\r
+import org.opendaylight.yangtools.yang.binding.annotations.RoutingContext\r
+import org.opendaylight.yangtools.sal.binding.model.api.type.builder.AnnotationTypeBuilder\r
+import org.opendaylight.yangtools.yang.model.api.ModuleImport\r
+import org.opendaylight.yangtools.yang.binding.DataContainer\rimport java.util.Iterator
+import org.opendaylight.yangtools.yang.model.api.AugmentationTarget
+import java.util.Collection
+import org.opendaylight.yangtools.yang.model.api.YangNode
 
-\r
 public class BindingGeneratorImpl implements BindingGenerator {\r
-\r
     /**\r
      * Outter key represents the package name. Outter value represents map of\r
      * all builders in the same package. Inner key represents the schema node\r
@@ -94,8 +100,7 @@ public class BindingGeneratorImpl implements BindingGenerator {
      * case is used this map.\r
      */\r
     private val allGroupings = new HashMap<SchemaPath, GeneratedType>();\r
-    \r
-    \r
+\r
     private val yangToJavaMapping = new HashMap<SchemaPath, Type>();\r
 \r
     /**\r
@@ -108,7 +113,6 @@ public class BindingGeneratorImpl implements BindingGenerator {
      */\r
     private val static String AUGMENT_IDENTIFIER_NAME = "augment-identifier";\r
 \r
-    \r
     /**\r
      * Resolves generated types from <code>context</code> schema nodes of all\r
      * modules.\r
@@ -128,33 +132,12 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             if <code>context</code> contain no modules\r
      */\r
     override generateTypes(SchemaContext context) {\r
-        checkArgument(context !== null,"Schema Context reference cannot be NULL.");\r
-        checkState(context.modules !== null,"Schema Context does not contain defined modules.");\r
-        val List<Type> generatedTypes = new ArrayList();\r
+        checkArgument(context !== null, "Schema Context reference cannot be NULL.");\r
+        checkState(context.modules !== null, "Schema Context does not contain defined modules.");\r
         schemaContext = context;\r
         typeProvider = new TypeProviderImpl(context);\r
         val Set<Module> modules = context.modules;\r
-        genTypeBuilders = new HashMap();\r
-        for (module : modules) {\r
-\r
-            generatedTypes.addAll(allGroupingsToGenTypes(module));\r
-\r
-            if (false == module.childNodes.isEmpty()) {\r
-                generatedTypes.add(moduleToDataType(module));\r
-            }\r
-            generatedTypes.addAll(allTypeDefinitionsToGenTypes(module));\r
-            generatedTypes.addAll(allContainersToGenTypes(module));\r
-            generatedTypes.addAll(allListsToGenTypes(module));\r
-            generatedTypes.addAll(allChoicesToGenTypes(module));\r
-            generatedTypes.addAll(allRPCMethodsToGenType(module));\r
-            generatedTypes.addAll(allNotificationsToGenType(module));\r
-            generatedTypes.addAll(allIdentitiesToGenTypes(module, context));\r
-        }\r
-        for (module : modules) {\r
-            generatedTypes.addAll(allAugmentsToGenTypes(module));\r
-            \r
-        }\r
-        return generatedTypes;\r
+        return generateTypes(context, modules);\r
     }\r
 \r
     /**\r
@@ -186,41 +169,41 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             if <code>context</code> contain no modules\r
      */\r
     override generateTypes(SchemaContext context, Set<Module> modules) {\r
-        checkArgument(context !== null,"Schema Context reference cannot be NULL.");\r
-        checkState(context.modules !== null,"Schema Context does not contain defined modules.");\r
-        checkArgument(modules !== null,"Set of Modules cannot be NULL.");\r
+        checkArgument(context !== null, "Schema Context reference cannot be NULL.");\r
+        checkState(context.modules !== null, "Schema Context does not contain defined modules.");\r
+        checkArgument(modules !== null, "Set of Modules cannot be NULL.");\r
 \r
         val List<Type> filteredGenTypes = new ArrayList();\r
+\r
         schemaContext = context;\r
         typeProvider = new TypeProviderImpl(context);\r
-        val Set<Module> contextModules = context.modules;\r
+        val contextModules = ModuleDependencySort.sort(context.modules);\r
         genTypeBuilders = new HashMap();\r
-        \r
+\r
         for (contextModule : contextModules) {\r
             val List<Type> generatedTypes = new ArrayList();\r
-\r
+            generatedTypes.addAll(allTypeDefinitionsToGenTypes(contextModule));\r
             generatedTypes.addAll(allGroupingsToGenTypes(contextModule));\r
-            if (false == contextModule.childNodes.isEmpty()) {\r
+            if(false == contextModule.childNodes.isEmpty()) {\r
                 generatedTypes.add(moduleToDataType(contextModule));\r
             }\r
-            generatedTypes.addAll(allTypeDefinitionsToGenTypes(contextModule));\r
             generatedTypes.addAll(allContainersToGenTypes(contextModule));\r
             generatedTypes.addAll(allListsToGenTypes(contextModule));\r
             generatedTypes.addAll(allChoicesToGenTypes(contextModule));\r
             generatedTypes.addAll(allRPCMethodsToGenType(contextModule));\r
             generatedTypes.addAll(allNotificationsToGenType(contextModule));\r
             generatedTypes.addAll(allIdentitiesToGenTypes(contextModule, context));\r
-            \r
-            if (modules.contains(contextModule)) {\r
+\r
+            if(modules.contains(contextModule)) {\r
                 filteredGenTypes.addAll(generatedTypes);\r
             }\r
         }\r
         for (contextModule : contextModules) {\r
             val generatedTypes = (allAugmentsToGenTypes(contextModule));\r
-            if (modules.contains(contextModule)) {\r
+            if(modules.contains(contextModule)) {\r
                 filteredGenTypes.addAll(generatedTypes);\r
             }\r
-            \r
+\r
         }\r
         return filteredGenTypes;\r
     }\r
@@ -241,24 +224,76 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def List<Type> allTypeDefinitionsToGenTypes( Module module) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
-        checkArgument(module.name !== null,"Module name cannot be NULL.");\r
-        val Set<TypeDefinition<?>> typeDefinitions = module.typeDefinitions;\r
-        checkState(typeDefinitions !== null,'''Type Definitions for module «module.name» cannot be NULL.''');\r
+    private def List<Type> allTypeDefinitionsToGenTypes(Module module) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
+        checkArgument(module.name !== null, "Module name cannot be NULL.");\r
+        val it = new DataNodeIterator(module);\r
+        val List<TypeDefinition<?>> typeDefinitions = it.allTypedefs;\r
+        checkState(typeDefinitions !== null, '''Type Definitions for module «module.name» cannot be NULL.''');\r
 \r
-        \r
         val List<Type> generatedTypes = new ArrayList();\r
-        for ( TypeDefinition<?> typedef : typeDefinitions) {\r
-            if (typedef !== null) {\r
+        for (TypeDefinition<?> typedef : typeDefinitions) {\r
+            if(typedef !== null) {\r
                 val type = (typeProvider as TypeProviderImpl).generatedTypeForExtendedDefinitionType(typedef, typedef);\r
-                if ((type !== null) && !generatedTypes.contains(type)) {\r
+                if((type !== null) && !generatedTypes.contains(type)) {\r
                     generatedTypes.add(type);\r
                 }\r
             }\r
         }\r
         return generatedTypes;\r
     }\r
+    \r
+    private def List<Type> dataNodeContainerToGenType(String basePackageName, DataNodeContainer node, Module module) {\r
+        if (node === null) {\r
+            return null;\r
+        }\r
+        if (!(node instanceof SchemaNode)) {\r
+            throw new IllegalArgumentException("node to generate must be instance of SchemaNode");\r
+        }\r
+        val List<Type> result = new ArrayList();\r
+\r
+        result.addAll(processUsesAugments(node, module));\r
+\r
+        val packageName = packageNameForGeneratedType(basePackageName, (node as SchemaNode).path);\r
+        val typeBuilder = addDefaultInterfaceDefinition(packageName, node as SchemaNode);\r
+        val schemaNodes = node.childNodes;\r
+        if (node instanceof ContainerSchemaNode) {\r
+            resolveDataSchemaNodes(basePackageName, typeBuilder, schemaNodes);\r
+            result.add(typeBuilder.toInstance());\r
+        } else if (node instanceof ListSchemaNode) {\r
+            val List<String> listKeys = listKeys(node as ListSchemaNode);\r
+            val genTOBuilder = resolveListKeyTOBuilder(packageName, node as ListSchemaNode);\r
+\r
+            if(genTOBuilder !== null) {\r
+                val identifierMarker = IDENTIFIER.parameterizedTypeFor(typeBuilder);\r
+                val identifiableMarker = IDENTIFIABLE.parameterizedTypeFor(genTOBuilder);\r
+                genTOBuilder.addImplementsType(identifierMarker);\r
+                typeBuilder.addImplementsType(identifiableMarker);\r
+            }\r
+            for (schemaNode : schemaNodes) {\r
+                if (!schemaNode.isAugmenting()) {\r
+                    addSchemaNodeToListBuilders(basePackageName, schemaNode, typeBuilder, genTOBuilder, listKeys);\r
+                }\r
+            }\r
+            result.addAll(typeBuildersToGenTypes(typeBuilder, genTOBuilder));\r
+        }\r
+        \r
+        return result;\r
+    }\r
+\r
+    private def List<Type> processUsesAugments(DataNodeContainer node, Module module) {\r
+        val List<Type> result = new ArrayList();\r
+        val basePackageName = moduleNamespaceToPackageName(module);\r
+\r
+        for (usesNode : node.uses) {\r
+            for (augment : usesNode.augmentations) {\r
+                result.addAll(augmentationToGenTypes(basePackageName, augment, module, usesNode));\r
+                result.addAll(processUsesAugments(augment, module));\r
+            }\r
+        }\r
+\r
+        return result;\r
+    }\r
 \r
     /**\r
      * Converts all <b>containers</b> of the module to the list of\r
@@ -277,14 +312,14 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def List<Type> allContainersToGenTypes( Module module) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
+    private def List<Type> allContainersToGenTypes(Module module) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
 \r
-        checkArgument(module.name !== null,"Module name cannot be NULL.");\r
+        checkArgument(module.name !== null, "Module name cannot be NULL.");\r
 \r
-        if (module.childNodes === null) {\r
-            throw new IllegalArgumentException("Reference to Set of Child Nodes in module " + module.name\r
-                    + " cannot be NULL.");\r
+        if(module.childNodes === null) {\r
+            throw new IllegalArgumentException(\r
+                "Reference to Set of Child Nodes in module " + module.name + " cannot be NULL.");\r
         }\r
 \r
         val List<Type> generatedTypes = new ArrayList();\r
@@ -292,8 +327,8 @@ public class BindingGeneratorImpl implements BindingGenerator {
         val List<ContainerSchemaNode> schemaContainers = it.allContainers();\r
         val basePackageName = moduleNamespaceToPackageName(module);\r
         for (container : schemaContainers) {\r
-            if (!container.isAddedByUses()) {\r
-                generatedTypes.add(containerToGenType(basePackageName, container));\r
+            if(!container.isAddedByUses()) {\r
+                generatedTypes.addAll(dataNodeContainerToGenType(basePackageName, container, module));\r
             }\r
         }\r
         return generatedTypes;\r
@@ -316,23 +351,23 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def List<Type> allListsToGenTypes( Module module) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
-        checkArgument(module.name !== null,"Module name cannot be NULL.");\r
+    private def List<Type> allListsToGenTypes(Module module) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
+        checkArgument(module.name !== null, "Module name cannot be NULL.");\r
 \r
-        if (module.childNodes === null) {\r
-            throw new IllegalArgumentException("Reference to Set of Child Nodes in module " + module.name\r
-                    + " cannot be NULL.");\r
+        if(module.childNodes === null) {\r
+            throw new IllegalArgumentException(\r
+                "Reference to Set of Child Nodes in module " + module.name + " cannot be NULL.");\r
         }\r
 \r
         val List<Type> generatedTypes = new ArrayList();\r
         val it = new DataNodeIterator(module);\r
         val List<ListSchemaNode> schemaLists = it.allLists();\r
         val basePackageName = moduleNamespaceToPackageName(module);\r
-        if (schemaLists !== null) {\r
+        if(schemaLists !== null) {\r
             for (list : schemaLists) {\r
-                if (!list.isAddedByUses()) {\r
-                    generatedTypes.addAll(listToGenType(basePackageName, list));\r
+                if(!list.isAddedByUses()) {\r
+                    generatedTypes.addAll(dataNodeContainerToGenType(basePackageName, list, module));\r
                 }\r
             }\r
         }\r
@@ -355,18 +390,18 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def List<GeneratedType> allChoicesToGenTypes( Module module) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
-        checkArgument(module.name !== null,"Module name cannot be NULL.");\r
+    private def List<Type> allChoicesToGenTypes(Module module) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
+        checkArgument(module.name !== null, "Module name cannot be NULL.");\r
 \r
         val it = new DataNodeIterator(module);\r
         val choiceNodes = it.allChoices();\r
         val basePackageName = moduleNamespaceToPackageName(module);\r
 \r
-        val List<GeneratedType> generatedTypes = new ArrayList();\r
+        val List<Type> generatedTypes = new ArrayList();\r
         for (choice : choiceNodes) {\r
-            if ((choice !== null) && !choice.isAddedByUses()) {\r
-                generatedTypes.addAll(choiceToGeneratedType(basePackageName, choice));\r
+            if((choice !== null) && !choice.isAddedByUses()) {\r
+                generatedTypes.addAll(choiceToGeneratedType(basePackageName, choice, module));\r
             }\r
         }\r
         return generatedTypes;\r
@@ -389,19 +424,19 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def List<Type> allAugmentsToGenTypes( Module module) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
-        checkArgument(module.name !== null,"Module name cannot be NULL.");\r
-        if (module.childNodes === null) {\r
-            throw new IllegalArgumentException("Reference to Set of Augmentation Definitions in module "\r
-                    + module.name + " cannot be NULL.");\r
+    private def List<Type> allAugmentsToGenTypes(Module module) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
+        checkArgument(module.name !== null, "Module name cannot be NULL.");\r
+        if(module.childNodes === null) {\r
+            throw new IllegalArgumentException(\r
+                "Reference to Set of Augmentation Definitions in module " + module.name + " cannot be NULL.");\r
         }\r
 \r
         val List<Type> generatedTypes = new ArrayList();\r
         val basePackageName = moduleNamespaceToPackageName(module);\r
         val List<AugmentationSchema> augmentations = resolveAugmentations(module);\r
         for (augment : augmentations) {\r
-            generatedTypes.addAll(augmentationToGenTypes(basePackageName, augment));\r
+            generatedTypes.addAll(augmentationToGenTypes(basePackageName, augment, module, null));\r
         }\r
         return generatedTypes;\r
     }\r
@@ -422,21 +457,21 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def List<AugmentationSchema> resolveAugmentations( Module module) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
-        checkState(module.augmentations !== null,"Augmentations Set cannot be NULL.");\r
+    private def List<AugmentationSchema> resolveAugmentations(Module module) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
+        checkState(module.augmentations !== null, "Augmentations Set cannot be NULL.");\r
 \r
         val Set<AugmentationSchema> augmentations = module.augmentations;\r
         val List<AugmentationSchema> sortedAugmentations = new ArrayList(augmentations);\r
-        Collections.sort(sortedAugmentations, [augSchema1, augSchema2 |\r
-\r
-                if (augSchema1.targetPath.path.size() > augSchema2.targetPath.path.size()) {\r
+        Collections.sort(sortedAugmentations,\r
+            [ augSchema1, augSchema2 |\r
+                if(augSchema1.targetPath.path.size() > augSchema2.targetPath.path.size()) {\r
                     return 1;\r
-                } else if (augSchema1.targetPath.path.size() < augSchema2.targetPath.path.size()) {\r
+                } else if(augSchema1.targetPath.path.size() < augSchema2.targetPath.path.size()) {\r
                     return -1;\r
                 }\r
                 return 0;\r
-           ]);\r
+            ]);\r
         return sortedAugmentations;\r
     }\r
 \r
@@ -454,15 +489,15 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             if the module equals null\r
      *\r
      */\r
-    private def GeneratedType moduleToDataType( Module module) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
+    private def GeneratedType moduleToDataType(Module module) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
 \r
         val moduleDataTypeBuilder = moduleTypeBuilder(module, "Data");\r
         addImplementedInterfaceFromUses(module, moduleDataTypeBuilder);\r
         moduleDataTypeBuilder.addImplementsType(DATA_ROOT);\r
 \r
         val basePackageName = moduleNamespaceToPackageName(module);\r
-        if (moduleDataTypeBuilder !== null) {\r
+        if(moduleDataTypeBuilder !== null) {\r
             val Set<DataSchemaNode> dataNodes = module.childNodes;\r
             resolveDataSchemaNodes(basePackageName, moduleDataTypeBuilder, dataNodes);\r
         }\r
@@ -487,20 +522,20 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def List<Type> allRPCMethodsToGenType( Module module) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
+    private def List<Type> allRPCMethodsToGenType(Module module) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
 \r
-        checkArgument(module.name !== null,"Module name cannot be NULL.");\r
+        checkArgument(module.name !== null, "Module name cannot be NULL.");\r
 \r
-        if (module.childNodes === null) {\r
-            throw new IllegalArgumentException("Reference to Set of RPC Method Definitions in module "\r
-                    + module.name + " cannot be NULL.");\r
+        if(module.childNodes === null) {\r
+            throw new IllegalArgumentException(\r
+                "Reference to Set of RPC Method Definitions in module " + module.name + " cannot be NULL.");\r
         }\r
 \r
         val basePackageName = moduleNamespaceToPackageName(module);\r
         val Set<RpcDefinition> rpcDefinitions = module.rpcs;\r
 \r
-        if (rpcDefinitions.isEmpty()) {\r
+        if(rpcDefinitions.isEmpty()) {\r
             return Collections.emptyList();\r
         }\r
 \r
@@ -508,7 +543,7 @@ public class BindingGeneratorImpl implements BindingGenerator {
         val interfaceBuilder = moduleTypeBuilder(module, "Service");\r
         interfaceBuilder.addImplementsType(Types.typeForClass(RpcService));\r
         for (rpc : rpcDefinitions) {\r
-            if (rpc !== null) {\r
+            if(rpc !== null) {\r
 \r
                 val rpcName = parseToClassName(rpc.QName.localName);\r
                 val rpcMethodName = parseToValidParamName(rpcName);\r
@@ -519,7 +554,7 @@ public class BindingGeneratorImpl implements BindingGenerator {
                 val input = rpc.input;\r
                 val output = rpc.output;\r
 \r
-                if (input !== null) {\r
+                if(input !== null) {\r
                     rpcInOut.add(new DataNodeIterator(input));\r
                     val inType = addRawInterfaceDefinition(basePackageName, input, rpcName);\r
                     addImplementedInterfaceFromUses(input, inType);\r
@@ -532,7 +567,7 @@ public class BindingGeneratorImpl implements BindingGenerator {
                 }\r
 \r
                 var Type outTypeInstance = VOID;\r
-                if (output !== null) {\r
+                if(output !== null) {\r
                     rpcInOut.add(new DataNodeIterator(output));\r
                     val outType = addRawInterfaceDefinition(basePackageName, output, rpcName);\r
                     addImplementedInterfaceFromUses(output, outType);\r
@@ -549,18 +584,26 @@ public class BindingGeneratorImpl implements BindingGenerator {
                 method.setReturnType(Types.parameterizedTypeFor(FUTURE, rpcRes));\r
                 for (iter : rpcInOut) {\r
                     val List<ContainerSchemaNode> nContainers = iter.allContainers();\r
-                    if ((nContainers !== null) && !nContainers.isEmpty()) {\r
+                    if((nContainers !== null) && !nContainers.isEmpty()) {\r
                         for (container : nContainers) {\r
-                            if (!container.isAddedByUses()) {\r
-                                genRPCTypes.add(containerToGenType(basePackageName, container));\r
+                            if(!container.isAddedByUses()) {\r
+                                genRPCTypes.addAll(dataNodeContainerToGenType(basePackageName, container, module));\r
                             }\r
                         }\r
                     }\r
                     val List<ListSchemaNode> nLists = iter.allLists();\r
-                    if ((nLists !== null) && !nLists.isEmpty()) {\r
+                    if((nLists !== null) && !nLists.isEmpty()) {\r
                         for (list : nLists) {\r
-                            if (!list.isAddedByUses()) {\r
-                                genRPCTypes.addAll(listToGenType(basePackageName, list));\r
+                            if(!list.isAddedByUses()) {\r
+                                genRPCTypes.addAll(dataNodeContainerToGenType(basePackageName, list, module));\r
+                            }\r
+                        }\r
+                    }\r
+                    val List<ChoiceNode> nChoices = iter.allChoices();\r
+                    if((nChoices !== null) && !nChoices.isEmpty()) {\r
+                        for (choice : nChoices) {\r
+                            if(!choice.isAddedByUses()) {\r
+                                genRPCTypes.addAll(choiceToGeneratedType(basePackageName, choice, module));\r
                             }\r
                         }\r
                     }\r
@@ -589,54 +632,55 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def List<Type> allNotificationsToGenType( Module module) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
+    private def List<Type> allNotificationsToGenType(Module module) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
 \r
-        checkArgument(module.name !== null,"Module name cannot be NULL.");\r
+        checkArgument(module.name !== null, "Module name cannot be NULL.");\r
 \r
-        if (module.childNodes === null) {\r
-            throw new IllegalArgumentException("Reference to Set of Notification Definitions in module "\r
-                    + module.name + " cannot be NULL.");\r
+        if(module.childNodes === null) {\r
+            throw new IllegalArgumentException(\r
+                "Reference to Set of Notification Definitions in module " + module.name + " cannot be NULL.");\r
         }\r
         val notifications = module.notifications;\r
         if(notifications.isEmpty()) return Collections.emptyList();\r
-        \r
+\r
         val listenerInterface = moduleTypeBuilder(module, "Listener");\r
         listenerInterface.addImplementsType(BindingTypes.NOTIFICATION_LISTENER);\r
-        \r
-        \r
-        \r
+\r
         val basePackageName = moduleNamespaceToPackageName(module);\r
         val List<Type> generatedTypes = new ArrayList();\r
+\r
         \r
-        \r
-        for ( notification : notifications) {\r
-            if (notification !== null) {\r
+\r
+        for (notification : notifications) {\r
+            if(notification !== null) {\r
+                generatedTypes.addAll(processUsesAugments(notification, module));\r
+\r
                 val iter = new DataNodeIterator(notification);\r
 \r
                 // Containers\r
                 for (node : iter.allContainers()) {\r
-                    if (!node.isAddedByUses()) {\r
-                        generatedTypes.add(containerToGenType(basePackageName, node));\r
+                    if(!node.isAddedByUses()) {\r
+                        generatedTypes.addAll(dataNodeContainerToGenType(basePackageName, node, module));\r
                     }\r
                 }\r
+\r
                 // Lists\r
                 for (node : iter.allLists()) {\r
-                    if (!node.isAddedByUses()) {\r
-                        generatedTypes.addAll(listToGenType(basePackageName, node));\r
+                    if(!node.isAddedByUses()) {\r
+                        generatedTypes.addAll(dataNodeContainerToGenType(basePackageName, node, module));\r
                     }\r
                 }\r
-                val notificationInterface = addDefaultInterfaceDefinition(basePackageName,\r
-                        notification);\r
+                val notificationInterface = addDefaultInterfaceDefinition(basePackageName, notification);\r
                 notificationInterface.addImplementsType(NOTIFICATION);\r
+\r
                 // Notification object\r
                 resolveDataSchemaNodes(basePackageName, notificationInterface, notification.childNodes);\r
-                \r
-                listenerInterface.addMethod("on"+notificationInterface.name) //\r
-                    .setAccessModifier(AccessModifier.PUBLIC)\r
-                    .addParameter(notificationInterface, "notification")\r
-                    .setReturnType(Types.VOID);\r
-                \r
+\r
+                listenerInterface.addMethod("on" + notificationInterface.name) //\r
+                .setAccessModifier(AccessModifier.PUBLIC).addParameter(notificationInterface, "notification").\r
+                    setReturnType(Types.VOID);\r
+\r
                 generatedTypes.add(notificationInterface.toInstance());\r
             }\r
         }\r
@@ -658,14 +702,14 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         (object of type <code>IdentitySchemaNode</code>\r
      *\r
      */\r
-    private def List<Type> allIdentitiesToGenTypes( Module module, SchemaContext context) {\r
+    private def List<Type> allIdentitiesToGenTypes(Module module, SchemaContext context) {\r
         val List<Type> genTypes = new ArrayList();\r
 \r
         val Set<IdentitySchemaNode> schemaIdentities = module.identities;\r
 \r
         val basePackageName = moduleNamespaceToPackageName(module);\r
 \r
-        if (schemaIdentities !== null && !schemaIdentities.isEmpty()) {\r
+        if(schemaIdentities !== null && !schemaIdentities.isEmpty()) {\r
             for (identity : schemaIdentities) {\r
                 genTypes.add(identityToGenType(basePackageName, identity, context));\r
             }\r
@@ -694,8 +738,8 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *\r
      */\r
     private def GeneratedType identityToGenType(String basePackageName, IdentitySchemaNode identity,\r
-            SchemaContext context) {\r
-        if (identity === null) {\r
+        SchemaContext context) {\r
+        if(identity === null) {\r
             return null;\r
         }\r
 \r
@@ -704,7 +748,7 @@ public class BindingGeneratorImpl implements BindingGenerator {
         val newType = new GeneratedTOBuilderImpl(packageName, genTypeName);\r
 \r
         val baseIdentity = identity.baseIdentity;\r
-        if (baseIdentity !== null) {\r
+        if(baseIdentity !== null) {\r
             val baseIdentityParentModule = SchemaContextUtil.findParentModule(context, baseIdentity);\r
 \r
             val returnTypePkgName = moduleNamespaceToPackageName(baseIdentityParentModule);\r
@@ -733,16 +777,19 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         (object of type <code>GroupingDefinition</code>)\r
      *\r
      */\r
-    private def List<Type> allGroupingsToGenTypes( Module module) {\r
-        checkArgument(module !== null,"Module parameter can not be null");\r
+    private def List<Type> allGroupingsToGenTypes(Module module) {\r
+        checkArgument(module !== null, "Module parameter can not be null");\r
         val List<Type> genTypes = new ArrayList();\r
         val basePackageName = moduleNamespaceToPackageName(module);\r
-        val Set<GroupingDefinition> groupings = module.groupings;\r
-        val List<GroupingDefinition> groupingsSortedByDependencies = new GroupingDefinitionDependencySort().sort(groupings);\r
+        val it = new DataNodeIterator(module);\r
+        val List<GroupingDefinition> groupings = it.allGroupings();\r
+        val List<GroupingDefinition> groupingsSortedByDependencies = new GroupingDefinitionDependencySort().sort(\r
+            groupings);\r
 \r
         for (grouping : groupingsSortedByDependencies) {\r
-            val genType = groupingToGenType(basePackageName, grouping);\r
+            val genType = groupingToGenType(basePackageName, grouping, module);\r
             genTypes.add(genType);\r
+            genTypes.addAll(processUsesAugments(grouping, module));\r
             val schemaPath = grouping.path;\r
             allGroupings.put(schemaPath, genType);\r
         }\r
@@ -761,8 +808,8 @@ public class BindingGeneratorImpl implements BindingGenerator {
      * @return GeneratedType which is generated from grouping (object of type\r
      *         <code>GroupingDefinition</code>)\r
      */\r
-    private def GeneratedType groupingToGenType( String basePackageName, GroupingDefinition grouping) {\r
-        if (grouping === null) {\r
+    private def GeneratedType groupingToGenType(String basePackageName, GroupingDefinition grouping, Module module) {\r
+        if(grouping === null) {\r
             return null;\r
         }\r
 \r
@@ -771,6 +818,7 @@ public class BindingGeneratorImpl implements BindingGenerator {
         val typeBuilder = addDefaultInterfaceDefinition(packageName, grouping);\r
 \r
         resolveDataSchemaNodes(basePackageName, typeBuilder, schemaNodes);\r
+\r
         return typeBuilder.toInstance();\r
     }\r
 \r
@@ -785,11 +833,11 @@ public class BindingGeneratorImpl implements BindingGenerator {
      * @return EnumTypeDefinition if it is found inside\r
      *         <code>typeDefinition</code> or <code>null</code> in other case\r
      */\r
-    private def EnumTypeDefinition enumTypeDefFromExtendedType( TypeDefinition<?> typeDefinition) {\r
-        if (typeDefinition !== null) {\r
-            if (typeDefinition.baseType instanceof EnumTypeDefinition) {\r
+    private def EnumTypeDefinition enumTypeDefFromExtendedType(TypeDefinition<?> typeDefinition) {\r
+        if(typeDefinition !== null) {\r
+            if(typeDefinition.baseType instanceof EnumTypeDefinition) {\r
                 return typeDefinition.baseType as EnumTypeDefinition;\r
-            } else if (typeDefinition.baseType instanceof ExtendedType) {\r
+            } else if(typeDefinition.baseType instanceof ExtendedType) {\r
                 return enumTypeDefFromExtendedType(typeDefinition.baseType);\r
             }\r
         }\r
@@ -813,10 +861,10 @@ public class BindingGeneratorImpl implements BindingGenerator {
      * @return enumeration builder which contais data from\r
      *         <code>enumTypeDef</code>\r
      */\r
-    private def EnumBuilder resolveInnerEnumFromTypeDefinition( EnumTypeDefinition enumTypeDef, String enumName,\r
-            GeneratedTypeBuilder typeBuilder) {\r
-        if ((enumTypeDef !== null) && (typeBuilder !== null) && (enumTypeDef.QName !== null)\r
-                && (enumTypeDef.QName.localName !== null)) {\r
+    private def EnumBuilder resolveInnerEnumFromTypeDefinition(EnumTypeDefinition enumTypeDef, String enumName,\r
+        GeneratedTypeBuilder typeBuilder) {\r
+        if((enumTypeDef !== null) && (typeBuilder !== null) && (enumTypeDef.QName !== null) &&\r
+            (enumTypeDef.QName.localName !== null)) {\r
 \r
             val enumerationName = parseToClassName(enumName);\r
             val enumBuilder = typeBuilder.addEnumeration(enumerationName);\r
@@ -841,13 +889,12 @@ public class BindingGeneratorImpl implements BindingGenerator {
      * @throws IllegalArgumentException\r
      *             if <code>module</code> equals null\r
      */\r
-    private def GeneratedTypeBuilder moduleTypeBuilder( Module module, String postfix) {\r
-        checkArgument(module !== null,"Module reference cannot be NULL.");\r
+    private def GeneratedTypeBuilder moduleTypeBuilder(Module module, String postfix) {\r
+        checkArgument(module !== null, "Module reference cannot be NULL.");\r
         val packageName = moduleNamespaceToPackageName(module);\r
         val moduleName = parseToClassName(module.name) + postfix;\r
 \r
         return new GeneratedTypeBuilderImpl(packageName, moduleName);\r
-\r
     }\r
 \r
     /**\r
@@ -872,42 +919,225 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             <li>if target path of <code>augSchema</code> equals null</li>\r
      *             </ul>\r
      */\r
-    private def List<Type> augmentationToGenTypes(String augmentPackageName, AugmentationSchema augSchema) {\r
-        checkArgument(augmentPackageName !== null,"Package Name cannot be NULL.");\r
-        checkArgument(augSchema !== null,"Augmentation Schema cannot be NULL.");\r
-        checkState(augSchema.targetPath !== null,"Augmentation Schema does not contain Target Path (Target Path is NULL).");\r
-        val List<Type> genTypes = new ArrayList();\r
+    private def List<Type> augmentationToGenTypes(String augmentPackageName, AugmentationSchema augSchema, Module module,
+        UsesNode parentUsesNode) {
+        checkArgument(augmentPackageName !== null, "Package Name cannot be NULL.");
+        checkArgument(augSchema !== null, "Augmentation Schema cannot be NULL.");
+        checkState(augSchema.targetPath !== null,
+            "Augmentation Schema does not contain Target Path (Target Path is NULL).");
+        val List<Type> genTypes = new ArrayList();
+        genTypes.addAll(processUsesAugments(augSchema, module));
+
         // EVERY augmented interface will extends Augmentation<T> interface\r
         // and DataObject interface!!!\r
-        val targetPath = augSchema.targetPath;\r
-        val targetSchemaNode = findDataSchemaNode(schemaContext, targetPath);\r
-        var targetType = yangToJavaMapping.get(targetSchemaNode.path);\r
-        if(targetType == null) {\r
-            // FIXME: augmentation should be added as last, all types should already be generated\r
-            // and have assigned Java Types,\r
-            val targetModule = findParentModule(schemaContext, targetSchemaNode);\r
-            val targetBasePackage = moduleNamespaceToPackageName(targetModule);\r
-            val typePackage = packageNameForGeneratedType(targetBasePackage, targetSchemaNode.getPath());\r
-            val targetSchemaNodeName = targetSchemaNode.getQName().getLocalName();\r
-            val typeName = parseToClassName(targetSchemaNodeName);\r
-            targetType = new ReferencedTypeImpl(typePackage,typeName);\r
-        }\r
-        if (targetSchemaNode !== null) {\r
-            val augChildNodes = augSchema.childNodes;\r
-            if (!(targetSchemaNode instanceof ChoiceNode)) {\r
-                val augTypeBuilder = addRawAugmentGenTypeDefinition(augmentPackageName,\r
-                        targetType, augSchema);\r
-                val augType = augTypeBuilder.toInstance();\r
-                genTypes.add(augType);\r
+        val targetPath = augSchema.targetPath;
+        var targetSchemaNode = findDataSchemaNode(schemaContext, targetPath);
+        if (targetSchemaNode instanceof DataSchemaNode && (targetSchemaNode as DataSchemaNode).isAddedByUses()) {\r
+            if (parentUsesNode == null) {\r
+                targetSchemaNode = findOriginal(targetSchemaNode as DataSchemaNode);\r
             } else {\r
-                val choiceTarget = targetSchemaNode as ChoiceNode;\r
-                val choiceCaseNodes = choiceTarget.cases;\r
-                genTypes.addAll(generateTypesFromAugmentedChoiceCases(augmentPackageName, targetType,\r
-                        choiceCaseNodes));\r
+                targetSchemaNode = findOriginalTargetFromGrouping(targetSchemaNode.QName.localName, parentUsesNode);\r
+            }\r
+            if (targetSchemaNode == null) {\r
+                throw new NullPointerException(\r
+                    "Failed to find target node from grouping for augmentation " + augSchema + " in module " + module.name);\r
+            }
+        }
+\r
+        if (targetSchemaNode !== null) {
+            var targetType = yangToJavaMapping.get(targetSchemaNode.path);
+            if (targetType == null) {
+                // FIXME: augmentation should be added as last, all types should already be generated\r
+                // and have assigned Java Types,\r
+                val targetModule = findParentModule(schemaContext, targetSchemaNode);
+                val targetBasePackage = moduleNamespaceToPackageName(targetModule);
+                val typePackage = packageNameForGeneratedType(targetBasePackage, targetSchemaNode.getPath());
+                val targetSchemaNodeName = targetSchemaNode.getQName().getLocalName();
+                val typeName = parseToClassName(targetSchemaNodeName);
+                targetType = new ReferencedTypeImpl(typePackage, typeName);
+            }
+            val augChildNodes = augSchema.childNodes;
+
+            if (!(targetSchemaNode instanceof ChoiceNode)) {
+                var packageName = augmentPackageName;
+                if (parentUsesNode != null) {
+                    packageName = packageNameForGeneratedType(augmentPackageName, augSchema.targetPath);
+                }
+                val augTypeBuilder = addRawAugmentGenTypeDefinition(packageName, augmentPackageName, targetType,
+                    augSchema);
+                val augType = augTypeBuilder.toInstance();
+                genTypes.add(augType);
+            } else {
+                genTypes.addAll(
+                    generateTypesFromAugmentedChoiceCases(augmentPackageName, targetType, augChildNodes,
+                        targetSchemaNode as ChoiceNode));
+            }
+            genTypes.addAll(augmentationBodyToGenTypes(augmentPackageName, augChildNodes, module));
+        }
+
+        return genTypes;
+    }\r
+\r
+    /**\r
+     * Utility method which search for original node defined in grouping.\r
+     */\r
+    private def DataSchemaNode findOriginal(DataSchemaNode node) {\r
+        var DataSchemaNode result = findCorrectTargetFromGrouping(node);\r
+        if (result == null) {\r
+            result = findCorrectTargetFromAugment(node);\r
+            if (result != null) {\r
+                if (result.addedByUses) {\r
+                    result = findOriginal(result);\r
+                }\r
             }\r
-            genTypes.addAll(augmentationBodyToGenTypes(augmentPackageName, augChildNodes));\r
         }\r
-        return genTypes;\r
+        return result;\r
+    }\r
+\r
+    private def DataSchemaNode findCorrectTargetFromAugment(DataSchemaNode node) {\r
+        if (!node.augmenting) {\r
+            return null;\r
+        }\r
+\r
+        var String currentName = node.QName.localName;\r
+        var tmpPath = new ArrayList<String>();\r
+        var YangNode parent = node;\r
+        var AugmentationSchema augment = null;\r
+        do {\r
+            parent = (parent as DataSchemaNode).parent;\r
+            if (parent instanceof AugmentationTarget) {\r
+                tmpPath.add(currentName);\r
+                augment = findNodeInAugment((parent as AugmentationTarget).availableAugmentations, currentName);\r
+                if (augment == null) {\r
+                    currentName = (parent as DataSchemaNode).QName.localName; \r
+                }\r
+            }\r
+        } while ((parent as DataSchemaNode).augmenting && augment == null);\r
+\r
+        if (augment == null) {\r
+            return null;\r
+        } else {\r
+            Collections.reverse(tmpPath);\r
+            var Object actualParent = augment;\r
+            var DataSchemaNode result = null;\r
+            for (name : tmpPath) {\r
+                if (actualParent instanceof DataNodeContainer) {\r
+                    result = (actualParent as DataNodeContainer).getDataChildByName(name);\r
+                    actualParent = (actualParent as DataNodeContainer).getDataChildByName(name);\r
+                } else {\r
+                    if (actualParent instanceof ChoiceNode) {\r
+                        result = (actualParent as ChoiceNode).getCaseNodeByName(name);\r
+                        actualParent = (actualParent as ChoiceNode).getCaseNodeByName(name); \r
+                    }\r
+                }\r
+            }\r
+\r
+            if (result.addedByUses) {\r
+                result = findCorrectTargetFromGrouping(result);\r
+            }\r
+\r
+            return result;\r
+        }\r
+    }\r
+\r
+    private def AugmentationSchema findNodeInAugment(Collection<AugmentationSchema> augments, String name) {\r
+        for (augment : augments) {\r
+            if (augment.getDataChildByName(name) != null) {\r
+                return augment;\r
+            }\r
+        }\r
+        return null;\r
+    }\r
+\r
+    private def DataSchemaNode findCorrectTargetFromGrouping(DataSchemaNode node) {\r
+        if (node.path.path.size == 1) {\r
+            // uses is under module statement\r
+            val Module m = findParentModule(schemaContext, node);\r
+            var DataSchemaNode result = null;\r
+            for (u : m.uses) {\r
+                var SchemaNode targetGrouping = findNodeInSchemaContext(schemaContext, u.groupingPath.path);\r
+                if (!(targetGrouping instanceof GroupingDefinition)) {\r
+                    throw new IllegalArgumentException("Failed to generate code for augment in " + u);\r
+                }\r
+                var gr = targetGrouping as GroupingDefinition;\r
+                result = gr.getDataChildByName(node.QName.localName);\r
+            }\r
+            if (result == null) {\r
+                throw new IllegalArgumentException("Failed to generate code for augment");\r
+            }\r
+            return result;\r
+        } else {\r
+            var DataSchemaNode result = null;\r
+            var String currentName = node.QName.localName;\r
+            var tmpPath = new ArrayList<String>();\r
+            var YangNode parent = node.parent; \r
+            do {\r
+                tmpPath.add(currentName);\r
+                val dataNodeParent = parent as DataNodeContainer;\r
+                for (u : dataNodeParent.uses) {\r
+                    var SchemaNode targetGrouping = findNodeInSchemaContext(schemaContext, u.groupingPath.path);\r
+                    if (!(targetGrouping instanceof GroupingDefinition)) {\r
+                        throw new IllegalArgumentException("Failed to generate code for augment in " + u);\r
+                    }\r
+                    var gr = targetGrouping as GroupingDefinition;\r
+                    result = gr.getDataChildByName(currentName);\r
+                }\r
+                if (result == null) {\r
+                    currentName = (parent as SchemaNode).QName.localName;\r
+                    if (parent instanceof DataSchemaNode) {\r
+                        parent = (parent as DataSchemaNode).parent;\r
+                    } else {\r
+                        parent = (parent as DataNodeContainer).parent;\r
+                    }\r
+                }\r
+            } while (result == null && !(parent instanceof Module));\r
+\r
+            if (result != null) {\r
+                if (tmpPath.size == 1) {\r
+                    return result;\r
+                } else {\r
+                    var DataSchemaNode newParent = result;\r
+                    Collections.reverse(tmpPath);\r
+                    tmpPath.remove(0);\r
+                    for (name : tmpPath) {\r
+                        newParent = (newParent as DataNodeContainer).getDataChildByName(name);\r
+                    }\r
+                    return newParent;\r
+                }\r
+            }\r
+\r
+            return result;\r
+        }\r
+    }\r
+\r
+    /**\r
+     * Convenient method to find node added by uses statement.\r
+     */\r
+    private def DataSchemaNode findOriginalTargetFromGrouping(String targetSchemaNodeName, UsesNode parentUsesNode) {\r
+        var SchemaNode targetGrouping = findNodeInSchemaContext(schemaContext, parentUsesNode.groupingPath.path);\r
+        if (!(targetGrouping instanceof GroupingDefinition)) {\r
+            throw new IllegalArgumentException("Failed to generate code for augment in " + parentUsesNode);\r
+        }\r
+\r
+        var grouping = targetGrouping as GroupingDefinition;\r
+        var result = grouping.getDataChildByName(targetSchemaNodeName);\r
+        if (result == null) {\r
+            return null;\r
+        }\r
+        var boolean fromUses = result.addedByUses;\r
+\r
+        var Iterator<UsesNode> groupingUses = grouping.uses.iterator;\r
+        while (fromUses) {\r
+            if (groupingUses.hasNext()) {\r
+                grouping = findNodeInSchemaContext(schemaContext, groupingUses.next().groupingPath.path) as GroupingDefinition;\r
+                result = grouping.getDataChildByName(targetSchemaNodeName);\r
+                fromUses = result.addedByUses;\r
+            } else {\r
+                throw new NullPointerException("Failed to generate code for augment in " + parentUsesNode);\r
+            }\r
+        }\r
+\r
+        return result;\r
     }\r
 \r
     /**\r
@@ -929,8 +1159,8 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *            and uses of augment\r
      * @return generated type builder for augment\r
      */\r
-    private def GeneratedTypeBuilder addRawAugmentGenTypeDefinition( String augmentPackageName,\r
-            Type targetTypeRef, AugmentationSchema augSchema) {\r
+    private def GeneratedTypeBuilder addRawAugmentGenTypeDefinition(String augmentPackageName, String basePackageName,\r
+        Type targetTypeRef, AugmentationSchema augSchema) {\r
         var Map<String, GeneratedTypeBuilder> augmentBuilders = genTypeBuilders.get(augmentPackageName);\r
         if (augmentBuilders === null) {\r
             augmentBuilders = new HashMap();\r
@@ -938,12 +1168,11 @@ public class BindingGeneratorImpl implements BindingGenerator {
         }\r
         val augIdentifier = getAugmentIdentifier(augSchema.unknownSchemaNodes);\r
 \r
-        val augTypeName = if (augIdentifier !== null ) { \r
-            parseToClassName(augIdentifier)\r
-        } else {\r
-            augGenTypeName(augmentBuilders, targetTypeRef.name);\r
-        }\r
-        val Set<DataSchemaNode> augChildNodes = augSchema.childNodes;\r
+        val augTypeName = if (augIdentifier !== null) {\r
+                parseToClassName(augIdentifier)\r
+            } else {\r
+                augGenTypeName(augmentBuilders, targetTypeRef.name);\r
+            }\r
 \r
         val augTypeBuilder = new GeneratedTypeBuilderImpl(augmentPackageName, augTypeName);\r
 \r
@@ -951,7 +1180,7 @@ public class BindingGeneratorImpl implements BindingGenerator {
         augTypeBuilder.addImplementsType(Types.augmentationTypeFor(targetTypeRef));\r
         addImplementedInterfaceFromUses(augSchema, augTypeBuilder);\r
 \r
-        augSchemaNodeToMethods(augmentPackageName, augTypeBuilder, augChildNodes);\r
+        augSchemaNodeToMethods(basePackageName, augTypeBuilder, augSchema.childNodes);\r
         augmentBuilders.put(augTypeName, augTypeBuilder);\r
         return augTypeBuilder;\r
     }\r
@@ -959,13 +1188,13 @@ public class BindingGeneratorImpl implements BindingGenerator {
     /**\r
      *\r
      * @param unknownSchemaNodes\r
-     * @return\r
+     * @return nodeParameter of UnknownSchemaNode\r
      */\r
     private def String getAugmentIdentifier(List<UnknownSchemaNode> unknownSchemaNodes) {\r
         for (unknownSchemaNode : unknownSchemaNodes) {\r
             val nodeType = unknownSchemaNode.nodeType;\r
-            if (AUGMENT_IDENTIFIER_NAME.equals(nodeType.localName)\r
-                    && YANG_EXT_NAMESPACE.equals(nodeType.namespace.toString())) {\r
+            if(AUGMENT_IDENTIFIER_NAME.equals(nodeType.localName) &&\r
+                YANG_EXT_NAMESPACE.equals(nodeType.namespace.toString())) {\r
                 return unknownSchemaNode.nodeParameter;\r
             }\r
         }\r
@@ -985,25 +1214,30 @@ public class BindingGeneratorImpl implements BindingGenerator {
      * @return list of <code>Type</code> which represents container, list and\r
      *         choice subnodes of augment\r
      */\r
-    private def List<Type> augmentationBodyToGenTypes( String augBasePackageName,\r
-            Set<DataSchemaNode> augChildNodes) {\r
+    private def List<Type> augmentationBodyToGenTypes(String augBasePackageName, Set<DataSchemaNode> augChildNodes, Module module) {\r
         val List<Type> genTypes = new ArrayList();\r
         val List<DataNodeIterator> augSchemaIts = new ArrayList();\r
         for (childNode : augChildNodes) {\r
-            if (childNode instanceof DataNodeContainer) {\r
+            if (!childNode.addedByUses) {\r
+                \r
+            \r
+            if(childNode instanceof DataNodeContainer) {\r
                 augSchemaIts.add(new DataNodeIterator(childNode as DataNodeContainer));\r
 \r
-                if (childNode instanceof ContainerSchemaNode) {\r
-                    genTypes.add(containerToGenType(augBasePackageName, childNode as ContainerSchemaNode));\r
-                } else if (childNode instanceof ListSchemaNode) {\r
-                    genTypes.addAll(listToGenType(augBasePackageName, childNode as ListSchemaNode));\r
+                if(childNode instanceof ContainerSchemaNode) {\r
+                    genTypes.addAll(dataNodeContainerToGenType(augBasePackageName, childNode as ContainerSchemaNode, module));\r
+                } else if(childNode instanceof ListSchemaNode) {\r
+                    genTypes.addAll(dataNodeContainerToGenType(augBasePackageName, childNode as ListSchemaNode, module));\r
                 }\r
-            } else if (childNode instanceof ChoiceNode) {\r
+            } else if(childNode instanceof ChoiceNode) {\r
                 val choice = childNode as ChoiceNode;\r
                 for (caseNode : choice.cases) {\r
                     augSchemaIts.add(new DataNodeIterator(caseNode));\r
                 }\r
-                genTypes.addAll(choiceToGeneratedType(augBasePackageName, childNode as ChoiceNode));\r
+                genTypes.addAll(choiceToGeneratedType(augBasePackageName, childNode as ChoiceNode, module));\r
+            }\r
+            \r
+            \r
             }\r
         }\r
 \r
@@ -1012,19 +1246,19 @@ public class BindingGeneratorImpl implements BindingGenerator {
             val List<ListSchemaNode> augLists = it.allLists();\r
             val List<ChoiceNode> augChoices = it.allChoices();\r
 \r
-            if (augContainers !== null) {\r
+            if(augContainers !== null) {\r
                 for (container : augContainers) {\r
-                    genTypes.add(containerToGenType(augBasePackageName, container));\r
+                    genTypes.addAll(dataNodeContainerToGenType(augBasePackageName, container, module));\r
                 }\r
             }\r
-            if (augLists !== null) {\r
+            if(augLists !== null) {\r
                 for (list : augLists) {\r
-                    genTypes.addAll(listToGenType(augBasePackageName, list));\r
+                    genTypes.addAll(dataNodeContainerToGenType(augBasePackageName, list, module));\r
                 }\r
             }\r
-            if (augChoices !== null) {\r
+            if(augChoices !== null) {\r
                 for (choice : augChoices) {\r
-                    genTypes.addAll(choiceToGeneratedType(augBasePackageName, choice));\r
+                    genTypes.addAll(choiceToGeneratedType(augBasePackageName, choice, module));\r
                 }\r
             }\r
         }\r
@@ -1043,40 +1277,14 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *            string with name of augmented node\r
      * @return string with unique name for augmentation builder\r
      */\r
-    private def String augGenTypeName( Map<String, GeneratedTypeBuilder> builders, String genTypeName) {\r
+    private def String augGenTypeName(Map<String, GeneratedTypeBuilder> builders, String genTypeName) {\r
         var index = 1;\r
-        while ((builders !== null) && builders.containsKey(genTypeName + index)) {\r
+        while((builders !== null) && builders.containsKey(genTypeName + index)) {\r
             index = index + 1;\r
         }\r
         return genTypeName + index;\r
     }\r
 \r
-    /**\r
-     * Converts <code>containerNode</code> to generated type. Firstly the\r
-     * generated type builder is created. The subnodes of\r
-     * <code>containerNode</code> are added as methods and the instance of\r
-     * <code>GeneratedType</code> is returned.\r
-     *\r
-     * @param basePackageName\r
-     *            string contains the module package name\r
-     * @param containerNode\r
-     *            container schema node with the data about childs nodes and\r
-     *            schema paths\r
-     * @return generated type for <code>containerNode</code>\r
-     */\r
-    private def GeneratedType containerToGenType( String basePackageName, ContainerSchemaNode containerNode) {\r
-        if (containerNode === null) {\r
-            return null;\r
-        }\r
-\r
-        val packageName = packageNameForGeneratedType(basePackageName, containerNode.path);\r
-        val schemaNodes = containerNode.childNodes;\r
-        val typeBuilder = addDefaultInterfaceDefinition(packageName, containerNode);\r
-\r
-        resolveDataSchemaNodes(basePackageName, typeBuilder, schemaNodes);\r
-        return typeBuilder.toInstance();\r
-    }\r
-\r
     /**\r
      * Adds the methods to <code>typeBuilder</code> which represent subnodes of\r
      * node for which <code>typeBuilder</code> was created.\r
@@ -1098,18 +1306,30 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         parameter. The getter methods (representing child nodes) could be\r
      *         added to it.\r
      */\r
-    private def GeneratedTypeBuilder resolveDataSchemaNodes( String basePackageName,\r
-            GeneratedTypeBuilder typeBuilder, Set<DataSchemaNode> schemaNodes) {\r
-        if ((schemaNodes !== null) && (typeBuilder !== null)) {\r
+    private def GeneratedTypeBuilder resolveDataSchemaNodes(String basePackageName, GeneratedTypeBuilder typeBuilder,\r
+        Set<DataSchemaNode> schemaNodes) {\r
+        if((schemaNodes !== null) && (typeBuilder !== null)) {\r
             for (schemaNode : schemaNodes) {\r
-                if (!schemaNode.isAugmenting() && !schemaNode.isAddedByUses()) {\r
+                if(!schemaNode.isAugmenting() && !schemaNode.isAddedByUses()) {\r
                     addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);\r
                 }\r
-                \r
+\r
             }\r
         }\r
         return typeBuilder;\r
     }\r
+       \r
+       private def GeneratedTypeBuilder resolveDataSchemaNodesAugmented(String basePackageName, GeneratedTypeBuilder typeBuilder,\r
+               Set<DataSchemaNode> schemaNodes) {\r
+               if ((schemaNodes !== null) && (typeBuilder !== null)) {\r
+                       for (schemaNode : schemaNodes) {\r
+                               if (!schemaNode.isAddedByUses()) {\r
+                                       addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);\r
+                               }\r
+                       }\r
+               }\r
+               return typeBuilder;\r
+       }\r
 \r
     /**\r
      * Adds the methods to <code>typeBuilder</code> what represents subnodes of\r
@@ -1129,17 +1349,17 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         parameter <code>typeBuilder</code>. The getter method could be\r
      *         added to it.\r
      */\r
-    private def GeneratedTypeBuilder augSchemaNodeToMethods( String basePackageName,\r
-             GeneratedTypeBuilder typeBuilder,  Set<DataSchemaNode> schemaNodes) {\r
-        if ((schemaNodes !== null) && (typeBuilder !== null)) {\r
+    private def GeneratedTypeBuilder augSchemaNodeToMethods(String basePackageName, GeneratedTypeBuilder typeBuilder,\r
+        Set<DataSchemaNode> schemaNodes) {\r
+        if((schemaNodes !== null) && (typeBuilder !== null)) {\r
             for (schemaNode : schemaNodes) {\r
-                if (schemaNode.isAugmenting()) {\r
-                    addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);\r
-                }\r
-            }\r
-        }\r
-        return typeBuilder;\r
-    }\r
+                           if (!schemaNode.isAugmenting()) {\r
+                                   addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);\r
+                               }\r
+                       }\r
+               }\r
+               return typeBuilder;\r
+       }\r
 \r
     /**\r
      * Adds to <code>typeBuilder</code> a method which is derived from\r
@@ -1154,15 +1374,20 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *            generated type builder to which is <code>schemaNode</code>\r
      *            added as a method.\r
      */\r
-    private def void addSchemaNodeToBuilderAsMethod( String basePackageName,  DataSchemaNode node,\r
-             GeneratedTypeBuilder typeBuilder) {\r
-        if (node !== null && typeBuilder !== null) {\r
-            switch(node) {\r
-                case node instanceof LeafSchemaNode: resolveLeafSchemaNodeAsMethod(typeBuilder, node as LeafSchemaNode)\r
-                case node instanceof LeafListSchemaNode: resolveLeafListSchemaNode(typeBuilder, node as LeafListSchemaNode)\r
-                case node instanceof ContainerSchemaNode: resolveContainerSchemaNode(basePackageName, typeBuilder, node as ContainerSchemaNode)\r
-                case node instanceof ListSchemaNode: resolveListSchemaNode(basePackageName, typeBuilder, node as ListSchemaNode)\r
-                case node instanceof ChoiceNode: resolveChoiceSchemaNode(basePackageName, typeBuilder, node as ChoiceNode)\r
+    private def void addSchemaNodeToBuilderAsMethod(String basePackageName, DataSchemaNode node,\r
+        GeneratedTypeBuilder typeBuilder) {\r
+        if(node !== null && typeBuilder !== null) {\r
+            switch (node) {\r
+                case node instanceof LeafSchemaNode:\r
+                    resolveLeafSchemaNodeAsMethod(typeBuilder, node as LeafSchemaNode)\r
+                case node instanceof LeafListSchemaNode:\r
+                    resolveLeafListSchemaNode(typeBuilder, node as LeafListSchemaNode)\r
+                case node instanceof ContainerSchemaNode:\r
+                    resolveContainerSchemaNode(basePackageName, typeBuilder, node as ContainerSchemaNode)\r
+                case node instanceof ListSchemaNode:\r
+                    resolveListSchemaNode(basePackageName, typeBuilder, node as ListSchemaNode)\r
+                case node instanceof ChoiceNode:\r
+                    resolveChoiceSchemaNode(basePackageName, typeBuilder, node as ChoiceNode)\r
             }\r
         }\r
     }\r
@@ -1192,14 +1417,14 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def void resolveChoiceSchemaNode( String basePackageName,  GeneratedTypeBuilder typeBuilder,\r
-             ChoiceNode choiceNode) {\r
-        checkArgument(basePackageName !== null,"Base Package Name cannot be NULL.");\r
-        checkArgument(typeBuilder !== null,"Generated Type Builder cannot be NULL.");\r
-        checkArgument(choiceNode !== null,"Choice Schema Node cannot be NULL.");\r
+    private def void resolveChoiceSchemaNode(String basePackageName, GeneratedTypeBuilder typeBuilder,\r
+        ChoiceNode choiceNode) {\r
+        checkArgument(basePackageName !== null, "Base Package Name cannot be NULL.");\r
+        checkArgument(typeBuilder !== null, "Generated Type Builder cannot be NULL.");\r
+        checkArgument(choiceNode !== null, "Choice Schema Node cannot be NULL.");\r
 \r
         val choiceName = choiceNode.QName.localName;\r
-        if (choiceName !== null && !choiceNode.isAddedByUses()) {\r
+        if(choiceName !== null && !choiceNode.isAddedByUses()) {\r
             val packageName = packageNameForGeneratedType(basePackageName, choiceNode.path);\r
             val choiceType = addDefaultInterfaceDefinition(packageName, choiceNode);\r
             constructGetter(typeBuilder, choiceName, choiceNode.description, choiceType);\r
@@ -1229,20 +1454,21 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def List<GeneratedType> choiceToGeneratedType( String basePackageName,  ChoiceNode choiceNode) {\r
-        checkArgument(basePackageName !== null,"Base Package Name cannot be NULL.");\r
-        checkArgument(choiceNode !== null,"Choice Schema Node cannot be NULL.");\r
+    private def List<Type> choiceToGeneratedType(String basePackageName, ChoiceNode choiceNode, Module module) {\r
+        checkArgument(basePackageName !== null, "Base Package Name cannot be NULL.");\r
+        checkArgument(choiceNode !== null, "Choice Schema Node cannot be NULL.");\r
 \r
-        val List<GeneratedType> generatedTypes = new ArrayList();\r
+        val List<Type> generatedTypes = new ArrayList();\r
         val packageName = packageNameForGeneratedType(basePackageName, choiceNode.path);\r
         val choiceTypeBuilder = addRawInterfaceDefinition(packageName, choiceNode);\r
-        //choiceTypeBuilder.addImplementsType(DATA_OBJECT);\r
+\r
+        choiceTypeBuilder.addImplementsType(DataContainer.typeForClass);\r
         val choiceType = choiceTypeBuilder.toInstance();\r
 \r
         generatedTypes.add(choiceType);\r
         val Set<ChoiceCaseNode> caseNodes = choiceNode.cases;\r
-        if ((caseNodes !== null) && !caseNodes.isEmpty()) {\r
-            generatedTypes.addAll(generateTypesFromChoiceCases(basePackageName, choiceType, caseNodes));\r
+        if((caseNodes !== null) && !caseNodes.isEmpty()) {\r
+            generatedTypes.addAll(generateTypesFromChoiceCases(basePackageName, choiceType, caseNodes, module));\r
         }\r
         return generatedTypes;\r
     }\r
@@ -1273,25 +1499,27 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *             *\r
      */\r
-    private def List<GeneratedType> generateTypesFromChoiceCases( String basePackageName,  Type refChoiceType,\r
-             Set<ChoiceCaseNode> caseNodes) {\r
-        checkArgument(basePackageName !== null,"Base Package Name cannot be NULL.");\r
-        checkArgument(refChoiceType !== null,"Referenced Choice Type cannot be NULL.");\r
-        checkArgument(caseNodes !== null,"Set of Choice Case Nodes cannot be NULL.");\r
+    private def List<Type> generateTypesFromChoiceCases(String basePackageName, Type refChoiceType,\r
+        Set<ChoiceCaseNode> caseNodes, Module module) {\r
+        checkArgument(basePackageName !== null, "Base Package Name cannot be NULL.");\r
+        checkArgument(refChoiceType !== null, "Referenced Choice Type cannot be NULL.");\r
+        checkArgument(caseNodes !== null, "Set of Choice Case Nodes cannot be NULL.");\r
 \r
-        val List<GeneratedType> generatedTypes = new ArrayList();\r
+        val List<Type> generatedTypes = new ArrayList();\r
         for (caseNode : caseNodes) {\r
-            if (caseNode !== null && !caseNode.isAddedByUses() && !caseNode.isAugmenting()) {\r
+            if(caseNode !== null && !caseNode.isAddedByUses() && !caseNode.isAugmenting()) {\r
                 val packageName = packageNameForGeneratedType(basePackageName, caseNode.path);\r
                 val caseTypeBuilder = addDefaultInterfaceDefinition(packageName, caseNode);\r
                 caseTypeBuilder.addImplementsType(refChoiceType);\r
 \r
                 val Set<DataSchemaNode> childNodes = caseNode.childNodes;\r
-                if (childNodes !== null) {\r
+                if(childNodes !== null) {\r
                     resolveDataSchemaNodes(basePackageName, caseTypeBuilder, childNodes);\r
                 }\r
                 generatedTypes.add(caseTypeBuilder.toInstance());\r
             }\r
+            \r
+            generatedTypes.addAll(processUsesAugments(caseNode, module));\r
         }\r
 \r
         return generatedTypes;\r
@@ -1322,23 +1550,33 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             <li>if <code>caseNodes</code> equals null</li>\r
      *             </ul>\r
      */\r
-    private def List<GeneratedType> generateTypesFromAugmentedChoiceCases( String basePackageName,\r
-             Type refChoiceType,  Set<ChoiceCaseNode> caseNodes) {\r
-        checkArgument(basePackageName !== null,"Base Package Name cannot be NULL.");\r
-        checkArgument(refChoiceType !== null,"Referenced Choice Type cannot be NULL.");\r
-        checkArgument(caseNodes !== null,"Set of Choice Case Nodes cannot be NULL.");\r
+    private def List<GeneratedType> generateTypesFromAugmentedChoiceCases(String basePackageName, Type refChoiceType,\r
+        Set<DataSchemaNode> caseNodes, ChoiceNode targetNode) {\r
+        checkArgument(basePackageName !== null, "Base Package Name cannot be NULL.");\r
+        checkArgument(refChoiceType !== null, "Referenced Choice Type cannot be NULL.");\r
+        checkArgument(caseNodes !== null, "Set of Choice Case Nodes cannot be NULL.");\r
 \r
         val List<GeneratedType> generatedTypes = new ArrayList();\r
         for (caseNode : caseNodes) {\r
-            if (caseNode !== null && caseNode.isAugmenting()) {\r
+               if(caseNode !== null) {\r
                 val packageName = packageNameForGeneratedType(basePackageName, caseNode.path);\r
                 val caseTypeBuilder = addDefaultInterfaceDefinition(packageName, caseNode);\r
                 caseTypeBuilder.addImplementsType(refChoiceType);\r
 \r
-                val Set<DataSchemaNode> childNodes = caseNode.childNodes;\r
-                if (childNodes !== null) {\r
-                    resolveDataSchemaNodes(basePackageName, caseTypeBuilder, childNodes);\r
+                if (caseNode instanceof DataNodeContainer) {\r
+                       val DataNodeContainer dataNodeCase = caseNode as DataNodeContainer;\r
+                       val Set<DataSchemaNode> childNodes = dataNodeCase.childNodes;\r
+                    if(childNodes !== null) {\r
+                                           resolveDataSchemaNodesAugmented(basePackageName, caseTypeBuilder, childNodes);\r
+                    }\r
+                } else {\r
+                       val ChoiceCaseNode node = targetNode.getCaseNodeByName(caseNode.getQName().getLocalName());\r
+                       val Set<DataSchemaNode> childNodes = node.childNodes;\r
+                    if(childNodes !== null) {\r
+                                           resolveDataSchemaNodesAugmented(basePackageName, caseTypeBuilder, childNodes);\r
+                    }\r
                 }\r
+                \r
                 generatedTypes.add(caseTypeBuilder.toInstance());\r
             }\r
         }\r
@@ -1363,44 +1601,44 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         <li>true - in other cases</li>\r
      *         </ul>\r
      */\r
-    private def boolean resolveLeafSchemaNodeAsMethod( GeneratedTypeBuilder typeBuilder,  LeafSchemaNode leaf) {\r
-        if ((leaf !== null) && (typeBuilder !== null)) {\r
+    private def boolean resolveLeafSchemaNodeAsMethod(GeneratedTypeBuilder typeBuilder, LeafSchemaNode leaf) {\r
+        if((leaf !== null) && (typeBuilder !== null)) {\r
             val leafName = leaf.QName.localName;\r
             var String leafDesc = leaf.description;\r
-            if (leafDesc === null) {\r
+            if(leafDesc === null) {\r
                 leafDesc = "";\r
             }\r
 \r
             val parentModule = findParentModule(schemaContext, leaf);\r
-            if (leafName !== null && !leaf.isAddedByUses()) {\r
+            if(leafName !== null && !leaf.isAddedByUses()) {\r
                 val TypeDefinition<?> typeDef = leaf.type;\r
 \r
                 var Type returnType = null;\r
-                if (typeDef instanceof EnumTypeDefinition) {\r
+                if(typeDef instanceof EnumTypeDefinition) {\r
                     returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef, leaf);\r
-                    val enumTypeDef = enumTypeDefFromExtendedType(typeDef);\r
-                    val enumBuilder = resolveInnerEnumFromTypeDefinition(enumTypeDef, leafName,\r
-                            typeBuilder);\r
+                    val enumTypeDef = typeDef as EnumTypeDefinition;\r
+                    val enumBuilder = resolveInnerEnumFromTypeDefinition(enumTypeDef, leafName, typeBuilder);\r
 \r
-                    if (enumBuilder !== null) {\r
+                    if(enumBuilder !== null) {\r
                         returnType = new ReferencedTypeImpl(enumBuilder.packageName, enumBuilder.name);\r
                     }\r
                     (typeProvider as TypeProviderImpl).putReferencedType(leaf.path, returnType);\r
-                } else if (typeDef instanceof UnionType) {\r
+                } else if(typeDef instanceof UnionType) {\r
                     val genTOBuilder = addTOToTypeBuilder(typeDef, typeBuilder, leafName, leaf, parentModule);\r
-                    if (genTOBuilder !== null) {\r
+                    if(genTOBuilder !== null) {\r
                         returnType = new ReferencedTypeImpl(genTOBuilder.packageName, genTOBuilder.name);\r
                     }\r
-                } else if (typeDef instanceof BitsTypeDefinition) {\r
+                } else if(typeDef instanceof BitsTypeDefinition) {\r
                     val genTOBuilder = addTOToTypeBuilder(typeDef, typeBuilder, leafName, leaf, parentModule);\r
-                    if (genTOBuilder !== null) {\r
+                    if(genTOBuilder !== null) {\r
                         returnType = new ReferencedTypeImpl(genTOBuilder.packageName, genTOBuilder.name);\r
                     }\r
                 } else {\r
                     returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef, leaf);\r
                 }\r
-                if (returnType !== null) {\r
-                    constructGetter(typeBuilder, leafName, leafDesc, returnType);\r
+                if(returnType !== null) {\r
+                    val MethodSignatureBuilder getter = constructGetter(typeBuilder, leafName, leafDesc, returnType);\r
+                    processContextRefExtension(leaf, getter, parentModule);\r
                     return true;\r
                 }\r
             }\r
@@ -1408,6 +1646,59 @@ public class BindingGeneratorImpl implements BindingGenerator {
         return false;\r
     }\r
 \r
+    private def void processContextRefExtension(LeafSchemaNode leaf, MethodSignatureBuilder getter, Module module) {\r
+        for (node : leaf.unknownSchemaNodes) {\r
+            val nodeType = node.nodeType;\r
+            if ("context-reference".equals(nodeType.localName)) {\r
+                val nodeParam = node.nodeParameter;\r
+                var IdentitySchemaNode identity = null;\r
+                var String basePackageName = null;\r
+                val String[] splittedElement = nodeParam.split(":");\r
+                if (splittedElement.length == 1) {\r
+                    identity = findIdentityByName(module.identities, splittedElement.get(0));\r
+                    basePackageName = moduleNamespaceToPackageName(module);\r
+                } else if (splittedElement.length == 2) {\r
+                    var prefix = splittedElement.get(0);\r
+                    val Module dependentModule = findModuleFromImports(module.imports, prefix)\r
+                    if (dependentModule == null) {\r
+                        throw new IllegalArgumentException("Failed to process context-reference: unknown prefix " + prefix);\r
+                    }\r
+                    identity = findIdentityByName(dependentModule.identities, splittedElement.get(1));\r
+                    basePackageName = moduleNamespaceToPackageName(dependentModule);\r
+                } else {\r
+                    throw new IllegalArgumentException("Failed to process context-reference: unknown identity " + nodeParam);\r
+                }\r
+                if (identity == null) {\r
+                    throw new IllegalArgumentException("Failed to process context-reference: unknown identity " + nodeParam);\r
+                }\r
+\r
+                val Class<RoutingContext> clazz = typeof(RoutingContext);\r
+                val AnnotationTypeBuilder rc = getter.addAnnotation(clazz.package.name, clazz.simpleName);\r
+                val packageName = packageNameForGeneratedType(basePackageName, identity.path);\r
+                val genTypeName = parseToClassName(identity.QName.localName);\r
+                rc.addParameter("value", packageName + "." + genTypeName + ".class");\r
+            }\r
+        }\r
+    }\r
+\r
+    private def IdentitySchemaNode findIdentityByName(Set<IdentitySchemaNode> identities, String name) {\r
+        for (id : identities) {\r
+            if (id.QName.localName.equals(name)) {\r
+                return id;\r
+            }\r
+        }\r
+        return null;\r
+    }\r
+\r
+    private def Module findModuleFromImports(Set<ModuleImport> imports, String prefix) {\r
+        for (imp : imports) {\r
+            if (imp.prefix.equals(prefix)) {\r
+                return schemaContext.findModuleByName(imp.moduleName, imp.revision);\r
+            }\r
+        }\r
+        return null;\r
+    }\r
+\r
     /**\r
      * Converts <code>leaf</code> schema node to property of generated TO\r
      * builder.\r
@@ -1427,22 +1718,22 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         <li>true - other cases</li>\r
      *         </ul>\r
      */\r
-    private def boolean resolveLeafSchemaNodeAsProperty( GeneratedTOBuilder toBuilder, LeafSchemaNode leaf,\r
-            boolean isReadOnly) {\r
-        if ((leaf !== null) && (toBuilder !== null)) {\r
+    private def boolean resolveLeafSchemaNodeAsProperty(GeneratedTOBuilder toBuilder, LeafSchemaNode leaf,\r
+        boolean isReadOnly) {\r
+        if((leaf !== null) && (toBuilder !== null)) {\r
             val leafName = leaf.QName.localName;\r
             var String leafDesc = leaf.description;\r
-            if (leafDesc === null) {\r
+            if(leafDesc === null) {\r
                 leafDesc = "";\r
             }\r
 \r
-            if (leafName !== null && !leaf.isAddedByUses()) {\r
+            if(leafName !== null) {\r
                 val TypeDefinition<?> typeDef = leaf.type;\r
 \r
                 // TODO: properly resolve enum types\r
                 val returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef, leaf);\r
 \r
-                if (returnType !== null) {\r
+                if(returnType !== null) {\r
                     val propBuilder = toBuilder.addProperty(parseToClassName(leafName));\r
 \r
                     propBuilder.setReadOnly(isReadOnly);\r
@@ -1477,15 +1768,15 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         <li>false - other cases</li>\r
      *         </ul>\r
      */\r
-    private def boolean resolveLeafListSchemaNode( GeneratedTypeBuilder typeBuilder, LeafListSchemaNode node) {\r
-        if ((node !== null) && (typeBuilder !== null)) {\r
+    private def boolean resolveLeafListSchemaNode(GeneratedTypeBuilder typeBuilder, LeafListSchemaNode node) {\r
+        if((node !== null) && (typeBuilder !== null)) {\r
             val nodeName = node.QName.localName;\r
             var String nodeDesc = node.description;\r
-            if (nodeDesc === null) {\r
+            if(nodeDesc === null) {\r
                 nodeDesc = "";\r
             }\r
 \r
-            if (nodeName !== null && !node.isAddedByUses()) {\r
+            if(nodeName !== null && !node.isAddedByUses()) {\r
                 val TypeDefinition<?> type = node.type;\r
                 val listType = Types.listTypeFor(typeProvider.javaTypeForSchemaDefinitionType(type, node));\r
 \r
@@ -1522,12 +1813,12 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         <li>true - other cases</li>\r
      *         </ul>\r
      */\r
-    private def boolean resolveContainerSchemaNode( String basePackageName, GeneratedTypeBuilder typeBuilder,\r
-            ContainerSchemaNode containerNode) {\r
-        if ((containerNode !== null) && (typeBuilder !== null)) {\r
+    private def boolean resolveContainerSchemaNode(String basePackageName, GeneratedTypeBuilder typeBuilder,\r
+        ContainerSchemaNode containerNode) {\r
+        if((containerNode !== null) && (typeBuilder !== null)) {\r
             val nodeName = containerNode.QName.localName;\r
 \r
-            if (nodeName !== null && !containerNode.isAddedByUses()) {\r
+            if(nodeName !== null && !containerNode.isAddedByUses()) {\r
                 val packageName = packageNameForGeneratedType(basePackageName, containerNode.path);\r
 \r
                 val rawGenType = addDefaultInterfaceDefinition(packageName, containerNode);\r
@@ -1564,12 +1855,12 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         <li>true - other cases</li>\r
      *         </ul>\r
      */\r
-    private def boolean resolveListSchemaNode( String basePackageName, GeneratedTypeBuilder typeBuilder,\r
-            ListSchemaNode listNode) {\r
-        if ((listNode !== null) && (typeBuilder !== null)) {\r
+    private def boolean resolveListSchemaNode(String basePackageName, GeneratedTypeBuilder typeBuilder,\r
+        ListSchemaNode listNode) {\r
+        if((listNode !== null) && (typeBuilder !== null)) {\r
             val listName = listNode.QName.localName;\r
 \r
-            if (listName !== null && !listNode.isAddedByUses()) {\r
+            if(listName !== null && !listNode.isAddedByUses()) {\r
                 val packageName = packageNameForGeneratedType(basePackageName, listNode.path);\r
                 val rawGenType = addDefaultInterfaceDefinition(packageName, listNode);\r
                 constructGetter(typeBuilder, listName, listNode.description, Types.listTypeFor(rawGenType));\r
@@ -1600,15 +1891,15 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *            schema node for which is created generated type builder\r
      * @return generated type builder <code>schemaNode</code>\r
      */\r
-    private def GeneratedTypeBuilder addDefaultInterfaceDefinition( String packageName, SchemaNode schemaNode) {\r
+    private def GeneratedTypeBuilder addDefaultInterfaceDefinition(String packageName, SchemaNode schemaNode) {\r
         val builder = addRawInterfaceDefinition(packageName, schemaNode, "");\r
         builder.addImplementsType(DATA_OBJECT);\r
-        if (!(schemaNode instanceof GroupingDefinition)) {\r
+        if(!(schemaNode instanceof GroupingDefinition)) {\r
             builder.addImplementsType(augmentable(builder));\r
         }\r
 \r
-        if (schemaNode instanceof DataNodeContainer) {\r
-            addImplementedInterfaceFromUses( schemaNode as DataNodeContainer, builder);\r
+        if(schemaNode instanceof DataNodeContainer) {\r
+            addImplementedInterfaceFromUses(schemaNode as DataNodeContainer, builder);\r
         }\r
 \r
         return builder;\r
@@ -1624,7 +1915,7 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *            schema node which provide data about the schema node name\r
      * @return generated type builder for <code>schemaNode</code>\r
      */\r
-    private def GeneratedTypeBuilder addRawInterfaceDefinition( String packageName,  SchemaNode schemaNode) {\r
+    private def GeneratedTypeBuilder addRawInterfaceDefinition(String packageName, SchemaNode schemaNode) {\r
         return addRawInterfaceDefinition(packageName, schemaNode, "");\r
     }\r
 \r
@@ -1651,30 +1942,31 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             </ul>\r
      *\r
      */\r
-    private def GeneratedTypeBuilder addRawInterfaceDefinition( String packageName,  SchemaNode schemaNode,\r
-             String prefix) {\r
-        checkArgument(schemaNode !== null,"Data Schema Node cannot be NULL.");\r
-        checkArgument(packageName !== null,"Package Name for Generated Type cannot be NULL.");\r
-        checkArgument(schemaNode.QName !== null,"QName for Data Schema Node cannot be NULL.");\r
+    private def GeneratedTypeBuilder addRawInterfaceDefinition(String packageName, SchemaNode schemaNode,\r
+        String prefix) {\r
+        checkArgument(schemaNode !== null, "Data Schema Node cannot be NULL.");\r
+        checkArgument(packageName !== null, "Package Name for Generated Type cannot be NULL.");\r
+        checkArgument(schemaNode.QName !== null, "QName for Data Schema Node cannot be NULL.");\r
         val schemaNodeName = schemaNode.QName.localName;\r
-        checkArgument(schemaNodeName !== null,"Local Name of QName for Data Schema Node cannot be NULL.");\r
+        checkArgument(schemaNodeName !== null, "Local Name of QName for Data Schema Node cannot be NULL.");\r
 \r
         var String genTypeName;\r
-        if (prefix === null) {\r
+        if(prefix === null) {\r
             genTypeName = parseToClassName(schemaNodeName);\r
         } else {\r
             genTypeName = prefix + parseToClassName(schemaNodeName);\r
         }\r
+\r
         //FIXME: Validation of name conflict\r
         val newType = new GeneratedTypeBuilderImpl(packageName, genTypeName);\r
-        yangToJavaMapping.put(schemaNode.path,newType);\r
-        if (!genTypeBuilders.containsKey(packageName)) {\r
+        yangToJavaMapping.put(schemaNode.path, newType);\r
+        if(!genTypeBuilders.containsKey(packageName)) {\r
             val Map<String, GeneratedTypeBuilder> builders = new HashMap();\r
             builders.put(genTypeName, newType);\r
             genTypeBuilders.put(packageName, builders);\r
         } else {\r
             val Map<String, GeneratedTypeBuilder> builders = genTypeBuilders.get(packageName);\r
-            if (!builders.containsKey(genTypeName)) {\r
+            if(!builders.containsKey(genTypeName)) {\r
                 builders.put(genTypeName, newType);\r
             }\r
         }\r
@@ -1689,7 +1981,7 @@ public class BindingGeneratorImpl implements BindingGenerator {
      * @return string with the name of the getter method for\r
      *         <code>methodName</code> in JAVA method format\r
      */\r
-    private def String getterMethodName( String methodName,Type returnType) {\r
+    private def String getterMethodName(String methodName, Type returnType) {\r
         val method = new StringBuilder();\r
         if(BOOLEAN.equals(returnType)) {\r
             method.append("is");\r
@@ -1721,10 +2013,10 @@ public class BindingGeneratorImpl implements BindingGenerator {
      * @return method signature builder which represents the getter method of\r
      *         <code>interfaceBuilder</code>\r
      */\r
-    private def MethodSignatureBuilder constructGetter( GeneratedTypeBuilder interfaceBuilder,\r
-             String schemaNodeName,  String comment,  Type returnType) {\r
+    private def MethodSignatureBuilder constructGetter(GeneratedTypeBuilder interfaceBuilder, String schemaNodeName,\r
+        String comment, Type returnType) {\r
 \r
-        val getMethod = interfaceBuilder.addMethod(getterMethodName(schemaNodeName,returnType));\r
+        val getMethod = interfaceBuilder.addMethod(getterMethodName(schemaNodeName, returnType));\r
 \r
         getMethod.setComment(comment);\r
         getMethod.setReturnType(returnType);\r
@@ -1732,34 +2024,6 @@ public class BindingGeneratorImpl implements BindingGenerator {
         return getMethod;\r
     }\r
 \r
-    private def listToGenType( String basePackageName,  ListSchemaNode list) {\r
-        checkArgument(basePackageName !== null,"Package Name for Generated Type cannot be NULL.");\r
-        checkArgument(list !== null,"List Schema Node cannot be NULL.");\r
-\r
-        val packageName = packageNameForGeneratedType(basePackageName, list.path);\r
-        // val typeBuilder =\r
-        // resolveListTypeBuilder(packageName, list);\r
-        val typeBuilder = addDefaultInterfaceDefinition(packageName, list);\r
-\r
-        val List<String> listKeys = listKeys(list);\r
-        val genTOBuilder = resolveListKeyTOBuilder(packageName, list);\r
-\r
-        if (genTOBuilder !== null) {\r
-            val identifierMarker = IDENTIFIER.parameterizedTypeFor(typeBuilder);\r
-            val identifiableMarker = IDENTIFIABLE.parameterizedTypeFor(genTOBuilder);\r
-            genTOBuilder.addImplementsType(identifierMarker);\r
-            typeBuilder.addImplementsType(identifiableMarker);\r
-        }\r
-        val schemaNodes = list.childNodes;\r
-\r
-        for (schemaNode : schemaNodes) {\r
-            if (!schemaNode.isAugmenting()) {\r
-                addSchemaNodeToListBuilders(basePackageName, schemaNode, typeBuilder, genTOBuilder, listKeys);\r
-            }\r
-        }\r
-        return typeBuildersToGenTypes(typeBuilder, genTOBuilder);\r
-    }\r
-\r
     /**\r
      * Adds <code>schemaNode</code> to <code>typeBuilder</code> as getter method\r
      * or to <code>genTOBuilder</code> as property.\r
@@ -1782,34 +2046,38 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *             <li>if <code>typeBuilder</code> equals null</li>\r
      *             </ul>\r
      */\r
-    private def void addSchemaNodeToListBuilders( String basePackageName,  DataSchemaNode schemaNode,\r
-             GeneratedTypeBuilder typeBuilder,  GeneratedTOBuilder genTOBuilder,  List<String> listKeys) {\r
-        checkArgument(schemaNode !== null,"Data Schema Node cannot be NULL.");\r
+    private def void addSchemaNodeToListBuilders(String basePackageName, DataSchemaNode schemaNode,\r
+        GeneratedTypeBuilder typeBuilder, GeneratedTOBuilder genTOBuilder, List<String> listKeys) {\r
+        checkArgument(schemaNode !== null, "Data Schema Node cannot be NULL.");\r
 \r
-        checkArgument(typeBuilder !== null,"Generated Type Builder cannot be NULL.");\r
+        checkArgument(typeBuilder !== null, "Generated Type Builder cannot be NULL.");\r
 \r
-        if (schemaNode instanceof LeafSchemaNode) {\r
+        if(schemaNode instanceof LeafSchemaNode) {\r
             val leaf = schemaNode as LeafSchemaNode;\r
             val leafName = leaf.QName.localName;\r
-            if (!listKeys.contains(leafName)) {\r
+            if(!listKeys.contains(leafName)) {\r
                 resolveLeafSchemaNodeAsMethod(typeBuilder, leaf);\r
             } else {\r
                 resolveLeafSchemaNodeAsProperty(genTOBuilder, leaf, true);\r
             }\r
-        } else if (schemaNode instanceof LeafListSchemaNode) {\r
-            resolveLeafListSchemaNode(typeBuilder,  schemaNode as LeafListSchemaNode);\r
-        } else if (schemaNode instanceof ContainerSchemaNode) {\r
-            resolveContainerSchemaNode(basePackageName, typeBuilder, schemaNode as ContainerSchemaNode);\r
-        } else if (schemaNode instanceof ListSchemaNode) {\r
-            resolveListSchemaNode(basePackageName, typeBuilder, schemaNode as ListSchemaNode);\r
+        } else if (!schemaNode.addedByUses) {\r
+            if (schemaNode instanceof LeafListSchemaNode) {\r
+                resolveLeafListSchemaNode(typeBuilder, schemaNode as LeafListSchemaNode);\r
+            } else if(schemaNode instanceof ContainerSchemaNode) {\r
+                resolveContainerSchemaNode(basePackageName, typeBuilder, schemaNode as ContainerSchemaNode);\r
+            } else if(schemaNode instanceof ChoiceNode) {\r
+                resolveChoiceSchemaNode(basePackageName,typeBuilder,schemaNode as ChoiceNode);\r
+            } else if(schemaNode instanceof ListSchemaNode) {\r
+                resolveListSchemaNode(basePackageName, typeBuilder, schemaNode as ListSchemaNode);\r
+            }\r
         }\r
     }\r
 \r
-    private def typeBuildersToGenTypes( GeneratedTypeBuilder typeBuilder, GeneratedTOBuilder genTOBuilder) {\r
+    private def typeBuildersToGenTypes(GeneratedTypeBuilder typeBuilder, GeneratedTOBuilder genTOBuilder) {\r
         val List<Type> genTypes = new ArrayList();\r
-        checkArgument(typeBuilder !== null,"Generated Type Builder cannot be NULL.");\r
+        checkArgument(typeBuilder !== null, "Generated Type Builder cannot be NULL.");\r
 \r
-        if (genTOBuilder !== null) {\r
+        if(genTOBuilder !== null) {\r
             val genTO = genTOBuilder.toInstance();\r
             constructGetter(typeBuilder, "key", "Returns Primary Key of Yang List Type", genTO);\r
             genTypes.add(genTO);\r
@@ -1828,10 +2096,10 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         <code>list</code> contains no keys then the empty list is\r
      *         returned.\r
      */\r
-    private def listKeys( ListSchemaNode list) {\r
+    private def listKeys(ListSchemaNode list) {\r
         val List<String> listKeys = new ArrayList();\r
 \r
-        if (list.keyDefinition !== null) {\r
+        if(list.keyDefinition !== null) {\r
             val keyDefinitions = list.keyDefinition;\r
             for (keyDefinition : keyDefinitions) {\r
                 listKeys.add(keyDefinition.localName);\r
@@ -1852,12 +2120,13 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *         <code>list</code> or null if <code>list</code> is null or list of\r
      *         key definitions is null or empty.\r
      */\r
-    private def GeneratedTOBuilder resolveListKeyTOBuilder( String packageName,  ListSchemaNode list) {\r
+    private def GeneratedTOBuilder resolveListKeyTOBuilder(String packageName, ListSchemaNode list) {\r
         var GeneratedTOBuilder genTOBuilder = null;\r
-        if ((list.keyDefinition !== null) && (!list.keyDefinition.isEmpty())) {\r
-            if (list !== null) {\r
+        if((list.keyDefinition !== null) && (!list.keyDefinition.isEmpty())) {\r
+            if(list !== null) {\r
                 val listName = list.QName.localName + "Key";\r
-                genTOBuilder = schemaNodeToTransferObjectBuilder(packageName, listName);\r
+                val String genTOName = parseToClassName(listName);\r
+                genTOBuilder = new GeneratedTOBuilderImpl(packageName, genTOName);\r
             }\r
         }\r
         return genTOBuilder;\r
@@ -1886,18 +2155,21 @@ public class BindingGeneratorImpl implements BindingGenerator {
      * @return generated TO builder for <code>typeDef</code>\r
      */\r
     private def GeneratedTOBuilder addTOToTypeBuilder(TypeDefinition<?> typeDef, GeneratedTypeBuilder typeBuilder,\r
-            String leafName, LeafSchemaNode leaf, Module parentModule) {\r
+        String leafName, LeafSchemaNode leaf, Module parentModule) {\r
         val classNameFromLeaf = parseToClassName(leafName);\r
         val List<GeneratedTOBuilder> genTOBuilders = new ArrayList();\r
         val packageName = typeBuilder.fullyQualifiedName;\r
-        if (typeDef instanceof UnionTypeDefinition) {\r
-            genTOBuilders.addAll((typeProvider as TypeProviderImpl).provideGeneratedTOBuildersForUnionTypeDef(\r
-                    packageName, typeDef, classNameFromLeaf, leaf));\r
-        } else if (typeDef instanceof BitsTypeDefinition) {\r
-            genTOBuilders.add(((typeProvider as TypeProviderImpl) ).provideGeneratedTOBuilderForBitsTypeDefinition(\r
-                    packageName, typeDef, classNameFromLeaf));\r
-        }\r
-        if (genTOBuilders !== null && !genTOBuilders.isEmpty()) {\r
+        if(typeDef instanceof UnionTypeDefinition) {\r
+            genTOBuilders.addAll(\r
+                (typeProvider as TypeProviderImpl).\r
+                    provideGeneratedTOBuildersForUnionTypeDef(packageName, (typeDef as UnionTypeDefinition),\r
+                        classNameFromLeaf, leaf));\r
+        } else if(typeDef instanceof BitsTypeDefinition) {\r
+            genTOBuilders.add(\r
+                ((typeProvider as TypeProviderImpl) ).\r
+                    provideGeneratedTOBuilderForBitsTypeDefinition(packageName, typeDef, classNameFromLeaf));\r
+        }\r
+        if(genTOBuilders !== null && !genTOBuilders.isEmpty()) {\r
             for (genTOBuilder : genTOBuilders) {\r
                 typeBuilder.addEnclosingTransferObject(genTOBuilder);\r
             }\r
@@ -1923,18 +2195,17 @@ public class BindingGeneratorImpl implements BindingGenerator {
      *            <code>dataNodeContainer</code>\r
      * @return generated type builder with all implemented types\r
      */\r
-    private def addImplementedInterfaceFromUses( DataNodeContainer dataNodeContainer,\r
-            GeneratedTypeBuilder builder) {\r
+    private def addImplementedInterfaceFromUses(DataNodeContainer dataNodeContainer, GeneratedTypeBuilder builder) {\r
         for (usesNode : dataNodeContainer.uses) {\r
-            if (usesNode.groupingPath !== null) {\r
+            if(usesNode.groupingPath !== null) {\r
                 val genType = allGroupings.get(usesNode.groupingPath);\r
-                if (genType === null) {\r
-                    throw new IllegalStateException("Grouping " + usesNode.groupingPath + "is not resolved for "\r
-                            + builder.name);\r
+                if(genType === null) {\r
+                    throw new IllegalStateException(\r
+                        "Grouping " + usesNode.groupingPath + "is not resolved for " + builder.name);\r
                 }\r
                 builder.addImplementsType(genType);\r
             }\r
         }\r
         return builder;\r
     }\r
-}
+}\r