Implemented range validation (Bug 178).
[mdsal.git] / code-generator / binding-generator-impl / src / main / java / org / opendaylight / yangtools / sal / binding / generator / impl / BindingGeneratorImpl.xtend
index 751930767de48ff0fce1ad2798fffe1ca329d95f..4d694bba9270a351159ba1aa7f7a9bfaaa22727f 100644 (file)
@@ -1,9 +1,9 @@
-/*
- * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
- *
- * This program and the accompanying materials are made available under the
- * terms of the Eclipse Public License v1.0 which accompanies this distribution,
- * and is available at http://www.eclipse.org/legal/epl-v10.html
+/*\r
+ * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.\r
+ *\r
+ * This program and the accompanying materials are made available under the\r
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
+ * and is available at http://www.eclipse.org/legal/epl-v10.html\r
  */
 package org.opendaylight.yangtools.sal.binding.generator.impl;
 
@@ -21,7 +21,6 @@ import org.opendaylight.yangtools.binding.generator.util.generated.type.builder.
 import org.opendaylight.yangtools.sal.binding.generator.api.BindingGenerator;
 import org.opendaylight.yangtools.sal.binding.generator.spi.TypeProvider;
 import org.opendaylight.yangtools.sal.binding.model.api.AccessModifier;
-import org.opendaylight.yangtools.sal.binding.model.api.GeneratedType;
 import org.opendaylight.yangtools.sal.binding.model.api.Type;
 import org.opendaylight.yangtools.sal.binding.model.api.type.builder.EnumBuilder;
 import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTOBuilder;
@@ -62,66 +61,72 @@ import static org.opendaylight.yangtools.binding.generator.util.BindingTypes.*;
 import static org.opendaylight.yangtools.yang.model.util.SchemaContextUtil.*;
 import org.opendaylight.yangtools.yang.parser.util.ModuleDependencySort
 import org.opendaylight.yangtools.yang.model.util.ExtendedType;
-
+import org.opendaylight.yangtools.yang.model.api.UsesNode
+import org.opendaylight.yangtools.yang.binding.annotations.RoutingContext
+import org.opendaylight.yangtools.sal.binding.model.api.type.builder.AnnotationTypeBuilder
+import org.opendaylight.yangtools.yang.model.api.ModuleImport
+import org.opendaylight.yangtools.yang.binding.DataContainer
+import java.util.Iterator
+import org.opendaylight.yangtools.yang.model.api.AugmentationTarget
+import java.util.Collection
+import org.opendaylight.yangtools.yang.model.api.YangNode
+import org.opendaylight.yangtools.yang.model.api.NotificationDefinition
+import org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil
+import org.opendaylight.yangtools.sal.binding.model.api.Restrictions
+import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedPropertyBuilder
+import org.opendaylight.yangtools.binding.generator.util.generated.type.builder.GeneratedPropertyBuilderImpl
+import org.opendaylight.yangtools.yang.common.QName\r
+\r
 public class BindingGeneratorImpl implements BindingGenerator {
-    /**
-     * Outter key represents the package name. Outter value represents map of
-     * all builders in the same package. Inner key represents the schema node
-     * name (in JAVA class/interface name format). Inner value represents
-     * instance of builder for schema node specified in key part.
+
+    private final Map<Module, ModuleContext> genCtx = new HashMap()
+
+    /**\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
+     * name (in JAVA class/interface name format). Inner value represents\r
+     * instance of builder for schema node specified in key part.\r
      */
     private Map<String, Map<String, GeneratedTypeBuilder>> genTypeBuilders;
 
-    /**
-     * Provide methods for converting YANG types to JAVA types.
+    /**\r
+     * Provide methods for converting YANG types to JAVA types.\r
      */
     private var TypeProvider typeProvider;
 
-    /**
-     * Holds reference to schema context to resolve data of augmented elemnt
-     * when creating augmentation builder
+    /**\r
+     * Holds reference to schema context to resolve data of augmented elemnt\r
+     * when creating augmentation builder\r
      */
     private var SchemaContext schemaContext;
 
-    /**
-     * Each grouping which is converted from schema node to generated type is
-     * added to this map with its Schema path as key to make it easier to get
-     * reference to it. In schema nodes in <code>uses</code> attribute there is
-     * only Schema Path but when building list of implemented interfaces for
-     * Schema node the object of type <code>Type</code> is required. So in this
-     * case is used this map.
-     */
-    private val allGroupings = new HashMap<SchemaPath, GeneratedType>();
-
-    private val yangToJavaMapping = new HashMap<SchemaPath, Type>();
-
-    /**
-     * Constant with the concrete name of namespace.
+    /**\r
+     * Constant with the concrete name of namespace.\r
      */
     private val static String YANG_EXT_NAMESPACE = "urn:opendaylight:yang:extension:yang-ext";
 
-    /**
-     * Constant with the concrete name of identifier.
+    /**\r
+     * Constant with the concrete name of identifier.\r
      */
     private val static String AUGMENT_IDENTIFIER_NAME = "augment-identifier";
 
-    /**
-     * Resolves generated types from <code>context</code> schema nodes of all
-     * modules.
-     *
-     * Generated types are created for modules, groupings, types, containers,
-     * lists, choices, augments, rpcs, notification, identities.
-     *
-     * @param context
-     *            schema context which contains data about all schema nodes
-     *            saved in modules
-     * @return list of types (usually <code>GeneratedType</code>
-     *         <code>GeneratedTransferObject</code>which are generated from
-     *         <code>context</code> data.
-     * @throws IllegalArgumentException
-     *             if param <code>context</code> is null
-     * @throws IllegalStateException
-     *             if <code>context</code> contain no modules
+    /**\r
+     * Resolves generated types from <code>context</code> schema nodes of all\r
+     * modules.\r
+     *\r
+     * Generated types are created for modules, groupings, types, containers,\r
+     * lists, choices, augments, rpcs, notification, identities.\r
+     *\r
+     * @param context\r
+     *            schema context which contains data about all schema nodes\r
+     *            saved in modules\r
+     * @return list of types (usually <code>GeneratedType</code>\r
+     *         <code>GeneratedTransferObject</code>which are generated from\r
+     *         <code>context</code> data.\r
+     * @throws IllegalArgumentException\r
+     *             if param <code>context</code> is null\r
+     * @throws IllegalStateException\r
+     *             if <code>context</code> contain no modules\r
      */
     override generateTypes(SchemaContext context) {
         checkArgument(context !== null, "Schema Context reference cannot be NULL.");
@@ -132,269 +137,220 @@ public class BindingGeneratorImpl implements BindingGenerator {
         return generateTypes(context, modules);
     }
 
-    /**
-     * Resolves generated types from <code>context</code> schema nodes only for
-     * modules specified in <code>modules</code>
-     *
-     * Generated types are created for modules, groupings, types, containers,
-     * lists, choices, augments, rpcs, notification, identities.
-     *
-     * @param context
-     *            schema context which contains data about all schema nodes
-     *            saved in modules
-     * @param modules
-     *            set of modules for which schema nodes should be generated
-     *            types
-     * @return list of types (usually <code>GeneratedType</code> or
-     *         <code>GeneratedTransferObject</code>) which:
-     *         <ul>
-     *         <li>are generated from <code>context</code> schema nodes and</li>
-     *         <li>are also part of some of the module in <code>modules</code>
-     *         set</li>.
-     *         </ul>
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if param <code>context</code> is null or</li>
-     *             <li>if param <code>modules</code> is null</li>
-     *             </ul>
-     * @throws IllegalStateException
-     *             if <code>context</code> contain no modules
+    /**\r
+     * Resolves generated types from <code>context</code> schema nodes only for\r
+     * modules specified in <code>modules</code>\r
+     *\r
+     * Generated types are created for modules, groupings, types, containers,\r
+     * lists, choices, augments, rpcs, notification, identities.\r
+     *\r
+     * @param context\r
+     *            schema context which contains data about all schema nodes\r
+     *            saved in modules\r
+     * @param modules\r
+     *            set of modules for which schema nodes should be generated\r
+     *            types\r
+     * @return list of types (usually <code>GeneratedType</code> or\r
+     *         <code>GeneratedTransferObject</code>) which:\r
+     *         <ul>\r
+     *         <li>are generated from <code>context</code> schema nodes and</li>\r
+     *         <li>are also part of some of the module in <code>modules</code>\r
+     *         set</li>.\r
+     *         </ul>\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if param <code>context</code> is null or</li>\r
+     *             <li>if param <code>modules</code> is null</li>\r
+     *             </ul>\r
+     * @throws IllegalStateException\r
+     *             if <code>context</code> contain no modules\r
      */
     override generateTypes(SchemaContext context, Set<Module> modules) {
         checkArgument(context !== null, "Schema Context reference cannot be NULL.");
         checkState(context.modules !== null, "Schema Context does not contain defined modules.");
         checkArgument(modules !== null, "Set of Modules cannot be NULL.");
 
-        val List<Type> filteredGenTypes = new ArrayList();
-
         schemaContext = context;
         typeProvider = new TypeProviderImpl(context);
         val contextModules = ModuleDependencySort.sort(context.modules);
         genTypeBuilders = new HashMap();
 
         for (contextModule : contextModules) {
-            val List<Type> generatedTypes = new ArrayList();
-            generatedTypes.addAll(allTypeDefinitionsToGenTypes(contextModule));
-            generatedTypes.addAll(allGroupingsToGenTypes(contextModule));
-            if(false == contextModule.childNodes.isEmpty()) {
-                generatedTypes.add(moduleToDataType(contextModule));
-            }
-            generatedTypes.addAll(allContainersToGenTypes(contextModule));
-            generatedTypes.addAll(allListsToGenTypes(contextModule));
-            generatedTypes.addAll(allChoicesToGenTypes(contextModule));
-            generatedTypes.addAll(allRPCMethodsToGenType(contextModule));
-            generatedTypes.addAll(allNotificationsToGenType(contextModule));
-            generatedTypes.addAll(allIdentitiesToGenTypes(contextModule, context));
-
-            if(modules.contains(contextModule)) {
-                filteredGenTypes.addAll(generatedTypes);
-            }
+            moduleToGenTypes(contextModule, context);
         }
         for (contextModule : contextModules) {
-            val generatedTypes = (allAugmentsToGenTypes(contextModule));
-            if(modules.contains(contextModule)) {
-                filteredGenTypes.addAll(generatedTypes);
-            }
+            allAugmentsToGenTypes(contextModule);
+        }
+
+        val List<Type> filteredGenTypes = new ArrayList();
+        for (Module m : modules) {
+            filteredGenTypes.addAll(genCtx.get(m).generatedTypes);
 
         }
+        //genCtx.clear;
+
         return filteredGenTypes;
     }
 
-    /**
-     * Converts all extended type definitions of module to the list of
-     * <code>Type</code> objects.
-     *
-     * @param module
-     *            module from which is obtained set of type definitions
-     * @return list of <code>Type</code> which are generated from extended
-     *         definition types (object of type <code>ExtendedType</code>)
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if module equals null</li>
-     *             <li>if name of module equals null</li>
-     *             <li>if type definitions of module equal null</li>
-     *             </ul>
-     *
+    private def void moduleToGenTypes(Module m, SchemaContext context) {
+        genCtx.put(m, new ModuleContext)
+        allTypeDefinitionsToGenTypes(m)
+        groupingsToGenTypes(m, m.groupings)
+        rpcMethodsToGenType(m)
+        allIdentitiesToGenTypes(m, context)
+        notificationsToGenType(m)
+
+        if (!m.childNodes.isEmpty()) {
+            val moduleType = moduleToDataType(m)
+            genCtx.get(m).addModuleNode(moduleType)
+            val basePackageName = moduleNamespaceToPackageName(m);
+            resolveDataSchemaNodes(m, basePackageName, moduleType, moduleType, m.childNodes)
+        }
+    }
+
+    /**\r
+     * Converts all extended type definitions of module to the list of\r
+     * <code>Type</code> objects.\r
+     *\r
+     * @param module\r
+     *            module from which is obtained set of type definitions\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if module equals null</li>\r
+     *             <li>if name of module equals null</li>\r
+     *             <li>if type definitions of module equal null</li>\r
+     *             </ul>\r
+     *\r
      */
-    private def List<Type> allTypeDefinitionsToGenTypes(Module module) {
+    private def void allTypeDefinitionsToGenTypes(Module module) {
         checkArgument(module !== null, "Module reference cannot be NULL.");
         checkArgument(module.name !== null, "Module name cannot be NULL.");
-        val Set<TypeDefinition<?>> typeDefinitions = module.typeDefinitions;
+        val it = new DataNodeIterator(module);
+        val List<TypeDefinition<?>> typeDefinitions = it.allTypedefs;
         checkState(typeDefinitions !== null, '''Type Definitions for module «module.name» cannot be NULL.''');
 
-        val List<Type> generatedTypes = new ArrayList();
         for (TypeDefinition<?> typedef : typeDefinitions) {
-            if(typedef !== null) {
+            if (typedef !== null) {
                 val type = (typeProvider as TypeProviderImpl).generatedTypeForExtendedDefinitionType(typedef, typedef);
-                if((type !== null) && !generatedTypes.contains(type)) {
-                    generatedTypes.add(type);
+                if (type !== null) {
+                    genCtx.get(module).addTypedefType(typedef.path, type)
                 }
             }
         }
-        return generatedTypes;
     }
-
-    /**
-     * Converts all <b>containers</b> of the module to the list of
-     * <code>Type</code> objects.
-     *
-     * @param module
-     *            module from which is obtained DataNodeIterator to iterate over
-     *            all containers
-     * @return list of <code>Type</code> which are generated from containers
-     *         (objects of type <code>ContainerSchemaNode</code>)
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if the module equals null</li>
-     *             <li>if the name of module equals null</li>
-     *             <li>if the set of child nodes equals null</li>
-     *             </ul>
-     *
-     */
-    private def List<Type> allContainersToGenTypes(Module module) {
-        checkArgument(module !== null, "Module reference cannot be NULL.");
-
-        checkArgument(module.name !== null, "Module name cannot be NULL.");
-
-        if(module.childNodes === null) {
-            throw new IllegalArgumentException(
-                "Reference to Set of Child Nodes in module " + module.name + " cannot be NULL.");
+\r
+    private def GeneratedTypeBuilder processDataSchemaNode(Module module, String basePackageName,
+        GeneratedTypeBuilder parent, GeneratedTypeBuilder childOf, DataSchemaNode node) {
+        if (node.augmenting || node.addedByUses) {
+            return null
         }
-
-        val List<Type> generatedTypes = new ArrayList();
-        val it = new DataNodeIterator(module);
-        val List<ContainerSchemaNode> schemaContainers = it.allContainers();
-        val basePackageName = moduleNamespaceToPackageName(module);
-        for (container : schemaContainers) {
-            if(!container.isAddedByUses()) {
-                generatedTypes.add(containerToGenType(basePackageName, container));
-            }
+        val packageName = packageNameForGeneratedType(basePackageName, node.path)
+        val genType = addDefaultInterfaceDefinition(packageName, node, childOf)
+        if (node instanceof DataNodeContainer) {
+            genCtx.get(module).addChildNodeType(node.path, genType)
+            groupingsToGenTypes(module, (node as DataNodeContainer).groupings)
+            processUsesAugments(node as DataNodeContainer, module)
         }
-        return generatedTypes;
+        return genType
     }
 
-    /**
-     * Converts all <b>lists</b> of the module to the list of <code>Type</code>
-     * objects.
-     *
-     * @param module
-     *            module from which is obtained DataNodeIterator to iterate over
-     *            all lists
-     * @return list of <code>Type</code> which are generated from lists (objects
-     *         of type <code>ListSchemaNode</code>)
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if the module equals null</li>
-     *             <li>if the name of module equals null</li>
-     *             <li>if the set of child nodes equals null</li>
-     *             </ul>
-     *
-     */
-    private def List<Type> allListsToGenTypes(Module module) {
-        checkArgument(module !== null, "Module reference cannot be NULL.");
-        checkArgument(module.name !== null, "Module name cannot be NULL.");
-
-        if(module.childNodes === null) {
-            throw new IllegalArgumentException(
-                "Reference to Set of Child Nodes in module " + module.name + " cannot be NULL.");
+    private def void containerToGenType(Module module, String basePackageName, GeneratedTypeBuilder parent,
+        GeneratedTypeBuilder childOf, ContainerSchemaNode node) {
+        val genType = processDataSchemaNode(module, basePackageName, parent, childOf, node)
+        if (genType != null) {
+            constructGetter(parent, node.QName.localName, node.description, genType)
+            resolveDataSchemaNodes(module, basePackageName, genType, genType, node.childNodes)
         }
+    }
 
-        val List<Type> generatedTypes = new ArrayList();
-        val it = new DataNodeIterator(module);
-        val List<ListSchemaNode> schemaLists = it.allLists();
-        val basePackageName = moduleNamespaceToPackageName(module);
-        if(schemaLists !== null) {
-            for (list : schemaLists) {
-                if(!list.isAddedByUses()) {
-                    generatedTypes.addAll(listToGenType(basePackageName, list));
+    private def void listToGenType(Module module, String basePackageName, GeneratedTypeBuilder parent,
+        GeneratedTypeBuilder childOf, ListSchemaNode node) {
+        val genType = processDataSchemaNode(module, basePackageName, parent, childOf, node)
+        if (genType != null) {
+            constructGetter(parent, node.QName.localName, node.description, Types.listTypeFor(genType))
+
+            val List<String> listKeys = listKeys(node);
+            val packageName = packageNameForGeneratedType(basePackageName, (node).path)
+            val genTOBuilder = resolveListKeyTOBuilder(packageName, node);
+            if (genTOBuilder !== null) {
+                val identifierMarker = IDENTIFIER.parameterizedTypeFor(genType);
+                val identifiableMarker = IDENTIFIABLE.parameterizedTypeFor(genTOBuilder);
+                genTOBuilder.addImplementsType(identifierMarker);
+                genType.addImplementsType(identifiableMarker);
+            }
+
+            for (schemaNode : node.childNodes) {
+                if (!schemaNode.augmenting) {
+                    addSchemaNodeToListBuilders(basePackageName, schemaNode, genType, genTOBuilder, listKeys, module);
                 }
             }
+
+            // serialVersionUID\r
+            if (genTOBuilder !== null) {
+                val GeneratedPropertyBuilder prop = new GeneratedPropertyBuilderImpl("serialVersionUID");
+                prop.setValue(Long.toString(computeDefaultSUID(genTOBuilder as GeneratedTOBuilderImpl)));
+                genTOBuilder.setSUID(prop);
+            }
+
+            typeBuildersToGenTypes(module, genType, genTOBuilder);
         }
-        return generatedTypes;
     }
 
-    /**
-     * Converts all <b>choices</b> of the module to the list of
-     * <code>Type</code> objects.
-     *
-     * @param module
-     *            module from which is obtained DataNodeIterator to iterate over
-     *            all choices
-     * @return list of <code>Type</code> which are generated from choices
-     *         (objects of type <code>ChoiceNode</code>)
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if the module equals null</li>
-     *             <li>if the name of module equals null</li> *
-     *             </ul>
-     *
-     */
-    private def List<GeneratedType> allChoicesToGenTypes(Module module) {
-        checkArgument(module !== null, "Module reference cannot be NULL.");
-        checkArgument(module.name !== null, "Module name cannot be NULL.");
-
-        val it = new DataNodeIterator(module);
-        val choiceNodes = it.allChoices();
+    private def void processUsesAugments(DataNodeContainer node, Module module) {
         val basePackageName = moduleNamespaceToPackageName(module);
-
-        val List<GeneratedType> generatedTypes = new ArrayList();
-        for (choice : choiceNodes) {
-            if((choice !== null) && !choice.isAddedByUses()) {
-                generatedTypes.addAll(choiceToGeneratedType(basePackageName, choice));
+        for (usesNode : node.uses) {
+            for (augment : usesNode.augmentations) {
+                augmentationToGenTypes(basePackageName, augment, module, usesNode);
+                processUsesAugments(augment, module);
             }
         }
-        return generatedTypes;
     }
 
-    /**
-     * Converts all <b>augmentation</b> of the module to the list
-     * <code>Type</code> objects.
-     *
-     * @param module
-     *            module from which is obtained list of all augmentation objects
-     *            to iterate over them
-     * @return list of <code>Type</code> which are generated from augments
-     *         (objects of type <code>AugmentationSchema</code>)
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if the module equals null</li>
-     *             <li>if the name of module equals null</li>
-     *             <li>if the set of child nodes equals null</li>
-     *             </ul>
-     *
+    /**\r
+     * Converts all <b>augmentation</b> of the module to the list\r
+     * <code>Type</code> objects.\r
+     *\r
+     * @param module\r
+     *            module from which is obtained list of all augmentation objects\r
+     *            to iterate over them\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if the module equals null</li>\r
+     *             <li>if the name of module equals null</li>\r
+     *             <li>if the set of child nodes equals null</li>\r
+     *             </ul>\r
+     *\r
      */
-    private def List<Type> allAugmentsToGenTypes(Module module) {
+    private def void allAugmentsToGenTypes(Module module) {
         checkArgument(module !== null, "Module reference cannot be NULL.");
         checkArgument(module.name !== null, "Module name cannot be NULL.");
-        if(module.childNodes === null) {
+        if (module.childNodes === null) {
             throw new IllegalArgumentException(
                 "Reference to Set of Augmentation Definitions in module " + module.name + " cannot be NULL.");
         }
 
-        val List<Type> generatedTypes = new ArrayList();
         val basePackageName = moduleNamespaceToPackageName(module);
         val List<AugmentationSchema> augmentations = resolveAugmentations(module);
         for (augment : augmentations) {
-            generatedTypes.addAll(augmentationToGenTypes(basePackageName, augment));
+            augmentationToGenTypes(basePackageName, augment, module, null);
         }
-        return generatedTypes;
     }
 
-    /**
-     * Returns list of <code>AugmentationSchema</code> objects. The objects are
-     * sorted according to the length of their target path from the shortest to
-     * the longest.
-     *
-     * @param module
-     *            module from which is obtained list of all augmentation objects
-     * @return list of sorted <code>AugmentationSchema</code> objects obtained
-     *         from <code>module</code>
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if the module equals null</li>
-     *             <li>if the set of augmentation equals null</li>
-     *             </ul>
-     *
+    /**\r
+     * Returns list of <code>AugmentationSchema</code> objects. The objects are\r
+     * sorted according to the length of their target path from the shortest to\r
+     * the longest.\r
+     *\r
+     * @param module\r
+     *            module from which is obtained list of all augmentation objects\r
+     * @return list of sorted <code>AugmentationSchema</code> objects obtained\r
+     *         from <code>module</code>\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if the module equals null</li>\r
+     *             <li>if the set of augmentation equals null</li>\r
+     *             </ul>\r
+     *\r
      */
     private def List<AugmentationSchema> resolveAugmentations(Module module) {
         checkArgument(module !== null, "Module reference cannot be NULL.");
@@ -404,9 +360,9 @@ public class BindingGeneratorImpl implements BindingGenerator {
         val List<AugmentationSchema> sortedAugmentations = new ArrayList(augmentations);
         Collections.sort(sortedAugmentations,
             [ augSchema1, augSchema2 |
-                if(augSchema1.targetPath.path.size() > augSchema2.targetPath.path.size()) {
+                if (augSchema1.targetPath.path.size() > augSchema2.targetPath.path.size()) {
                     return 1;
-                } else if(augSchema1.targetPath.path.size() < augSchema2.targetPath.path.size()) {
+                } else if (augSchema1.targetPath.path.size() < augSchema2.targetPath.path.size()) {
                     return -1;
                 }
                 return 0;
@@ -414,510 +370,605 @@ public class BindingGeneratorImpl implements BindingGenerator {
         return sortedAugmentations;
     }
 
-    /**
-     * Converts whole <b>module</b> to <code>GeneratedType</code> object.
-     * Firstly is created the module builder object from which is vally
-     * obtained reference to <code>GeneratedType</code> object.
-     *
-     * @param module
-     *            module from which are obtained the module name, child nodes,
-     *            uses and is derived package name
-     * @return <code>GeneratedType</code> which is internal representation of
-     *         the module
-     * @throws IllegalArgumentException
-     *             if the module equals null
-     *
+    /**\r
+     * Converts whole <b>module</b> to <code>GeneratedType</code> object.\r
+     * Firstly is created the module builder object from which is vally\r
+     * obtained reference to <code>GeneratedType</code> object.\r
+     *\r
+     * @param module\r
+     *            module from which are obtained the module name, child nodes,\r
+     *            uses and is derived package name\r
+     * @return <code>GeneratedType</code> which is internal representation of\r
+     *         the module\r
+     * @throws IllegalArgumentException\r
+     *             if the module equals null\r
+     *\r
      */
-    private def GeneratedType moduleToDataType(Module module) {
+    private def GeneratedTypeBuilder moduleToDataType(Module module) {
         checkArgument(module !== null, "Module reference cannot be NULL.");
 
         val moduleDataTypeBuilder = moduleTypeBuilder(module, "Data");
         addImplementedInterfaceFromUses(module, moduleDataTypeBuilder);
         moduleDataTypeBuilder.addImplementsType(DATA_ROOT);
-
-        val basePackageName = moduleNamespaceToPackageName(module);
-        if(moduleDataTypeBuilder !== null) {
-            val Set<DataSchemaNode> dataNodes = module.childNodes;
-            resolveDataSchemaNodes(basePackageName, moduleDataTypeBuilder, dataNodes);
-        }
-        return moduleDataTypeBuilder.toInstance();
+        return moduleDataTypeBuilder;
     }
 
-    /**
-     * Converts all <b>rpcs</b> inputs and outputs substatements of the module
-     * to the list of <code>Type</code> objects. In addition are to containers
-     * and lists which belong to input or output also part of returning list.
-     *
-     * @param module
-     *            module from which is obtained set of all rpc objects to
-     *            iterate over them
-     * @return list of <code>Type</code> which are generated from rpcs inputs,
-     *         outputs + container and lists which are part of inputs or outputs
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if the module equals null</li>
-     *             <li>if the name of module equals null</li>
-     *             <li>if the set of child nodes equals null</li>
-     *             </ul>
-     *
+    /**\r
+     * Converts all <b>rpcs</b> inputs and outputs substatements of the module\r
+     * to the list of <code>Type</code> objects. In addition are to containers\r
+     * and lists which belong to input or output also part of returning list.\r
+     *\r
+     * @param module\r
+     *            module from which is obtained set of all rpc objects to\r
+     *            iterate over them\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if the module equals null</li>\r
+     *             <li>if the name of module equals null</li>\r
+     *             <li>if the set of child nodes equals null</li>\r
+     *             </ul>\r
+     *\r
      */
-    private def List<Type> allRPCMethodsToGenType(Module module) {
+    private def void rpcMethodsToGenType(Module module) {
         checkArgument(module !== null, "Module reference cannot be NULL.");
-
         checkArgument(module.name !== null, "Module name cannot be NULL.");
-
-        if(module.childNodes === null) {
-            throw new IllegalArgumentException(
-                "Reference to Set of RPC Method Definitions in module " + module.name + " cannot be NULL.");
-        }
+        checkArgument(module.childNodes !== null,
+            "Reference to Set of RPC Method Definitions in module " + module.name + " cannot be NULL.");
 
         val basePackageName = moduleNamespaceToPackageName(module);
         val Set<RpcDefinition> rpcDefinitions = module.rpcs;
-
-        if(rpcDefinitions.isEmpty()) {
-            return Collections.emptyList();
+        if (rpcDefinitions.isEmpty()) {
+            return;
         }
 
-        val List<Type> genRPCTypes = new ArrayList();
         val interfaceBuilder = moduleTypeBuilder(module, "Service");
         interfaceBuilder.addImplementsType(Types.typeForClass(RpcService));
         for (rpc : rpcDefinitions) {
-            if(rpc !== null) {
-
+            if (rpc !== null) {
                 val rpcName = parseToClassName(rpc.QName.localName);
                 val rpcMethodName = parseToValidParamName(rpcName);
                 val method = interfaceBuilder.addMethod(rpcMethodName);
-
-                val rpcInOut = new ArrayList();
-
                 val input = rpc.input;
                 val output = rpc.output;
 
-                if(input !== null) {
-                    rpcInOut.add(new DataNodeIterator(input));
+                if (input !== null) {
                     val inType = addRawInterfaceDefinition(basePackageName, input, rpcName);
                     addImplementedInterfaceFromUses(input, inType);
                     inType.addImplementsType(DATA_OBJECT);
                     inType.addImplementsType(augmentable(inType));
-                    resolveDataSchemaNodes(basePackageName, inType, input.childNodes);
+                    resolveDataSchemaNodes(module, basePackageName, inType, inType, input.childNodes);
+                    genCtx.get(module).addChildNodeType(input.path, inType)
                     val inTypeInstance = inType.toInstance();
-                    genRPCTypes.add(inTypeInstance);
                     method.addParameter(inTypeInstance, "input");
                 }
 
                 var Type outTypeInstance = VOID;
-                if(output !== null) {
-                    rpcInOut.add(new DataNodeIterator(output));
+                if (output !== null) {
                     val outType = addRawInterfaceDefinition(basePackageName, output, rpcName);
                     addImplementedInterfaceFromUses(output, outType);
                     outType.addImplementsType(DATA_OBJECT);
                     outType.addImplementsType(augmentable(outType));
-
-                    resolveDataSchemaNodes(basePackageName, outType, output.childNodes);
+                    resolveDataSchemaNodes(module, basePackageName, outType, outType, output.childNodes);
+                    genCtx.get(module).addChildNodeType(output.path, outType)
                     outTypeInstance = outType.toInstance();
-                    genRPCTypes.add(outTypeInstance);
-
                 }
 
                 val rpcRes = Types.parameterizedTypeFor(Types.typeForClass(RpcResult), outTypeInstance);
                 method.setReturnType(Types.parameterizedTypeFor(FUTURE, rpcRes));
-                for (iter : rpcInOut) {
-                    val List<ContainerSchemaNode> nContainers = iter.allContainers();
-                    if((nContainers !== null) && !nContainers.isEmpty()) {
-                        for (container : nContainers) {
-                            if(!container.isAddedByUses()) {
-                                genRPCTypes.add(containerToGenType(basePackageName, container));
-                            }
-                        }
-                    }
-                    val List<ListSchemaNode> nLists = iter.allLists();
-                    if((nLists !== null) && !nLists.isEmpty()) {
-                        for (list : nLists) {
-                            if(!list.isAddedByUses()) {
-                                genRPCTypes.addAll(listToGenType(basePackageName, list));
-                            }
-                        }
-                    }
-                }
             }
         }
-        genRPCTypes.add(interfaceBuilder.toInstance());
-        return genRPCTypes;
+
+        genCtx.get(module).addTopLevelNodeType(interfaceBuilder)
     }
 
-    /**
-     * Converts all <b>notifications</b> of the module to the list of
-     * <code>Type</code> objects. In addition are to this list added containers
-     * and lists which are part of this notification.
-     *
-     * @param module
-     *            module from which is obtained set of all notification objects
-     *            to iterate over them
-     * @return list of <code>Type</code> which are generated from notification
-     *         (object of type <code>NotificationDefinition</code>
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if the module equals null</li>
-     *             <li>if the name of module equals null</li>
-     *             <li>if the set of child nodes equals null</li>
-     *             </ul>
-     *
+    /**\r
+     * Converts all <b>notifications</b> of the module to the list of\r
+     * <code>Type</code> objects. In addition are to this list added containers\r
+     * and lists which are part of this notification.\r
+     *\r
+     * @param module\r
+     *            module from which is obtained set of all notification objects\r
+     *            to iterate over them\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if the module equals null</li>\r
+     *             <li>if the name of module equals null</li>\r
+     *             <li>if the set of child nodes equals null</li>\r
+     *             </ul>\r
+     *\r
      */
-    private def List<Type> allNotificationsToGenType(Module module) {
+    private def void notificationsToGenType(Module module) {
         checkArgument(module !== null, "Module reference cannot be NULL.");
-
         checkArgument(module.name !== null, "Module name cannot be NULL.");
 
-        if(module.childNodes === null) {
+        if (module.childNodes === null) {
             throw new IllegalArgumentException(
                 "Reference to Set of Notification Definitions in module " + module.name + " cannot be NULL.");
         }
         val notifications = module.notifications;
-        if(notifications.isEmpty()) return Collections.emptyList();
+        if(notifications.empty) return;
 
         val listenerInterface = moduleTypeBuilder(module, "Listener");
         listenerInterface.addImplementsType(BindingTypes.NOTIFICATION_LISTENER);
-
         val basePackageName = moduleNamespaceToPackageName(module);
-        val List<Type> generatedTypes = new ArrayList();
 
         for (notification : notifications) {
-            if(notification !== null) {
-                val iter = new DataNodeIterator(notification);
+            if (notification !== null) {
+                processUsesAugments(notification, module);
 
-                // Containers
-                for (node : iter.allContainers()) {
-                    if(!node.isAddedByUses()) {
-                        generatedTypes.add(containerToGenType(basePackageName, node));
-                    }
-                }
-
-                // Lists
-                for (node : iter.allLists()) {
-                    if(!node.isAddedByUses()) {
-                        generatedTypes.addAll(listToGenType(basePackageName, node));
-                    }
-                }
-                val notificationInterface = addDefaultInterfaceDefinition(basePackageName, notification);
+                val notificationInterface = addDefaultInterfaceDefinition(basePackageName, notification,
+                    BindingTypes.DATA_OBJECT);
                 notificationInterface.addImplementsType(NOTIFICATION);
+                genCtx.get(module).addChildNodeType(notification.path, notificationInterface)
 
-                // Notification object
-                resolveDataSchemaNodes(basePackageName, notificationInterface, notification.childNodes);
+                // Notification object\r
+                resolveDataSchemaNodes(module, basePackageName, notificationInterface, notificationInterface,
+                    notification.childNodes);
 
-                listenerInterface.addMethod("on" + notificationInterface.name) //
+                listenerInterface.addMethod("on" + notificationInterface.name) //\r
                 .setAccessModifier(AccessModifier.PUBLIC).addParameter(notificationInterface, "notification").
                     setReturnType(Types.VOID);
-
-                generatedTypes.add(notificationInterface.toInstance());
             }
         }
-        generatedTypes.add(listenerInterface.toInstance());
-        return generatedTypes;
+
+        genCtx.get(module).addTopLevelNodeType(listenerInterface)
     }
 
-    /**
-     * Converts all <b>identities</b> of the module to the list of
-     * <code>Type</code> objects.
-     *
-     * @param module
-     *            module from which is obtained set of all identity objects to
-     *            iterate over them
-     * @param context
-     *            schema context only used as input parameter for method
-     *            {@link identityToGenType}
-     * @return list of <code>Type</code> which are generated from identities
-     *         (object of type <code>IdentitySchemaNode</code>
-     *
+    /**\r
+     * Converts all <b>identities</b> of the module to the list of\r
+     * <code>Type</code> objects.\r
+     *\r
+     * @param module\r
+     *            module from which is obtained set of all identity objects to\r
+     *            iterate over them\r
+     * @param context\r
+     *            schema context only used as input parameter for method\r
+     *            {@link identityToGenType}\r
+     *\r
      */
-    private def List<Type> allIdentitiesToGenTypes(Module module, SchemaContext context) {
-        val List<Type> genTypes = new ArrayList();
-
+    private def void allIdentitiesToGenTypes(Module module, SchemaContext context) {
         val Set<IdentitySchemaNode> schemaIdentities = module.identities;
-
         val basePackageName = moduleNamespaceToPackageName(module);
 
-        if(schemaIdentities !== null && !schemaIdentities.isEmpty()) {
+        if (schemaIdentities !== null && !schemaIdentities.isEmpty()) {
             for (identity : schemaIdentities) {
-                genTypes.add(identityToGenType(basePackageName, identity, context));
+                identityToGenType(module, basePackageName, identity, context);
             }
         }
-        return genTypes;
     }
 
-    /**
-     * Converts the <b>identity</b> object to GeneratedType. Firstly it is
-     * created transport object builder. If identity contains base identity then
-     * reference to base identity is added to superior identity as its extend.
-     * If identity doesn't contain base identity then only reference to abstract
-     * class {@link org.opendaylight.yangtools.yang.model.api.BaseIdentity
-     * BaseIdentity} is added
-     *
-     * @param basePackageName
-     *            string contains the module package name
-     * @param identity
-     *            IdentitySchemaNode which contains data about identity
-     * @param context
-     *            SchemaContext which is used to get package and name
-     *            information about base of identity
-     *
-     * @return GeneratedType which is generated from identity (object of type
-     *         <code>IdentitySchemaNode</code>
-     *
+    /**\r
+     * Converts the <b>identity</b> object to GeneratedType. Firstly it is\r
+     * created transport object builder. If identity contains base identity then\r
+     * reference to base identity is added to superior identity as its extend.\r
+     * If identity doesn't contain base identity then only reference to abstract\r
+     * class {@link org.opendaylight.yangtools.yang.model.api.BaseIdentity\r
+     * BaseIdentity} is added\r
+     *\r
+     * @param module current module\r
+     * @param basePackageName\r
+     *            string contains the module package name\r
+     * @param identity\r
+     *            IdentitySchemaNode which contains data about identity\r
+     * @param context\r
+     *            SchemaContext which is used to get package and name\r
+     *            information about base of identity\r
+     *\r
      */
-    private def GeneratedType identityToGenType(String basePackageName, IdentitySchemaNode identity,
+    private def void identityToGenType(Module module, String basePackageName, IdentitySchemaNode identity,
         SchemaContext context) {
-        if(identity === null) {
-            return null;
+        if (identity === null) {
+            return;
         }
-
         val packageName = packageNameForGeneratedType(basePackageName, identity.path);
         val genTypeName = parseToClassName(identity.QName.localName);
         val newType = new GeneratedTOBuilderImpl(packageName, genTypeName);
-
         val baseIdentity = identity.baseIdentity;
-        if(baseIdentity !== null) {
+        if (baseIdentity === null) {
+            newType.setExtendsType(Types.baseIdentityTO);
+        } else {
             val baseIdentityParentModule = SchemaContextUtil.findParentModule(context, baseIdentity);
-
             val returnTypePkgName = moduleNamespaceToPackageName(baseIdentityParentModule);
             val returnTypeName = parseToClassName(baseIdentity.QName.localName);
-
             val gto = new GeneratedTOBuilderImpl(returnTypePkgName, returnTypeName).toInstance();
             newType.setExtendsType(gto);
-        } else {
-            newType.setExtendsType(Types.baseIdentityTO);
         }
         newType.setAbstract(true);
-        return newType.toInstance();
+        genCtx.get(module).addIdentityType(newType)
     }
 
-    /**
-     * Converts all <b>groupings</b> of the module to the list of
-     * <code>Type</code> objects. Firstly are groupings sorted according mutual
-     * dependencies. At least dependend (indepedent) groupings are in the list
-     * saved at first positions. For every grouping the record is added to map
-     * {@link BindingGeneratorImpl#allGroupings allGroupings}
-     *
-     * @param module
-     *            module from which is obtained set of all grouping objects to
-     *            iterate over them
-     * @return list of <code>Type</code> which are generated from groupings
-     *         (object of type <code>GroupingDefinition</code>)
-     *
+    /**\r
+     * Converts all <b>groupings</b> of the module to the list of\r
+     * <code>Type</code> objects. Firstly are groupings sorted according mutual\r
+     * dependencies. At least dependend (indepedent) groupings are in the list\r
+     * saved at first positions. For every grouping the record is added to map\r
+     * {@link BindingGeneratorImpl#allGroupings allGroupings}\r
+     *\r
+     * @param module\r
+     *            current module\r
+     * @param collection of groupings from which types will be generated\r
+     *\r
      */
-    private def List<Type> allGroupingsToGenTypes(Module module) {
-        checkArgument(module !== null, "Module parameter can not be null");
-        val List<Type> genTypes = new ArrayList();
+    private def void groupingsToGenTypes(Module module, Collection<GroupingDefinition> groupings) {
         val basePackageName = moduleNamespaceToPackageName(module);
-        val Set<GroupingDefinition> groupings = module.groupings;
         val List<GroupingDefinition> groupingsSortedByDependencies = new GroupingDefinitionDependencySort().sort(
             groupings);
-
         for (grouping : groupingsSortedByDependencies) {
-            val genType = groupingToGenType(basePackageName, grouping);
-            genTypes.add(genType);
-            val schemaPath = grouping.path;
-            allGroupings.put(schemaPath, genType);
+            groupingToGenType(basePackageName, grouping, module);
         }
-        return genTypes;
     }
 
-    /**
-     * Converts individual grouping to GeneratedType. Firstly generated type
-     * builder is created and every child node of grouping is resolved to the
-     * method.
-     *
-     * @param basePackageName
-     *            string contains the module package name
-     * @param grouping
-     *            GroupingDefinition which contains data about grouping
-     * @return GeneratedType which is generated from grouping (object of type
-     *         <code>GroupingDefinition</code>)
+    /**\r
+     * Converts individual grouping to GeneratedType. Firstly generated type\r
+     * builder is created and every child node of grouping is resolved to the\r
+     * method.\r
+     *\r
+     * @param basePackageName\r
+     *            string contains the module package name\r
+     * @param grouping\r
+     *            GroupingDefinition which contains data about grouping\r
+     * @param module current module\r
+     * @return GeneratedType which is generated from grouping (object of type\r
+     *         <code>GroupingDefinition</code>)\r
      */
-    private def GeneratedType groupingToGenType(String basePackageName, GroupingDefinition grouping) {
-        if(grouping === null) {
-            return null;
-        }
-
+    private def void groupingToGenType(String basePackageName, GroupingDefinition grouping, Module module) {
         val packageName = packageNameForGeneratedType(basePackageName, grouping.path);
-        val Set<DataSchemaNode> schemaNodes = grouping.childNodes;
-        val typeBuilder = addDefaultInterfaceDefinition(packageName, grouping);
-
-        resolveDataSchemaNodes(basePackageName, typeBuilder, schemaNodes);
-        return typeBuilder.toInstance();
+        val genType = addDefaultInterfaceDefinition(packageName, grouping);
+        genCtx.get(module).addGroupingType(grouping.path, genType)
+        resolveDataSchemaNodes(module, basePackageName, genType, genType, grouping.childNodes);
+        groupingsToGenTypes(module, grouping.groupings);
+        processUsesAugments(grouping, module);
     }
 
-    /**
-     * Tries to find EnumTypeDefinition in <code>typeDefinition</code>. If base
-     * type of <code>typeDefinition</code> is of the type ExtendedType then this
-     * method is recursivelly called with this base type.
-     *
-     * @param typeDefinition
-     *            TypeDefinition in which should be EnumTypeDefinition found as
-     *            base type
-     * @return EnumTypeDefinition if it is found inside
-     *         <code>typeDefinition</code> or <code>null</code> in other case
+    /**\r
+     * Tries to find EnumTypeDefinition in <code>typeDefinition</code>. If base\r
+     * type of <code>typeDefinition</code> is of the type ExtendedType then this\r
+     * method is recursivelly called with this base type.\r
+     *\r
+     * @param typeDefinition\r
+     *            TypeDefinition in which should be EnumTypeDefinition found as\r
+     *            base type\r
+     * @return EnumTypeDefinition if it is found inside\r
+     *         <code>typeDefinition</code> or <code>null</code> in other case\r
      */
     private def EnumTypeDefinition enumTypeDefFromExtendedType(TypeDefinition<?> typeDefinition) {
-        if(typeDefinition !== null) {
-            if(typeDefinition.baseType instanceof EnumTypeDefinition) {
+        if (typeDefinition !== null) {
+            if (typeDefinition.baseType instanceof EnumTypeDefinition) {
                 return typeDefinition.baseType as EnumTypeDefinition;
-            } else if(typeDefinition.baseType instanceof ExtendedType) {
+            } else if (typeDefinition.baseType instanceof ExtendedType) {
                 return enumTypeDefFromExtendedType(typeDefinition.baseType);
             }
         }
         return null;
     }
 
-    /**
-     * Adds enumeration builder created from <code>enumTypeDef</code> to
-     * <code>typeBuilder</code>.
-     *
-     * Each <code>enumTypeDef</code> item is added to builder with its name and
-     * value.
-     *
-     * @param enumTypeDef
-     *            EnumTypeDefinition contains enum data
-     * @param enumName
-     *            string contains name which will be assigned to enumeration
-     *            builder
-     * @param typeBuilder
-     *            GeneratedTypeBuilder to which will be enum builder assigned
-     * @return enumeration builder which contais data from
-     *         <code>enumTypeDef</code>
+    /**\r
+     * Adds enumeration builder created from <code>enumTypeDef</code> to\r
+     * <code>typeBuilder</code>.\r
+     *\r
+     * Each <code>enumTypeDef</code> item is added to builder with its name and\r
+     * value.\r
+     *\r
+     * @param enumTypeDef\r
+     *            EnumTypeDefinition contains enum data\r
+     * @param enumName\r
+     *            string contains name which will be assigned to enumeration\r
+     *            builder\r
+     * @param typeBuilder\r
+     *            GeneratedTypeBuilder to which will be enum builder assigned\r
+     * @return enumeration builder which contais data from\r
+     *         <code>enumTypeDef</code>\r
      */
     private def EnumBuilder resolveInnerEnumFromTypeDefinition(EnumTypeDefinition enumTypeDef, String enumName,
         GeneratedTypeBuilder typeBuilder) {
-        if((enumTypeDef !== null) && (typeBuilder !== null) && (enumTypeDef.QName !== null) &&
+        if ((enumTypeDef !== null) && (typeBuilder !== null) && (enumTypeDef.QName !== null) &&
             (enumTypeDef.QName.localName !== null)) {
-
             val enumerationName = parseToClassName(enumName);
             val enumBuilder = typeBuilder.addEnumeration(enumerationName);
             enumBuilder.updateEnumPairsFromEnumTypeDef(enumTypeDef);
-
             return enumBuilder;
         }
         return null;
     }
 
-    /**
-     * Generates type builder for <code>module</code>.
-     *
-     * @param module
-     *            Module which is source of package name for generated type
-     *            builder
-     * @param postfix
-     *            string which is added to the module class name representation
-     *            as suffix
-     * @return instance of GeneratedTypeBuilder which represents
-     *         <code>module</code>.
-     * @throws IllegalArgumentException
-     *             if <code>module</code> equals null
+    /**\r
+     * Generates type builder for <code>module</code>.\r
+     *\r
+     * @param module\r
+     *            Module which is source of package name for generated type\r
+     *            builder\r
+     * @param postfix\r
+     *            string which is added to the module class name representation\r
+     *            as suffix\r
+     * @return instance of GeneratedTypeBuilder which represents\r
+     *         <code>module</code>.\r
+     * @throws IllegalArgumentException\r
+     *             if <code>module</code> equals null\r
      */
     private def GeneratedTypeBuilder moduleTypeBuilder(Module module, String postfix) {
         checkArgument(module !== null, "Module reference cannot be NULL.");
         val packageName = moduleNamespaceToPackageName(module);
         val moduleName = parseToClassName(module.name) + postfix;
-
         return new GeneratedTypeBuilderImpl(packageName, moduleName);
-
     }
 
-    /**
-     * Converts <code>augSchema</code> to list of <code>Type</code> which
-     * contains generated type for augmentation. In addition there are also
-     * generated types for all containers, list and choices which are child of
-     * <code>augSchema</code> node or a generated types for cases are added if
-     * augmented node is choice.
-     *
-     * @param augmentPackageName
-     *            string with the name of the package to which the augmentation
-     *            belongs
-     * @param augSchema
-     *            AugmentationSchema which is contains data about agumentation
-     *            (target path, childs...)
-     * @return list of <code>Type</code> objects which contains generated type
-     *         for augmentation and for container, list and choice child nodes
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if <code>augmentPackageName</code> equals null</li>
-     *             <li>if <code>augSchema</code> equals null</li>
-     *             <li>if target path of <code>augSchema</code> equals null</li>
-     *             </ul>
+    /**\r
+     * Converts <code>augSchema</code> to list of <code>Type</code> which\r
+     * contains generated type for augmentation. In addition there are also\r
+     * generated types for all containers, list and choices which are child of\r
+     * <code>augSchema</code> node or a generated types for cases are added if\r
+     * augmented node is choice.\r
+     *\r
+     * @param augmentPackageName\r
+     *            string with the name of the package to which the augmentation\r
+     *            belongs\r
+     * @param augSchema\r
+     *            AugmentationSchema which is contains data about agumentation\r
+     *            (target path, childs...)\r
+     * @param module current module\r
+     * @param parentUsesNode parent uses node of this augment (can be null if this augment is not defined under uses statement)\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if <code>augmentPackageName</code> equals null</li>\r
+     *             <li>if <code>augSchema</code> equals null</li>\r
+     *             <li>if target path of <code>augSchema</code> equals null</li>\r
+     *             </ul>\r
      */
-    private def List<Type> augmentationToGenTypes(String augmentPackageName, AugmentationSchema augSchema) {
+    private def void 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();
 
-        // EVERY augmented interface will extends Augmentation<T> interface
-        // and DataObject interface!!!
+        processUsesAugments(augSchema, module);
+
+        // EVERY augmented interface will extends Augmentation<T> interface\r
+        // and DataObject interface\r
         val targetPath = augSchema.targetPath;
-        val targetSchemaNode = findDataSchemaNode(schemaContext, targetPath);
-        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
-                // and have assigned Java Types,
-                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);
+        var targetSchemaNode = findDataSchemaNode(schemaContext, targetPath);
+        if (targetSchemaNode instanceof DataSchemaNode && (targetSchemaNode as DataSchemaNode).isAddedByUses()) {
+            if (parentUsesNode == null) {
+                targetSchemaNode = findOriginal(targetSchemaNode as DataSchemaNode);
+            } else {
+                targetSchemaNode = findOriginalTargetFromGrouping(targetSchemaNode.QName.localName, parentUsesNode);
+            }
+            if (targetSchemaNode == null) {
+                throw new NullPointerException(
+                    "Failed to find target node from grouping for augmentation " + augSchema + " in module " +
+                        module.name);
+            }
+        }
+
+        if (targetSchemaNode !== null) {
+            var targetTypeBuilder = findChildNodeByPath(targetSchemaNode.path)
+            if (targetTypeBuilder === null) {
+                targetTypeBuilder = findCaseByPath(targetSchemaNode.path)
+            }
+            if (targetTypeBuilder === null) {
+                throw new NullPointerException("Target type not yet generated: " + targetSchemaNode);
+            }
+            if (!(targetSchemaNode instanceof ChoiceNode)) {
+                var packageName = augmentPackageName;
+                if (parentUsesNode != null) {
+                    packageName = packageNameForGeneratedType(augmentPackageName, augSchema.targetPath);
+                }
+                val augTypeBuilder = addRawAugmentGenTypeDefinition(module, packageName, augmentPackageName,
+                    targetTypeBuilder.toInstance, augSchema);
+                genCtx.get(module).addAugmentType(augTypeBuilder)\r
+                genCtx.get(module).addTypeToAugmentation(augTypeBuilder,augSchema);
+            } else {
+                generateTypesFromAugmentedChoiceCases(module, augmentPackageName, targetTypeBuilder.toInstance,
+                    targetSchemaNode as ChoiceNode, augSchema.childNodes);
+            }
+        }
+    }
+
+    /**\r
+     * Utility method which search for original node defined in grouping.\r
+     */
+    private def DataSchemaNode findOriginal(DataSchemaNode node) {
+        var DataSchemaNode result = findCorrectTargetFromGrouping(node);
+        if (result == null) {
+            result = findCorrectTargetFromAugment(node);
+            if (result != null) {
+                if (result.addedByUses) {
+                    result = findOriginal(result);
+                }
+            }
+        }
+        return result;
+    }
+
+    private def DataSchemaNode findCorrectTargetFromAugment(DataSchemaNode node) {
+        if (!node.augmenting) {
+            return null;
+        }
+
+        var String currentName = node.QName.localName;
+        var tmpPath = new ArrayList<String>();
+        var YangNode parent = node;
+        var AugmentationSchema augment = null;
+        do {
+            parent = (parent as DataSchemaNode).parent;
+            if (parent instanceof AugmentationTarget) {
+                tmpPath.add(currentName);
+                augment = findNodeInAugment((parent as AugmentationTarget).availableAugmentations, currentName);
+                if (augment == null) {
+                    currentName = (parent as DataSchemaNode).QName.localName;
+                }
+            }
+        } while ((parent as DataSchemaNode).augmenting && augment == null);
+
+        if (augment == null) {
+            return null;
+        } else {
+            Collections.reverse(tmpPath);
+            var Object actualParent = augment;
+            var DataSchemaNode result = null;
+            for (name : tmpPath) {
+                if (actualParent instanceof DataNodeContainer) {
+                    result = (actualParent as DataNodeContainer).getDataChildByName(name);
+                    actualParent = (actualParent as DataNodeContainer).getDataChildByName(name);
+                } else {
+                    if (actualParent instanceof ChoiceNode) {
+                        result = (actualParent as ChoiceNode).getCaseNodeByName(name);
+                        actualParent = (actualParent as ChoiceNode).getCaseNodeByName(name);
+                    }
+                }
+            }
+
+            if (result.addedByUses) {
+                result = findCorrectTargetFromGrouping(result);
+            }
+
+            return result;
+        }
+    }
+
+    private def AugmentationSchema findNodeInAugment(Collection<AugmentationSchema> augments, String name) {
+        for (augment : augments) {
+            if (augment.getDataChildByName(name) != null) {
+                return augment;
+            }
+        }
+        return null;
+    }
+
+    private def DataSchemaNode findCorrectTargetFromGrouping(DataSchemaNode node) {
+        if (node.path.path.size == 1) {
+
+            // uses is under module statement\r
+            val Module m = findParentModule(schemaContext, node);
+            var DataSchemaNode result = null;
+            for (u : m.uses) {
+                var SchemaNode targetGrouping = findNodeInSchemaContext(schemaContext, u.groupingPath.path);
+                if (!(targetGrouping instanceof GroupingDefinition)) {
+                    throw new IllegalArgumentException("Failed to generate code for augment in " + u);
+                }
+                var gr = targetGrouping as GroupingDefinition;
+                result = gr.getDataChildByName(node.QName.localName);
+            }
+            if (result == null) {
+                throw new IllegalArgumentException("Failed to generate code for augment");
+            }
+            return result;
+        } else {
+            var DataSchemaNode result = null;
+            var String currentName = node.QName.localName;
+            var tmpPath = new ArrayList<String>();
+            var YangNode parent = node.parent;
+            do {
+                tmpPath.add(currentName);
+                val dataNodeParent = parent as DataNodeContainer;
+                for (u : dataNodeParent.uses) {
+                    if (result == null) {\r
+                        var SchemaNode targetGrouping = findNodeInSchemaContext(schemaContext, u.groupingPath.path);
+                        if (!(targetGrouping instanceof GroupingDefinition)) {
+                            throw new IllegalArgumentException("Failed to generate code for augment in " + u);
+                        }
+                        var gr = targetGrouping as GroupingDefinition;
+                        result = gr.getDataChildByName(currentName);\r
+                    }
+                }
+                if (result == null) {
+                    currentName = (parent as SchemaNode).QName.localName;
+                    if (parent instanceof DataSchemaNode) {
+                        parent = (parent as DataSchemaNode).parent;
+                    } else {
+                        parent = (parent as DataNodeContainer).parent;
+                    }
+                }
+            } while (result == null && !(parent instanceof Module));
+
+            if (result != null) {
+                if (tmpPath.size == 1) {\r
+                    if (result != null && result.addedByUses) {\r
+                        result = findOriginal(result);\r
+                    }
+                    return result;
+                } else {
+                    var DataSchemaNode newParent = result;
+                    Collections.reverse(tmpPath);
+                    tmpPath.remove(0);
+                    for (name : tmpPath) {
+                        newParent = (newParent as DataNodeContainer).getDataChildByName(name);
+                    }\r
+                    if (newParent != null && newParent.addedByUses) {\r
+                        newParent = findOriginal(newParent);\r
+                    }
+                    return newParent;
+                }
             }
-            val augChildNodes = augSchema.childNodes;
-            if(!(targetSchemaNode instanceof ChoiceNode)) {
-                val augTypeBuilder = addRawAugmentGenTypeDefinition(augmentPackageName, targetType, augSchema);
-                val augType = augTypeBuilder.toInstance();
-                genTypes.add(augType);
+
+            return result;
+        }
+    }
+
+    /**\r
+     * Convenient method to find node added by uses statement.\r
+     */
+    private def DataSchemaNode findOriginalTargetFromGrouping(String targetSchemaNodeName, UsesNode parentUsesNode) {
+        var SchemaNode targetGrouping = findNodeInSchemaContext(schemaContext, parentUsesNode.groupingPath.path);
+        if (!(targetGrouping instanceof GroupingDefinition)) {
+            throw new IllegalArgumentException("Failed to generate code for augment in " + parentUsesNode);
+        }
+
+        var grouping = targetGrouping as GroupingDefinition;
+        var result = grouping.getDataChildByName(targetSchemaNodeName);
+        if (result == null) {
+            return null;
+        }
+        var boolean fromUses = result.addedByUses;
+
+        var Iterator<UsesNode> groupingUses = grouping.uses.iterator;
+        while (fromUses) {
+            if (groupingUses.hasNext()) {
+                grouping = findNodeInSchemaContext(schemaContext, groupingUses.next().groupingPath.path) as GroupingDefinition;
+                result = grouping.getDataChildByName(targetSchemaNodeName);
+                fromUses = result.addedByUses;
             } else {
-                val choiceTarget = targetSchemaNode as ChoiceNode;
-                val choiceCaseNodes = choiceTarget.cases;
-                genTypes.addAll(
-                    generateTypesFromAugmentedChoiceCases(augmentPackageName, targetType, choiceCaseNodes));
+                throw new NullPointerException("Failed to generate code for augment in " + parentUsesNode);
             }
-            genTypes.addAll(augmentationBodyToGenTypes(augmentPackageName, augChildNodes));
         }
-        return genTypes;
+
+        return result;
     }
 
-    /**
-     * Returns a generated type builder for an augmentation.
-     *
-     * The name of the type builder is equal to the name of augmented node with
-     * serial number as suffix.
-     *
-     * @param augmentPackageName
-     *            string with contains the package name to which the augment
-     *            belongs
-     * @param targetPackageName
-     *            string with the package name to which the augmented node
-     *            belongs
-     * @param targetSchemaNodeName
-     *            string with the name of the augmented node
-     * @param augSchema
-     *            augmentation schema which contains data about the child nodes
-     *            and uses of augment
-     * @return generated type builder for augment
+    /**\r
+     * Returns a generated type builder for an augmentation.\r
+     *\r
+     * The name of the type builder is equal to the name of augmented node with\r
+     * serial number as suffix.\r
+     *\r
+     * @param module current module\r
+     * @param augmentPackageName\r
+     *            string with contains the package name to which the augment\r
+     *            belongs\r
+     * @param basePackageName\r
+     *            string with the package name to which the augmented node\r
+     *            belongs\r
+     * @param targetTypeRef\r
+     *            target type\r
+     * @param augSchema\r
+     *            augmentation schema which contains data about the child nodes\r
+     *            and uses of augment\r
+     * @return generated type builder for augment\r
      */
-    private def GeneratedTypeBuilder addRawAugmentGenTypeDefinition(String augmentPackageName, Type targetTypeRef,
-        AugmentationSchema augSchema) {
+    private def GeneratedTypeBuilder addRawAugmentGenTypeDefinition(Module module, String augmentPackageName,
+        String basePackageName, Type targetTypeRef, AugmentationSchema augSchema) {
         var Map<String, GeneratedTypeBuilder> augmentBuilders = genTypeBuilders.get(augmentPackageName);
-        if(augmentBuilders === null) {
+        if (augmentBuilders === null) {
             augmentBuilders = new HashMap();
             genTypeBuilders.put(augmentPackageName, augmentBuilders);
         }
         val augIdentifier = getAugmentIdentifier(augSchema.unknownSchemaNodes);
 
-        val augTypeName = if(augIdentifier !== null) {
+        val augTypeName = if (augIdentifier !== null) {
                 parseToClassName(augIdentifier)
             } else {
                 augGenTypeName(augmentBuilders, targetTypeRef.name);
             }
-        val Set<DataSchemaNode> augChildNodes = augSchema.childNodes;
 
         val augTypeBuilder = new GeneratedTypeBuilderImpl(augmentPackageName, augTypeName);
 
@@ -925,20 +976,20 @@ public class BindingGeneratorImpl implements BindingGenerator {
         augTypeBuilder.addImplementsType(Types.augmentationTypeFor(targetTypeRef));
         addImplementedInterfaceFromUses(augSchema, augTypeBuilder);
 
-        augSchemaNodeToMethods(augmentPackageName, augTypeBuilder, augChildNodes);
+        augSchemaNodeToMethods(module, basePackageName, augTypeBuilder, augTypeBuilder, augSchema.childNodes);
         augmentBuilders.put(augTypeName, augTypeBuilder);
         return augTypeBuilder;
     }
 
-    /**
-     *
-     * @param unknownSchemaNodes
-     * @return nodeParameter of UnknownSchemaNode
+    /**\r
+     *\r
+     * @param unknownSchemaNodes\r
+     * @return nodeParameter of UnknownSchemaNode\r
      */
     private def String getAugmentIdentifier(List<UnknownSchemaNode> unknownSchemaNodes) {
         for (unknownSchemaNode : unknownSchemaNodes) {
             val nodeType = unknownSchemaNode.nodeType;
-            if(AUGMENT_IDENTIFIER_NAME.equals(nodeType.localName) &&
+            if (AUGMENT_IDENTIFIER_NAME.equals(nodeType.localName) &&
                 YANG_EXT_NAMESPACE.equals(nodeType.namespace.toString())) {
                 return unknownSchemaNode.nodeParameter;
             }
@@ -946,439 +997,364 @@ public class BindingGeneratorImpl implements BindingGenerator {
         return null;
     }
 
-    /**
-     * Convert a container, list and choice subnodes (and recursivelly their
-     * subnodes) of augment to generated types
-     *
-     * @param augBasePackageName
-     *            string with the augment package name
-     * @param augChildNodes
-     *            set of data schema nodes which represents child nodes of the
-     *            augment
-     *
-     * @return list of <code>Type</code> which represents container, list and
-     *         choice subnodes of augment
-     */
-    private def List<Type> augmentationBodyToGenTypes(String augBasePackageName, Set<DataSchemaNode> augChildNodes) {
-        val List<Type> genTypes = new ArrayList();
-        val List<DataNodeIterator> augSchemaIts = new ArrayList();
-        for (childNode : augChildNodes) {
-            if(childNode instanceof DataNodeContainer) {
-                augSchemaIts.add(new DataNodeIterator(childNode as DataNodeContainer));
-
-                if(childNode instanceof ContainerSchemaNode) {
-                    genTypes.add(containerToGenType(augBasePackageName, childNode as ContainerSchemaNode));
-                } else if(childNode instanceof ListSchemaNode) {
-                    genTypes.addAll(listToGenType(augBasePackageName, childNode as ListSchemaNode));
-                }
-            } else if(childNode instanceof ChoiceNode) {
-                val choice = childNode as ChoiceNode;
-                for (caseNode : choice.cases) {
-                    augSchemaIts.add(new DataNodeIterator(caseNode));
-                }
-                genTypes.addAll(choiceToGeneratedType(augBasePackageName, childNode as ChoiceNode));
-            }
-        }
-
-        for (it : augSchemaIts) {
-            val List<ContainerSchemaNode> augContainers = it.allContainers();
-            val List<ListSchemaNode> augLists = it.allLists();
-            val List<ChoiceNode> augChoices = it.allChoices();
-
-            if(augContainers !== null) {
-                for (container : augContainers) {
-                    genTypes.add(containerToGenType(augBasePackageName, container));
-                }
-            }
-            if(augLists !== null) {
-                for (list : augLists) {
-                    genTypes.addAll(listToGenType(augBasePackageName, list));
-                }
-            }
-            if(augChoices !== null) {
-                for (choice : augChoices) {
-                    genTypes.addAll(choiceToGeneratedType(augBasePackageName, choice));
-                }
-            }
-        }
-        return genTypes;
-    }
-
-    /**
-     * Returns first unique name for the augment generated type builder. The
-     * generated type builder name for augment consists from name of augmented
-     * node and serial number of its augmentation.
-     *
-     * @param builders
-     *            map of builders which were created in the package to which the
-     *            augmentation belongs
-     * @param genTypeName
-     *            string with name of augmented node
-     * @return string with unique name for augmentation builder
+    /**\r
+     * Returns first unique name for the augment generated type builder. The\r
+     * generated type builder name for augment consists from name of augmented\r
+     * node and serial number of its augmentation.\r
+     *\r
+     * @param builders\r
+     *            map of builders which were created in the package to which the\r
+     *            augmentation belongs\r
+     * @param genTypeName\r
+     *            string with name of augmented node\r
+     * @return string with unique name for augmentation builder\r
      */
     private def String augGenTypeName(Map<String, GeneratedTypeBuilder> builders, String genTypeName) {
         var index = 1;
-        while((builders !== null) && builders.containsKey(genTypeName + index)) {
+        while ((builders !== null) && builders.containsKey(genTypeName + index)) {
             index = index + 1;
         }
         return genTypeName + index;
     }
 
-    /**
-     * Converts <code>containerNode</code> to generated type. Firstly the
-     * generated type builder is created. The subnodes of
-     * <code>containerNode</code> are added as methods and the instance of
-     * <code>GeneratedType</code> is returned.
-     *
-     * @param basePackageName
-     *            string contains the module package name
-     * @param containerNode
-     *            container schema node with the data about childs nodes and
-     *            schema paths
-     * @return generated type for <code>containerNode</code>
-     */
-    private def GeneratedType containerToGenType(String basePackageName, ContainerSchemaNode containerNode) {
-        if(containerNode === null) {
-            return null;
-        }
-
-        val packageName = packageNameForGeneratedType(basePackageName, containerNode.path);
-        val schemaNodes = containerNode.childNodes;
-        val typeBuilder = addDefaultInterfaceDefinition(packageName, containerNode);
-
-        resolveDataSchemaNodes(basePackageName, typeBuilder, schemaNodes);
-        return typeBuilder.toInstance();
-    }
-
-    /**
-     * Adds the methods to <code>typeBuilder</code> which represent subnodes of
-     * node for which <code>typeBuilder</code> was created.
-     *
-     * The subnodes aren't mapped to the methods if they are part of grouping or
-     * augment (in this case are already part of them).
-     *
-     * @param basePackageName
-     *            string contains the module package name
-     * @param typeBuilder
-     *            generated type builder which represents any node. The subnodes
-     *            of this node are added to the <code>typeBuilder</code> as
-     *            methods. The subnode can be of type leaf, leaf-list, list,
-     *            container, choice.
-     * @param schemaNodes
-     *            set of data schema nodes which are the children of the node
-     *            for which <code>typeBuilder</code> was created
-     * @return generated type builder which is the same builder as input
-     *         parameter. The getter methods (representing child nodes) could be
-     *         added to it.
+    /**\r
+     * Adds the methods to <code>typeBuilder</code> which represent subnodes of\r
+     * node for which <code>typeBuilder</code> was created.\r
+     *\r
+     * The subnodes aren't mapped to the methods if they are part of grouping or\r
+     * augment (in this case are already part of them).\r
+     *\r
+     * @param module current module\r
+     * @param basePackageName\r
+     *            string contains the module package name\r
+     * @param parent\r
+     *            generated type builder which represents any node. The subnodes\r
+     *            of this node are added to the <code>typeBuilder</code> as\r
+     *            methods. The subnode can be of type leaf, leaf-list, list,\r
+     *            container, choice.\r
+     * @param childOf parent type\r
+     * @param schemaNodes\r
+     *            set of data schema nodes which are the children of the node\r
+     *            for which <code>typeBuilder</code> was created\r
+     * @return generated type builder which is the same builder as input\r
+     *         parameter. The getter methods (representing child nodes) could be\r
+     *         added to it.\r
      */
-    private def GeneratedTypeBuilder resolveDataSchemaNodes(String basePackageName, GeneratedTypeBuilder typeBuilder,
-        Set<DataSchemaNode> schemaNodes) {
-        if((schemaNodes !== null) && (typeBuilder !== null)) {
+    private def GeneratedTypeBuilder resolveDataSchemaNodes(Module module, String basePackageName,
+        GeneratedTypeBuilder parent, GeneratedTypeBuilder childOf, Set<DataSchemaNode> schemaNodes) {
+        if ((schemaNodes !== null) && (parent !== null)) {
             for (schemaNode : schemaNodes) {
-                if(!schemaNode.isAugmenting() && !schemaNode.isAddedByUses()) {
-                    addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);
+                if (!schemaNode.augmenting && !schemaNode.addedByUses) {
+                    addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, parent, childOf, module);
                 }
-
             }
         }
-        return typeBuilder;
+        return parent;
     }
 
-    /**
-     * Adds the methods to <code>typeBuilder</code> what represents subnodes of
-     * node for which <code>typeBuilder</code> was created.
-     *
-     * @param basePackageName
-     *            string contains the module package name
-     * @param typeBuilder
-     *            generated type builder which represents any node. The subnodes
-     *            of this node are added to the <code>typeBuilder</code> as
-     *            methods. The subnode can be of type leaf, leaf-list, list,
-     *            container, choice.
-     * @param schemaNodes
-     *            set of data schema nodes which are the children of the node
-     *            for which <code>typeBuilder</code> was created
-     * @return generated type builder which is the same object as the input
-     *         parameter <code>typeBuilder</code>. The getter method could be
-     *         added to it.
+    /**\r
+     * Adds the methods to <code>typeBuilder</code> what represents subnodes of\r
+     * node for which <code>typeBuilder</code> was created.\r
+     *\r
+     * @param module current module\r
+     * @param basePackageName\r
+     *            string contains the module package name\r
+     * @param typeBuilder\r
+     *            generated type builder which represents any node. The subnodes\r
+     *            of this node are added to the <code>typeBuilder</code> as\r
+     *            methods. The subnode can be of type leaf, leaf-list, list,\r
+     *            container, choice.\r
+     * @param childOf parent type\r
+     * @param schemaNodes\r
+     *            set of data schema nodes which are the children of the node\r
+     *            for which <code>typeBuilder</code> was created\r
+     * @return generated type builder which is the same object as the input\r
+     *         parameter <code>typeBuilder</code>. The getter method could be\r
+     *         added to it.\r
      */
-    private def GeneratedTypeBuilder augSchemaNodeToMethods(String basePackageName, GeneratedTypeBuilder typeBuilder,
-        Set<DataSchemaNode> schemaNodes) {
-        if((schemaNodes !== null) && (typeBuilder !== null)) {
+    private def GeneratedTypeBuilder augSchemaNodeToMethods(Module module, String basePackageName,
+        GeneratedTypeBuilder typeBuilder, GeneratedTypeBuilder childOf, Set<DataSchemaNode> schemaNodes) {
+        if ((schemaNodes !== null) && (typeBuilder !== null)) {
             for (schemaNode : schemaNodes) {
-                if(schemaNode.isAugmenting()) {
-                    addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);
+                if (!schemaNode.isAugmenting()) {
+                    addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder, childOf, module);
                 }
             }
         }
         return typeBuilder;
     }
 
-    /**
-     * Adds to <code>typeBuilder</code> a method which is derived from
-     * <code>schemaNode</code>.
-     *
-     * @param basePackageName
-     *            string with the module package name
-     * @param schemaNode
-     *            data schema node which is added to <code>typeBuilder</code> as
-     *            a method
-     * @param typeBuilder
-     *            generated type builder to which is <code>schemaNode</code>
-     *            added as a method.
+    /**\r
+     * Adds to <code>typeBuilder</code> a method which is derived from\r
+     * <code>schemaNode</code>.\r
+     *\r
+     * @param basePackageName\r
+     *            string with the module package name\r
+     * @param node\r
+     *            data schema node which is added to <code>typeBuilder</code> as\r
+     *            a method\r
+     * @param typeBuilder\r
+     *            generated type builder to which is <code>schemaNode</code>\r
+     *            added as a method.\r
+     * @param childOf parent type\r
+     * @param module current module\r
      */
     private def void addSchemaNodeToBuilderAsMethod(String basePackageName, DataSchemaNode node,
-        GeneratedTypeBuilder typeBuilder) {
-        if(node !== null && typeBuilder !== null) {
+        GeneratedTypeBuilder typeBuilder, GeneratedTypeBuilder childOf, Module module) {
+        if (node !== null && typeBuilder !== null) {
             switch (node) {
                 case node instanceof LeafSchemaNode:
                     resolveLeafSchemaNodeAsMethod(typeBuilder, node as LeafSchemaNode)
                 case node instanceof LeafListSchemaNode:
                     resolveLeafListSchemaNode(typeBuilder, node as LeafListSchemaNode)
                 case node instanceof ContainerSchemaNode:
-                    resolveContainerSchemaNode(basePackageName, typeBuilder, node as ContainerSchemaNode)
+                    containerToGenType(module, basePackageName, typeBuilder, childOf, node as ContainerSchemaNode)
                 case node instanceof ListSchemaNode:
-                    resolveListSchemaNode(basePackageName, typeBuilder, node as ListSchemaNode)
+                    listToGenType(module, basePackageName, typeBuilder, childOf, node as ListSchemaNode)
                 case node instanceof ChoiceNode:
-                    resolveChoiceSchemaNode(basePackageName, typeBuilder, node as ChoiceNode)
+                    choiceToGeneratedType(module, basePackageName, typeBuilder, node as ChoiceNode)
             }
         }
     }
 
-    /**
-     * Creates a getter method for a choice node.
-     *
-     * Firstly generated type builder for choice is created or found in
-     * {@link BindingGeneratorImpl#allGroupings allGroupings}. The package name
-     * in the builder is created as concatenation of module package name and
-     * names of all parent nodes. In the end the getter method for choice is
-     * added to <code>typeBuilder</code> and return type is set to choice
-     * builder.
-     *
-     * @param basePackageName
-     *            string with the module package name
-     * @param typeBuilder
-     *            generated type builder to which is <code>choiceNode</code>
-     *            added as getter method
-     * @param choiceNode
-     *            choice node which is mapped as a getter method
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if <code>basePackageName</code> equals null</li>
-     *             <li>if <code>typeBuilder</code> equals null</li>
-     *             <li>if <code>choiceNode</code> equals null</li>
-     *             </ul>
-     *
+    /**\r
+     * Converts <code>choiceNode</code> to the list of generated types for\r
+     * choice and its cases.\r
+     *\r
+     * The package names for choice and for its cases are created as\r
+     * concatenation of the module package (<code>basePackageName</code>) and\r
+     * names of all parents node.\r
+     *\r
+     * @param module current module\r
+     * @param basePackageName\r
+     *            string with the module package name\r
+     * @param parent parent type\r
+     * @param childOf concrete parent for case child nodes\r
+     * @param choiceNode\r
+     *            choice node which is mapped to generated type. Also child\r
+     *            nodes - cases are mapped to generated types.\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if <code>basePackageName</code> equals null</li>\r
+     *             <li>if <code>choiceNode</code> equals null</li>\r
+     *             </ul>\r
+     *\r
      */
-    private def void resolveChoiceSchemaNode(String basePackageName, GeneratedTypeBuilder typeBuilder,
+    private def void choiceToGeneratedType(Module module, String basePackageName, GeneratedTypeBuilder parent,
         ChoiceNode choiceNode) {
         checkArgument(basePackageName !== null, "Base Package Name cannot be NULL.");
-        checkArgument(typeBuilder !== null, "Generated Type Builder cannot be NULL.");
         checkArgument(choiceNode !== null, "Choice Schema Node cannot be NULL.");
 
-        val choiceName = choiceNode.QName.localName;
-        if(choiceName !== null && !choiceNode.isAddedByUses()) {
-            val packageName = packageNameForGeneratedType(basePackageName, choiceNode.path);
-            val choiceType = addDefaultInterfaceDefinition(packageName, choiceNode);
-            constructGetter(typeBuilder, choiceName, choiceNode.description, choiceType);
-        }
-    }
-
-    /**
-     * Converts <code>choiceNode</code> to the list of generated types for
-     * choice and its cases.
-     *
-     * The package names for choice and for its cases are created as
-     * concatenation of the module package (<code>basePackageName</code>) and
-     * names of all parents node.
-     *
-     * @param basePackageName
-     *            string with the module package name
-     * @param choiceNode
-     *            choice node which is mapped to generated type. Also child
-     *            nodes - cases are mapped to generated types.
-     * @return list of generated types which contains generated type for choice
-     *         and generated types for all cases which aren't added do choice
-     *         through <i>uses</i>.
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if <code>basePackageName</code> equals null</li>
-     *             <li>if <code>choiceNode</code> equals null</li>
-     *             </ul>
-     *
-     */
-    private def List<GeneratedType> choiceToGeneratedType(String basePackageName, ChoiceNode choiceNode) {
-        checkArgument(basePackageName !== null, "Base Package Name cannot be NULL.");
-        checkArgument(choiceNode !== null, "Choice Schema Node cannot be NULL.");
-
-        val List<GeneratedType> generatedTypes = new ArrayList();
         val packageName = packageNameForGeneratedType(basePackageName, choiceNode.path);
         val choiceTypeBuilder = addRawInterfaceDefinition(packageName, choiceNode);
-
-        //choiceTypeBuilder.addImplementsType(DATA_OBJECT);
-        val choiceType = choiceTypeBuilder.toInstance();
-
-        generatedTypes.add(choiceType);
-        val Set<ChoiceCaseNode> caseNodes = choiceNode.cases;
-        if((caseNodes !== null) && !caseNodes.isEmpty()) {
-            generatedTypes.addAll(generateTypesFromChoiceCases(basePackageName, choiceType, caseNodes));
-        }
-        return generatedTypes;
+        constructGetter(parent, choiceNode.QName.localName, choiceNode.description, choiceTypeBuilder);
+        choiceTypeBuilder.addImplementsType(DataContainer.typeForClass);
+        genCtx.get(module).addChildNodeType(choiceNode.path, choiceTypeBuilder)
+        generateTypesFromChoiceCases(module, basePackageName, parent, choiceTypeBuilder.toInstance, choiceNode);
     }
 
-    /**
-     * Converts <code>caseNodes</code> set to list of corresponding generated
-     * types.
-     *
-     * For every <i>case</i> which isn't added through augment or <i>uses</i> is
-     * created generated type builder. The package names for the builder is
-     * created as concatenation of the module package (
-     * <code>basePackageName</code>) and names of all parents nodes of the
-     * concrete <i>case</i>. There is also relation "<i>implements type</i>"
-     * between every case builder and <i>choice</i> type
-     *
-     * @param basePackageName
-     *            string with the module package name
-     * @param refChoiceType
-     *            type which represents superior <i>case</i>
-     * @param caseNodes
-     *            set of choice case nodes which are mapped to generated types
-     * @return list of generated types for <code>caseNodes</code>.
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if <code>basePackageName</code> equals null</li>
-     *             <li>if <code>refChoiceType</code> equals null</li>
-     *             <li>if <code>caseNodes</code> equals null</li>
-     *             </ul>
-     *             *
+    /**\r
+     * Converts <code>caseNodes</code> set to list of corresponding generated\r
+     * types.\r
+     *\r
+     * For every <i>case</i> which isn't added through augment or <i>uses</i> is\r
+     * created generated type builder. The package names for the builder is\r
+     * created as concatenation of the module package (\r
+     * <code>basePackageName</code>) and names of all parents nodes of the\r
+     * concrete <i>case</i>. There is also relation "<i>implements type</i>"\r
+     * between every case builder and <i>choice</i> type\r
+     *\r
+     * @param basePackageName\r
+     *            string with the module package name\r
+     * @param refChoiceType\r
+     *            type which represents superior <i>case</i>\r
+     * @param caseNodes\r
+     *            set of choice case nodes which are mapped to generated types\r
+     * @return list of generated types for <code>caseNodes</code>.\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if <code>basePackageName</code> equals null</li>\r
+     *             <li>if <code>refChoiceType</code> equals null</li>\r
+     *             <li>if <code>caseNodes</code> equals null</li>\r
+     *             </ul>\r
+     *             *\r
      */
-    private def List<GeneratedType> generateTypesFromChoiceCases(String basePackageName, Type refChoiceType,
-        Set<ChoiceCaseNode> caseNodes) {
+    private def void generateTypesFromChoiceCases(Module module, String basePackageName,
+        GeneratedTypeBuilder choiceParent, Type refChoiceType, ChoiceNode choiceNode) {
         checkArgument(basePackageName !== null, "Base Package Name cannot be NULL.");
         checkArgument(refChoiceType !== null, "Referenced Choice Type cannot be NULL.");
-        checkArgument(caseNodes !== null, "Set of Choice Case Nodes cannot be NULL.");
+        checkArgument(choiceNode !== null, "ChoiceNode cannot be NULL.");
 
-        val List<GeneratedType> generatedTypes = new ArrayList();
-        for (caseNode : caseNodes) {
-            if(caseNode !== null && !caseNode.isAddedByUses() && !caseNode.isAugmenting()) {
-                val packageName = packageNameForGeneratedType(basePackageName, caseNode.path);
-                val caseTypeBuilder = addDefaultInterfaceDefinition(packageName, caseNode);
-                caseTypeBuilder.addImplementsType(refChoiceType);
+        val Set<ChoiceCaseNode> caseNodes = choiceNode.cases;
+        if (caseNodes == null) {
+            return
+        }
 
-                val Set<DataSchemaNode> childNodes = caseNode.childNodes;
-                if(childNodes !== null) {
-                    resolveDataSchemaNodes(basePackageName, caseTypeBuilder, childNodes);
+        for (caseNode : caseNodes) {
+            if (caseNode !== null && !caseNode.isAddedByUses() && !caseNode.isAugmenting()) {
+                val packageName = packageNameForGeneratedType(basePackageName, caseNode.path)
+                val caseTypeBuilder = addDefaultInterfaceDefinition(packageName, caseNode)
+                caseTypeBuilder.addImplementsType(refChoiceType)
+                genCtx.get(module).addCaseType(caseNode.path, caseTypeBuilder)
+                val Set<DataSchemaNode> caseChildNodes = caseNode.childNodes
+                if (caseChildNodes !== null) {
+                    val parentNode = choiceNode.parent
+                    var SchemaNode parent
+                    if (parentNode instanceof AugmentationSchema) {
+                        val augSchema = parentNode as AugmentationSchema;
+                        val targetPath = augSchema.targetPath;
+                        var targetSchemaNode = findDataSchemaNode(schemaContext, targetPath)
+                        if (targetSchemaNode instanceof DataSchemaNode &&
+                            (targetSchemaNode as DataSchemaNode).isAddedByUses()) {
+                            targetSchemaNode = findOriginal(targetSchemaNode as DataSchemaNode);
+                            if (targetSchemaNode == null) {
+                                throw new NullPointerException(
+                                    "Failed to find target node from grouping for augmentation " + augSchema +
+                                        " in module " + module.name);
+                            }
+                        }
+                        parent = targetSchemaNode
+                    } else {
+                        parent = choiceNode.parent as SchemaNode
+                    }
+                    var GeneratedTypeBuilder childOfType = findChildNodeByPath(parent.path)
+                    resolveDataSchemaNodes(module, basePackageName, caseTypeBuilder, childOfType, caseChildNodes)
                 }
-                generatedTypes.add(caseTypeBuilder.toInstance());
             }
-        }
 
-        return generatedTypes;
+            processUsesAugments(caseNode, module);
+        }
     }
 
-    /**
-     * Generates list of generated types for all the cases of a choice which are
-     * added to the choice through the augment.
-     *
-     *
-     * @param basePackageName
-     *            string contains name of package to which augment belongs. If
-     *            an augmented choice is from an other package (pcg1) than an
-     *            augmenting choice (pcg2) then case's of the augmenting choice
-     *            will belong to pcg2.
-     * @param refChoiceType
-     *            Type which represents the choice to which case belongs. Every
-     *            case has to contain its choice in extend part.
-     * @param caseNodes
-     *            set of choice case nodes for which is checked if are/aren't
-     *            added to choice through augmentation
-     * @return list of generated types which represents augmented cases of
-     *         choice <code>refChoiceType</code>
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if <code>basePackageName</code> equals null</li>
-     *             <li>if <code>refChoiceType</code> equals null</li>
-     *             <li>if <code>caseNodes</code> equals null</li>
-     *             </ul>
+    /**\r
+     * Generates list of generated types for all the cases of a choice which are\r
+     * added to the choice through the augment.\r
+     *\r
+     *\r
+     * @param basePackageName\r
+     *            string contains name of package to which augment belongs. If\r
+     *            an augmented choice is from an other package (pcg1) than an\r
+     *            augmenting choice (pcg2) then case's of the augmenting choice\r
+     *            will belong to pcg2.\r
+     * @param refChoiceType\r
+     *            Type which represents the choice to which case belongs. Every\r
+     *            case has to contain its choice in extend part.\r
+     * @param caseNodes\r
+     *            set of choice case nodes for which is checked if are/aren't\r
+     *            added to choice through augmentation\r
+     * @return list of generated types which represents augmented cases of\r
+     *         choice <code>refChoiceType</code>\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if <code>basePackageName</code> equals null</li>\r
+     *             <li>if <code>refChoiceType</code> equals null</li>\r
+     *             <li>if <code>caseNodes</code> equals null</li>\r
+     *             </ul>\r
      */
-    private def List<GeneratedType> generateTypesFromAugmentedChoiceCases(String basePackageName, Type refChoiceType,
-        Set<ChoiceCaseNode> caseNodes) {
+    private def void generateTypesFromAugmentedChoiceCases(Module module, String basePackageName, Type targetType,
+        ChoiceNode targetNode, Set<DataSchemaNode> augmentedNodes) {
         checkArgument(basePackageName !== null, "Base Package Name cannot be NULL.");
-        checkArgument(refChoiceType !== null, "Referenced Choice Type cannot be NULL.");
-        checkArgument(caseNodes !== null, "Set of Choice Case Nodes cannot be NULL.");
+        checkArgument(targetType !== null, "Referenced Choice Type cannot be NULL.");
+        checkArgument(augmentedNodes !== null, "Set of Choice Case Nodes cannot be NULL.");
 
-        val List<GeneratedType> generatedTypes = new ArrayList();
-        for (caseNode : caseNodes) {
-            if(caseNode !== null && caseNode.isAugmenting()) {
+        for (caseNode : augmentedNodes) {
+            if (caseNode !== null) {
                 val packageName = packageNameForGeneratedType(basePackageName, caseNode.path);
                 val caseTypeBuilder = addDefaultInterfaceDefinition(packageName, caseNode);
-                caseTypeBuilder.addImplementsType(refChoiceType);
+                caseTypeBuilder.addImplementsType(targetType);
+
+                val SchemaNode parent = targetNode.parent as SchemaNode;
+                var GeneratedTypeBuilder childOfType = null;
+                if (parent instanceof Module) {
+                    childOfType = genCtx.get(parent as Module).moduleNode
+                } else if (parent instanceof ChoiceCaseNode) {
+                    childOfType = findCaseByPath(parent.path)
+                } else if (parent instanceof DataSchemaNode || parent instanceof NotificationDefinition) {
+                    childOfType = findChildNodeByPath(parent.path)
+                } else if (parent instanceof GroupingDefinition) {
+                    childOfType = findGroupingByPath(parent.path);
+                }
 
-                val Set<DataSchemaNode> childNodes = caseNode.childNodes;
-                if(childNodes !== null) {
-                    resolveDataSchemaNodes(basePackageName, caseTypeBuilder, childNodes);
+                if (childOfType == null) {
+                    throw new IllegalArgumentException("Failed to find parent type of choice " + targetNode);
                 }
-                generatedTypes.add(caseTypeBuilder.toInstance());
+
+                if (caseNode instanceof DataNodeContainer) {
+                    val DataNodeContainer dataNodeCase = caseNode as DataNodeContainer;
+                    val Set<DataSchemaNode> childNodes = dataNodeCase.childNodes;
+                    if (childNodes !== null) {
+                        resolveDataSchemaNodes(module, basePackageName, caseTypeBuilder, childOfType, childNodes);
+                    }
+                } else {
+                    val ChoiceCaseNode node = targetNode.getCaseNodeByName(caseNode.getQName().getLocalName());
+                    val Set<DataSchemaNode> childNodes = node.childNodes;
+                    if (childNodes !== null) {
+                        resolveDataSchemaNodes(module, basePackageName, caseTypeBuilder, childOfType, childNodes);
+                    }
+                }
+
+                genCtx.get(module).addCaseType(caseNode.path, caseTypeBuilder)
             }
         }
 
-        return generatedTypes;
     }
 
-    /**
-     * Converts <code>leaf</code> to the getter method which is added to
-     * <code>typeBuilder</code>.
-     *
-     * @param typeBuilder
-     *            generated type builder to which is added getter method as
-     *            <code>leaf</code> mapping
-     * @param leaf
-     *            leaf schema node which is mapped as getter method which is
-     *            added to <code>typeBuilder</code>
-     * @return boolean value
-     *         <ul>
-     *         <li>false - if <code>leaf</code> or <code>typeBuilder</code> are
-     *         null</li>
-     *         <li>true - in other cases</li>
-     *         </ul>
+    /**\r
+     * Converts <code>leaf</code> to the getter method which is added to\r
+     * <code>typeBuilder</code>.\r
+     *\r
+     * @param typeBuilder\r
+     *            generated type builder to which is added getter method as\r
+     *            <code>leaf</code> mapping\r
+     * @param leaf\r
+     *            leaf schema node which is mapped as getter method which is\r
+     *            added to <code>typeBuilder</code>\r
+     * @return boolean value\r
+     *         <ul>\r
+     *         <li>false - if <code>leaf</code> or <code>typeBuilder</code> are\r
+     *         null</li>\r
+     *         <li>true - in other cases</li>\r
+     *         </ul>\r
      */
     private def boolean resolveLeafSchemaNodeAsMethod(GeneratedTypeBuilder typeBuilder, LeafSchemaNode leaf) {
-        if((leaf !== null) && (typeBuilder !== null)) {
+        if ((leaf !== null) && (typeBuilder !== null)) {
             val leafName = leaf.QName.localName;
             var String leafDesc = leaf.description;
-            if(leafDesc === null) {
+            if (leafDesc === null) {
                 leafDesc = "";
             }
 
             val parentModule = findParentModule(schemaContext, leaf);
-            if(leafName !== null && !leaf.isAddedByUses()) {
+            if (leafName !== null && !leaf.isAddedByUses()) {
                 val TypeDefinition<?> typeDef = leaf.type;
 
                 var Type returnType = null;
-                if(typeDef instanceof EnumTypeDefinition) {
+                if (typeDef instanceof EnumTypeDefinition) {
                     returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef, leaf);
                     val enumTypeDef = typeDef as EnumTypeDefinition;
                     val enumBuilder = resolveInnerEnumFromTypeDefinition(enumTypeDef, leafName, typeBuilder);
 
-                    if(enumBuilder !== null) {
+                    if (enumBuilder !== null) {
                         returnType = new ReferencedTypeImpl(enumBuilder.packageName, enumBuilder.name);
                     }
                     (typeProvider as TypeProviderImpl).putReferencedType(leaf.path, returnType);
-                } else if(typeDef instanceof UnionType) {
+                } else if (typeDef instanceof UnionType) {
                     val genTOBuilder = addTOToTypeBuilder(typeDef, typeBuilder, leafName, leaf, parentModule);
-                    if(genTOBuilder !== null) {
+                    if (genTOBuilder !== null) {
                         returnType = new ReferencedTypeImpl(genTOBuilder.packageName, genTOBuilder.name);
                     }
-                } else if(typeDef instanceof BitsTypeDefinition) {
+                } else if (typeDef instanceof BitsTypeDefinition) {
                     val genTOBuilder = addTOToTypeBuilder(typeDef, typeBuilder, leafName, leaf, parentModule);
-                    if(genTOBuilder !== null) {
+                    if (genTOBuilder !== null) {
                         returnType = new ReferencedTypeImpl(genTOBuilder.packageName, genTOBuilder.name);
                     }
-                } else {
-                    returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef, leaf);
+                } else {\r
+                    val Restrictions restrictions = BindingGeneratorUtil.getRestrictions(typeDef);\r
+                    returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef, leaf, restrictions);
                 }
-                if(returnType !== null) {
-                    constructGetter(typeBuilder, leafName, leafDesc, returnType);
+                if (returnType !== null) {
+                    val MethodSignatureBuilder getter = constructGetter(typeBuilder, leafName, leafDesc, returnType);
+                    processContextRefExtension(leaf, getter, parentModule);
                     return true;
                 }
             }
@@ -1386,51 +1362,103 @@ public class BindingGeneratorImpl implements BindingGenerator {
         return false;
     }
 
-    /**
-     * Converts <code>leaf</code> schema node to property of generated TO
-     * builder.
-     *
-     * @param toBuilder
-     *            generated TO builder to which is <code>leaf</code> added as
-     *            property
-     * @param leaf
-     *            leaf schema node which is added to <code>toBuilder</code> as
-     *            property
-     * @param isReadOnly
-     *            boolean value which says if leaf property is|isn't read only
-     * @return boolean value
-     *         <ul>
-     *         <li>false - if <code>leaf</code>, <code>toBuilder</code> or leaf
-     *         name equals null or if leaf is added by <i>uses</i>.</li>
-     *         <li>true - other cases</li>
-     *         </ul>
+    private def void processContextRefExtension(LeafSchemaNode leaf, MethodSignatureBuilder getter, Module module) {
+        for (node : leaf.unknownSchemaNodes) {
+            val nodeType = node.nodeType;
+            if ("context-reference".equals(nodeType.localName)) {
+                val nodeParam = node.nodeParameter;
+                var IdentitySchemaNode identity = null;
+                var String basePackageName = null;
+                val String[] splittedElement = nodeParam.split(":");
+                if (splittedElement.length == 1) {
+                    identity = findIdentityByName(module.identities, splittedElement.get(0));
+                    basePackageName = moduleNamespaceToPackageName(module);
+                } else if (splittedElement.length == 2) {
+                    var prefix = splittedElement.get(0);
+                    val Module dependentModule = findModuleFromImports(module.imports, prefix)
+                    if (dependentModule == null) {
+                        throw new IllegalArgumentException(
+                            "Failed to process context-reference: unknown prefix " + prefix);
+                    }
+                    identity = findIdentityByName(dependentModule.identities, splittedElement.get(1));
+                    basePackageName = moduleNamespaceToPackageName(dependentModule);
+                } else {
+                    throw new IllegalArgumentException(
+                        "Failed to process context-reference: unknown identity " + nodeParam);
+                }
+                if (identity == null) {
+                    throw new IllegalArgumentException(
+                        "Failed to process context-reference: unknown identity " + nodeParam);
+                }
+
+                val Class<RoutingContext> clazz = typeof(RoutingContext);
+                val AnnotationTypeBuilder rc = getter.addAnnotation(clazz.package.name, clazz.simpleName);
+                val packageName = packageNameForGeneratedType(basePackageName, identity.path);
+                val genTypeName = parseToClassName(identity.QName.localName);
+                rc.addParameter("value", packageName + "." + genTypeName + ".class");
+            }
+        }
+    }
+
+    private def IdentitySchemaNode findIdentityByName(Set<IdentitySchemaNode> identities, String name) {
+        for (id : identities) {
+            if (id.QName.localName.equals(name)) {
+                return id;
+            }
+        }
+        return null;
+    }
+
+    private def Module findModuleFromImports(Set<ModuleImport> imports, String prefix) {
+        for (imp : imports) {
+            if (imp.prefix.equals(prefix)) {
+                return schemaContext.findModuleByName(imp.moduleName, imp.revision);
+            }
+        }
+        return null;
+    }
+
+    /**\r
+     * Converts <code>leaf</code> schema node to property of generated TO\r
+     * builder.\r
+     *\r
+     * @param toBuilder\r
+     *            generated TO builder to which is <code>leaf</code> added as\r
+     *            property\r
+     * @param leaf\r
+     *            leaf schema node which is added to <code>toBuilder</code> as\r
+     *            property\r
+     * @param isReadOnly\r
+     *            boolean value which says if leaf property is|isn't read only\r
+     * @return boolean value\r
+     *         <ul>\r
+     *         <li>false - if <code>leaf</code>, <code>toBuilder</code> or leaf\r
+     *         name equals null or if leaf is added by <i>uses</i>.</li>\r
+     *         <li>true - other cases</li>\r
+     *         </ul>\r
      */
     private def boolean resolveLeafSchemaNodeAsProperty(GeneratedTOBuilder toBuilder, LeafSchemaNode leaf,
         boolean isReadOnly) {
-        if((leaf !== null) && (toBuilder !== null)) {
+        if ((leaf !== null) && (toBuilder !== null)) {
             val leafName = leaf.QName.localName;
             var String leafDesc = leaf.description;
-            if(leafDesc === null) {
+            if (leafDesc === null) {
                 leafDesc = "";
             }
 
-            if(leafName !== null) {
+            if (leafName !== null) {
                 val TypeDefinition<?> typeDef = leaf.type;
 
-                // TODO: properly resolve enum types
+                // TODO: properly resolve enum types\r
                 val returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef, leaf);
-
-                if(returnType !== null) {
-                    val propBuilder = toBuilder.addProperty(parseToClassName(leafName));
-
+                if (returnType !== null) {
+                    val propBuilder = toBuilder.addProperty(parseToValidParamName(leafName));
                     propBuilder.setReadOnly(isReadOnly);
                     propBuilder.setReturnType(returnType);
                     propBuilder.setComment(leafDesc);
-
                     toBuilder.addEqualsIdentity(propBuilder);
                     toBuilder.addHashIdentity(propBuilder);
                     toBuilder.addToStringProperty(propBuilder);
-
                     return true;
                 }
             }
@@ -1438,35 +1466,53 @@ public class BindingGeneratorImpl implements BindingGenerator {
         return false;
     }
 
-    /**
-     * Converts <code>node</code> leaf list schema node to getter method of
-     * <code>typeBuilder</code>.
-     *
-     * @param typeBuilder
-     *            generated type builder to which is <code>node</code> added as
-     *            getter method
-     * @param node
-     *            leaf list schema node which is added to
-     *            <code>typeBuilder</code> as getter method
-     * @return boolean value
-     *         <ul>
-     *         <li>true - if <code>node</code>, <code>typeBuilder</code>,
-     *         nodeName equal null or <code>node</code> is added by <i>uses</i></li>
-     *         <li>false - other cases</li>
-     *         </ul>
+    /**\r
+     * Converts <code>node</code> leaf list schema node to getter method of\r
+     * <code>typeBuilder</code>.\r
+     *\r
+     * @param typeBuilder\r
+     *            generated type builder to which is <code>node</code> added as\r
+     *            getter method\r
+     * @param node\r
+     *            leaf list schema node which is added to\r
+     *            <code>typeBuilder</code> as getter method\r
+     * @return boolean value\r
+     *         <ul>\r
+     *         <li>true - if <code>node</code>, <code>typeBuilder</code>,\r
+     *         nodeName equal null or <code>node</code> is added by <i>uses</i></li>\r
+     *         <li>false - other cases</li>\r
+     *         </ul>\r
      */
     private def boolean resolveLeafListSchemaNode(GeneratedTypeBuilder typeBuilder, LeafListSchemaNode node) {
-        if((node !== null) && (typeBuilder !== null)) {
+        if ((node !== null) && (typeBuilder !== null)) {
             val nodeName = node.QName.localName;
             var String nodeDesc = node.description;
-            if(nodeDesc === null) {
+            if (nodeDesc === null) {
                 nodeDesc = "";
             }
-
-            if(nodeName !== null && !node.isAddedByUses()) {
-                val TypeDefinition<?> type = node.type;
-                val listType = Types.listTypeFor(typeProvider.javaTypeForSchemaDefinitionType(type, node));
-
+            if (nodeName !== null && !node.isAddedByUses()) {
+                val TypeDefinition<?> typeDef = node.type;\r
+                val parentModule = findParentModule(schemaContext, node);\r
+\r
+                var Type returnType = null;\r
+                if (typeDef instanceof EnumTypeDefinition) {\r
+                    returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef, node);\r
+                    val enumTypeDef = typeDef as EnumTypeDefinition;\r
+                    val enumBuilder = resolveInnerEnumFromTypeDefinition(enumTypeDef, nodeName, typeBuilder);\r
+                    returnType = new ReferencedTypeImpl(enumBuilder.packageName, enumBuilder.name);\r
+                    (typeProvider as TypeProviderImpl).putReferencedType(node.path, returnType);\r
+                } else if (typeDef instanceof UnionType) {\r
+                    val genTOBuilder = addTOToTypeBuilder(typeDef, typeBuilder, nodeName, node, parentModule);\r
+                    returnType = new ReferencedTypeImpl(genTOBuilder.packageName, genTOBuilder.name);\r
+                } else if (typeDef instanceof BitsTypeDefinition) {\r
+                    val genTOBuilder = addTOToTypeBuilder(typeDef, typeBuilder, nodeName, node, parentModule);\r
+                    returnType = new ReferencedTypeImpl(genTOBuilder.packageName, genTOBuilder.name);\r
+                } else {\r
+                    val Restrictions restrictions = BindingGeneratorUtil.getRestrictions(typeDef);\r
+                    returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef, node, restrictions);\r
+                }\r
+
+                val listType = Types.listTypeFor(returnType);
                 constructGetter(typeBuilder, nodeName, nodeDesc, listType);
                 return true;
             }
@@ -1474,160 +1520,92 @@ public class BindingGeneratorImpl implements BindingGenerator {
         return false;
     }
 
-    /**
-     * Creates a getter method for a container node.
-     *
-     * Firstly generated type builder for container is created or found in
-     * {@link BindingGeneratorImpl#allGroupings allGroupings}. The package name
-     * in the builder is created as concatenation of module package name and
-     * names of all parent nodes. In the end the getter method for container is
-     * added to <code>typeBuilder</code> and return type is set to container
-     * type builder.
-     *
-     * @param basePackageName
-     *            string with the module package name
-     * @param typeBuilder
-     *            generated type builder to which is <code>containerNode</code>
-     *            added as getter method
-     * @param containerNode
-     *            container schema node which is mapped as getter method to
-     *            <code>typeBuilder</code>
-     * @return boolean value
-     *         <ul>
-     *         <li>false - if <code>containerNode</code>,
-     *         <code>typeBuilder</code>, container node name equal null or
-     *         <code>containerNode</code> is added by uses</li>
-     *         <li>true - other cases</li>
-     *         </ul>
-     */
-    private def boolean resolveContainerSchemaNode(String basePackageName, GeneratedTypeBuilder typeBuilder,
-        ContainerSchemaNode containerNode) {
-        if((containerNode !== null) && (typeBuilder !== null)) {
-            val nodeName = containerNode.QName.localName;
-
-            if(nodeName !== null && !containerNode.isAddedByUses()) {
-                val packageName = packageNameForGeneratedType(basePackageName, containerNode.path);
-
-                val rawGenType = addDefaultInterfaceDefinition(packageName, containerNode);
-                constructGetter(typeBuilder, nodeName, containerNode.description, rawGenType);
-
-                return true;
-            }
-        }
-        return false;
+    private def GeneratedTypeBuilder addDefaultInterfaceDefinition(String packageName, SchemaNode schemaNode) {
+        return addDefaultInterfaceDefinition(packageName, schemaNode, null);
     }
 
-    /**
-     * Creates a getter method for a list node.
-     *
-     * Firstly generated type builder for list is created or found in
-     * {@link BindingGeneratorImpl#allGroupings allGroupings}. The package name
-     * in the builder is created as concatenation of module package name and
-     * names of all parent nodes. In the end the getter method for list is added
-     * to <code>typeBuilder</code> and return type is set to list type builder.
-     *
-     * @param basePackageName
-     *            string with the module package name
-     * @param typeBuilder
-     *            generated type builder to which is <code></code> added as
-     *            getter method
-     * @param listNode
-     *            list schema node which is mapped as getter method to
-     *            <code>typeBuilder</code>
-     * @return boolean value
-     *         <ul>
-     *         <li>false - if <code>listNode</code>, <code>typeBuilder</code>,
-     *         list node name equal null or <code>listNode</code> is added by
-     *         uses</li>
-     *         <li>true - other cases</li>
-     *         </ul>
+    /**\r
+     * Instantiates generated type builder with <code>packageName</code> and\r
+     * <code>schemaNode</code>.\r
+     *\r
+     * The new builder always implements\r
+     * {@link org.opendaylight.yangtools.yang.binding.DataObject DataObject}.<br />\r
+     * If <code>schemaNode</code> is instance of GroupingDefinition it also\r
+     * implements {@link org.opendaylight.yangtools.yang.binding.Augmentable\r
+     * Augmentable}.<br />\r
+     * If <code>schemaNode</code> is instance of\r
+     * {@link org.opendaylight.yangtools.yang.model.api.DataNodeContainer\r
+     * DataNodeContainer} it can also implement nodes which are specified in\r
+     * <i>uses</i>.\r
+     *\r
+     * @param packageName\r
+     *            string with the name of the package to which\r
+     *            <code>schemaNode</code> belongs.\r
+     * @param schemaNode\r
+     *            schema node for which is created generated type builder\r
+     * @param parent parent type (can be null)\r
+     * @return generated type builder <code>schemaNode</code>\r
      */
-    private def boolean resolveListSchemaNode(String basePackageName, GeneratedTypeBuilder typeBuilder,
-        ListSchemaNode listNode) {
-        if((listNode !== null) && (typeBuilder !== null)) {
-            val listName = listNode.QName.localName;
-
-            if(listName !== null && !listNode.isAddedByUses()) {
-                val packageName = packageNameForGeneratedType(basePackageName, listNode.path);
-                val rawGenType = addDefaultInterfaceDefinition(packageName, listNode);
-                constructGetter(typeBuilder, listName, listNode.description, Types.listTypeFor(rawGenType));
-                return true;
-            }
+    private def GeneratedTypeBuilder addDefaultInterfaceDefinition(String packageName, SchemaNode schemaNode,
+        Type parent) {
+        val it = addRawInterfaceDefinition(packageName, schemaNode, "");\r
+        val qname = schemaNode.QName;
+        addConstant(QName.typeForClass,"QNAME",'''\r
+            org.opendaylight.yangtools.yang.common.QName.create("«qname.namespace»","«qname.formattedRevision»","«qname.localName»")\r
+        ''');\r
+        if (parent === null) {
+            addImplementsType(DATA_OBJECT);
+        } else {
+            addImplementsType(BindingTypes.childOf(parent));
         }
-        return false;
-    }
-
-    /**
-     * Instantiates generated type builder with <code>packageName</code> and
-     * <code>schemaNode</code>.
-     *
-     * The new builder always implements
-     * {@link org.opendaylight.yangtools.yang.binding.DataObject DataObject}.<br />
-     * If <code>schemaNode</code> is instance of GroupingDefinition it also
-     * implements {@link org.opendaylight.yangtools.yang.binding.Augmentable
-     * Augmentable}.<br />
-     * If <code>schemaNode</code> is instance of
-     * {@link org.opendaylight.yangtools.yang.model.api.DataNodeContainer
-     * DataNodeContainer} it can also implement nodes which are specified in
-     * <i>uses</i>.
-     *
-     * @param packageName
-     *            string with the name of the package to which
-     *            <code>schemaNode</code> belongs.
-     * @param schemaNode
-     *            schema node for which is created generated type builder
-     * @return generated type builder <code>schemaNode</code>
-     */
-    private def GeneratedTypeBuilder addDefaultInterfaceDefinition(String packageName, SchemaNode schemaNode) {
-        val builder = addRawInterfaceDefinition(packageName, schemaNode, "");
-        builder.addImplementsType(DATA_OBJECT);
-        if(!(schemaNode instanceof GroupingDefinition)) {
-            builder.addImplementsType(augmentable(builder));
+        if (!(schemaNode instanceof GroupingDefinition)) {
+            addImplementsType(augmentable(it));
         }
 
-        if(schemaNode instanceof DataNodeContainer) {
-            addImplementedInterfaceFromUses(schemaNode as DataNodeContainer, builder);
+        if (schemaNode instanceof DataNodeContainer) {
+            addImplementedInterfaceFromUses(schemaNode as DataNodeContainer, it);
         }
 
-        return builder;
+        return it;
     }
 
-    /**
-     * Wraps the calling of the same overloaded method.
-     *
-     * @param packageName
-     *            string with the package name to which returning generated type
-     *            builder belongs
-     * @param schemaNode
-     *            schema node which provide data about the schema node name
-     * @return generated type builder for <code>schemaNode</code>
+    /**\r
+     * Wraps the calling of the same overloaded method.\r
+     *\r
+     * @param packageName\r
+     *            string with the package name to which returning generated type\r
+     *            builder belongs\r
+     * @param schemaNode\r
+     *            schema node which provide data about the schema node name\r
+     * @return generated type builder for <code>schemaNode</code>\r
      */
     private def GeneratedTypeBuilder addRawInterfaceDefinition(String packageName, SchemaNode schemaNode) {
         return addRawInterfaceDefinition(packageName, schemaNode, "");
     }
 
-    /**
-     * Returns reference to generated type builder for specified
-     * <code>schemaNode</code> with <code>packageName</code>.
-     *
-     * Firstly the generated type builder is searched in
-     * {@link BindingGeneratorImpl#genTypeBuilders genTypeBuilders}. If it isn't
-     * found it is created and added to <code>genTypeBuilders</code>.
-     *
-     * @param packageName
-     *            string with the package name to which returning generated type
-     *            builder belongs
-     * @param schemaNode
-     *            schema node which provide data about the schema node name
-     * @return generated type builder for <code>schemaNode</code>
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if <code>schemaNode</code> equals null</li>
-     *             <li>if <code>packageName</code> equals null</li>
-     *             <li>if Q name of schema node is null</li>
-     *             <li>if schema node name is nul</li>
-     *             </ul>
-     *
+    /**\r
+     * Returns reference to generated type builder for specified\r
+     * <code>schemaNode</code> with <code>packageName</code>.\r
+     *\r
+     * Firstly the generated type builder is searched in\r
+     * {@link BindingGeneratorImpl#genTypeBuilders genTypeBuilders}. If it isn't\r
+     * found it is created and added to <code>genTypeBuilders</code>.\r
+     *\r
+     * @param packageName\r
+     *            string with the package name to which returning generated type\r
+     *            builder belongs\r
+     * @param schemaNode\r
+     *            schema node which provide data about the schema node name\r
+     * @param prefix return type name prefix\r
+     * @return generated type builder for <code>schemaNode</code>\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if <code>schemaNode</code> equals null</li>\r
+     *             <li>if <code>packageName</code> equals null</li>\r
+     *             <li>if Q name of schema node is null</li>\r
+     *             <li>if schema node name is nul</li>\r
+     *             </ul>\r
+     *\r
      */
     private def GeneratedTypeBuilder addRawInterfaceDefinition(String packageName, SchemaNode schemaNode,
         String prefix) {
@@ -1638,180 +1616,148 @@ public class BindingGeneratorImpl implements BindingGenerator {
         checkArgument(schemaNodeName !== null, "Local Name of QName for Data Schema Node cannot be NULL.");
 
         var String genTypeName;
-        if(prefix === null) {
+        if (prefix === null) {
             genTypeName = parseToClassName(schemaNodeName);
         } else {
             genTypeName = prefix + parseToClassName(schemaNodeName);
         }
 
-        //FIXME: Validation of name conflict
+        //FIXME: Validation of name conflict\r
         val newType = new GeneratedTypeBuilderImpl(packageName, genTypeName);
-        yangToJavaMapping.put(schemaNode.path, newType);
-        if(!genTypeBuilders.containsKey(packageName)) {
+        if (!genTypeBuilders.containsKey(packageName)) {
             val Map<String, GeneratedTypeBuilder> builders = new HashMap();
             builders.put(genTypeName, newType);
             genTypeBuilders.put(packageName, builders);
         } else {
             val Map<String, GeneratedTypeBuilder> builders = genTypeBuilders.get(packageName);
-            if(!builders.containsKey(genTypeName)) {
+            if (!builders.containsKey(genTypeName)) {
                 builders.put(genTypeName, newType);
             }
         }
         return newType;
     }
 
-    /**
-     * Creates the name of the getter method from <code>methodName</code>.
-     *
-     * @param methodName
-     *            string with the name of the getter method
-     * @return string with the name of the getter method for
-     *         <code>methodName</code> in JAVA method format
+    /**\r
+     * Creates the name of the getter method from <code>methodName</code>.\r
+     *\r
+     * @param methodName\r
+     *            string with the name of the getter method\r
+     * @param returnType return type\r
+     * @return string with the name of the getter method for\r
+     *         <code>methodName</code> in JAVA method format\r
      */
-    private def String getterMethodName(String methodName, Type returnType) {
+    public static def String getterMethodName(String localName, Type returnType) {
         val method = new StringBuilder();
-        if(BOOLEAN.equals(returnType)) {
+        if (BOOLEAN.equals(returnType)) {
             method.append("is");
         } else {
             method.append("get");
         }
-        method.append(parseToClassName(methodName));
+        method.append(parseToClassName(localName));
         return method.toString();
     }
 
-    /**
-     * Created a method signature builder as part of
-     * <code>interfaceBuilder</code>.
-     *
-     * The method signature builder is created for the getter method of
-     * <code>schemaNodeName</code>. Also <code>comment</code> and
-     * <code>returnType</code> information are added to the builder.
-     *
-     * @param interfaceBuilder
-     *            generated type builder for which the getter method should be
-     *            created
-     * @param schemaNodeName
-     *            string with schema node name. The name will be the part of the
-     *            getter method name.
-     * @param comment
-     *            string with comment for the getter method
-     * @param returnType
-     *            type which represents the return type of the getter method
-     * @return method signature builder which represents the getter method of
-     *         <code>interfaceBuilder</code>
+    /**\r
+     * Created a method signature builder as part of\r
+     * <code>interfaceBuilder</code>.\r
+     *\r
+     * The method signature builder is created for the getter method of\r
+     * <code>schemaNodeName</code>. Also <code>comment</code> and\r
+     * <code>returnType</code> information are added to the builder.\r
+     *\r
+     * @param interfaceBuilder\r
+     *            generated type builder for which the getter method should be\r
+     *            created\r
+     * @param schemaNodeName\r
+     *            string with schema node name. The name will be the part of the\r
+     *            getter method name.\r
+     * @param comment\r
+     *            string with comment for the getter method\r
+     * @param returnType\r
+     *            type which represents the return type of the getter method\r
+     * @return method signature builder which represents the getter method of\r
+     *         <code>interfaceBuilder</code>\r
      */
     private def MethodSignatureBuilder constructGetter(GeneratedTypeBuilder interfaceBuilder, String schemaNodeName,
         String comment, Type returnType) {
-
         val getMethod = interfaceBuilder.addMethod(getterMethodName(schemaNodeName, returnType));
-
         getMethod.setComment(comment);
         getMethod.setReturnType(returnType);
-
         return getMethod;
     }
 
-    private def listToGenType(String basePackageName, ListSchemaNode list) {
-        checkArgument(basePackageName !== null, "Package Name for Generated Type cannot be NULL.");
-        checkArgument(list !== null, "List Schema Node cannot be NULL.");
-
-        val packageName = packageNameForGeneratedType(basePackageName, list.path);
-
-        // val typeBuilder =
-        // resolveListTypeBuilder(packageName, list);
-        val typeBuilder = addDefaultInterfaceDefinition(packageName, list);
-
-        val List<String> listKeys = listKeys(list);
-        val genTOBuilder = resolveListKeyTOBuilder(packageName, list);
-
-        if(genTOBuilder !== null) {
-            val identifierMarker = IDENTIFIER.parameterizedTypeFor(typeBuilder);
-            val identifiableMarker = IDENTIFIABLE.parameterizedTypeFor(genTOBuilder);
-            genTOBuilder.addImplementsType(identifierMarker);
-            typeBuilder.addImplementsType(identifiableMarker);
-        }
-        val schemaNodes = list.childNodes;
-
-        for (schemaNode : schemaNodes) {
-            if(!schemaNode.isAugmenting()) {
-                addSchemaNodeToListBuilders(basePackageName, schemaNode, typeBuilder, genTOBuilder, listKeys);
-            }
-        }
-        return typeBuildersToGenTypes(typeBuilder, genTOBuilder);
-    }
-
-    /**
-     * Adds <code>schemaNode</code> to <code>typeBuilder</code> as getter method
-     * or to <code>genTOBuilder</code> as property.
-     *
-     * @param basePackageName
-     *            string contains the module package name
-     * @param schemaNode
-     *            data schema node which should be added as getter method to
-     *            <code>typeBuilder</code> or as a property to
-     *            <code>genTOBuilder</code> if is part of the list key
-     * @param typeBuilder
-     *            generated type builder for the list schema node
-     * @param genTOBuilder
-     *            generated TO builder for the list keys
-     * @param listKeys
-     *            list of string which contains names of the list keys
-     * @throws IllegalArgumentException
-     *             <ul>
-     *             <li>if <code>schemaNode</code> equals null</li>
-     *             <li>if <code>typeBuilder</code> equals null</li>
-     *             </ul>
+    /**\r
+     * Adds <code>schemaNode</code> to <code>typeBuilder</code> as getter method\r
+     * or to <code>genTOBuilder</code> as property.\r
+     *\r
+     * @param basePackageName\r
+     *            string contains the module package name\r
+     * @param schemaNode\r
+     *            data schema node which should be added as getter method to\r
+     *            <code>typeBuilder</code> or as a property to\r
+     *            <code>genTOBuilder</code> if is part of the list key\r
+     * @param typeBuilder\r
+     *            generated type builder for the list schema node\r
+     * @param genTOBuilder\r
+     *            generated TO builder for the list keys\r
+     * @param listKeys\r
+     *            list of string which contains names of the list keys\r
+     * @param module current module\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if <code>schemaNode</code> equals null</li>\r
+     *             <li>if <code>typeBuilder</code> equals null</li>\r
+     *             </ul>\r
      */
     private def void addSchemaNodeToListBuilders(String basePackageName, DataSchemaNode schemaNode,
-        GeneratedTypeBuilder typeBuilder, GeneratedTOBuilder genTOBuilder, List<String> listKeys) {
+        GeneratedTypeBuilder typeBuilder, GeneratedTOBuilder genTOBuilder, List<String> listKeys, Module module) {
         checkArgument(schemaNode !== null, "Data Schema Node cannot be NULL.");
-
         checkArgument(typeBuilder !== null, "Generated Type Builder cannot be NULL.");
 
-        if(schemaNode instanceof LeafSchemaNode) {
+        if (schemaNode instanceof LeafSchemaNode) {
             val leaf = schemaNode as LeafSchemaNode;
-            val leafName = leaf.QName.localName;
-            if(!listKeys.contains(leafName)) {
-                resolveLeafSchemaNodeAsMethod(typeBuilder, leaf);
-            } else {
+            val leafName = leaf.QName.localName;\r
+            resolveLeafSchemaNodeAsMethod(typeBuilder, leaf);
+            if (listKeys.contains(leafName)) {
                 resolveLeafSchemaNodeAsProperty(genTOBuilder, leaf, true);
             }
-        } else if(schemaNode instanceof LeafListSchemaNode) {
-            resolveLeafListSchemaNode(typeBuilder, schemaNode as LeafListSchemaNode);
-        } else if(schemaNode instanceof ContainerSchemaNode) {
-            resolveContainerSchemaNode(basePackageName, typeBuilder, schemaNode as ContainerSchemaNode);
-        } else if(schemaNode instanceof ListSchemaNode) {
-            resolveListSchemaNode(basePackageName, typeBuilder, schemaNode as ListSchemaNode);
+        } else if (!schemaNode.addedByUses) {
+            if (schemaNode instanceof LeafListSchemaNode) {
+                resolveLeafListSchemaNode(typeBuilder, schemaNode as LeafListSchemaNode);
+            } else if (schemaNode instanceof ContainerSchemaNode) {
+                containerToGenType(module, basePackageName, typeBuilder, typeBuilder, schemaNode as ContainerSchemaNode);
+            } else if (schemaNode instanceof ChoiceNode) {
+                choiceToGeneratedType(module, basePackageName, typeBuilder, schemaNode as ChoiceNode);
+            } else if (schemaNode instanceof ListSchemaNode) {
+                listToGenType(module, basePackageName, typeBuilder, typeBuilder, schemaNode as ListSchemaNode);
+            }
         }
     }
 
-    private def typeBuildersToGenTypes(GeneratedTypeBuilder typeBuilder, GeneratedTOBuilder genTOBuilder) {
-        val List<Type> genTypes = new ArrayList();
+    private def typeBuildersToGenTypes(Module module, GeneratedTypeBuilder typeBuilder, GeneratedTOBuilder genTOBuilder) {
         checkArgument(typeBuilder !== null, "Generated Type Builder cannot be NULL.");
 
-        if(genTOBuilder !== null) {
+        if (genTOBuilder !== null) {
             val genTO = genTOBuilder.toInstance();
             constructGetter(typeBuilder, "key", "Returns Primary Key of Yang List Type", genTO);
-            genTypes.add(genTO);
+            genCtx.get(module).addGeneratedTOBuilder(genTOBuilder)
         }
-        genTypes.add(typeBuilder.toInstance());
-        return genTypes;
     }
 
-    /**
-     * Selects the names of the list keys from <code>list</code> and returns
-     * them as the list of the strings
-     *
-     * @param list
-     *            of string with names of the list keys
-     * @return list of string which represents names of the list keys. If the
-     *         <code>list</code> contains no keys then the empty list is
-     *         returned.
+    /**\r
+     * Selects the names of the list keys from <code>list</code> and returns\r
+     * them as the list of the strings\r
+     *\r
+     * @param list\r
+     *            of string with names of the list keys\r
+     * @return list of string which represents names of the list keys. If the\r
+     *         <code>list</code> contains no keys then the empty list is\r
+     *         returned.\r
      */
     private def listKeys(ListSchemaNode list) {
         val List<String> listKeys = new ArrayList();
 
-        if(list.keyDefinition !== null) {
+        if (list.keyDefinition !== null) {
             val keyDefinitions = list.keyDefinition;
             for (keyDefinition : keyDefinitions) {
                 listKeys.add(keyDefinition.localName);
@@ -1820,68 +1766,67 @@ public class BindingGeneratorImpl implements BindingGenerator {
         return listKeys;
     }
 
-    /**
-     * Generates for the <code>list</code> which contains any list keys special
-     * generated TO builder.
-     *
-     * @param packageName
-     *            string with package name to which the list belongs
-     * @param list
-     *            list schema node which is source of data about the list name
-     * @return generated TO builder which represents the keys of the
-     *         <code>list</code> or null if <code>list</code> is null or list of
-     *         key definitions is null or empty.
+    /**\r
+     * Generates for the <code>list</code> which contains any list keys special\r
+     * generated TO builder.\r
+     *\r
+     * @param packageName\r
+     *            string with package name to which the list belongs\r
+     * @param list\r
+     *            list schema node which is source of data about the list name\r
+     * @return generated TO builder which represents the keys of the\r
+     *         <code>list</code> or null if <code>list</code> is null or list of\r
+     *         key definitions is null or empty.\r
      */
     private def GeneratedTOBuilder resolveListKeyTOBuilder(String packageName, ListSchemaNode list) {
         var GeneratedTOBuilder genTOBuilder = null;
-        if((list.keyDefinition !== null) && (!list.keyDefinition.isEmpty())) {
-            if(list !== null) {
-                val listName = list.QName.localName + "Key";
-                val String genTOName = parseToClassName(listName);
-                genTOBuilder = new GeneratedTOBuilderImpl(packageName, genTOName);
-            }
+        if ((list.keyDefinition !== null) && (!list.keyDefinition.isEmpty())) {
+            val listName = list.QName.localName + "Key";
+            val String genTOName = parseToClassName(listName);
+            genTOBuilder = new GeneratedTOBuilderImpl(packageName, genTOName);
         }
         return genTOBuilder;
-
     }
 
-    /**
-     * Builds generated TO builders for <code>typeDef</code> of type
-     * {@link org.opendaylight.yangtools.yang.model.util.UnionType UnionType} or
-     * {@link org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition
-     * BitsTypeDefinition} which are also added to <code>typeBuilder</code> as
-     * enclosing transfer object.
-     *
-     * If more then one generated TO builder is created for enclosing then all
-     * of the generated TO builders are added to <code>typeBuilder</code> as
-     * enclosing transfer objects.
-     *
-     * @param typeDef
-     *            type definition which can be of type <code>UnionType</code> or
-     *            <code>BitsTypeDefinition</code>
-     * @param typeBuilder
-     *            generated type builder to which is added generated TO created
-     *            from <code>typeDef</code>
-     * @param leafName
-     *            string with name for generated TO builder
-     * @return generated TO builder for <code>typeDef</code>
+    /**\r
+     * Builds generated TO builders for <code>typeDef</code> of type\r
+     * {@link org.opendaylight.yangtools.yang.model.util.UnionType UnionType} or\r
+     * {@link org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition\r
+     * BitsTypeDefinition} which are also added to <code>typeBuilder</code> as\r
+     * enclosing transfer object.\r
+     *\r
+     * If more then one generated TO builder is created for enclosing then all\r
+     * of the generated TO builders are added to <code>typeBuilder</code> as\r
+     * enclosing transfer objects.\r
+     *\r
+     * @param typeDef\r
+     *            type definition which can be of type <code>UnionType</code> or\r
+     *            <code>BitsTypeDefinition</code>\r
+     * @param typeBuilder\r
+     *            generated type builder to which is added generated TO created\r
+     *            from <code>typeDef</code>\r
+     * @param leafName\r
+     *            string with name for generated TO builder\r
+     * @param leaf\r
+     * @param parentModule\r
+     * @return generated TO builder for <code>typeDef</code>\r
      */
     private def GeneratedTOBuilder addTOToTypeBuilder(TypeDefinition<?> typeDef, GeneratedTypeBuilder typeBuilder,
-        String leafName, LeafSchemaNode leaf, Module parentModule) {
+        String leafName, DataSchemaNode leaf, Module parentModule) {
         val classNameFromLeaf = parseToClassName(leafName);
         val List<GeneratedTOBuilder> genTOBuilders = new ArrayList();
         val packageName = typeBuilder.fullyQualifiedName;
-        if(typeDef instanceof UnionTypeDefinition) {
+        if (typeDef instanceof UnionTypeDefinition) {
             genTOBuilders.addAll(
                 (typeProvider as TypeProviderImpl).
                     provideGeneratedTOBuildersForUnionTypeDef(packageName, (typeDef as UnionTypeDefinition),
                         classNameFromLeaf, leaf));
-        } else if(typeDef instanceof BitsTypeDefinition) {
+        } else if (typeDef instanceof BitsTypeDefinition) {
             genTOBuilders.add(
                 ((typeProvider as TypeProviderImpl) ).
                     provideGeneratedTOBuilderForBitsTypeDefinition(packageName, typeDef, classNameFromLeaf));
         }
-        if(genTOBuilders !== null && !genTOBuilders.isEmpty()) {
+        if (genTOBuilders !== null && !genTOBuilders.isEmpty()) {
             for (genTOBuilder : genTOBuilders) {
                 typeBuilder.addEnclosingTransferObject(genTOBuilder);
             }
@@ -1891,27 +1836,27 @@ public class BindingGeneratorImpl implements BindingGenerator {
 
     }
 
-    /**
-     * Adds the implemented types to type builder.
-     *
-     * The method passes through the list of <i>uses</i> in
-     * {@code dataNodeContainer}. For every <i>use</i> is obtained coresponding
-     * generated type from {@link BindingGeneratorImpl#allGroupings
-     * allGroupings} which is added as <i>implements type</i> to
-     * <code>builder</code>
-     *
-     * @param dataNodeContainer
-     *            element which contains the list of used YANG groupings
-     * @param builder
-     *            builder to which are added implemented types according to
-     *            <code>dataNodeContainer</code>
-     * @return generated type builder with all implemented types
+    /**\r
+     * Adds the implemented types to type builder.\r
+     *\r
+     * The method passes through the list of <i>uses</i> in\r
+     * {@code dataNodeContainer}. For every <i>use</i> is obtained coresponding\r
+     * generated type from {@link BindingGeneratorImpl#allGroupings\r
+     * allGroupings} which is added as <i>implements type</i> to\r
+     * <code>builder</code>\r
+     *\r
+     * @param dataNodeContainer\r
+     *            element which contains the list of used YANG groupings\r
+     * @param builder\r
+     *            builder to which are added implemented types according to\r
+     *            <code>dataNodeContainer</code>\r
+     * @return generated type builder with all implemented types\r
      */
     private def addImplementedInterfaceFromUses(DataNodeContainer dataNodeContainer, GeneratedTypeBuilder builder) {
         for (usesNode : dataNodeContainer.uses) {
-            if(usesNode.groupingPath !== null) {
-                val genType = allGroupings.get(usesNode.groupingPath);
-                if(genType === null) {
+            if (usesNode.groupingPath !== null) {
+                val genType = findGroupingByPath(usesNode.groupingPath).toInstance
+                if (genType === null) {
                     throw new IllegalStateException(
                         "Grouping " + usesNode.groupingPath + "is not resolved for " + builder.name);
                 }
@@ -1920,4 +1865,39 @@ public class BindingGeneratorImpl implements BindingGenerator {
         }
         return builder;
     }
+
+    private def GeneratedTypeBuilder findChildNodeByPath(SchemaPath path) {
+        for (ctx : genCtx.values) {
+            var result = ctx.getChildNode(path)
+            if (result !== null) {
+                return result
+            }
+        }
+        return null
+    }
+
+    private def GeneratedTypeBuilder findGroupingByPath(SchemaPath path) {
+        for (ctx : genCtx.values) {
+            var result = ctx.getGrouping(path)
+            if (result !== null) {
+                return result
+            }
+        }
+        return null
+    }
+
+    private def GeneratedTypeBuilder findCaseByPath(SchemaPath path) {
+        for (ctx : genCtx.values) {
+            var result = ctx.getCase(path)
+            if (result !== null) {
+                return result
+            }
+        }
+        return null
+    }
+\r
+\r
+    public def getModuleContexts() {\r
+        genCtx;\r
+    }
 }