Added instance identifier support. 00/900/1
authorTony Tkacik <ttkacik@cisco.com>
Mon, 19 Aug 2013 10:50:11 +0000 (12:50 +0200)
committerTony Tkacik <ttkacik@cisco.com>
Mon, 19 Aug 2013 10:50:11 +0000 (12:50 +0200)
- Added Identifiable and Identifier concepts to YANG binding
- Updated binding generator to use Identifiable and Identifier for list statements.

Signed-off-by: Tony Tkacik <ttkacik@cisco.com>
code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/generator/impl/BindingGeneratorImpl.java
code-generator/binding-generator-impl/src/main/java/org/opendaylight/yangtools/sal/binding/yang/types/BaseYangTypes.java
code-generator/binding-generator-impl/src/test/java/org/opendaylight/yangtools/sal/binding/generator/impl/AugmentedTypeTest.java
code-generator/binding-generator-impl/src/test/java/org/opendaylight/yangtools/sal/binding/generator/impl/GeneratedTypesTest.java
yang/yang-binding/src/main/java/org/opendaylight/yangtools/yang/binding/Identifiable.java [new file with mode: 0644]
yang/yang-binding/src/main/java/org/opendaylight/yangtools/yang/binding/Identifier.java [new file with mode: 0644]
yang/yang-binding/src/main/java/org/opendaylight/yangtools/yang/binding/InstanceIdentifier.java

index 48df06b1374a66614eab7fe0392dc32a3490639c..f20e01dc989f41266e3c1e228a4cf3b2e3691864 100644 (file)
-/*
- * 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
- */
-package org.opendaylight.yangtools.sal.binding.generator.impl;
-
-import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.moduleNamespaceToPackageName;
-import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.packageNameForGeneratedType;
-import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.parseToClassName;
-import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.parseToValidParamName;
-import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.schemaNodeToTransferObjectBuilder;
-import static org.opendaylight.yangtools.yang.model.util.SchemaContextUtil.findDataSchemaNode;
-import static org.opendaylight.yangtools.yang.model.util.SchemaContextUtil.findParentModule;
-
-import java.util.ArrayList;
-import java.util.Collections;
-import java.util.Comparator;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
-import java.util.concurrent.Future;
-
-import org.opendaylight.yangtools.binding.generator.util.ReferencedTypeImpl;
-import org.opendaylight.yangtools.binding.generator.util.Types;
-import org.opendaylight.yangtools.binding.generator.util.generated.type.builder.GeneratedTOBuilderImpl;
-import org.opendaylight.yangtools.binding.generator.util.generated.type.builder.GeneratedTypeBuilderImpl;
-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.GeneratedTransferObject;
-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.GeneratedPropertyBuilder;
-import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTOBuilder;
-import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTypeBuilder;
-import org.opendaylight.yangtools.sal.binding.model.api.type.builder.MethodSignatureBuilder;
-import org.opendaylight.yangtools.sal.binding.yang.types.GroupingDefinitionDependencySort;
-import org.opendaylight.yangtools.sal.binding.yang.types.TypeProviderImpl;
-import org.opendaylight.yangtools.yang.binding.DataRoot;
-import org.opendaylight.yangtools.yang.binding.RpcService;
-import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.common.RpcResult;
-import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
-import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
-import org.opendaylight.yangtools.yang.model.api.ChoiceNode;
-import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
-import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
-import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
-import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
-import org.opendaylight.yangtools.yang.model.api.IdentitySchemaNode;
-import org.opendaylight.yangtools.yang.model.api.LeafListSchemaNode;
-import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
-import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
-import org.opendaylight.yangtools.yang.model.api.Module;
-import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
-import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
-import org.opendaylight.yangtools.yang.model.api.SchemaNode;
-import org.opendaylight.yangtools.yang.model.api.SchemaPath;
-import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
-import org.opendaylight.yangtools.yang.model.api.UsesNode;
-import org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition;
-import org.opendaylight.yangtools.yang.model.api.type.EnumTypeDefinition;
-import org.opendaylight.yangtools.yang.model.api.type.EnumTypeDefinition.EnumPair;
-import org.opendaylight.yangtools.yang.model.util.DataNodeIterator;
-import org.opendaylight.yangtools.yang.model.util.ExtendedType;
-import org.opendaylight.yangtools.yang.model.util.SchemaContextUtil;
-import org.opendaylight.yangtools.yang.model.util.UnionType;
-
-public final 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 Map<String, Map<String, GeneratedTypeBuilder>> genTypeBuilders;
-
-    /**
-     * Provide methods for converting YANG types to JAVA types.
-     */
-    private TypeProvider typeProvider;
-
-    /**
-     * Holds reference to schema context to resolve data of augmented elemnt
-     * when creating augmentation builder
-     */
-    private 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 final Map<SchemaPath, GeneratedType> allGroupings = new HashMap<SchemaPath, GeneratedType>();
-
-    /**
-     * Only parent constructor is invoked.
-     */
-    public BindingGeneratorImpl() {
-        super();
-    }
-
-    /**
-     * 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
-     */
-    @Override
-    public List<Type> generateTypes(final SchemaContext context) {
-        if (context == null) {
-            throw new IllegalArgumentException("Schema Context reference cannot be NULL!");
-        }
-        if (context.getModules() == null) {
-            throw new IllegalStateException("Schema Context does not contain defined modules!");
-        }
-
-        final List<Type> generatedTypes = new ArrayList<>();
-        schemaContext = context;
-        typeProvider = new TypeProviderImpl(context);
-        final Set<Module> modules = context.getModules();
-        genTypeBuilders = new HashMap<>();
-        for (final Module module : modules) {
-
-            generatedTypes.addAll(allGroupingsToGenTypes(module));
-
-            if (false == module.getChildNodes().isEmpty()) {
-                generatedTypes.add(moduleToDataType(module));
-            }
-            generatedTypes.addAll(allTypeDefinitionsToGenTypes(module));
-            generatedTypes.addAll(allContainersToGenTypes(module));
-            generatedTypes.addAll(allListsToGenTypes(module));
-            generatedTypes.addAll(allChoicesToGenTypes(module));
-            generatedTypes.addAll(allAugmentsToGenTypes(module));
-            generatedTypes.addAll(allRPCMethodsToGenType(module));
-            generatedTypes.addAll(allNotificationsToGenType(module));
-            generatedTypes.addAll(allIdentitiesToGenTypes(module, context));
-
-        }
-        return generatedTypes;
-    }
-
-    /**
-     * 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
-     */
-    @Override
-    public List<Type> generateTypes(final SchemaContext context, final Set<Module> modules) {
-        if (context == null) {
-            throw new IllegalArgumentException("Schema Context reference cannot be NULL!");
-        }
-        if (context.getModules() == null) {
-            throw new IllegalStateException("Schema Context does not contain defined modules!");
-        }
-        if (modules == null) {
-            throw new IllegalArgumentException("Sef of Modules cannot be NULL!");
-        }
-
-        final List<Type> filteredGenTypes = new ArrayList<>();
-        schemaContext = context;
-        typeProvider = new TypeProviderImpl(context);
-        final Set<Module> contextModules = context.getModules();
-        genTypeBuilders = new HashMap<>();
-        for (final Module contextModule : contextModules) {
-            final List<Type> generatedTypes = new ArrayList<>();
-
-            generatedTypes.addAll(allGroupingsToGenTypes(contextModule));
-            if (false == contextModule.getChildNodes().isEmpty()) {
-                generatedTypes.add(moduleToDataType(contextModule));
-            }
-            generatedTypes.addAll(allTypeDefinitionsToGenTypes(contextModule));
-            generatedTypes.addAll(allContainersToGenTypes(contextModule));
-            generatedTypes.addAll(allListsToGenTypes(contextModule));
-            generatedTypes.addAll(allChoicesToGenTypes(contextModule));
-            generatedTypes.addAll(allAugmentsToGenTypes(contextModule));
-            generatedTypes.addAll(allRPCMethodsToGenType(contextModule));
-            generatedTypes.addAll(allNotificationsToGenType(contextModule));
-            generatedTypes.addAll(allIdentitiesToGenTypes(contextModule, context));
-
-            if (modules.contains(contextModule)) {
-                filteredGenTypes.addAll(generatedTypes);
-            }
-        }
-        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 List<Type> allTypeDefinitionsToGenTypes(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-        if (module.getName() == null) {
-            throw new IllegalArgumentException("Module name cannot be NULL!");
-        }
-        if (module.getTypeDefinitions() == null) {
-            throw new IllegalArgumentException("Type Definitions for module " + module.getName() + " cannot be NULL!");
-        }
-
-        final Set<TypeDefinition<?>> typeDefinitions = module.getTypeDefinitions();
-        final List<Type> generatedTypes = new ArrayList<>();
-        for (final TypeDefinition<?> typedef : typeDefinitions) {
-            if (typedef != null) {
-                final Type type = ((TypeProviderImpl) typeProvider).generatedTypeForExtendedDefinitionType(typedef);
-                if ((type != null) && !generatedTypes.contains(type)) {
-                    generatedTypes.add(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 List<Type> allContainersToGenTypes(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-
-        if (module.getName() == null) {
-            throw new IllegalArgumentException("Module name cannot be NULL!");
-        }
-
-        if (module.getChildNodes() == null) {
-            throw new IllegalArgumentException("Reference to Set of Child Nodes in module " + module.getName()
-                    + " cannot be NULL!");
-        }
-
-        final List<Type> generatedTypes = new ArrayList<>();
-        final DataNodeIterator it = new DataNodeIterator(module);
-        final List<ContainerSchemaNode> schemaContainers = it.allContainers();
-        final String basePackageName = moduleNamespaceToPackageName(module);
-        for (final ContainerSchemaNode container : schemaContainers) {
-            if (!container.isAddedByUses()) {
-                generatedTypes.add(containerToGenType(basePackageName, container));
-            }
-        }
-        return generatedTypes;
-    }
-
-    /**
-     * 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 List<Type> allListsToGenTypes(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-
-        if (module.getName() == null) {
-            throw new IllegalArgumentException("Module name cannot be NULL!");
-        }
-
-        if (module.getChildNodes() == null) {
-            throw new IllegalArgumentException("Reference to Set of Child Nodes in module " + module.getName()
-                    + " cannot be NULL!");
-        }
-
-        final List<Type> generatedTypes = new ArrayList<>();
-        final DataNodeIterator it = new DataNodeIterator(module);
-        final List<ListSchemaNode> schemaLists = it.allLists();
-        final String basePackageName = moduleNamespaceToPackageName(module);
-        if (schemaLists != null) {
-            for (final ListSchemaNode list : schemaLists) {
-                if (!list.isAddedByUses()) {
-                    generatedTypes.addAll(listToGenType(basePackageName, list));
-                }
-            }
-        }
-        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 List<GeneratedType> allChoicesToGenTypes(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-        if (module.getName() == null) {
-            throw new IllegalArgumentException("Module name cannot be NULL!");
-        }
-
-        final DataNodeIterator it = new DataNodeIterator(module);
-        final List<ChoiceNode> choiceNodes = it.allChoices();
-        final String basePackageName = moduleNamespaceToPackageName(module);
-
-        final List<GeneratedType> generatedTypes = new ArrayList<>();
-        for (final ChoiceNode choice : choiceNodes) {
-            if ((choice != null) && !choice.isAddedByUses()) {
-                generatedTypes.addAll(choiceToGeneratedType(basePackageName, choice));
-            }
-        }
-        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>
-     * 
-     */
-    private List<Type> allAugmentsToGenTypes(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-        if (module.getName() == null) {
-            throw new IllegalArgumentException("Module name cannot be NULL!");
-        }
-        if (module.getChildNodes() == null) {
-            throw new IllegalArgumentException("Reference to Set of Augmentation Definitions in module "
-                    + module.getName() + " cannot be NULL!");
-        }
-
-        final List<Type> generatedTypes = new ArrayList<>();
-        final String basePackageName = moduleNamespaceToPackageName(module);
-        final List<AugmentationSchema> augmentations = resolveAugmentations(module);
-        for (final AugmentationSchema augment : augmentations) {
-            generatedTypes.addAll(augmentationToGenTypes(basePackageName, augment));
-        }
-        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>
-     * 
-     */
-    private List<AugmentationSchema> resolveAugmentations(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-        if (module.getAugmentations() == null) {
-            throw new IllegalStateException("Augmentations Set cannot be NULL!");
-        }
-
-        final Set<AugmentationSchema> augmentations = module.getAugmentations();
-        final List<AugmentationSchema> sortedAugmentations = new ArrayList<>(augmentations);
-        Collections.sort(sortedAugmentations, new Comparator<AugmentationSchema>() {
-
-            @Override
-            public int compare(AugmentationSchema augSchema1, AugmentationSchema augSchema2) {
-
-                if (augSchema1.getTargetPath().getPath().size() > augSchema2.getTargetPath().getPath().size()) {
-                    return 1;
-                } else if (augSchema1.getTargetPath().getPath().size() < augSchema2.getTargetPath().getPath().size()) {
-                    return -1;
-                }
-                return 0;
-
-            }
-        });
-
-        return sortedAugmentations;
-    }
-
-    /**
-     * Converts whole <b>module</b> to <code>GeneratedType</code> object.
-     * Firstly is created the module builder object from which is finally
-     * 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
-     * 
-     */
-    private GeneratedType moduleToDataType(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-
-        final GeneratedTypeBuilder moduleDataTypeBuilder = moduleTypeBuilder(module, "Data");
-        addImplementedInterfaceFromUses(module, moduleDataTypeBuilder);
-        moduleDataTypeBuilder.addImplementsType(Types.typeForClass(DataRoot.class));
-
-        final String basePackageName = moduleNamespaceToPackageName(module);
-        if (moduleDataTypeBuilder != null) {
-            final Set<DataSchemaNode> dataNodes = module.getChildNodes();
-            resolveDataSchemaNodes(basePackageName, moduleDataTypeBuilder, dataNodes);
-        }
-        return moduleDataTypeBuilder.toInstance();
-    }
-
-    /**
-     * 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>
-     * 
-     */
-    private List<Type> allRPCMethodsToGenType(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-
-        if (module.getName() == null) {
-            throw new IllegalArgumentException("Module name cannot be NULL!");
-        }
-
-        if (module.getChildNodes() == null) {
-            throw new IllegalArgumentException("Reference to Set of RPC Method Definitions in module "
-                    + module.getName() + " cannot be NULL!");
-        }
-
-        final String basePackageName = moduleNamespaceToPackageName(module);
-        final Set<RpcDefinition> rpcDefinitions = module.getRpcs();
-
-        if (rpcDefinitions.isEmpty()) {
-            return Collections.emptyList();
-        }
-
-        final List<Type> genRPCTypes = new ArrayList<>();
-        final GeneratedTypeBuilder interfaceBuilder = moduleTypeBuilder(module, "Service");
-        interfaceBuilder.addImplementsType(Types.typeForClass(RpcService.class));
-        final Type future = Types.typeForClass(Future.class);
-        for (final RpcDefinition rpc : rpcDefinitions) {
-            if (rpc != null) {
-
-                String rpcName = parseToClassName(rpc.getQName().getLocalName());
-                String rpcMethodName = parseToValidParamName(rpcName);
-                MethodSignatureBuilder method = interfaceBuilder.addMethod(rpcMethodName);
-
-                final List<DataNodeIterator> rpcInOut = new ArrayList<>();
-
-                ContainerSchemaNode input = rpc.getInput();
-                ContainerSchemaNode output = rpc.getOutput();
-
-                if (input != null) {
-                    rpcInOut.add(new DataNodeIterator(input));
-                    GeneratedTypeBuilder inType = addRawInterfaceDefinition(basePackageName, input, rpcName);
-                    addImplementedInterfaceFromUses(input, inType);
-                    inType.addImplementsType(Types.DATA_OBJECT);
-                    resolveDataSchemaNodes(basePackageName, inType, input.getChildNodes());
-                    Type inTypeInstance = inType.toInstance();
-                    genRPCTypes.add(inTypeInstance);
-                    method.addParameter(inTypeInstance, "input");
-                }
-
-                Type outTypeInstance = Types.typeForClass(Void.class);
-                if (output != null) {
-                    rpcInOut.add(new DataNodeIterator(output));
-                    GeneratedTypeBuilder outType = addRawInterfaceDefinition(basePackageName, output, rpcName);
-                    addImplementedInterfaceFromUses(output, outType);
-                    outType.addImplementsType(Types.DATA_OBJECT);
-                    resolveDataSchemaNodes(basePackageName, outType, output.getChildNodes());
-                    outTypeInstance = outType.toInstance();
-                    genRPCTypes.add(outTypeInstance);
-
-                }
-
-                final Type rpcRes = Types.parameterizedTypeFor(Types.typeForClass(RpcResult.class), outTypeInstance);
-                method.setReturnType(Types.parameterizedTypeFor(future, rpcRes));
-                for (DataNodeIterator it : rpcInOut) {
-                    List<ContainerSchemaNode> nContainers = it.allContainers();
-                    if ((nContainers != null) && !nContainers.isEmpty()) {
-                        for (final ContainerSchemaNode container : nContainers) {
-                            if (!container.isAddedByUses()) {
-                                genRPCTypes.add(containerToGenType(basePackageName, container));
-                            }
-                        }
-                    }
-                    List<ListSchemaNode> nLists = it.allLists();
-                    if ((nLists != null) && !nLists.isEmpty()) {
-                        for (final ListSchemaNode list : nLists) {
-                            if (!list.isAddedByUses()) {
-                                genRPCTypes.addAll(listToGenType(basePackageName, list));
-                            }
-                        }
-                    }
-                }
-            }
-        }
-        genRPCTypes.add(interfaceBuilder.toInstance());
-        return genRPCTypes;
-    }
-
-    /**
-     * 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>
-     * 
-     */
-    private List<Type> allNotificationsToGenType(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-
-        if (module.getName() == null) {
-            throw new IllegalArgumentException("Module name cannot be NULL!");
-        }
-
-        if (module.getChildNodes() == null) {
-            throw new IllegalArgumentException("Reference to Set of Notification Definitions in module "
-                    + module.getName() + " cannot be NULL!");
-        }
-
-        final String basePackageName = moduleNamespaceToPackageName(module);
-        final List<Type> genNotifyTypes = new ArrayList<>();
-        final Set<NotificationDefinition> notifications = module.getNotifications();
-
-        for (final NotificationDefinition notification : notifications) {
-            if (notification != null) {
-                DataNodeIterator it = new DataNodeIterator(notification);
-
-                // Containers
-                for (ContainerSchemaNode node : it.allContainers()) {
-                    if (!node.isAddedByUses()) {
-                        genNotifyTypes.add(containerToGenType(basePackageName, node));
-                    }
-                }
-                // Lists
-                for (ListSchemaNode node : it.allLists()) {
-                    if (!node.isAddedByUses()) {
-                        genNotifyTypes.addAll(listToGenType(basePackageName, node));
-                    }
-                }
-                final GeneratedTypeBuilder notificationTypeBuilder = addDefaultInterfaceDefinition(basePackageName,
-                        notification);
-                notificationTypeBuilder.addImplementsType(Types
-                        .typeForClass(org.opendaylight.yangtools.yang.binding.Notification.class));
-                // Notification object
-                resolveDataSchemaNodes(basePackageName, notificationTypeBuilder, notification.getChildNodes());
-                genNotifyTypes.add(notificationTypeBuilder.toInstance());
-            }
-        }
-        return genNotifyTypes;
-    }
-
-    /**
-     * 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>
-     * 
-     */
-    private List<Type> allIdentitiesToGenTypes(final Module module, final SchemaContext context) {
-        List<Type> genTypes = new ArrayList<>();
-
-        final Set<IdentitySchemaNode> schemaIdentities = module.getIdentities();
-
-        final String basePackageName = moduleNamespaceToPackageName(module);
-
-        if (schemaIdentities != null && !schemaIdentities.isEmpty()) {
-            for (final IdentitySchemaNode identity : schemaIdentities) {
-                genTypes.add(identityToGenType(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 containing package name to which identity belongs
-     * @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>
-     * 
-     */
-    private GeneratedType identityToGenType(final String basePackageName, final IdentitySchemaNode identity,
-            final SchemaContext context) {
-        if (identity == null) {
-            return null;
-        }
-
-        final String packageName = packageNameForGeneratedType(basePackageName, identity.getPath());
-        final String genTypeName = parseToClassName(identity.getQName().getLocalName());
-        final GeneratedTOBuilderImpl newType = new GeneratedTOBuilderImpl(packageName, genTypeName);
-
-        IdentitySchemaNode baseIdentity = identity.getBaseIdentity();
-        if (baseIdentity != null) {
-            Module baseIdentityParentModule = SchemaContextUtil.findParentModule(context, baseIdentity);
-
-            final String returnTypePkgName = moduleNamespaceToPackageName(baseIdentityParentModule);
-            final String returnTypeName = parseToClassName(baseIdentity.getQName().getLocalName());
-
-            GeneratedTransferObject gto = new GeneratedTOBuilderImpl(returnTypePkgName, returnTypeName).toInstance();
-            newType.setExtendsType(gto);
-        } else {
-            newType.setExtendsType(Types.getBaseIdentityTO());
-        }
-        newType.setAbstract(true);
-        return newType.toInstance();
-    }
-
-    /**
-     * 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>)
-     * 
-     */
-    private List<Type> allGroupingsToGenTypes(final Module module) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module parameter can not be null");
-        }
-        final List<Type> genTypes = new ArrayList<>();
-        final String basePackageName = moduleNamespaceToPackageName(module);
-        final Set<GroupingDefinition> groupings = module.getGroupings();
-        List<GroupingDefinition> groupingsSortedByDependencies;
-
-        groupingsSortedByDependencies = GroupingDefinitionDependencySort.sort(groupings);
-
-        for (final GroupingDefinition grouping : groupingsSortedByDependencies) {
-            GeneratedType genType = groupingToGenType(basePackageName, grouping);
-            genTypes.add(genType);
-            SchemaPath schemaPath = grouping.getPath();
-            allGroupings.put(schemaPath, genType);
-        }
-        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 containing name of package to which grouping belongs.
-     * @param grouping
-     *            GroupingDefinition which contains data about grouping
-     * @return GeneratedType which is generated from grouping (object of type
-     *         <code>GroupingDefinition</code>)
-     */
-    private GeneratedType groupingToGenType(final String basePackageName, GroupingDefinition grouping) {
-        if (grouping == null) {
-            return null;
-        }
-
-        final String packageName = packageNameForGeneratedType(basePackageName, grouping.getPath());
-        final Set<DataSchemaNode> schemaNodes = grouping.getChildNodes();
-        final GeneratedTypeBuilder typeBuilder = addDefaultInterfaceDefinition(packageName, grouping);
-
-        resolveDataSchemaNodes(basePackageName, typeBuilder, schemaNodes);
-        return typeBuilder.toInstance();
-    }
-
-    /**
-     * 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
-     */
-    private EnumTypeDefinition enumTypeDefFromExtendedType(final TypeDefinition<?> typeDefinition) {
-        if (typeDefinition != null) {
-            if (typeDefinition.getBaseType() instanceof EnumTypeDefinition) {
-                return (EnumTypeDefinition) typeDefinition.getBaseType();
-            } else if (typeDefinition.getBaseType() instanceof ExtendedType) {
-                return enumTypeDefFromExtendedType(typeDefinition.getBaseType());
-            }
-        }
-        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>
-     */
-    private EnumBuilder resolveInnerEnumFromTypeDefinition(final EnumTypeDefinition enumTypeDef, final String enumName,
-            final GeneratedTypeBuilder typeBuilder) {
-        if ((enumTypeDef != null) && (typeBuilder != null) && (enumTypeDef.getQName() != null)
-                && (enumTypeDef.getQName().getLocalName() != null)) {
-
-            final String enumerationName = parseToClassName(enumName);
-            final EnumBuilder enumBuilder = typeBuilder.addEnumeration(enumerationName);
-
-            if (enumBuilder != null) {
-                final List<EnumPair> enums = enumTypeDef.getValues();
-                if (enums != null) {
-                    int listIndex = 0;
-                    for (final EnumPair enumPair : enums) {
-                        if (enumPair != null) {
-                            final String enumPairName = parseToClassName(enumPair.getName());
-                            Integer enumPairValue = enumPair.getValue();
-
-                            if (enumPairValue == null) {
-                                enumPairValue = listIndex;
-                            }
-                            enumBuilder.addValue(enumPairName, enumPairValue);
-                            listIndex++;
-                        }
-                    }
-                }
-                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
-     */
-    private GeneratedTypeBuilder moduleTypeBuilder(final Module module, final String postfix) {
-        if (module == null) {
-            throw new IllegalArgumentException("Module reference cannot be NULL!");
-        }
-        String packageName = moduleNamespaceToPackageName(module);
-        final String moduleName = parseToClassName(module.getName()) + 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>
-     */
-    private List<Type> augmentationToGenTypes(final String augmentPackageName, final AugmentationSchema augSchema) {
-        if (augmentPackageName == null) {
-            throw new IllegalArgumentException("Package Name cannot be NULL!");
-        }
-        if (augSchema == null) {
-            throw new IllegalArgumentException("Augmentation Schema cannot be NULL!");
-        }
-        if (augSchema.getTargetPath() == null) {
-            throw new IllegalStateException("Augmentation Schema does not contain Target Path (Target Path is NULL).");
-        }
-
-        final List<Type> genTypes = new ArrayList<>();
-
-        // EVERY augmented interface will extends Augmentation<T> interface
-        // and DataObject interface!!!
-        final SchemaPath targetPath = augSchema.getTargetPath();
-        final DataSchemaNode targetSchemaNode = findDataSchemaNode(schemaContext, targetPath);
-        if ((targetSchemaNode != null) && (targetSchemaNode.getQName() != null)
-                && (targetSchemaNode.getQName().getLocalName() != null)) {
-            final Module targetModule = findParentModule(schemaContext, targetSchemaNode);
-            final String targetBasePackage = moduleNamespaceToPackageName(targetModule);
-            final String targetPackageName = packageNameForGeneratedType(targetBasePackage, targetSchemaNode.getPath());
-            final String targetSchemaNodeName = targetSchemaNode.getQName().getLocalName();
-            final Set<DataSchemaNode> augChildNodes = augSchema.getChildNodes();
-
-            if (!(targetSchemaNode instanceof ChoiceNode)) {
-                final GeneratedTypeBuilder augTypeBuilder = addRawAugmentGenTypeDefinition(augmentPackageName,
-                        targetPackageName, targetSchemaNodeName, augSchema);
-                final GeneratedType augType = augTypeBuilder.toInstance();
-                genTypes.add(augType);
-            } else {
-                final Type refChoiceType = new ReferencedTypeImpl(targetPackageName,
-                        parseToClassName(targetSchemaNodeName));
-                final ChoiceNode choiceTarget = (ChoiceNode) targetSchemaNode;
-                final Set<ChoiceCaseNode> choiceCaseNodes = choiceTarget.getCases();
-                genTypes.addAll(generateTypesFromAugmentedChoiceCases(augmentPackageName, refChoiceType,
-                        choiceCaseNodes));
-            }
-            genTypes.addAll(augmentationBodyToGenTypes(augmentPackageName, augChildNodes));
-        }
-        return genTypes;
-    }
-
-    /**
-     * 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
-     */
-    private GeneratedTypeBuilder addRawAugmentGenTypeDefinition(final String augmentPackageName,
-            final String targetPackageName, final String targetSchemaNodeName, final AugmentationSchema augSchema) {
-        final String targetTypeName = parseToClassName(targetSchemaNodeName);
-        Map<String, GeneratedTypeBuilder> augmentBuilders = genTypeBuilders.get(augmentPackageName);
-        if (augmentBuilders == null) {
-            augmentBuilders = new HashMap<>();
-            genTypeBuilders.put(augmentPackageName, augmentBuilders);
-        }
-
-        final String augTypeName = augGenTypeName(augmentBuilders, targetTypeName);
-        final Type targetTypeRef = new ReferencedTypeImpl(targetPackageName, targetTypeName);
-        final Set<DataSchemaNode> augChildNodes = augSchema.getChildNodes();
-
-        final GeneratedTypeBuilder augTypeBuilder = new GeneratedTypeBuilderImpl(augmentPackageName, augTypeName);
-
-        augTypeBuilder.addImplementsType(Types.DATA_OBJECT);
-        augTypeBuilder.addImplementsType(Types.augmentationTypeFor(targetTypeRef));
-        addImplementedInterfaceFromUses(augSchema, augTypeBuilder);
-
-        augSchemaNodeToMethods(augmentPackageName, augTypeBuilder, augChildNodes);
-        augmentBuilders.put(augTypeName, augTypeBuilder);
-        return augTypeBuilder;
-    }
-
-    /**
-     * 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 List<Type> augmentationBodyToGenTypes(final String augBasePackageName,
-            final Set<DataSchemaNode> augChildNodes) {
-        final List<Type> genTypes = new ArrayList<>();
-        final List<DataNodeIterator> augSchemaIts = new ArrayList<>();
-        for (final DataSchemaNode childNode : augChildNodes) {
-            if (childNode instanceof DataNodeContainer) {
-                augSchemaIts.add(new DataNodeIterator((DataNodeContainer) childNode));
-
-                if (childNode instanceof ContainerSchemaNode) {
-                    genTypes.add(containerToGenType(augBasePackageName, (ContainerSchemaNode) childNode));
-                } else if (childNode instanceof ListSchemaNode) {
-                    genTypes.addAll(listToGenType(augBasePackageName, (ListSchemaNode) childNode));
-                }
-            } else if (childNode instanceof ChoiceNode) {
-                final ChoiceNode choice = (ChoiceNode) childNode;
-                for (final ChoiceCaseNode caseNode : choice.getCases()) {
-                    augSchemaIts.add(new DataNodeIterator(caseNode));
-                }
-                genTypes.addAll(choiceToGeneratedType(augBasePackageName, (ChoiceNode) childNode));
-            }
-        }
-
-        for (final DataNodeIterator it : augSchemaIts) {
-            final List<ContainerSchemaNode> augContainers = it.allContainers();
-            final List<ListSchemaNode> augLists = it.allLists();
-            final List<ChoiceNode> augChoices = it.allChoices();
-
-            if (augContainers != null) {
-                for (final ContainerSchemaNode container : augContainers) {
-                    genTypes.add(containerToGenType(augBasePackageName, container));
-                }
-            }
-            if (augLists != null) {
-                for (final ListSchemaNode list : augLists) {
-                    genTypes.addAll(listToGenType(augBasePackageName, list));
-                }
-            }
-            if (augChoices != null) {
-                for (final ChoiceNode 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
-     */
-    private String augGenTypeName(final Map<String, GeneratedTypeBuilder> builders, final String genTypeName) {
-        String augTypeName = genTypeName;
-
-        int index = 1;
-        while ((builders != null) && builders.containsKey(genTypeName + index)) {
-            index++;
-        }
-        augTypeName += index;
-        return augTypeName;
-    }
-
-    /**
-     * 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 with name of the package to which the superior node
-     *            belongs
-     * @param containerNode
-     *            container schema node with the data about childs nodes and
-     *            schema paths
-     * @return generated type for <code>containerNode</code>
-     */
-    private GeneratedType containerToGenType(final String basePackageName, ContainerSchemaNode containerNode) {
-        if (containerNode == null) {
-            return null;
-        }
-
-        final String packageName = packageNameForGeneratedType(basePackageName, containerNode.getPath());
-        final Set<DataSchemaNode> schemaNodes = containerNode.getChildNodes();
-        final GeneratedTypeBuilder typeBuilder = addDefaultInterfaceDefinition(packageName, containerNode);
-
-        resolveDataSchemaNodes(basePackageName, typeBuilder, schemaNodes);
-        return typeBuilder.toInstance();
-    }
-
-    /**
-     * 
-     * @param basePackageName
-     * @param typeBuilder
-     * @param schemaNodes
-     * @return
-     */
-    private GeneratedTypeBuilder resolveDataSchemaNodes(final String basePackageName,
-            final GeneratedTypeBuilder typeBuilder, final Set<DataSchemaNode> schemaNodes) {
-        if ((schemaNodes != null) && (typeBuilder != null)) {
-            for (final DataSchemaNode schemaNode : schemaNodes) {
-                if (schemaNode.isAugmenting() || schemaNode.isAddedByUses()) {
-                    continue;
-                }
-                addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);
-            }
-        }
-        return typeBuilder;
-    }
-
-    private GeneratedTypeBuilder augSchemaNodeToMethods(final String basePackageName,
-            final GeneratedTypeBuilder typeBuilder, final Set<DataSchemaNode> schemaNodes) {
-        if ((schemaNodes != null) && (typeBuilder != null)) {
-            for (final DataSchemaNode schemaNode : schemaNodes) {
-                if (schemaNode.isAugmenting()) {
-                    addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);
-                }
-            }
-        }
-        return typeBuilder;
-    }
-
-    private void addSchemaNodeToBuilderAsMethod(final String basePackageName, final DataSchemaNode schemaNode,
-            final GeneratedTypeBuilder typeBuilder) {
-        if (schemaNode != null && typeBuilder != null) {
-            if (schemaNode instanceof LeafSchemaNode) {
-                resolveLeafSchemaNodeAsMethod(typeBuilder, (LeafSchemaNode) schemaNode);
-            } else if (schemaNode instanceof LeafListSchemaNode) {
-                resolveLeafListSchemaNode(typeBuilder, (LeafListSchemaNode) schemaNode);
-            } else if (schemaNode instanceof ContainerSchemaNode) {
-                resolveContainerSchemaNode(basePackageName, typeBuilder, (ContainerSchemaNode) schemaNode);
-            } else if (schemaNode instanceof ListSchemaNode) {
-                resolveListSchemaNode(basePackageName, typeBuilder, (ListSchemaNode) schemaNode);
-            } else if (schemaNode instanceof ChoiceNode) {
-                resolveChoiceSchemaNode(basePackageName, typeBuilder, (ChoiceNode) schemaNode);
-            }
-        }
-    }
-
-    private void resolveChoiceSchemaNode(final String basePackageName, final GeneratedTypeBuilder typeBuilder,
-            final ChoiceNode choiceNode) {
-        if (basePackageName == null) {
-            throw new IllegalArgumentException("Base Package Name cannot be NULL!");
-        }
-        if (typeBuilder == null) {
-            throw new IllegalArgumentException("Generated Type Builder cannot be NULL!");
-        }
-        if (choiceNode == null) {
-            throw new IllegalArgumentException("Choice Schema Node cannot be NULL!");
-        }
-
-        final String choiceName = choiceNode.getQName().getLocalName();
-        if (choiceName != null && !choiceNode.isAddedByUses()) {
-            final String packageName = packageNameForGeneratedType(basePackageName, choiceNode.getPath());
-            final GeneratedTypeBuilder choiceType = addDefaultInterfaceDefinition(packageName, choiceNode);
-            constructGetter(typeBuilder, choiceName, choiceNode.getDescription(), choiceType);
-        }
-    }
-
-    private List<GeneratedType> choiceToGeneratedType(final String basePackageName, final ChoiceNode choiceNode) {
-        if (basePackageName == null) {
-            throw new IllegalArgumentException("Base Package Name cannot be NULL!");
-        }
-        if (choiceNode == null) {
-            throw new IllegalArgumentException("Choice Schema Node cannot be NULL!");
-        }
-
-        final List<GeneratedType> generatedTypes = new ArrayList<>();
-        final String packageName = packageNameForGeneratedType(basePackageName, choiceNode.getPath());
-        final GeneratedTypeBuilder choiceTypeBuilder = addRawInterfaceDefinition(packageName, choiceNode);
-        choiceTypeBuilder.addImplementsType(Types.DATA_OBJECT);
-        final GeneratedType choiceType = choiceTypeBuilder.toInstance();
-
-        generatedTypes.add(choiceType);
-        final Set<ChoiceCaseNode> caseNodes = choiceNode.getCases();
-        if ((caseNodes != null) && !caseNodes.isEmpty()) {
-            generatedTypes.addAll(generateTypesFromChoiceCases(basePackageName, choiceType, caseNodes));
-        }
-        return generatedTypes;
-    }
-
-    private List<GeneratedType> generateTypesFromChoiceCases(final String basePackageName, final Type refChoiceType,
-            final Set<ChoiceCaseNode> caseNodes) {
-        if (basePackageName == null) {
-            throw new IllegalArgumentException("Base Package Name cannot be NULL!");
-        }
-        if (refChoiceType == null) {
-            throw new IllegalArgumentException("Referenced Choice Type cannot be NULL!");
-        }
-        if (caseNodes == null) {
-            throw new IllegalArgumentException("Set of Choice Case Nodes cannot be NULL!");
-        }
-
-        final List<GeneratedType> generatedTypes = new ArrayList<>();
-        for (final ChoiceCaseNode caseNode : caseNodes) {
-            if (caseNode != null && !caseNode.isAddedByUses() && !caseNode.isAugmenting()) {
-                final String packageName = packageNameForGeneratedType(basePackageName, caseNode.getPath());
-                final GeneratedTypeBuilder caseTypeBuilder = addDefaultInterfaceDefinition(packageName, caseNode);
-                caseTypeBuilder.addImplementsType(refChoiceType);
-
-                final Set<DataSchemaNode> childNodes = caseNode.getChildNodes();
-                if (childNodes != null) {
-                    resolveDataSchemaNodes(basePackageName, caseTypeBuilder, childNodes);
-                }
-                generatedTypes.add(caseTypeBuilder.toInstance());
-            }
-        }
-
-        return generatedTypes;
-    }
-
-    /**
-     * 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>
-     */
-    private List<GeneratedType> generateTypesFromAugmentedChoiceCases(final String basePackageName,
-            final Type refChoiceType, final Set<ChoiceCaseNode> caseNodes) {
-        if (basePackageName == null) {
-            throw new IllegalArgumentException("Base Package Name cannot be NULL!");
-        }
-        if (refChoiceType == null) {
-            throw new IllegalArgumentException("Referenced Choice Type cannot be NULL!");
-        }
-        if (caseNodes == null) {
-            throw new IllegalArgumentException("Set of Choice Case Nodes cannot be NULL!");
-        }
-
-        final List<GeneratedType> generatedTypes = new ArrayList<>();
-        for (final ChoiceCaseNode caseNode : caseNodes) {
-            if (caseNode != null && caseNode.isAugmenting()) {
-                final String packageName = packageNameForGeneratedType(basePackageName, caseNode.getPath());
-                final GeneratedTypeBuilder caseTypeBuilder = addDefaultInterfaceDefinition(packageName, caseNode);
-                caseTypeBuilder.addImplementsType(refChoiceType);
-
-                final Set<DataSchemaNode> childNodes = caseNode.getChildNodes();
-                if (childNodes != null) {
-                    resolveDataSchemaNodes(basePackageName, caseTypeBuilder, childNodes);
-                }
-                generatedTypes.add(caseTypeBuilder.toInstance());
-            }
-        }
-
-        return generatedTypes;
-    }
-
-    private boolean resolveLeafSchemaNodeAsMethod(final GeneratedTypeBuilder typeBuilder, final LeafSchemaNode leaf) {
-        if ((leaf != null) && (typeBuilder != null)) {
-            final String leafName = leaf.getQName().getLocalName();
-            String leafDesc = leaf.getDescription();
-            if (leafDesc == null) {
-                leafDesc = "";
-            }
-
-            if (leafName != null && !leaf.isAddedByUses()) {
-                final TypeDefinition<?> typeDef = leaf.getType();
-
-                Type returnType = null;
-                if (typeDef instanceof EnumTypeDefinition) {
-                    returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef);
-                    final EnumTypeDefinition enumTypeDef = enumTypeDefFromExtendedType(typeDef);
-                    final EnumBuilder enumBuilder = resolveInnerEnumFromTypeDefinition(enumTypeDef, leafName,
-                            typeBuilder);
-
-                    if (enumBuilder != null) {
-                        returnType = new ReferencedTypeImpl(enumBuilder.getPackageName(), enumBuilder.getName());
-                    }
-                    ((TypeProviderImpl) typeProvider).putReferencedType(leaf.getPath(), returnType);
-                } else if (typeDef instanceof UnionType) {
-                    GeneratedTOBuilder genTOBuilder = addEnclosedTOToTypeBuilder(typeDef, typeBuilder, leafName);
-                    if (genTOBuilder != null) {
-                        returnType = new ReferencedTypeImpl(genTOBuilder.getPackageName(), genTOBuilder.getName());
-                    }
-                } else if (typeDef instanceof BitsTypeDefinition) {
-                    GeneratedTOBuilder genTOBuilder = addEnclosedTOToTypeBuilder(typeDef, typeBuilder, leafName);
-                    if (genTOBuilder != null) {
-                        returnType = new ReferencedTypeImpl(genTOBuilder.getPackageName(), genTOBuilder.getName());
-                    }
-                } else {
-                    returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef);
-                }
-                if (returnType != null) {
-                    constructGetter(typeBuilder, leafName, leafDesc, returnType);
-                    return true;
-                }
-            }
-        }
-        return false;
-    }
-
-    private boolean resolveLeafSchemaNodeAsProperty(final GeneratedTOBuilder toBuilder, final LeafSchemaNode leaf,
-            boolean isReadOnly) {
-        if ((leaf != null) && (toBuilder != null)) {
-            final String leafName = leaf.getQName().getLocalName();
-            String leafDesc = leaf.getDescription();
-            if (leafDesc == null) {
-                leafDesc = "";
-            }
-
-            if (leafName != null && !leaf.isAddedByUses()) {
-                final TypeDefinition<?> typeDef = leaf.getType();
-
-                // TODO: properly resolve enum types
-                final Type returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef);
-
-                if (returnType != null) {
-                    final GeneratedPropertyBuilder propBuilder = toBuilder.addProperty(parseToClassName(leafName));
-
-                    propBuilder.setReadOnly(isReadOnly);
-                    propBuilder.setReturnType(returnType);
-                    propBuilder.setComment(leafDesc);
-
-                    toBuilder.addEqualsIdentity(propBuilder);
-                    toBuilder.addHashIdentity(propBuilder);
-                    toBuilder.addToStringProperty(propBuilder);
-
-                    return true;
-                }
-            }
-        }
-        return false;
-    }
-
-    private boolean resolveLeafListSchemaNode(final GeneratedTypeBuilder typeBuilder, final LeafListSchemaNode node) {
-        if ((node != null) && (typeBuilder != null)) {
-            final String nodeName = node.getQName().getLocalName();
-            String nodeDesc = node.getDescription();
-            if (nodeDesc == null) {
-                nodeDesc = "";
-            }
-
-            if (nodeName != null && !node.isAddedByUses()) {
-                final TypeDefinition<?> type = node.getType();
-                final Type listType = Types.listTypeFor(typeProvider.javaTypeForSchemaDefinitionType(type));
-
-                constructGetter(typeBuilder, nodeName, nodeDesc, listType);
-                return true;
-            }
-        }
-        return false;
-    }
-
-    private boolean resolveContainerSchemaNode(final String basePackageName, final GeneratedTypeBuilder typeBuilder,
-            final ContainerSchemaNode containerNode) {
-        if ((containerNode != null) && (typeBuilder != null)) {
-            final String nodeName = containerNode.getQName().getLocalName();
-
-            if (nodeName != null && !containerNode.isAddedByUses()) {
-                final String packageName = packageNameForGeneratedType(basePackageName, containerNode.getPath());
-
-                final GeneratedTypeBuilder rawGenType = addDefaultInterfaceDefinition(packageName, containerNode);
-                constructGetter(typeBuilder, nodeName, containerNode.getDescription(), rawGenType);
-
-                return true;
-            }
-        }
-        return false;
-    }
-
-    private boolean resolveListSchemaNode(final String basePackageName, final GeneratedTypeBuilder typeBuilder,
-            final ListSchemaNode schemaNode) {
-        if ((schemaNode != null) && (typeBuilder != null)) {
-            final String listName = schemaNode.getQName().getLocalName();
-
-            if (listName != null && !schemaNode.isAddedByUses()) {
-                final String packageName = packageNameForGeneratedType(basePackageName, schemaNode.getPath());
-                final GeneratedTypeBuilder rawGenType = addDefaultInterfaceDefinition(packageName, schemaNode);
-                constructGetter(typeBuilder, listName, schemaNode.getDescription(), Types.listTypeFor(rawGenType));
-                return true;
-            }
-        }
-        return false;
-    }
-
-    /**
-     * Method instantiates new Generated Type Builder and sets the implements
-     * definitions of Data Object and Augmentable.
-     * 
-     * @param packageName
-     *            Generated Type Package Name
-     * @param schemaNode
-     *            Schema Node definition
-     * @return Generated Type Builder instance for Schema Node definition
-     */
-    private GeneratedTypeBuilder addDefaultInterfaceDefinition(final String packageName, final SchemaNode schemaNode) {
-        final GeneratedTypeBuilder builder = addRawInterfaceDefinition(packageName, schemaNode, "");
-        builder.addImplementsType(Types.DATA_OBJECT);
-        if (!(schemaNode instanceof GroupingDefinition)) {
-            builder.addImplementsType(Types.augmentableTypeFor(builder));
-        }
-
-        if (schemaNode instanceof DataNodeContainer) {
-            addImplementedInterfaceFromUses((DataNodeContainer) schemaNode, builder);
-        }
-
-        return builder;
-    }
-
-    /**
-     * 
-     * @param packageName
-     * @param schemaNode
-     * @return
-     */
-    private GeneratedTypeBuilder addRawInterfaceDefinition(final String packageName, final SchemaNode schemaNode) {
-        return addRawInterfaceDefinition(packageName, schemaNode, "");
-    }
-
-    private GeneratedTypeBuilder addRawInterfaceDefinition(final String packageName, final SchemaNode schemaNode,
-            final String prefix) {
-        if (schemaNode == null) {
-            throw new IllegalArgumentException("Data Schema Node cannot be NULL!");
-        }
-        if (packageName == null) {
-            throw new IllegalArgumentException("Package Name for Generated Type cannot be NULL!");
-        }
-        if (schemaNode.getQName() == null) {
-            throw new IllegalArgumentException("QName for Data Schema Node cannot be NULL!");
-        }
-        final String schemaNodeName = schemaNode.getQName().getLocalName();
-        if (schemaNodeName == null) {
-            throw new IllegalArgumentException("Local Name of QName for Data Schema Node cannot be NULL!");
-        }
-
-        final String genTypeName;
-        if (prefix == null) {
-            genTypeName = parseToClassName(schemaNodeName);
-        } else {
-            genTypeName = prefix + parseToClassName(schemaNodeName);
-        }
-
-        final GeneratedTypeBuilder newType = new GeneratedTypeBuilderImpl(packageName, genTypeName);
-        if (!genTypeBuilders.containsKey(packageName)) {
-            final Map<String, GeneratedTypeBuilder> builders = new HashMap<>();
-            builders.put(genTypeName, newType);
-            genTypeBuilders.put(packageName, builders);
-        } else {
-            final Map<String, GeneratedTypeBuilder> builders = genTypeBuilders.get(packageName);
-            if (!builders.containsKey(genTypeName)) {
-                builders.put(genTypeName, newType);
-            }
-        }
-        return newType;
-    }
-
-    private String getterMethodName(final String methodName) {
-        final StringBuilder method = new StringBuilder();
-        method.append("get");
-        method.append(parseToClassName(methodName));
-        return method.toString();
-    }
-
-    private String setterMethodName(final String methodName) {
-        final StringBuilder method = new StringBuilder();
-        method.append("set");
-        method.append(parseToClassName(methodName));
-        return method.toString();
-    }
-
-    private MethodSignatureBuilder constructGetter(final GeneratedTypeBuilder interfaceBuilder,
-            final String schemaNodeName, final String comment, final Type returnType) {
-        final MethodSignatureBuilder getMethod = interfaceBuilder.addMethod(getterMethodName(schemaNodeName));
-
-        getMethod.setComment(comment);
-        getMethod.setReturnType(returnType);
-
-        return getMethod;
-    }
-
-    private MethodSignatureBuilder constructSetter(final GeneratedTypeBuilder interfaceBuilder,
-            final String schemaNodeName, final String comment, final Type parameterType) {
-        final MethodSignatureBuilder setMethod = interfaceBuilder.addMethod(setterMethodName(schemaNodeName));
-
-        setMethod.setComment(comment);
-        setMethod.addParameter(parameterType, parseToValidParamName(schemaNodeName));
-        setMethod.setReturnType(Types.voidType());
-
-        return setMethod;
-    }
-
-    private List<Type> listToGenType(final String basePackageName, final ListSchemaNode list) {
-        if (basePackageName == null) {
-            throw new IllegalArgumentException("Package Name for Generated Type cannot be NULL!");
-        }
-        if (list == null) {
-            throw new IllegalArgumentException("List Schema Node cannot be NULL!");
-        }
-
-        final String packageName = packageNameForGeneratedType(basePackageName, list.getPath());
-        final GeneratedTypeBuilder typeBuilder = resolveListTypeBuilder(packageName, list);
-        final List<String> listKeys = listKeys(list);
-        GeneratedTOBuilder genTOBuilder = resolveListKeyTOBuilder(packageName, list, listKeys);
-
-        final Set<DataSchemaNode> schemaNodes = list.getChildNodes();
-
-        for (final DataSchemaNode schemaNode : schemaNodes) {
-            if (schemaNode.isAugmenting()) {
-                continue;
-            }
-            addSchemaNodeToListBuilders(basePackageName, schemaNode, typeBuilder, genTOBuilder, listKeys);
-        }
-        return typeBuildersToGenTypes(typeBuilder, genTOBuilder);
-    }
-
-    private void addSchemaNodeToListBuilders(final String basePackageName, final DataSchemaNode schemaNode,
-            final GeneratedTypeBuilder typeBuilder, final GeneratedTOBuilder genTOBuilder, final List<String> listKeys) {
-        if (schemaNode == null) {
-            throw new IllegalArgumentException("Data Schema Node cannot be NULL!");
-        }
-
-        if (typeBuilder == null) {
-            throw new IllegalArgumentException("Generated Type Builder cannot be NULL!");
-        }
-
-        if (schemaNode instanceof LeafSchemaNode) {
-            final LeafSchemaNode leaf = (LeafSchemaNode) schemaNode;
-            if (!isPartOfListKey(leaf, listKeys)) {
-                resolveLeafSchemaNodeAsMethod(typeBuilder, leaf);
-            } else {
-                resolveLeafSchemaNodeAsProperty(genTOBuilder, leaf, true);
-            }
-        } else if (schemaNode instanceof LeafListSchemaNode) {
-            resolveLeafListSchemaNode(typeBuilder, (LeafListSchemaNode) schemaNode);
-        } else if (schemaNode instanceof ContainerSchemaNode) {
-            resolveContainerSchemaNode(basePackageName, typeBuilder, (ContainerSchemaNode) schemaNode);
-        } else if (schemaNode instanceof ListSchemaNode) {
-            resolveListSchemaNode(basePackageName, typeBuilder, (ListSchemaNode) schemaNode);
-        }
-    }
-
-    private List<Type> typeBuildersToGenTypes(final GeneratedTypeBuilder typeBuilder, GeneratedTOBuilder genTOBuilder) {
-        final List<Type> genTypes = new ArrayList<>();
-        if (typeBuilder == null) {
-            throw new IllegalArgumentException("Generated Type Builder cannot be NULL!");
-        }
-
-        if (genTOBuilder != null) {
-            final GeneratedTransferObject genTO = genTOBuilder.toInstance();
-            constructGetter(typeBuilder, genTO.getName(), "Returns Primary Key of Yang List Type", genTO);
-            genTypes.add(genTO);
-        }
-        genTypes.add(typeBuilder.toInstance());
-        return genTypes;
-    }
-
-    /**
-     * @param list
-     * @return
-     */
-    private GeneratedTOBuilder resolveListKey(final String packageName, final ListSchemaNode list) {
-        final String listName = list.getQName().getLocalName() + "Key";
-        return schemaNodeToTransferObjectBuilder(packageName, list, listName);
-    }
-
-    private boolean isPartOfListKey(final LeafSchemaNode leaf, final List<String> keys) {
-        if ((leaf != null) && (keys != null) && (leaf.getQName() != null)) {
-            final String leafName = leaf.getQName().getLocalName();
-            if (keys.contains(leafName)) {
-                return true;
-            }
-        }
-        return false;
-    }
-
-    private List<String> listKeys(final ListSchemaNode list) {
-        final List<String> listKeys = new ArrayList<>();
-
-        if (list.getKeyDefinition() != null) {
-            final List<QName> keyDefinitions = list.getKeyDefinition();
-
-            for (final QName keyDefinition : keyDefinitions) {
-                listKeys.add(keyDefinition.getLocalName());
-            }
-        }
-        return listKeys;
-    }
-
-    private GeneratedTypeBuilder resolveListTypeBuilder(final String packageName, final ListSchemaNode list) {
-        if (packageName == null) {
-            throw new IllegalArgumentException("Package Name for Generated Type cannot be NULL!");
-        }
-        if (list == null) {
-            throw new IllegalArgumentException("List Schema Node cannot be NULL!");
-        }
-
-        final String schemaNodeName = list.getQName().getLocalName();
-        final String genTypeName = parseToClassName(schemaNodeName);
-
-        GeneratedTypeBuilder typeBuilder = null;
-        final Map<String, GeneratedTypeBuilder> builders = genTypeBuilders.get(packageName);
-        if (builders != null) {
-            typeBuilder = builders.get(genTypeName);
-        }
-        if (typeBuilder == null) {
-            typeBuilder = addDefaultInterfaceDefinition(packageName, list);
-        }
-        return typeBuilder;
-    }
-
-    private GeneratedTOBuilder resolveListKeyTOBuilder(final String packageName, final ListSchemaNode list,
-            final List<String> listKeys) {
-        GeneratedTOBuilder genTOBuilder = null;
-        if (listKeys.size() > 0) {
-            genTOBuilder = resolveListKey(packageName, list);
-        }
-        return genTOBuilder;
-    }
-
-    private GeneratedTOBuilder addEnclosedTOToTypeBuilder(TypeDefinition<?> typeDef, GeneratedTypeBuilder typeBuilder,
-            String leafName) {
-        String className = parseToClassName(leafName);
-        GeneratedTOBuilder genTOBuilder = null;
-        if (typeDef instanceof UnionType) {
-            genTOBuilder = ((TypeProviderImpl) typeProvider).addUnionGeneratedTypeDefinition(
-                    typeBuilder.getFullyQualifiedName(), typeDef, className);
-        } else if (typeDef instanceof BitsTypeDefinition) {
-            genTOBuilder = ((TypeProviderImpl) typeProvider).bitsTypedefToTransferObject(
-                    typeBuilder.getFullyQualifiedName(), typeDef, className);
-        }
-        if (genTOBuilder != null) {
-            typeBuilder.addEnclosingTransferObject(genTOBuilder);
-            return genTOBuilder;
-        }
-        return null;
-
-    }
-
-    /**
-     * Adds the implemented types to type builder. The method passes through the
-     * list of elements which contains {@code dataNodeContainer} and adds them
-     * 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 which contains implemented types
-     */
-    private GeneratedTypeBuilder addImplementedInterfaceFromUses(final DataNodeContainer dataNodeContainer,
-            final GeneratedTypeBuilder builder) {
-        for (UsesNode usesNode : dataNodeContainer.getUses()) {
-            if (usesNode.getGroupingPath() != null) {
-                GeneratedType genType = allGroupings.get(usesNode.getGroupingPath());
-                if (genType == null) {
-                    throw new IllegalStateException("Grouping " + usesNode.getGroupingPath() + "is not resolved for "
-                            + builder.getName());
-                }
-                builder.addImplementsType(genType);
-            }
-        }
-        return builder;
-    }
-
-}
+/*\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
+ */\r
+package org.opendaylight.yangtools.sal.binding.generator.impl;\r
+\r
+import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.moduleNamespaceToPackageName;\r
+import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.packageNameForGeneratedType;\r
+import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.parseToClassName;\r
+import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.parseToValidParamName;\r
+import static org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil.schemaNodeToTransferObjectBuilder;\r
+import static org.opendaylight.yangtools.yang.model.util.SchemaContextUtil.findDataSchemaNode;\r
+import static org.opendaylight.yangtools.yang.model.util.SchemaContextUtil.findParentModule;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collections;\r
+import java.util.Comparator;\r
+import java.util.HashMap;\r
+import java.util.List;\r
+import java.util.Map;\r
+import java.util.Set;\r
+import java.util.concurrent.Future;\r
+\r
+import org.opendaylight.yangtools.binding.generator.util.ReferencedTypeImpl;\r
+import org.opendaylight.yangtools.binding.generator.util.Types;\r
+import org.opendaylight.yangtools.binding.generator.util.generated.type.builder.GeneratedTOBuilderImpl;\r
+import org.opendaylight.yangtools.binding.generator.util.generated.type.builder.GeneratedTypeBuilderImpl;\r
+import org.opendaylight.yangtools.sal.binding.generator.api.BindingGenerator;\r
+import org.opendaylight.yangtools.sal.binding.generator.spi.TypeProvider;\r
+import org.opendaylight.yangtools.sal.binding.model.api.ConcreteType;\r
+import org.opendaylight.yangtools.sal.binding.model.api.GeneratedTransferObject;\r
+import org.opendaylight.yangtools.sal.binding.model.api.GeneratedType;\r
+import org.opendaylight.yangtools.sal.binding.model.api.ParameterizedType;\r
+import org.opendaylight.yangtools.sal.binding.model.api.Type;\r
+import org.opendaylight.yangtools.sal.binding.model.api.type.builder.EnumBuilder;\r
+import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedPropertyBuilder;\r
+import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTOBuilder;\r
+import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTypeBuilder;\r
+import org.opendaylight.yangtools.sal.binding.model.api.type.builder.MethodSignatureBuilder;\r
+import org.opendaylight.yangtools.sal.binding.yang.types.GroupingDefinitionDependencySort;\r
+import org.opendaylight.yangtools.sal.binding.yang.types.TypeProviderImpl;\r
+import org.opendaylight.yangtools.yang.binding.DataRoot;\r
+import org.opendaylight.yangtools.yang.binding.Identifiable;\r
+import org.opendaylight.yangtools.yang.binding.Identifier;\r
+import org.opendaylight.yangtools.yang.binding.RpcService;\r
+import org.opendaylight.yangtools.yang.common.QName;\r
+import org.opendaylight.yangtools.yang.common.RpcResult;\r
+import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;\r
+import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;\r
+import org.opendaylight.yangtools.yang.model.api.ChoiceNode;\r
+import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;\r
+import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;\r
+import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;\r
+import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;\r
+import org.opendaylight.yangtools.yang.model.api.IdentitySchemaNode;\r
+import org.opendaylight.yangtools.yang.model.api.LeafListSchemaNode;\r
+import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;\r
+import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;\r
+import org.opendaylight.yangtools.yang.model.api.Module;\r
+import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;\r
+import org.opendaylight.yangtools.yang.model.api.RpcDefinition;\r
+import org.opendaylight.yangtools.yang.model.api.SchemaContext;\r
+import org.opendaylight.yangtools.yang.model.api.SchemaNode;\r
+import org.opendaylight.yangtools.yang.model.api.SchemaPath;\r
+import org.opendaylight.yangtools.yang.model.api.TypeDefinition;\r
+import org.opendaylight.yangtools.yang.model.api.UsesNode;\r
+import org.opendaylight.yangtools.yang.model.api.type.BitsTypeDefinition;\r
+import org.opendaylight.yangtools.yang.model.api.type.EnumTypeDefinition;\r
+import org.opendaylight.yangtools.yang.model.api.type.EnumTypeDefinition.EnumPair;\r
+import org.opendaylight.yangtools.yang.model.util.DataNodeIterator;\r
+import org.opendaylight.yangtools.yang.model.util.ExtendedType;\r
+import org.opendaylight.yangtools.yang.model.util.SchemaContextUtil;\r
+import org.opendaylight.yangtools.yang.model.util.UnionType;\r
+\r
+public final class BindingGeneratorImpl implements BindingGenerator {\r
+\r
+    /**\r
+     * Outter key represents the package name. Outter value represents map of\r
+     * all builders in the same package. Inner key represents the schema node\r
+     * name (in JAVA class/interface name format). Inner value represents\r
+     * instance of builder for schema node specified in key part.\r
+     */\r
+    private Map<String, Map<String, GeneratedTypeBuilder>> genTypeBuilders;\r
+\r
+    /**\r
+     * Provide methods for converting YANG types to JAVA types.\r
+     */\r
+    private TypeProvider typeProvider;\r
+\r
+    /**\r
+     * Holds reference to schema context to resolve data of augmented elemnt\r
+     * when creating augmentation builder\r
+     */\r
+    private SchemaContext schemaContext;\r
+\r
+    /**\r
+     * Each grouping which is converted from schema node to generated type is\r
+     * added to this map with its Schema path as key to make it easier to get\r
+     * reference to it. In schema nodes in <code>uses</code> attribute there is\r
+     * only Schema Path but when building list of implemented interfaces for\r
+     * Schema node the object of type <code>Type</code> is required. So in this\r
+     * case is used this map.\r
+     */\r
+    private final Map<SchemaPath, GeneratedType> allGroupings = new HashMap<SchemaPath, GeneratedType>();\r
+\r
+    /**\r
+     * Only parent constructor is invoked.\r
+     */\r
+    public BindingGeneratorImpl() {\r
+        super();\r
+    }\r
+\r
+    /**\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
+     */\r
+    @Override\r
+    public List<Type> generateTypes(final SchemaContext context) {\r
+        if (context == null) {\r
+            throw new IllegalArgumentException("Schema Context reference cannot be NULL!");\r
+        }\r
+        if (context.getModules() == null) {\r
+            throw new IllegalStateException("Schema Context does not contain defined modules!");\r
+        }\r
+\r
+        final List<Type> generatedTypes = new ArrayList<>();\r
+        schemaContext = context;\r
+        typeProvider = new TypeProviderImpl(context);\r
+        final Set<Module> modules = context.getModules();\r
+        genTypeBuilders = new HashMap<>();\r
+        for (final Module module : modules) {\r
+\r
+            generatedTypes.addAll(allGroupingsToGenTypes(module));\r
+\r
+            if (false == module.getChildNodes().isEmpty()) {\r
+                generatedTypes.add(moduleToDataType(module));\r
+            }\r
+            generatedTypes.addAll(allTypeDefinitionsToGenTypes(module));\r
+            generatedTypes.addAll(allContainersToGenTypes(module));\r
+            generatedTypes.addAll(allListsToGenTypes(module));\r
+            generatedTypes.addAll(allChoicesToGenTypes(module));\r
+            generatedTypes.addAll(allAugmentsToGenTypes(module));\r
+            generatedTypes.addAll(allRPCMethodsToGenType(module));\r
+            generatedTypes.addAll(allNotificationsToGenType(module));\r
+            generatedTypes.addAll(allIdentitiesToGenTypes(module, context));\r
+\r
+        }\r
+        return generatedTypes;\r
+    }\r
+\r
+    /**\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
+     */\r
+    @Override\r
+    public List<Type> generateTypes(final SchemaContext context, final Set<Module> modules) {\r
+        if (context == null) {\r
+            throw new IllegalArgumentException("Schema Context reference cannot be NULL!");\r
+        }\r
+        if (context.getModules() == null) {\r
+            throw new IllegalStateException("Schema Context does not contain defined modules!");\r
+        }\r
+        if (modules == null) {\r
+            throw new IllegalArgumentException("Sef of Modules cannot be NULL!");\r
+        }\r
+\r
+        final List<Type> filteredGenTypes = new ArrayList<>();\r
+        schemaContext = context;\r
+        typeProvider = new TypeProviderImpl(context);\r
+        final Set<Module> contextModules = context.getModules();\r
+        genTypeBuilders = new HashMap<>();\r
+        for (final Module contextModule : contextModules) {\r
+            final List<Type> generatedTypes = new ArrayList<>();\r
+\r
+            generatedTypes.addAll(allGroupingsToGenTypes(contextModule));\r
+            if (false == contextModule.getChildNodes().isEmpty()) {\r
+                generatedTypes.add(moduleToDataType(contextModule));\r
+            }\r
+            generatedTypes.addAll(allTypeDefinitionsToGenTypes(contextModule));\r
+            generatedTypes.addAll(allContainersToGenTypes(contextModule));\r
+            generatedTypes.addAll(allListsToGenTypes(contextModule));\r
+            generatedTypes.addAll(allChoicesToGenTypes(contextModule));\r
+            generatedTypes.addAll(allAugmentsToGenTypes(contextModule));\r
+            generatedTypes.addAll(allRPCMethodsToGenType(contextModule));\r
+            generatedTypes.addAll(allNotificationsToGenType(contextModule));\r
+            generatedTypes.addAll(allIdentitiesToGenTypes(contextModule, context));\r
+\r
+            if (modules.contains(contextModule)) {\r
+                filteredGenTypes.addAll(generatedTypes);\r
+            }\r
+        }\r
+        return filteredGenTypes;\r
+    }\r
+\r
+    /**\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
+     * @return list of <code>Type</code> which are generated from extended\r
+     *         definition types (object of type <code>ExtendedType</code>)\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
+     */\r
+    private List<Type> allTypeDefinitionsToGenTypes(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+        if (module.getName() == null) {\r
+            throw new IllegalArgumentException("Module name cannot be NULL!");\r
+        }\r
+        if (module.getTypeDefinitions() == null) {\r
+            throw new IllegalArgumentException("Type Definitions for module " + module.getName() + " cannot be NULL!");\r
+        }\r
+\r
+        final Set<TypeDefinition<?>> typeDefinitions = module.getTypeDefinitions();\r
+        final List<Type> generatedTypes = new ArrayList<>();\r
+        for (final TypeDefinition<?> typedef : typeDefinitions) {\r
+            if (typedef != null) {\r
+                final Type type = ((TypeProviderImpl) typeProvider).generatedTypeForExtendedDefinitionType(typedef);\r
+                if ((type != null) && !generatedTypes.contains(type)) {\r
+                    generatedTypes.add(type);\r
+                }\r
+            }\r
+        }\r
+        return generatedTypes;\r
+    }\r
+\r
+    /**\r
+     * Converts all <b>containers</b> of the module to the list of\r
+     * <code>Type</code> objects.\r
+     * \r
+     * @param module\r
+     *            module from which is obtained DataNodeIterator to iterate over\r
+     *            all containers\r
+     * @return list of <code>Type</code> which are generated from containers\r
+     *         (objects of type <code>ContainerSchemaNode</code>)\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
+     */\r
+    private List<Type> allContainersToGenTypes(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+\r
+        if (module.getName() == null) {\r
+            throw new IllegalArgumentException("Module name cannot be NULL!");\r
+        }\r
+\r
+        if (module.getChildNodes() == null) {\r
+            throw new IllegalArgumentException("Reference to Set of Child Nodes in module " + module.getName()\r
+                    + " cannot be NULL!");\r
+        }\r
+\r
+        final List<Type> generatedTypes = new ArrayList<>();\r
+        final DataNodeIterator it = new DataNodeIterator(module);\r
+        final List<ContainerSchemaNode> schemaContainers = it.allContainers();\r
+        final String basePackageName = moduleNamespaceToPackageName(module);\r
+        for (final ContainerSchemaNode container : schemaContainers) {\r
+            if (!container.isAddedByUses()) {\r
+                generatedTypes.add(containerToGenType(basePackageName, container));\r
+            }\r
+        }\r
+        return generatedTypes;\r
+    }\r
+\r
+    /**\r
+     * Converts all <b>lists</b> of the module to the list of <code>Type</code>\r
+     * objects.\r
+     * \r
+     * @param module\r
+     *            module from which is obtained DataNodeIterator to iterate over\r
+     *            all lists\r
+     * @return list of <code>Type</code> which are generated from lists (objects\r
+     *         of type <code>ListSchemaNode</code>)\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
+     */\r
+    private List<Type> allListsToGenTypes(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+\r
+        if (module.getName() == null) {\r
+            throw new IllegalArgumentException("Module name cannot be NULL!");\r
+        }\r
+\r
+        if (module.getChildNodes() == null) {\r
+            throw new IllegalArgumentException("Reference to Set of Child Nodes in module " + module.getName()\r
+                    + " cannot be NULL!");\r
+        }\r
+\r
+        final List<Type> generatedTypes = new ArrayList<>();\r
+        final DataNodeIterator it = new DataNodeIterator(module);\r
+        final List<ListSchemaNode> schemaLists = it.allLists();\r
+        final String basePackageName = moduleNamespaceToPackageName(module);\r
+        if (schemaLists != null) {\r
+            for (final ListSchemaNode list : schemaLists) {\r
+                if (!list.isAddedByUses()) {\r
+                    generatedTypes.addAll(listToGenType(basePackageName, list));\r
+                }\r
+            }\r
+        }\r
+        return generatedTypes;\r
+    }\r
+\r
+    /**\r
+     * Converts all <b>choices</b> of the module to the list of\r
+     * <code>Type</code> objects.\r
+     * \r
+     * @param module\r
+     *            module from which is obtained DataNodeIterator to iterate over\r
+     *            all choices\r
+     * @return list of <code>Type</code> which are generated from choices\r
+     *         (objects of type <code>ChoiceNode</code>)\r
+     * @throws IllegalArgumentException\r
+     *             <ul>\r
+     *             <li>if the module equals null</li>\r
+     *             <li>if the name of module equals null</li> *\r
+     *             </ul>\r
+     * \r
+     */\r
+    private List<GeneratedType> allChoicesToGenTypes(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+        if (module.getName() == null) {\r
+            throw new IllegalArgumentException("Module name cannot be NULL!");\r
+        }\r
+\r
+        final DataNodeIterator it = new DataNodeIterator(module);\r
+        final List<ChoiceNode> choiceNodes = it.allChoices();\r
+        final String basePackageName = moduleNamespaceToPackageName(module);\r
+\r
+        final List<GeneratedType> generatedTypes = new ArrayList<>();\r
+        for (final ChoiceNode choice : choiceNodes) {\r
+            if ((choice != null) && !choice.isAddedByUses()) {\r
+                generatedTypes.addAll(choiceToGeneratedType(basePackageName, choice));\r
+            }\r
+        }\r
+        return generatedTypes;\r
+    }\r
+\r
+    /**\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
+     * @return list of <code>Type</code> which are generated from augments\r
+     *         (objects of type <code>AugmentationSchema</code>)\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
+     */\r
+    private List<Type> allAugmentsToGenTypes(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+        if (module.getName() == null) {\r
+            throw new IllegalArgumentException("Module name cannot be NULL!");\r
+        }\r
+        if (module.getChildNodes() == null) {\r
+            throw new IllegalArgumentException("Reference to Set of Augmentation Definitions in module "\r
+                    + module.getName() + " cannot be NULL!");\r
+        }\r
+\r
+        final List<Type> generatedTypes = new ArrayList<>();\r
+        final String basePackageName = moduleNamespaceToPackageName(module);\r
+        final List<AugmentationSchema> augmentations = resolveAugmentations(module);\r
+        for (final AugmentationSchema augment : augmentations) {\r
+            generatedTypes.addAll(augmentationToGenTypes(basePackageName, augment));\r
+        }\r
+        return generatedTypes;\r
+    }\r
+\r
+    /**\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
+     */\r
+    private List<AugmentationSchema> resolveAugmentations(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+        if (module.getAugmentations() == null) {\r
+            throw new IllegalStateException("Augmentations Set cannot be NULL!");\r
+        }\r
+\r
+        final Set<AugmentationSchema> augmentations = module.getAugmentations();\r
+        final List<AugmentationSchema> sortedAugmentations = new ArrayList<>(augmentations);\r
+        Collections.sort(sortedAugmentations, new Comparator<AugmentationSchema>() {\r
+\r
+            @Override\r
+            public int compare(AugmentationSchema augSchema1, AugmentationSchema augSchema2) {\r
+\r
+                if (augSchema1.getTargetPath().getPath().size() > augSchema2.getTargetPath().getPath().size()) {\r
+                    return 1;\r
+                } else if (augSchema1.getTargetPath().getPath().size() < augSchema2.getTargetPath().getPath().size()) {\r
+                    return -1;\r
+                }\r
+                return 0;\r
+\r
+            }\r
+        });\r
+\r
+        return sortedAugmentations;\r
+    }\r
+\r
+    /**\r
+     * Converts whole <b>module</b> to <code>GeneratedType</code> object.\r
+     * Firstly is created the module builder object from which is finally\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
+     */\r
+    private GeneratedType moduleToDataType(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+\r
+        final GeneratedTypeBuilder moduleDataTypeBuilder = moduleTypeBuilder(module, "Data");\r
+        addImplementedInterfaceFromUses(module, moduleDataTypeBuilder);\r
+        moduleDataTypeBuilder.addImplementsType(Types.typeForClass(DataRoot.class));\r
+\r
+        final String basePackageName = moduleNamespaceToPackageName(module);\r
+        if (moduleDataTypeBuilder != null) {\r
+            final Set<DataSchemaNode> dataNodes = module.getChildNodes();\r
+            resolveDataSchemaNodes(basePackageName, moduleDataTypeBuilder, dataNodes);\r
+        }\r
+        return moduleDataTypeBuilder.toInstance();\r
+    }\r
+\r
+    /**\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
+     * @return list of <code>Type</code> which are generated from rpcs inputs,\r
+     *         outputs + container and lists which are part of inputs or outputs\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
+     */\r
+    private List<Type> allRPCMethodsToGenType(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+\r
+        if (module.getName() == null) {\r
+            throw new IllegalArgumentException("Module name cannot be NULL!");\r
+        }\r
+\r
+        if (module.getChildNodes() == null) {\r
+            throw new IllegalArgumentException("Reference to Set of RPC Method Definitions in module "\r
+                    + module.getName() + " cannot be NULL!");\r
+        }\r
+\r
+        final String basePackageName = moduleNamespaceToPackageName(module);\r
+        final Set<RpcDefinition> rpcDefinitions = module.getRpcs();\r
+\r
+        if (rpcDefinitions.isEmpty()) {\r
+            return Collections.emptyList();\r
+        }\r
+\r
+        final List<Type> genRPCTypes = new ArrayList<>();\r
+        final GeneratedTypeBuilder interfaceBuilder = moduleTypeBuilder(module, "Service");\r
+        interfaceBuilder.addImplementsType(Types.typeForClass(RpcService.class));\r
+        final Type future = Types.typeForClass(Future.class);\r
+        for (final RpcDefinition rpc : rpcDefinitions) {\r
+            if (rpc != null) {\r
+\r
+                String rpcName = parseToClassName(rpc.getQName().getLocalName());\r
+                String rpcMethodName = parseToValidParamName(rpcName);\r
+                MethodSignatureBuilder method = interfaceBuilder.addMethod(rpcMethodName);\r
+\r
+                final List<DataNodeIterator> rpcInOut = new ArrayList<>();\r
+\r
+                ContainerSchemaNode input = rpc.getInput();\r
+                ContainerSchemaNode output = rpc.getOutput();\r
+\r
+                if (input != null) {\r
+                    rpcInOut.add(new DataNodeIterator(input));\r
+                    GeneratedTypeBuilder inType = addRawInterfaceDefinition(basePackageName, input, rpcName);\r
+                    addImplementedInterfaceFromUses(input, inType);\r
+                    inType.addImplementsType(Types.DATA_OBJECT);\r
+                    resolveDataSchemaNodes(basePackageName, inType, input.getChildNodes());\r
+                    Type inTypeInstance = inType.toInstance();\r
+                    genRPCTypes.add(inTypeInstance);\r
+                    method.addParameter(inTypeInstance, "input");\r
+                }\r
+\r
+                Type outTypeInstance = Types.typeForClass(Void.class);\r
+                if (output != null) {\r
+                    rpcInOut.add(new DataNodeIterator(output));\r
+                    GeneratedTypeBuilder outType = addRawInterfaceDefinition(basePackageName, output, rpcName);\r
+                    addImplementedInterfaceFromUses(output, outType);\r
+                    outType.addImplementsType(Types.DATA_OBJECT);\r
+                    resolveDataSchemaNodes(basePackageName, outType, output.getChildNodes());\r
+                    outTypeInstance = outType.toInstance();\r
+                    genRPCTypes.add(outTypeInstance);\r
+\r
+                }\r
+\r
+                final Type rpcRes = Types.parameterizedTypeFor(Types.typeForClass(RpcResult.class), outTypeInstance);\r
+                method.setReturnType(Types.parameterizedTypeFor(future, rpcRes));\r
+                for (DataNodeIterator it : rpcInOut) {\r
+                    List<ContainerSchemaNode> nContainers = it.allContainers();\r
+                    if ((nContainers != null) && !nContainers.isEmpty()) {\r
+                        for (final ContainerSchemaNode container : nContainers) {\r
+                            if (!container.isAddedByUses()) {\r
+                                genRPCTypes.add(containerToGenType(basePackageName, container));\r
+                            }\r
+                        }\r
+                    }\r
+                    List<ListSchemaNode> nLists = it.allLists();\r
+                    if ((nLists != null) && !nLists.isEmpty()) {\r
+                        for (final ListSchemaNode list : nLists) {\r
+                            if (!list.isAddedByUses()) {\r
+                                genRPCTypes.addAll(listToGenType(basePackageName, list));\r
+                            }\r
+                        }\r
+                    }\r
+                }\r
+            }\r
+        }\r
+        genRPCTypes.add(interfaceBuilder.toInstance());\r
+        return genRPCTypes;\r
+    }\r
+\r
+    /**\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
+     * @return list of <code>Type</code> which are generated from notification\r
+     *         (object of type <code>NotificationDefinition</code>\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
+     */\r
+    private List<Type> allNotificationsToGenType(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+\r
+        if (module.getName() == null) {\r
+            throw new IllegalArgumentException("Module name cannot be NULL!");\r
+        }\r
+\r
+        if (module.getChildNodes() == null) {\r
+            throw new IllegalArgumentException("Reference to Set of Notification Definitions in module "\r
+                    + module.getName() + " cannot be NULL!");\r
+        }\r
+\r
+        final String basePackageName = moduleNamespaceToPackageName(module);\r
+        final List<Type> genNotifyTypes = new ArrayList<>();\r
+        final Set<NotificationDefinition> notifications = module.getNotifications();\r
+\r
+        for (final NotificationDefinition notification : notifications) {\r
+            if (notification != null) {\r
+                DataNodeIterator it = new DataNodeIterator(notification);\r
+\r
+                // Containers\r
+                for (ContainerSchemaNode node : it.allContainers()) {\r
+                    if (!node.isAddedByUses()) {\r
+                        genNotifyTypes.add(containerToGenType(basePackageName, node));\r
+                    }\r
+                }\r
+                // Lists\r
+                for (ListSchemaNode node : it.allLists()) {\r
+                    if (!node.isAddedByUses()) {\r
+                        genNotifyTypes.addAll(listToGenType(basePackageName, node));\r
+                    }\r
+                }\r
+                final GeneratedTypeBuilder notificationTypeBuilder = addDefaultInterfaceDefinition(basePackageName,\r
+                        notification);\r
+                notificationTypeBuilder.addImplementsType(Types\r
+                        .typeForClass(org.opendaylight.yangtools.yang.binding.Notification.class));\r
+                // Notification object\r
+                resolveDataSchemaNodes(basePackageName, notificationTypeBuilder, notification.getChildNodes());\r
+                genNotifyTypes.add(notificationTypeBuilder.toInstance());\r
+            }\r
+        }\r
+        return genNotifyTypes;\r
+    }\r
+\r
+    /**\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
+     * @return list of <code>Type</code> which are generated from identities\r
+     *         (object of type <code>IdentitySchemaNode</code>\r
+     * \r
+     */\r
+    private List<Type> allIdentitiesToGenTypes(final Module module, final SchemaContext context) {\r
+        List<Type> genTypes = new ArrayList<>();\r
+\r
+        final Set<IdentitySchemaNode> schemaIdentities = module.getIdentities();\r
+\r
+        final String basePackageName = moduleNamespaceToPackageName(module);\r
+\r
+        if (schemaIdentities != null && !schemaIdentities.isEmpty()) {\r
+            for (final IdentitySchemaNode identity : schemaIdentities) {\r
+                genTypes.add(identityToGenType(basePackageName, identity, context));\r
+            }\r
+        }\r
+        return genTypes;\r
+    }\r
+\r
+    /**\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 basePackageName\r
+     *            string containing package name to which identity belongs\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
+     * @return GeneratedType which is generated from identity (object of type\r
+     *         <code>IdentitySchemaNode</code>\r
+     * \r
+     */\r
+    private GeneratedType identityToGenType(final String basePackageName, final IdentitySchemaNode identity,\r
+            final SchemaContext context) {\r
+        if (identity == null) {\r
+            return null;\r
+        }\r
+\r
+        final String packageName = packageNameForGeneratedType(basePackageName, identity.getPath());\r
+        final String genTypeName = parseToClassName(identity.getQName().getLocalName());\r
+        final GeneratedTOBuilderImpl newType = new GeneratedTOBuilderImpl(packageName, genTypeName);\r
+\r
+        IdentitySchemaNode baseIdentity = identity.getBaseIdentity();\r
+        if (baseIdentity != null) {\r
+            Module baseIdentityParentModule = SchemaContextUtil.findParentModule(context, baseIdentity);\r
+\r
+            final String returnTypePkgName = moduleNamespaceToPackageName(baseIdentityParentModule);\r
+            final String returnTypeName = parseToClassName(baseIdentity.getQName().getLocalName());\r
+\r
+            GeneratedTransferObject gto = new GeneratedTOBuilderImpl(returnTypePkgName, returnTypeName).toInstance();\r
+            newType.setExtendsType(gto);\r
+        } else {\r
+            newType.setExtendsType(Types.getBaseIdentityTO());\r
+        }\r
+        newType.setAbstract(true);\r
+        return newType.toInstance();\r
+    }\r
+\r
+    /**\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
+     *            module from which is obtained set of all grouping objects to\r
+     *            iterate over them\r
+     * @return list of <code>Type</code> which are generated from groupings\r
+     *         (object of type <code>GroupingDefinition</code>)\r
+     * \r
+     */\r
+    private List<Type> allGroupingsToGenTypes(final Module module) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module parameter can not be null");\r
+        }\r
+        final List<Type> genTypes = new ArrayList<>();\r
+        final String basePackageName = moduleNamespaceToPackageName(module);\r
+        final Set<GroupingDefinition> groupings = module.getGroupings();\r
+        List<GroupingDefinition> groupingsSortedByDependencies;\r
+\r
+        groupingsSortedByDependencies = GroupingDefinitionDependencySort.sort(groupings);\r
+\r
+        for (final GroupingDefinition grouping : groupingsSortedByDependencies) {\r
+            GeneratedType genType = groupingToGenType(basePackageName, grouping);\r
+            genTypes.add(genType);\r
+            SchemaPath schemaPath = grouping.getPath();\r
+            allGroupings.put(schemaPath, genType);\r
+        }\r
+        return genTypes;\r
+    }\r
+\r
+    /**\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 containing name of package to which grouping belongs.\r
+     * @param grouping\r
+     *            GroupingDefinition which contains data about grouping\r
+     * @return GeneratedType which is generated from grouping (object of type\r
+     *         <code>GroupingDefinition</code>)\r
+     */\r
+    private GeneratedType groupingToGenType(final String basePackageName, GroupingDefinition grouping) {\r
+        if (grouping == null) {\r
+            return null;\r
+        }\r
+\r
+        final String packageName = packageNameForGeneratedType(basePackageName, grouping.getPath());\r
+        final Set<DataSchemaNode> schemaNodes = grouping.getChildNodes();\r
+        final GeneratedTypeBuilder typeBuilder = addDefaultInterfaceDefinition(packageName, grouping);\r
+\r
+        resolveDataSchemaNodes(basePackageName, typeBuilder, schemaNodes);\r
+        return typeBuilder.toInstance();\r
+    }\r
+\r
+    /**\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
+     */\r
+    private EnumTypeDefinition enumTypeDefFromExtendedType(final TypeDefinition<?> typeDefinition) {\r
+        if (typeDefinition != null) {\r
+            if (typeDefinition.getBaseType() instanceof EnumTypeDefinition) {\r
+                return (EnumTypeDefinition) typeDefinition.getBaseType();\r
+            } else if (typeDefinition.getBaseType() instanceof ExtendedType) {\r
+                return enumTypeDefFromExtendedType(typeDefinition.getBaseType());\r
+            }\r
+        }\r
+        return null;\r
+    }\r
+\r
+    /**\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
+     */\r
+    private EnumBuilder resolveInnerEnumFromTypeDefinition(final EnumTypeDefinition enumTypeDef, final String enumName,\r
+            final GeneratedTypeBuilder typeBuilder) {\r
+        if ((enumTypeDef != null) && (typeBuilder != null) && (enumTypeDef.getQName() != null)\r
+                && (enumTypeDef.getQName().getLocalName() != null)) {\r
+\r
+            final String enumerationName = parseToClassName(enumName);\r
+            final EnumBuilder enumBuilder = typeBuilder.addEnumeration(enumerationName);\r
+\r
+            if (enumBuilder != null) {\r
+                final List<EnumPair> enums = enumTypeDef.getValues();\r
+                if (enums != null) {\r
+                    int listIndex = 0;\r
+                    for (final EnumPair enumPair : enums) {\r
+                        if (enumPair != null) {\r
+                            final String enumPairName = parseToClassName(enumPair.getName());\r
+                            Integer enumPairValue = enumPair.getValue();\r
+\r
+                            if (enumPairValue == null) {\r
+                                enumPairValue = listIndex;\r
+                            }\r
+                            enumBuilder.addValue(enumPairName, enumPairValue);\r
+                            listIndex++;\r
+                        }\r
+                    }\r
+                }\r
+                return enumBuilder;\r
+            }\r
+        }\r
+        return null;\r
+    }\r
+\r
+    /**\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
+     */\r
+    private GeneratedTypeBuilder moduleTypeBuilder(final Module module, final String postfix) {\r
+        if (module == null) {\r
+            throw new IllegalArgumentException("Module reference cannot be NULL!");\r
+        }\r
+        String packageName = moduleNamespaceToPackageName(module);\r
+        final String moduleName = parseToClassName(module.getName()) + postfix;\r
+\r
+        return new GeneratedTypeBuilderImpl(packageName, moduleName);\r
+\r
+    }\r
+\r
+    /**\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
+     * @return list of <code>Type</code> objects which contains generated type\r
+     *         for augmentation and for container, list and choice child nodes\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
+     */\r
+    private List<Type> augmentationToGenTypes(final String augmentPackageName, final AugmentationSchema augSchema) {\r
+        if (augmentPackageName == null) {\r
+            throw new IllegalArgumentException("Package Name cannot be NULL!");\r
+        }\r
+        if (augSchema == null) {\r
+            throw new IllegalArgumentException("Augmentation Schema cannot be NULL!");\r
+        }\r
+        if (augSchema.getTargetPath() == null) {\r
+            throw new IllegalStateException("Augmentation Schema does not contain Target Path (Target Path is NULL).");\r
+        }\r
+\r
+        final List<Type> genTypes = new ArrayList<>();\r
+\r
+        // EVERY augmented interface will extends Augmentation<T> interface\r
+        // and DataObject interface!!!\r
+        final SchemaPath targetPath = augSchema.getTargetPath();\r
+        final DataSchemaNode targetSchemaNode = findDataSchemaNode(schemaContext, targetPath);\r
+        if ((targetSchemaNode != null) && (targetSchemaNode.getQName() != null)\r
+                && (targetSchemaNode.getQName().getLocalName() != null)) {\r
+            final Module targetModule = findParentModule(schemaContext, targetSchemaNode);\r
+            final String targetBasePackage = moduleNamespaceToPackageName(targetModule);\r
+            final String targetPackageName = packageNameForGeneratedType(targetBasePackage, targetSchemaNode.getPath());\r
+            final String targetSchemaNodeName = targetSchemaNode.getQName().getLocalName();\r
+            final Set<DataSchemaNode> augChildNodes = augSchema.getChildNodes();\r
+\r
+            if (!(targetSchemaNode instanceof ChoiceNode)) {\r
+                final GeneratedTypeBuilder augTypeBuilder = addRawAugmentGenTypeDefinition(augmentPackageName,\r
+                        targetPackageName, targetSchemaNodeName, augSchema);\r
+                final GeneratedType augType = augTypeBuilder.toInstance();\r
+                genTypes.add(augType);\r
+            } else {\r
+                final Type refChoiceType = new ReferencedTypeImpl(targetPackageName,\r
+                        parseToClassName(targetSchemaNodeName));\r
+                final ChoiceNode choiceTarget = (ChoiceNode) targetSchemaNode;\r
+                final Set<ChoiceCaseNode> choiceCaseNodes = choiceTarget.getCases();\r
+                genTypes.addAll(generateTypesFromAugmentedChoiceCases(augmentPackageName, refChoiceType,\r
+                        choiceCaseNodes));\r
+            }\r
+            genTypes.addAll(augmentationBodyToGenTypes(augmentPackageName, augChildNodes));\r
+        }\r
+        return genTypes;\r
+    }\r
+\r
+    /**\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 augmentPackageName\r
+     *            string with contains the package name to which the augment\r
+     *            belongs\r
+     * @param targetPackageName\r
+     *            string with the package name to which the augmented node\r
+     *            belongs\r
+     * @param targetSchemaNodeName\r
+     *            string with the name of the augmented node\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
+     */\r
+    private GeneratedTypeBuilder addRawAugmentGenTypeDefinition(final String augmentPackageName,\r
+            final String targetPackageName, final String targetSchemaNodeName, final AugmentationSchema augSchema) {\r
+        final String targetTypeName = parseToClassName(targetSchemaNodeName);\r
+        Map<String, GeneratedTypeBuilder> augmentBuilders = genTypeBuilders.get(augmentPackageName);\r
+        if (augmentBuilders == null) {\r
+            augmentBuilders = new HashMap<>();\r
+            genTypeBuilders.put(augmentPackageName, augmentBuilders);\r
+        }\r
+\r
+        final String augTypeName = augGenTypeName(augmentBuilders, targetTypeName);\r
+        final Type targetTypeRef = new ReferencedTypeImpl(targetPackageName, targetTypeName);\r
+        final Set<DataSchemaNode> augChildNodes = augSchema.getChildNodes();\r
+\r
+        final GeneratedTypeBuilder augTypeBuilder = new GeneratedTypeBuilderImpl(augmentPackageName, augTypeName);\r
+\r
+        augTypeBuilder.addImplementsType(Types.DATA_OBJECT);\r
+        augTypeBuilder.addImplementsType(Types.augmentationTypeFor(targetTypeRef));\r
+        addImplementedInterfaceFromUses(augSchema, augTypeBuilder);\r
+\r
+        augSchemaNodeToMethods(augmentPackageName, augTypeBuilder, augChildNodes);\r
+        augmentBuilders.put(augTypeName, augTypeBuilder);\r
+        return augTypeBuilder;\r
+    }\r
+\r
+    /**\r
+     * Convert a container, list and choice subnodes (and recursivelly their\r
+     * subnodes) of augment to generated types\r
+     * \r
+     * @param augBasePackageName\r
+     *            string with the augment package name\r
+     * @param augChildNodes\r
+     *            set of data schema nodes which represents child nodes of the\r
+     *            augment\r
+     * \r
+     * @return list of <code>Type</code> which represents container, list and\r
+     *         choice subnodes of augment\r
+     */\r
+    private List<Type> augmentationBodyToGenTypes(final String augBasePackageName,\r
+            final Set<DataSchemaNode> augChildNodes) {\r
+        final List<Type> genTypes = new ArrayList<>();\r
+        final List<DataNodeIterator> augSchemaIts = new ArrayList<>();\r
+        for (final DataSchemaNode childNode : augChildNodes) {\r
+            if (childNode instanceof DataNodeContainer) {\r
+                augSchemaIts.add(new DataNodeIterator((DataNodeContainer) childNode));\r
+\r
+                if (childNode instanceof ContainerSchemaNode) {\r
+                    genTypes.add(containerToGenType(augBasePackageName, (ContainerSchemaNode) childNode));\r
+                } else if (childNode instanceof ListSchemaNode) {\r
+                    genTypes.addAll(listToGenType(augBasePackageName, (ListSchemaNode) childNode));\r
+                }\r
+            } else if (childNode instanceof ChoiceNode) {\r
+                final ChoiceNode choice = (ChoiceNode) childNode;\r
+                for (final ChoiceCaseNode caseNode : choice.getCases()) {\r
+                    augSchemaIts.add(new DataNodeIterator(caseNode));\r
+                }\r
+                genTypes.addAll(choiceToGeneratedType(augBasePackageName, (ChoiceNode) childNode));\r
+            }\r
+        }\r
+\r
+        for (final DataNodeIterator it : augSchemaIts) {\r
+            final List<ContainerSchemaNode> augContainers = it.allContainers();\r
+            final List<ListSchemaNode> augLists = it.allLists();\r
+            final List<ChoiceNode> augChoices = it.allChoices();\r
+\r
+            if (augContainers != null) {\r
+                for (final ContainerSchemaNode container : augContainers) {\r
+                    genTypes.add(containerToGenType(augBasePackageName, container));\r
+                }\r
+            }\r
+            if (augLists != null) {\r
+                for (final ListSchemaNode list : augLists) {\r
+                    genTypes.addAll(listToGenType(augBasePackageName, list));\r
+                }\r
+            }\r
+            if (augChoices != null) {\r
+                for (final ChoiceNode choice : augChoices) {\r
+                    genTypes.addAll(choiceToGeneratedType(augBasePackageName, choice));\r
+                }\r
+            }\r
+        }\r
+        return genTypes;\r
+    }\r
+\r
+    /**\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
+     */\r
+    private String augGenTypeName(final Map<String, GeneratedTypeBuilder> builders, final String genTypeName) {\r
+        String augTypeName = genTypeName;\r
+\r
+        int index = 1;\r
+        while ((builders != null) && builders.containsKey(genTypeName + index)) {\r
+            index++;\r
+        }\r
+        augTypeName += index;\r
+        return augTypeName;\r
+    }\r
+\r
+    /**\r
+     * Converts <code>containerNode</code> to generated type. Firstly the\r
+     * generated type builder is created. The subnodes of\r
+     * <code>containerNode</code> are added as methods and the instance of\r
+     * <code>GeneratedType</code> is returned.\r
+     * \r
+     * @param basePackageName\r
+     *            string with name of the package to which the superior node\r
+     *            belongs\r
+     * @param containerNode\r
+     *            container schema node with the data about childs nodes and\r
+     *            schema paths\r
+     * @return generated type for <code>containerNode</code>\r
+     */\r
+    private GeneratedType containerToGenType(final String basePackageName, ContainerSchemaNode containerNode) {\r
+        if (containerNode == null) {\r
+            return null;\r
+        }\r
+\r
+        final String packageName = packageNameForGeneratedType(basePackageName, containerNode.getPath());\r
+        final Set<DataSchemaNode> schemaNodes = containerNode.getChildNodes();\r
+        final GeneratedTypeBuilder typeBuilder = addDefaultInterfaceDefinition(packageName, containerNode);\r
+\r
+        resolveDataSchemaNodes(basePackageName, typeBuilder, schemaNodes);\r
+        return typeBuilder.toInstance();\r
+    }\r
+\r
+    /**\r
+     * \r
+     * @param basePackageName\r
+     * @param typeBuilder\r
+     * @param schemaNodes\r
+     * @return\r
+     */\r
+    private GeneratedTypeBuilder resolveDataSchemaNodes(final String basePackageName,\r
+            final GeneratedTypeBuilder typeBuilder, final Set<DataSchemaNode> schemaNodes) {\r
+        if ((schemaNodes != null) && (typeBuilder != null)) {\r
+            for (final DataSchemaNode schemaNode : schemaNodes) {\r
+                if (schemaNode.isAugmenting() || schemaNode.isAddedByUses()) {\r
+                    continue;\r
+                }\r
+                addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);\r
+            }\r
+        }\r
+        return typeBuilder;\r
+    }\r
+\r
+    private GeneratedTypeBuilder augSchemaNodeToMethods(final String basePackageName,\r
+            final GeneratedTypeBuilder typeBuilder, final Set<DataSchemaNode> schemaNodes) {\r
+        if ((schemaNodes != null) && (typeBuilder != null)) {\r
+            for (final DataSchemaNode schemaNode : schemaNodes) {\r
+                if (schemaNode.isAugmenting()) {\r
+                    addSchemaNodeToBuilderAsMethod(basePackageName, schemaNode, typeBuilder);\r
+                }\r
+            }\r
+        }\r
+        return typeBuilder;\r
+    }\r
+\r
+    private void addSchemaNodeToBuilderAsMethod(final String basePackageName, final DataSchemaNode schemaNode,\r
+            final GeneratedTypeBuilder typeBuilder) {\r
+        if (schemaNode != null && typeBuilder != null) {\r
+            if (schemaNode instanceof LeafSchemaNode) {\r
+                resolveLeafSchemaNodeAsMethod(typeBuilder, (LeafSchemaNode) schemaNode);\r
+            } else if (schemaNode instanceof LeafListSchemaNode) {\r
+                resolveLeafListSchemaNode(typeBuilder, (LeafListSchemaNode) schemaNode);\r
+            } else if (schemaNode instanceof ContainerSchemaNode) {\r
+                resolveContainerSchemaNode(basePackageName, typeBuilder, (ContainerSchemaNode) schemaNode);\r
+            } else if (schemaNode instanceof ListSchemaNode) {\r
+                resolveListSchemaNode(basePackageName, typeBuilder, (ListSchemaNode) schemaNode);\r
+            } else if (schemaNode instanceof ChoiceNode) {\r
+                resolveChoiceSchemaNode(basePackageName, typeBuilder, (ChoiceNode) schemaNode);\r
+            }\r
+        }\r
+    }\r
+\r
+    private void resolveChoiceSchemaNode(final String basePackageName, final GeneratedTypeBuilder typeBuilder,\r
+            final ChoiceNode choiceNode) {\r
+        if (basePackageName == null) {\r
+            throw new IllegalArgumentException("Base Package Name cannot be NULL!");\r
+        }\r
+        if (typeBuilder == null) {\r
+            throw new IllegalArgumentException("Generated Type Builder cannot be NULL!");\r
+        }\r
+        if (choiceNode == null) {\r
+            throw new IllegalArgumentException("Choice Schema Node cannot be NULL!");\r
+        }\r
+\r
+        final String choiceName = choiceNode.getQName().getLocalName();\r
+        if (choiceName != null && !choiceNode.isAddedByUses()) {\r
+            final String packageName = packageNameForGeneratedType(basePackageName, choiceNode.getPath());\r
+            final GeneratedTypeBuilder choiceType = addDefaultInterfaceDefinition(packageName, choiceNode);\r
+            constructGetter(typeBuilder, choiceName, choiceNode.getDescription(), choiceType);\r
+        }\r
+    }\r
+\r
+    private List<GeneratedType> choiceToGeneratedType(final String basePackageName, final ChoiceNode choiceNode) {\r
+        if (basePackageName == null) {\r
+            throw new IllegalArgumentException("Base Package Name cannot be NULL!");\r
+        }\r
+        if (choiceNode == null) {\r
+            throw new IllegalArgumentException("Choice Schema Node cannot be NULL!");\r
+        }\r
+\r
+        final List<GeneratedType> generatedTypes = new ArrayList<>();\r
+        final String packageName = packageNameForGeneratedType(basePackageName, choiceNode.getPath());\r
+        final GeneratedTypeBuilder choiceTypeBuilder = addRawInterfaceDefinition(packageName, choiceNode);\r
+        choiceTypeBuilder.addImplementsType(Types.DATA_OBJECT);\r
+        final GeneratedType choiceType = choiceTypeBuilder.toInstance();\r
+\r
+        generatedTypes.add(choiceType);\r
+        final Set<ChoiceCaseNode> caseNodes = choiceNode.getCases();\r
+        if ((caseNodes != null) && !caseNodes.isEmpty()) {\r
+            generatedTypes.addAll(generateTypesFromChoiceCases(basePackageName, choiceType, caseNodes));\r
+        }\r
+        return generatedTypes;\r
+    }\r
+\r
+    private List<GeneratedType> generateTypesFromChoiceCases(final String basePackageName, final Type refChoiceType,\r
+            final Set<ChoiceCaseNode> caseNodes) {\r
+        if (basePackageName == null) {\r
+            throw new IllegalArgumentException("Base Package Name cannot be NULL!");\r
+        }\r
+        if (refChoiceType == null) {\r
+            throw new IllegalArgumentException("Referenced Choice Type cannot be NULL!");\r
+        }\r
+        if (caseNodes == null) {\r
+            throw new IllegalArgumentException("Set of Choice Case Nodes cannot be NULL!");\r
+        }\r
+\r
+        final List<GeneratedType> generatedTypes = new ArrayList<>();\r
+        for (final ChoiceCaseNode caseNode : caseNodes) {\r
+            if (caseNode != null && !caseNode.isAddedByUses() && !caseNode.isAugmenting()) {\r
+                final String packageName = packageNameForGeneratedType(basePackageName, caseNode.getPath());\r
+                final GeneratedTypeBuilder caseTypeBuilder = addDefaultInterfaceDefinition(packageName, caseNode);\r
+                caseTypeBuilder.addImplementsType(refChoiceType);\r
+\r
+                final Set<DataSchemaNode> childNodes = caseNode.getChildNodes();\r
+                if (childNodes != null) {\r
+                    resolveDataSchemaNodes(basePackageName, caseTypeBuilder, childNodes);\r
+                }\r
+                generatedTypes.add(caseTypeBuilder.toInstance());\r
+            }\r
+        }\r
+\r
+        return generatedTypes;\r
+    }\r
+\r
+    /**\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
+     */\r
+    private List<GeneratedType> generateTypesFromAugmentedChoiceCases(final String basePackageName,\r
+            final Type refChoiceType, final Set<ChoiceCaseNode> caseNodes) {\r
+        if (basePackageName == null) {\r
+            throw new IllegalArgumentException("Base Package Name cannot be NULL!");\r
+        }\r
+        if (refChoiceType == null) {\r
+            throw new IllegalArgumentException("Referenced Choice Type cannot be NULL!");\r
+        }\r
+        if (caseNodes == null) {\r
+            throw new IllegalArgumentException("Set of Choice Case Nodes cannot be NULL!");\r
+        }\r
+\r
+        final List<GeneratedType> generatedTypes = new ArrayList<>();\r
+        for (final ChoiceCaseNode caseNode : caseNodes) {\r
+            if (caseNode != null && caseNode.isAugmenting()) {\r
+                final String packageName = packageNameForGeneratedType(basePackageName, caseNode.getPath());\r
+                final GeneratedTypeBuilder caseTypeBuilder = addDefaultInterfaceDefinition(packageName, caseNode);\r
+                caseTypeBuilder.addImplementsType(refChoiceType);\r
+\r
+                final Set<DataSchemaNode> childNodes = caseNode.getChildNodes();\r
+                if (childNodes != null) {\r
+                    resolveDataSchemaNodes(basePackageName, caseTypeBuilder, childNodes);\r
+                }\r
+                generatedTypes.add(caseTypeBuilder.toInstance());\r
+            }\r
+        }\r
+\r
+        return generatedTypes;\r
+    }\r
+\r
+    private boolean resolveLeafSchemaNodeAsMethod(final GeneratedTypeBuilder typeBuilder, final LeafSchemaNode leaf) {\r
+        if ((leaf != null) && (typeBuilder != null)) {\r
+            final String leafName = leaf.getQName().getLocalName();\r
+            String leafDesc = leaf.getDescription();\r
+            if (leafDesc == null) {\r
+                leafDesc = "";\r
+            }\r
+\r
+            if (leafName != null && !leaf.isAddedByUses()) {\r
+                final TypeDefinition<?> typeDef = leaf.getType();\r
+\r
+                Type returnType = null;\r
+                if (typeDef instanceof EnumTypeDefinition) {\r
+                    returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef);\r
+                    final EnumTypeDefinition enumTypeDef = enumTypeDefFromExtendedType(typeDef);\r
+                    final EnumBuilder enumBuilder = resolveInnerEnumFromTypeDefinition(enumTypeDef, leafName,\r
+                            typeBuilder);\r
+\r
+                    if (enumBuilder != null) {\r
+                        returnType = new ReferencedTypeImpl(enumBuilder.getPackageName(), enumBuilder.getName());\r
+                    }\r
+                    ((TypeProviderImpl) typeProvider).putReferencedType(leaf.getPath(), returnType);\r
+                } else if (typeDef instanceof UnionType) {\r
+                    GeneratedTOBuilder genTOBuilder = addEnclosedTOToTypeBuilder(typeDef, typeBuilder, leafName);\r
+                    if (genTOBuilder != null) {\r
+                        returnType = new ReferencedTypeImpl(genTOBuilder.getPackageName(), genTOBuilder.getName());\r
+                    }\r
+                } else if (typeDef instanceof BitsTypeDefinition) {\r
+                    GeneratedTOBuilder genTOBuilder = addEnclosedTOToTypeBuilder(typeDef, typeBuilder, leafName);\r
+                    if (genTOBuilder != null) {\r
+                        returnType = new ReferencedTypeImpl(genTOBuilder.getPackageName(), genTOBuilder.getName());\r
+                    }\r
+                } else {\r
+                    returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef);\r
+                }\r
+                if (returnType != null) {\r
+                    constructGetter(typeBuilder, leafName, leafDesc, returnType);\r
+                    return true;\r
+                }\r
+            }\r
+        }\r
+        return false;\r
+    }\r
+\r
+    private boolean resolveLeafSchemaNodeAsProperty(final GeneratedTOBuilder toBuilder, final LeafSchemaNode leaf,\r
+            boolean isReadOnly) {\r
+        if ((leaf != null) && (toBuilder != null)) {\r
+            final String leafName = leaf.getQName().getLocalName();\r
+            String leafDesc = leaf.getDescription();\r
+            if (leafDesc == null) {\r
+                leafDesc = "";\r
+            }\r
+\r
+            if (leafName != null && !leaf.isAddedByUses()) {\r
+                final TypeDefinition<?> typeDef = leaf.getType();\r
+\r
+                // TODO: properly resolve enum types\r
+                final Type returnType = typeProvider.javaTypeForSchemaDefinitionType(typeDef);\r
+\r
+                if (returnType != null) {\r
+                    final GeneratedPropertyBuilder propBuilder = toBuilder.addProperty(parseToClassName(leafName));\r
+\r
+                    propBuilder.setReadOnly(isReadOnly);\r
+                    propBuilder.setReturnType(returnType);\r
+                    propBuilder.setComment(leafDesc);\r
+\r
+                    toBuilder.addEqualsIdentity(propBuilder);\r
+                    toBuilder.addHashIdentity(propBuilder);\r
+                    toBuilder.addToStringProperty(propBuilder);\r
+\r
+                    return true;\r
+                }\r
+            }\r
+        }\r
+        return false;\r
+    }\r
+\r
+    private boolean resolveLeafListSchemaNode(final GeneratedTypeBuilder typeBuilder, final LeafListSchemaNode node) {\r
+        if ((node != null) && (typeBuilder != null)) {\r
+            final String nodeName = node.getQName().getLocalName();\r
+            String nodeDesc = node.getDescription();\r
+            if (nodeDesc == null) {\r
+                nodeDesc = "";\r
+            }\r
+\r
+            if (nodeName != null && !node.isAddedByUses()) {\r
+                final TypeDefinition<?> type = node.getType();\r
+                final Type listType = Types.listTypeFor(typeProvider.javaTypeForSchemaDefinitionType(type));\r
+\r
+                constructGetter(typeBuilder, nodeName, nodeDesc, listType);\r
+                return true;\r
+            }\r
+        }\r
+        return false;\r
+    }\r
+\r
+    private boolean resolveContainerSchemaNode(final String basePackageName, final GeneratedTypeBuilder typeBuilder,\r
+            final ContainerSchemaNode containerNode) {\r
+        if ((containerNode != null) && (typeBuilder != null)) {\r
+            final String nodeName = containerNode.getQName().getLocalName();\r
+\r
+            if (nodeName != null && !containerNode.isAddedByUses()) {\r
+                final String packageName = packageNameForGeneratedType(basePackageName, containerNode.getPath());\r
+\r
+                final GeneratedTypeBuilder rawGenType = addDefaultInterfaceDefinition(packageName, containerNode);\r
+                constructGetter(typeBuilder, nodeName, containerNode.getDescription(), rawGenType);\r
+\r
+                return true;\r
+            }\r
+        }\r
+        return false;\r
+    }\r
+\r
+    private boolean resolveListSchemaNode(final String basePackageName, final GeneratedTypeBuilder typeBuilder,\r
+            final ListSchemaNode schemaNode) {\r
+        if ((schemaNode != null) && (typeBuilder != null)) {\r
+            final String listName = schemaNode.getQName().getLocalName();\r
+\r
+            if (listName != null && !schemaNode.isAddedByUses()) {\r
+                final String packageName = packageNameForGeneratedType(basePackageName, schemaNode.getPath());\r
+                final GeneratedTypeBuilder rawGenType = addDefaultInterfaceDefinition(packageName, schemaNode);\r
+                constructGetter(typeBuilder, listName, schemaNode.getDescription(), Types.listTypeFor(rawGenType));\r
+                return true;\r
+            }\r
+        }\r
+        return false;\r
+    }\r
+\r
+    /**\r
+     * Method instantiates new Generated Type Builder and sets the implements\r
+     * definitions of Data Object and Augmentable.\r
+     * \r
+     * @param packageName\r
+     *            Generated Type Package Name\r
+     * @param schemaNode\r
+     *            Schema Node definition\r
+     * @return Generated Type Builder instance for Schema Node definition\r
+     */\r
+    private GeneratedTypeBuilder addDefaultInterfaceDefinition(final String packageName, final SchemaNode schemaNode) {\r
+        final GeneratedTypeBuilder builder = addRawInterfaceDefinition(packageName, schemaNode, "");\r
+        builder.addImplementsType(Types.DATA_OBJECT);\r
+        if (!(schemaNode instanceof GroupingDefinition)) {\r
+            builder.addImplementsType(Types.augmentableTypeFor(builder));\r
+        }\r
+\r
+        if (schemaNode instanceof DataNodeContainer) {\r
+            addImplementedInterfaceFromUses((DataNodeContainer) schemaNode, builder);\r
+        }\r
+\r
+        return builder;\r
+    }\r
+\r
+    /**\r
+     * \r
+     * @param packageName\r
+     * @param schemaNode\r
+     * @return\r
+     */\r
+    private GeneratedTypeBuilder addRawInterfaceDefinition(final String packageName, final SchemaNode schemaNode) {\r
+        return addRawInterfaceDefinition(packageName, schemaNode, "");\r
+    }\r
+\r
+    private GeneratedTypeBuilder addRawInterfaceDefinition(final String packageName, final SchemaNode schemaNode,\r
+            final String prefix) {\r
+        if (schemaNode == null) {\r
+            throw new IllegalArgumentException("Data Schema Node cannot be NULL!");\r
+        }\r
+        if (packageName == null) {\r
+            throw new IllegalArgumentException("Package Name for Generated Type cannot be NULL!");\r
+        }\r
+        if (schemaNode.getQName() == null) {\r
+            throw new IllegalArgumentException("QName for Data Schema Node cannot be NULL!");\r
+        }\r
+        final String schemaNodeName = schemaNode.getQName().getLocalName();\r
+        if (schemaNodeName == null) {\r
+            throw new IllegalArgumentException("Local Name of QName for Data Schema Node cannot be NULL!");\r
+        }\r
+\r
+        final String genTypeName;\r
+        if (prefix == null) {\r
+            genTypeName = parseToClassName(schemaNodeName);\r
+        } else {\r
+            genTypeName = prefix + parseToClassName(schemaNodeName);\r
+        }\r
+\r
+        final GeneratedTypeBuilder newType = new GeneratedTypeBuilderImpl(packageName, genTypeName);\r
+        if (!genTypeBuilders.containsKey(packageName)) {\r
+            final Map<String, GeneratedTypeBuilder> builders = new HashMap<>();\r
+            builders.put(genTypeName, newType);\r
+            genTypeBuilders.put(packageName, builders);\r
+        } else {\r
+            final Map<String, GeneratedTypeBuilder> builders = genTypeBuilders.get(packageName);\r
+            if (!builders.containsKey(genTypeName)) {\r
+                builders.put(genTypeName, newType);\r
+            }\r
+        }\r
+        return newType;\r
+    }\r
+\r
+    private String getterMethodName(final String methodName) {\r
+        final StringBuilder method = new StringBuilder();\r
+        method.append("get");\r
+        method.append(parseToClassName(methodName));\r
+        return method.toString();\r
+    }\r
+\r
+    private String setterMethodName(final String methodName) {\r
+        final StringBuilder method = new StringBuilder();\r
+        method.append("set");\r
+        method.append(parseToClassName(methodName));\r
+        return method.toString();\r
+    }\r
+\r
+    private MethodSignatureBuilder constructGetter(final GeneratedTypeBuilder interfaceBuilder,\r
+            final String schemaNodeName, final String comment, final Type returnType) {\r
+        final MethodSignatureBuilder getMethod = interfaceBuilder.addMethod(getterMethodName(schemaNodeName));\r
+\r
+        getMethod.setComment(comment);\r
+        getMethod.setReturnType(returnType);\r
+\r
+        return getMethod;\r
+    }\r
+\r
+    private MethodSignatureBuilder constructSetter(final GeneratedTypeBuilder interfaceBuilder,\r
+            final String schemaNodeName, final String comment, final Type parameterType) {\r
+        final MethodSignatureBuilder setMethod = interfaceBuilder.addMethod(setterMethodName(schemaNodeName));\r
+\r
+        setMethod.setComment(comment);\r
+        setMethod.addParameter(parameterType, parseToValidParamName(schemaNodeName));\r
+        setMethod.setReturnType(Types.voidType());\r
+\r
+        return setMethod;\r
+    }\r
+\r
+    private List<Type> listToGenType(final String basePackageName, final ListSchemaNode list) {\r
+        if (basePackageName == null) {\r
+            throw new IllegalArgumentException("Package Name for Generated Type cannot be NULL!");\r
+        }\r
+        if (list == null) {\r
+            throw new IllegalArgumentException("List Schema Node cannot be NULL!");\r
+        }\r
+\r
+        final String packageName = packageNameForGeneratedType(basePackageName, list.getPath());\r
+        final GeneratedTypeBuilder typeBuilder = resolveListTypeBuilder(packageName, list);\r
+        final List<String> listKeys = listKeys(list);\r
+        GeneratedTOBuilder genTOBuilder = resolveListKeyTOBuilder(packageName, list, listKeys);\r
+        \r
+        \r
+        if(genTOBuilder != null) {\r
+            ParameterizedType identifierMarker = Types.parameterizedTypeFor( Types.typeForClass(Identifier.class), typeBuilder);\r
+            ParameterizedType identifiableMarker = Types.parameterizedTypeFor(Types.typeForClass(Identifiable.class), genTOBuilder);\r
+            genTOBuilder.addImplementsType(identifierMarker);\r
+            typeBuilder.addImplementsType(identifiableMarker);\r
+        }\r
+        final Set<DataSchemaNode> schemaNodes = list.getChildNodes();\r
+\r
+        for (final DataSchemaNode schemaNode : schemaNodes) {\r
+            if (schemaNode.isAugmenting()) {\r
+                continue;\r
+            }\r
+            addSchemaNodeToListBuilders(basePackageName, schemaNode, typeBuilder, genTOBuilder, listKeys);\r
+        }\r
+        return typeBuildersToGenTypes(typeBuilder, genTOBuilder);\r
+    }\r
+\r
+    private void addSchemaNodeToListBuilders(final String basePackageName, final DataSchemaNode schemaNode,\r
+            final GeneratedTypeBuilder typeBuilder, final GeneratedTOBuilder genTOBuilder, final List<String> listKeys) {\r
+        if (schemaNode == null) {\r
+            throw new IllegalArgumentException("Data Schema Node cannot be NULL!");\r
+        }\r
+\r
+        if (typeBuilder == null) {\r
+            throw new IllegalArgumentException("Generated Type Builder cannot be NULL!");\r
+        }\r
+\r
+        if (schemaNode instanceof LeafSchemaNode) {\r
+            final LeafSchemaNode leaf = (LeafSchemaNode) schemaNode;\r
+            if (!isPartOfListKey(leaf, listKeys)) {\r
+                resolveLeafSchemaNodeAsMethod(typeBuilder, leaf);\r
+            } else {\r
+                resolveLeafSchemaNodeAsProperty(genTOBuilder, leaf, true);\r
+            }\r
+        } else if (schemaNode instanceof LeafListSchemaNode) {\r
+            resolveLeafListSchemaNode(typeBuilder, (LeafListSchemaNode) schemaNode);\r
+        } else if (schemaNode instanceof ContainerSchemaNode) {\r
+            resolveContainerSchemaNode(basePackageName, typeBuilder, (ContainerSchemaNode) schemaNode);\r
+        } else if (schemaNode instanceof ListSchemaNode) {\r
+            resolveListSchemaNode(basePackageName, typeBuilder, (ListSchemaNode) schemaNode);\r
+        }\r
+    }\r
+\r
+    private List<Type> typeBuildersToGenTypes(final GeneratedTypeBuilder typeBuilder, GeneratedTOBuilder genTOBuilder) {\r
+        final List<Type> genTypes = new ArrayList<>();\r
+        if (typeBuilder == null) {\r
+            throw new IllegalArgumentException("Generated Type Builder cannot be NULL!");\r
+        }\r
+\r
+        if (genTOBuilder != null) {\r
+            final GeneratedTransferObject genTO = genTOBuilder.toInstance();\r
+            constructGetter(typeBuilder, "key", "Returns Primary Key of Yang List Type", genTO);\r
+            genTypes.add(genTO);\r
+        }\r
+        genTypes.add(typeBuilder.toInstance());\r
+        return genTypes;\r
+    }\r
+\r
+    /**\r
+     * @param list\r
+     * @return\r
+     */\r
+    private GeneratedTOBuilder resolveListKey(final String packageName, final ListSchemaNode list) {\r
+        final String listName = list.getQName().getLocalName() + "Key";\r
+        return schemaNodeToTransferObjectBuilder(packageName, list, listName);\r
+    }\r
+\r
+    private boolean isPartOfListKey(final LeafSchemaNode leaf, final List<String> keys) {\r
+        if ((leaf != null) && (keys != null) && (leaf.getQName() != null)) {\r
+            final String leafName = leaf.getQName().getLocalName();\r
+            if (keys.contains(leafName)) {\r
+                return true;\r
+            }\r
+        }\r
+        return false;\r
+    }\r
+\r
+    private List<String> listKeys(final ListSchemaNode list) {\r
+        final List<String> listKeys = new ArrayList<>();\r
+\r
+        if (list.getKeyDefinition() != null) {\r
+            final List<QName> keyDefinitions = list.getKeyDefinition();\r
+\r
+            for (final QName keyDefinition : keyDefinitions) {\r
+                listKeys.add(keyDefinition.getLocalName());\r
+            }\r
+        }\r
+        return listKeys;\r
+    }\r
+\r
+    private GeneratedTypeBuilder resolveListTypeBuilder(final String packageName, final ListSchemaNode list) {\r
+        if (packageName == null) {\r
+            throw new IllegalArgumentException("Package Name for Generated Type cannot be NULL!");\r
+        }\r
+        if (list == null) {\r
+            throw new IllegalArgumentException("List Schema Node cannot be NULL!");\r
+        }\r
+\r
+        final String schemaNodeName = list.getQName().getLocalName();\r
+        final String genTypeName = parseToClassName(schemaNodeName);\r
+\r
+        GeneratedTypeBuilder typeBuilder = null;\r
+        final Map<String, GeneratedTypeBuilder> builders = genTypeBuilders.get(packageName);\r
+        if (builders != null) {\r
+            typeBuilder = builders.get(genTypeName);\r
+        }\r
+        if (typeBuilder == null) {\r
+            typeBuilder = addDefaultInterfaceDefinition(packageName, list);\r
+        }\r
+        return typeBuilder;\r
+    }\r
+\r
+    private GeneratedTOBuilder resolveListKeyTOBuilder(final String packageName, final ListSchemaNode list,\r
+            final List<String> listKeys) {\r
+        GeneratedTOBuilder genTOBuilder = null;\r
+        if (listKeys.size() > 0) {\r
+            genTOBuilder = resolveListKey(packageName, list);\r
+        }\r
+        return genTOBuilder;\r
+    }\r
+\r
+    private GeneratedTOBuilder addEnclosedTOToTypeBuilder(TypeDefinition<?> typeDef, GeneratedTypeBuilder typeBuilder,\r
+            String leafName) {\r
+        String className = parseToClassName(leafName);\r
+        GeneratedTOBuilder genTOBuilder = null;\r
+        if (typeDef instanceof UnionType) {\r
+            genTOBuilder = ((TypeProviderImpl) typeProvider).addUnionGeneratedTypeDefinition(\r
+                    typeBuilder.getFullyQualifiedName(), typeDef, className);\r
+        } else if (typeDef instanceof BitsTypeDefinition) {\r
+            genTOBuilder = ((TypeProviderImpl) typeProvider).bitsTypedefToTransferObject(\r
+                    typeBuilder.getFullyQualifiedName(), typeDef, className);\r
+        }\r
+        if (genTOBuilder != null) {\r
+            typeBuilder.addEnclosingTransferObject(genTOBuilder);\r
+            return genTOBuilder;\r
+        }\r
+        return null;\r
+\r
+    }\r
+\r
+    /**\r
+     * Adds the implemented types to type builder. The method passes through the\r
+     * list of elements which contains {@code dataNodeContainer} and adds them\r
+     * as <i>implements type</i> to <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 which contains implemented types\r
+     */\r
+    private GeneratedTypeBuilder addImplementedInterfaceFromUses(final DataNodeContainer dataNodeContainer,\r
+            final GeneratedTypeBuilder builder) {\r
+        for (UsesNode usesNode : dataNodeContainer.getUses()) {\r
+            if (usesNode.getGroupingPath() != null) {\r
+                GeneratedType genType = allGroupings.get(usesNode.getGroupingPath());\r
+                if (genType == null) {\r
+                    throw new IllegalStateException("Grouping " + usesNode.getGroupingPath() + "is not resolved for "\r
+                            + builder.getName());\r
+                }\r
+                builder.addImplementsType(genType);\r
+            }\r
+        }\r
+        return builder;\r
+    }\r
+\r
+}\r
index 6f9503ae4fca4c9252cf5961f92a71641b19cec4..964731d78206eab8eec72a94f11d3e0f88534088 100644 (file)
@@ -1,69 +1,70 @@
-/*
- * 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
- */
-package org.opendaylight.yangtools.sal.binding.yang.types;
-
-import java.math.BigInteger;
-import java.util.HashMap;
-import java.util.Map;
-
-import org.opendaylight.yangtools.binding.generator.util.Types;
-import org.opendaylight.yangtools.sal.binding.generator.spi.TypeProvider;
-import org.opendaylight.yangtools.sal.binding.model.api.Type;
-import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
-
-public final class BaseYangTypes {
-
-    private static Map<String, Type> typeMap = new HashMap<String, Type>();
-
-    public static final Type BOOLEAN_TYPE = Types.typeForClass(Boolean.class);
-    public static final Type EMPTY_TYPE = Types.typeForClass(Boolean.class);
-    public static final Type INT8_TYPE = Types.typeForClass(Byte.class);
-    public static final Type INT16_TYPE = Types.typeForClass(Short.class);
-    public static final Type INT32_TYPE = Types.typeForClass(Integer.class);
-    public static final Type INT64_TYPE = Types.typeForClass(Long.class);
-    public static final Type STRING_TYPE = Types.typeForClass(String.class);
-    public static final Type DECIMAL64_TYPE = Types.typeForClass(Double.class);
-    public static final Type UINT8_TYPE = Types.typeForClass(Short.class);
-    public static final Type UINT16_TYPE = Types.typeForClass(Integer.class);
-    public static final Type UINT32_TYPE = Types.typeForClass(Long.class);
-    public static final Type UINT64_TYPE = Types.typeForClass(BigInteger.class);
-    public static final Type BINARY_TYPE = Types.primitiveType("byte[]");
-
-    static {
-        typeMap.put("boolean", BOOLEAN_TYPE);
-        typeMap.put("empty", EMPTY_TYPE);
-        typeMap.put("int8", INT8_TYPE);
-        typeMap.put("int16", INT16_TYPE);
-        typeMap.put("int32", INT32_TYPE);
-        typeMap.put("int64", INT64_TYPE);
-        typeMap.put("string", STRING_TYPE);
-        typeMap.put("decimal64", DECIMAL64_TYPE);
-        typeMap.put("uint8", UINT8_TYPE);
-        typeMap.put("uint16", UINT16_TYPE);
-        typeMap.put("uint32", UINT32_TYPE);
-        typeMap.put("uint64", UINT64_TYPE);
-        typeMap.put("binary", BINARY_TYPE);
-    }
-
-    public static final TypeProvider BASE_YANG_TYPES_PROVIDER = new TypeProvider() {
-
-        @Override
-        public Type javaTypeForYangType(String type) {
-            return typeMap.get(type);
-        }
-
-        @Override
-        public Type javaTypeForSchemaDefinitionType(TypeDefinition<?> type) {
-            if (type != null) {
-                return typeMap.get(type.getQName().getLocalName());
-            }
-
-            return null;
-        }
-    };
-}
+/*\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
+ */\r
+package org.opendaylight.yangtools.sal.binding.yang.types;\r
+\r
+import java.math.BigDecimal;\r
+import java.math.BigInteger;\r
+import java.util.HashMap;\r
+import java.util.Map;\r
+\r
+import org.opendaylight.yangtools.binding.generator.util.Types;\r
+import org.opendaylight.yangtools.sal.binding.generator.spi.TypeProvider;\r
+import org.opendaylight.yangtools.sal.binding.model.api.Type;\r
+import org.opendaylight.yangtools.yang.model.api.TypeDefinition;\r
+\r
+public final class BaseYangTypes {\r
+\r
+    private static Map<String, Type> typeMap = new HashMap<String, Type>();\r
+\r
+    public static final Type BOOLEAN_TYPE = Types.typeForClass(Boolean.class);\r
+    public static final Type EMPTY_TYPE = Types.typeForClass(Boolean.class);\r
+    public static final Type INT8_TYPE = Types.typeForClass(Byte.class);\r
+    public static final Type INT16_TYPE = Types.typeForClass(Short.class);\r
+    public static final Type INT32_TYPE = Types.typeForClass(Integer.class);\r
+    public static final Type INT64_TYPE = Types.typeForClass(Long.class);\r
+    public static final Type STRING_TYPE = Types.typeForClass(String.class);\r
+    public static final Type DECIMAL64_TYPE = Types.typeForClass(BigDecimal.class);\r
+    public static final Type UINT8_TYPE = Types.typeForClass(Short.class);\r
+    public static final Type UINT16_TYPE = Types.typeForClass(Integer.class);\r
+    public static final Type UINT32_TYPE = Types.typeForClass(Long.class);\r
+    public static final Type UINT64_TYPE = Types.typeForClass(BigInteger.class);\r
+    public static final Type BINARY_TYPE = Types.primitiveType("byte[]");\r
+\r
+    static {\r
+        typeMap.put("boolean", BOOLEAN_TYPE);\r
+        typeMap.put("empty", EMPTY_TYPE);\r
+        typeMap.put("int8", INT8_TYPE);\r
+        typeMap.put("int16", INT16_TYPE);\r
+        typeMap.put("int32", INT32_TYPE);\r
+        typeMap.put("int64", INT64_TYPE);\r
+        typeMap.put("string", STRING_TYPE);\r
+        typeMap.put("decimal64", DECIMAL64_TYPE);\r
+        typeMap.put("uint8", UINT8_TYPE);\r
+        typeMap.put("uint16", UINT16_TYPE);\r
+        typeMap.put("uint32", UINT32_TYPE);\r
+        typeMap.put("uint64", UINT64_TYPE);\r
+        typeMap.put("binary", BINARY_TYPE);\r
+    }\r
+\r
+    public static final TypeProvider BASE_YANG_TYPES_PROVIDER = new TypeProvider() {\r
+\r
+        @Override\r
+        public Type javaTypeForYangType(String type) {\r
+            return typeMap.get(type);\r
+        }\r
+\r
+        @Override\r
+        public Type javaTypeForSchemaDefinitionType(TypeDefinition<?> type) {\r
+            if (type != null) {\r
+                return typeMap.get(type.getQName().getLocalName());\r
+            }\r
+\r
+            return null;\r
+        }\r
+    };\r
+}\r
index 5c569d3b4f2d986a042d97480a409995903f34f4..b8b8cf7b15ade3934939f55601f060b94a238b26 100644 (file)
-/*
- * 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
- */
-package org.opendaylight.yangtools.sal.binding.generator.impl;
-
-import static org.junit.Assert.*;
-
-import java.io.File;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Set;
-
-import org.junit.BeforeClass;
-import org.junit.Test;
-import org.opendaylight.yangtools.sal.binding.generator.api.BindingGenerator;
-import org.opendaylight.yangtools.sal.binding.model.api.GeneratedProperty;
-import org.opendaylight.yangtools.sal.binding.model.api.GeneratedTransferObject;
-import org.opendaylight.yangtools.sal.binding.model.api.GeneratedType;
-import org.opendaylight.yangtools.sal.binding.model.api.MethodSignature;
-import org.opendaylight.yangtools.sal.binding.model.api.Type;
-import org.opendaylight.yangtools.yang.model.api.Module;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
-import org.opendaylight.yangtools.yang.model.parser.api.YangModelParser;
-import org.opendaylight.yangtools.yang.parser.impl.YangParserImpl;
-
-public class AugmentedTypeTest {
-
-    private final static List<File> augmentModels = new ArrayList<>();
-    private final static String augmentFolderPath = AugmentedTypeTest.class
-            .getResource("/augment-test-models").getPath();
-
-    @BeforeClass
-    public static void loadTestResources() {
-        final File augFolder = new File(augmentFolderPath);
-
-        for (final File fileEntry : augFolder.listFiles()) {
-            if (fileEntry.isFile()) {
-                augmentModels.add(fileEntry);
-            }
-        }
-    }
-
-    @Test
-    public void augmentedAbstractTopologyTest() {
-        final YangModelParser parser = new YangParserImpl();
-        final Set<Module> modules = parser.parseYangModels(augmentModels);
-        final SchemaContext context = parser.resolveSchemaContext(modules);
-
-        assertNotNull("context is null", context);
-        final BindingGenerator bindingGen = new BindingGeneratorImpl();
-        final List<Type> genTypes = bindingGen.generateTypes(context);
-
-        assertNotNull("genTypes is null", genTypes);
-        assertFalse("genTypes is empty", genTypes.isEmpty());
-
-        GeneratedTransferObject gtInterfaceKey = null;
-        GeneratedType gtInterface = null;
-        GeneratedType gtTunnel = null;
-        GeneratedTransferObject gtTunnelKey = null;
-        GeneratedType gtNetworkLink2 = null;
-
-        for (final Type type : genTypes) {
-            if (type.getName().equals("InterfaceKey") && type.getPackageName().contains("augment._abstract.topology")) {
-                gtInterfaceKey = (GeneratedTransferObject) type;
-            } else if (type.getName().equals("Interface") && type.getPackageName().contains("augment._abstract.topology")) {
-                gtInterface = (GeneratedType) type;
-            } else if (type.getName().equals("Tunnel") && type.getPackageName().contains("augment._abstract.topology")) {
-                gtTunnel = (GeneratedType) type;
-            } else if (type.getName().equals("TunnelKey") && type.getPackageName().contains("augment._abstract.topology")) {
-                gtTunnelKey = (GeneratedTransferObject) type;
-            } else if (type.getName().equals("NetworkLink2") && type.getPackageName().contains("augment._abstract.topology")) {
-                gtNetworkLink2 = (GeneratedType) type;
-            }
-        }
-
-        // 'Interface
-        assertNotNull("gtInterface is null", gtInterface);
-        final List<MethodSignature> gtInterfaceMethods = gtInterface.getMethodDefinitions();
-        assertNotNull("gtInterfaceMethods is null", gtInterfaceMethods);
-        MethodSignature getIfcKeyMethod = null;
-        for (final MethodSignature method : gtInterfaceMethods) {
-            if (method.getName().equals("getInterfaceKey")) {
-                getIfcKeyMethod = method;
-                break;
-            }
-        }
-        assertNotNull("getIfcKeyMethod is null", getIfcKeyMethod);
-        assertNotNull("getIfcKeyMethod.getReturnType() is null", getIfcKeyMethod.getReturnType());
-        assertFalse("getIfcKeyMethod.getReturnType() should not be Void",
-                getIfcKeyMethod.getReturnType().equals("java.lang.Void"));
-        assertTrue("getIfcKeyMethod.getReturnType().getName() must be InterfaceKey",
-                getIfcKeyMethod.getReturnType().getName().equals("InterfaceKey"));
-
-        MethodSignature getHigherLayerIfMethod = null;
-        for (final MethodSignature method : gtInterfaceMethods) {
-            if (method.getName().equals("getHigherLayerIf")) {
-                getHigherLayerIfMethod = method;
-                break;
-            }
-        }
-        assertNotNull("getHigherLayerIfMethod is null", getHigherLayerIfMethod);
-        assertNotNull("getHigherLayerIfMethod.getReturnType() is null",
-                getHigherLayerIfMethod.getReturnType());
-        assertFalse("getHigherLayerIfMethod.getReturnType() should not be Void",
-                getHigherLayerIfMethod.getReturnType().equals("java.lang.Void"));
-        assertTrue("getHigherLayerIfMethod.getReturnType().getName() must be List",
-                getHigherLayerIfMethod.getReturnType().getName().equals("List"));
-
-        // 'InterfaceKey'
-        assertNotNull("gtInterfaceKey is null", gtInterfaceKey);
-        final List<GeneratedProperty> properties = gtInterfaceKey.getProperties();
-        assertNotNull("properties is null", properties);
-        GeneratedProperty gtInterfaceId = null;
-        for (final GeneratedProperty property : properties) {
-            if (property.getName().equals("InterfaceId")) {
-                gtInterfaceId = property;
-                break;
-            }
-        }
-        assertNotNull("gtInterfaceId is null", gtInterfaceId);
-        assertNotNull("gtInterfaceId.getReturnType() is null", gtInterfaceId.getReturnType());
-        assertFalse("gtInterfaceId.getReturnType() should not be Void",
-                gtInterfaceId.getReturnType().equals("java.lang.Void"));
-        assertTrue("gtInterfaceId.getReturnType().getName() must be String",
-                gtInterfaceId.getReturnType().getName().equals("String"));
-
-        // 'Tunnel'
-        assertNotNull("gtTunnel is null", gtTunnel);
-        final List<MethodSignature> tunnelMethods = gtTunnel.getMethodDefinitions();
-        assertNotNull("tunnelMethods is null", tunnelMethods);
-        MethodSignature getTunnelKeyMethod = null;
-        for (MethodSignature method : tunnelMethods) {
-            if (method.getName().equals("getTunnelKey")) {
-                getTunnelKeyMethod = method;
-                break;
-            }
-        }
-        assertNotNull("getTunnelKeyMethod is null", getTunnelKeyMethod);
-        assertNotNull("getTunnelKeyMethod.getReturnType()",
-                getTunnelKeyMethod.getReturnType());
-        assertFalse("getTunnelKeyMethod.getReturnType() should not be Void",
-                getTunnelKeyMethod.getReturnType().equals("java.lang.Void"));
-        assertTrue("getTunnelKeyMethod.getReturnType().getName() must be TunnelKey",
-                getTunnelKeyMethod.getReturnType().getName().equals("TunnelKey"));
-
-        // 'TunnelKey'
-        assertNotNull("gtTunnelKey is null", gtTunnelKey);
-        final List<GeneratedProperty> tunnelKeyProperties = gtTunnelKey.getProperties();
-        assertNotNull("tunnelKeyProperties is null", tunnelKeyProperties);
-
-        GeneratedProperty gtTunnelId = null;
-        for (final GeneratedProperty property : tunnelKeyProperties) {
-            if (property.getName().equals("TunnelId")) {
-                gtTunnelId = property;
-            }
-        }
-        assertNotNull("gtTunnelId is null", gtTunnelId);
-        assertNotNull("gtTunnelId.getReturnType() is null",
-                gtTunnelId.getReturnType());
-        assertFalse("gtTunnelId.getReturnType() should not be Void",
-                gtTunnelId.getReturnType().equals("java.lang.Void"));
-        assertTrue("gtTunnelId.getReturnType().getName() must be Integer",
-                gtTunnelId.getReturnType().getName().equals("Integer"));
-
-        // 'NetworkLink2'
-        assertNotNull("gtNetworkLink2 is null", gtNetworkLink2);
-
-        final List<MethodSignature> networkLink2Methods = gtNetworkLink2.getMethodDefinitions();
-        assertNotNull("networkLink2Methods is null", networkLink2Methods);
-
-//        FIXME: in some cases getIfcMethod is null which causes test fail. fix ASAP
-//      MethodSignature getIfcMethod = null;
-//      for (MethodSignature method : networkLink2Methods) {
-//          if (method.getName().equals("getInterface")) {
-//              getIfcMethod = method;
-//              break;
-//          }
-//      }
-//
-//      assertNotNull("getIfcMethod is null", getIfcMethod);
-//      assertNotNull("getIfcMethod.getReturnType() is null", getIfcMethod.getReturnType());
-//      assertFalse("getIfcMethod.getReturnType() should not be Void", getIfcMethod.getReturnType().equals("java.lang.Void"));
-//      assertTrue("getIfcMethod.getReturnType().getName() must be String", getIfcMethod.getReturnType().getName().equals("String"));
-    }
-
-    @Test
-    public void augmentedNetworkLinkTest() {
-
-    }
-
-    @Test
-    public void augmentedTopologyTunnelsTest() {
-
-    }
-}
+/*\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
+ */\r
+package org.opendaylight.yangtools.sal.binding.generator.impl;\r
+\r
+import static org.junit.Assert.*;\r
+\r
+import java.io.File;\r
+import java.util.ArrayList;\r
+import java.util.List;\r
+import java.util.Set;\r
+\r
+import org.junit.BeforeClass;\r
+import org.junit.Test;\r
+import org.opendaylight.yangtools.sal.binding.generator.api.BindingGenerator;\r
+import org.opendaylight.yangtools.sal.binding.model.api.GeneratedProperty;\r
+import org.opendaylight.yangtools.sal.binding.model.api.GeneratedTransferObject;\r
+import org.opendaylight.yangtools.sal.binding.model.api.GeneratedType;\r
+import org.opendaylight.yangtools.sal.binding.model.api.MethodSignature;\r
+import org.opendaylight.yangtools.sal.binding.model.api.Type;\r
+import org.opendaylight.yangtools.yang.model.api.Module;\r
+import org.opendaylight.yangtools.yang.model.api.SchemaContext;\r
+import org.opendaylight.yangtools.yang.model.parser.api.YangModelParser;\r
+import org.opendaylight.yangtools.yang.parser.impl.YangParserImpl;\r
+\r
+public class AugmentedTypeTest {\r
+\r
+    private final static List<File> augmentModels = new ArrayList<>();\r
+    private final static String augmentFolderPath = AugmentedTypeTest.class\r
+            .getResource("/augment-test-models").getPath();\r
+\r
+    @BeforeClass\r
+    public static void loadTestResources() {\r
+        final File augFolder = new File(augmentFolderPath);\r
+\r
+        for (final File fileEntry : augFolder.listFiles()) {\r
+            if (fileEntry.isFile()) {\r
+                augmentModels.add(fileEntry);\r
+            }\r
+        }\r
+    }\r
+\r
+    @Test\r
+    public void augmentedAbstractTopologyTest() {\r
+        final YangModelParser parser = new YangParserImpl();\r
+        final Set<Module> modules = parser.parseYangModels(augmentModels);\r
+        final SchemaContext context = parser.resolveSchemaContext(modules);\r
+\r
+        assertNotNull("context is null", context);\r
+        final BindingGenerator bindingGen = new BindingGeneratorImpl();\r
+        final List<Type> genTypes = bindingGen.generateTypes(context);\r
+\r
+        assertNotNull("genTypes is null", genTypes);\r
+        assertFalse("genTypes is empty", genTypes.isEmpty());\r
+\r
+        GeneratedTransferObject gtInterfaceKey = null;\r
+        GeneratedType gtInterface = null;\r
+        GeneratedType gtTunnel = null;\r
+        GeneratedTransferObject gtTunnelKey = null;\r
+        GeneratedType gtNetworkLink2 = null;\r
+\r
+        for (final Type type : genTypes) {\r
+            if (type.getName().equals("InterfaceKey") && type.getPackageName().contains("augment._abstract.topology")) {\r
+                gtInterfaceKey = (GeneratedTransferObject) type;\r
+            } else if (type.getName().equals("Interface") && type.getPackageName().contains("augment._abstract.topology")) {\r
+                gtInterface = (GeneratedType) type;\r
+            } else if (type.getName().equals("Tunnel") && type.getPackageName().contains("augment._abstract.topology")) {\r
+                gtTunnel = (GeneratedType) type;\r
+            } else if (type.getName().equals("TunnelKey") && type.getPackageName().contains("augment._abstract.topology")) {\r
+                gtTunnelKey = (GeneratedTransferObject) type;\r
+            } else if (type.getName().equals("NetworkLink2") && type.getPackageName().contains("augment._abstract.topology")) {\r
+                gtNetworkLink2 = (GeneratedType) type;\r
+            }\r
+        }\r
+\r
+        // 'Interface\r
+        assertNotNull("gtInterface is null", gtInterface);\r
+        final List<MethodSignature> gtInterfaceMethods = gtInterface.getMethodDefinitions();\r
+        assertNotNull("gtInterfaceMethods is null", gtInterfaceMethods);\r
+        MethodSignature getIfcKeyMethod = null;\r
+        for (final MethodSignature method : gtInterfaceMethods) {\r
+            if (method.getName().equals("getKey")) {\r
+                getIfcKeyMethod = method;\r
+                break;\r
+            }\r
+        }\r
+        assertNotNull("getIfcKeyMethod is null", getIfcKeyMethod);\r
+        assertNotNull("getIfcKeyMethod.getReturnType() is null", getIfcKeyMethod.getReturnType());\r
+        assertFalse("getIfcKeyMethod.getReturnType() should not be Void",\r
+                getIfcKeyMethod.getReturnType().equals("java.lang.Void"));\r
+        assertTrue("getIfcKeyMethod.getReturnType().getName() must be InterfaceKey",\r
+                getIfcKeyMethod.getReturnType().getName().equals("InterfaceKey"));\r
+\r
+        MethodSignature getHigherLayerIfMethod = null;\r
+        for (final MethodSignature method : gtInterfaceMethods) {\r
+            if (method.getName().equals("getHigherLayerIf")) {\r
+                getHigherLayerIfMethod = method;\r
+                break;\r
+            }\r
+        }\r
+        assertNotNull("getHigherLayerIfMethod is null", getHigherLayerIfMethod);\r
+        assertNotNull("getHigherLayerIfMethod.getReturnType() is null",\r
+                getHigherLayerIfMethod.getReturnType());\r
+        assertFalse("getHigherLayerIfMethod.getReturnType() should not be Void",\r
+                getHigherLayerIfMethod.getReturnType().equals("java.lang.Void"));\r
+        assertTrue("getHigherLayerIfMethod.getReturnType().getName() must be List",\r
+                getHigherLayerIfMethod.getReturnType().getName().equals("List"));\r
+\r
+        // 'InterfaceKey'\r
+        assertNotNull("gtInterfaceKey is null", gtInterfaceKey);\r
+        final List<GeneratedProperty> properties = gtInterfaceKey.getProperties();\r
+        assertNotNull("properties is null", properties);\r
+        GeneratedProperty gtInterfaceId = null;\r
+        for (final GeneratedProperty property : properties) {\r
+            if (property.getName().equals("InterfaceId")) {\r
+                gtInterfaceId = property;\r
+                break;\r
+            }\r
+        }\r
+        assertNotNull("gtInterfaceId is null", gtInterfaceId);\r
+        assertNotNull("gtInterfaceId.getReturnType() is null", gtInterfaceId.getReturnType());\r
+        assertFalse("gtInterfaceId.getReturnType() should not be Void",\r
+                gtInterfaceId.getReturnType().equals("java.lang.Void"));\r
+        assertTrue("gtInterfaceId.getReturnType().getName() must be String",\r
+                gtInterfaceId.getReturnType().getName().equals("String"));\r
+\r
+        // 'Tunnel'\r
+        assertNotNull("gtTunnel is null", gtTunnel);\r
+        final List<MethodSignature> tunnelMethods = gtTunnel.getMethodDefinitions();\r
+        assertNotNull("tunnelMethods is null", tunnelMethods);\r
+        MethodSignature getTunnelKeyMethod = null;\r
+        for (MethodSignature method : tunnelMethods) {\r
+            if (method.getName().equals("getKey")) {\r
+                getTunnelKeyMethod = method;\r
+                break;\r
+            }\r
+        }\r
+        assertNotNull("getTunnelKeyMethod is null", getTunnelKeyMethod);\r
+        assertNotNull("getTunnelKeyMethod.getReturnType()",\r
+                getTunnelKeyMethod.getReturnType());\r
+        assertFalse("getTunnelKeyMethod.getReturnType() should not be Void",\r
+                getTunnelKeyMethod.getReturnType().equals("java.lang.Void"));\r
+        assertTrue("getTunnelKeyMethod.getReturnType().getName() must be TunnelKey",\r
+                getTunnelKeyMethod.getReturnType().getName().equals("TunnelKey"));\r
+\r
+        // 'TunnelKey'\r
+        assertNotNull("gtTunnelKey is null", gtTunnelKey);\r
+        final List<GeneratedProperty> tunnelKeyProperties = gtTunnelKey.getProperties();\r
+        assertNotNull("tunnelKeyProperties is null", tunnelKeyProperties);\r
+\r
+        GeneratedProperty gtTunnelId = null;\r
+        for (final GeneratedProperty property : tunnelKeyProperties) {\r
+            if (property.getName().equals("TunnelId")) {\r
+                gtTunnelId = property;\r
+            }\r
+        }\r
+        assertNotNull("gtTunnelId is null", gtTunnelId);\r
+        assertNotNull("gtTunnelId.getReturnType() is null",\r
+                gtTunnelId.getReturnType());\r
+        assertFalse("gtTunnelId.getReturnType() should not be Void",\r
+                gtTunnelId.getReturnType().equals("java.lang.Void"));\r
+        assertTrue("gtTunnelId.getReturnType().getName() must be Integer",\r
+                gtTunnelId.getReturnType().getName().equals("Integer"));\r
+\r
+        // 'NetworkLink2'\r
+        assertNotNull("gtNetworkLink2 is null", gtNetworkLink2);\r
+\r
+        final List<MethodSignature> networkLink2Methods = gtNetworkLink2.getMethodDefinitions();\r
+        assertNotNull("networkLink2Methods is null", networkLink2Methods);\r
+\r
+//        FIXME: in some cases getIfcMethod is null which causes test fail. fix ASAP\r
+//      MethodSignature getIfcMethod = null;\r
+//      for (MethodSignature method : networkLink2Methods) {\r
+//          if (method.getName().equals("getInterface")) {\r
+//              getIfcMethod = method;\r
+//              break;\r
+//          }\r
+//      }\r
+//\r
+//      assertNotNull("getIfcMethod is null", getIfcMethod);\r
+//      assertNotNull("getIfcMethod.getReturnType() is null", getIfcMethod.getReturnType());\r
+//      assertFalse("getIfcMethod.getReturnType() should not be Void", getIfcMethod.getReturnType().equals("java.lang.Void"));\r
+//      assertTrue("getIfcMethod.getReturnType().getName() must be String", getIfcMethod.getReturnType().getName().equals("String"));\r
+    }\r
+\r
+    @Test\r
+    public void augmentedNetworkLinkTest() {\r
+\r
+    }\r
+\r
+    @Test\r
+    public void augmentedTopologyTunnelsTest() {\r
+\r
+    }\r
+}\r
index e82b24ba40d66687cc9b88d58ae30ec4fb40dde9..0658adaec3469117577cf2373432e5fd56200413 100644 (file)
@@ -158,7 +158,7 @@ public class GeneratedTypesTest {
         MethodSignature getIfcKey = null;\r
         MethodSignature getHigherLayerIf = null;\r
         for (final MethodSignature method : gtIfcMethods) {\r
-            if (method.getName().equals("getInterfaceKey")) {\r
+            if (method.getName().equals("getKey")) {\r
                 getIfcKey = method;\r
             } else if (method.getName().equals("getHigherLayerIf")) {\r
                 getHigherLayerIf = method;\r
@@ -226,7 +226,7 @@ public class GeneratedTypesTest {
         assertNotNull(gtTunnelMethods);\r
         MethodSignature getTunnelKey = null;\r
         for (MethodSignature method : gtTunnelMethods) {\r
-            if (method.getName().equals("getTunnelKey")) {\r
+            if (method.getName().equals("getKey")) {\r
                 getTunnelKey = method;\r
             }\r
         }\r
@@ -459,7 +459,7 @@ public class GeneratedTypesTest {
                     simpleListMethodsCount = genType.getMethodDefinitions().size();\r
                     final List<MethodSignature> methods = genType.getMethodDefinitions();\r
                     for (final MethodSignature method : methods) {\r
-                        if (method.getName().equals("getSimpleListKey")) {\r
+                        if (method.getName().equals("getKey")) {\r
                             getSimpleListKeyMethodCount++;\r
                             getSimpleListKeyMethodReturnTypeName = method.getReturnType().getName();\r
                         } else if (method.getName().equals("getListChildContainer")) {\r
diff --git a/yang/yang-binding/src/main/java/org/opendaylight/yangtools/yang/binding/Identifiable.java b/yang/yang-binding/src/main/java/org/opendaylight/yangtools/yang/binding/Identifiable.java
new file mode 100644 (file)
index 0000000..4eb04e4
--- /dev/null
@@ -0,0 +1,21 @@
+package org.opendaylight.yangtools.yang.binding;\r
+\r
+\r
+/**\r
+ * Object is uniquely identifiable in its scope by key\r
+ * \r
+ * \r
+ * \r
+ * @author ttkacik\r
+ *\r
+ * @param <T> Identifier class for this object\r
+ */\r
+public interface Identifiable<T extends Identifier<? extends Identifiable<T>>> {\r
+    \r
+    /**\r
+     * Returns an unique key for the object\r
+     * \r
+     * @return Key for the object\r
+     */\r
+    T getKey();\r
+}\r
diff --git a/yang/yang-binding/src/main/java/org/opendaylight/yangtools/yang/binding/Identifier.java b/yang/yang-binding/src/main/java/org/opendaylight/yangtools/yang/binding/Identifier.java
new file mode 100644 (file)
index 0000000..bdc386d
--- /dev/null
@@ -0,0 +1,13 @@
+package org.opendaylight.yangtools.yang.binding;\r
+\r
+/**\r
+ * \r
+ * Object is unique identifier for another object\r
+ * \r
+ * @author ttkacik\r
+ *\r
+ * @param <T> Class of object for which this object is identifier\r
+ */\r
+public interface Identifier<T extends Identifiable<?>> {\r
+\r
+}\r
index 42cc11a5540bfb855f37b951b399567ac991667c..832a9ed58af5531a5ced7677ba84fe00f0913489 100644 (file)
@@ -1,19 +1,81 @@
-/*
- * 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
- */
-package org.opendaylight.yangtools.yang.binding;
-
-/**
- * Created with IntelliJ IDEA.
- * User: lsedlak
- * Date: 27.6.2013
- * Time: 11:44
- * To change this template use File | Settings | File Templates.
- */
-public class InstanceIdentifier <T extends DataObject> {
-
-}
+/*\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
+ */\r
+package org.opendaylight.yangtools.yang.binding;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collections;\r
+import java.util.List;\r
+\r
+/**\r
+ * Uniquely identifies instance of data tree. \r
+ *\r
+ *\r
+ */\r
+public class InstanceIdentifier <T extends DataObject> {\r
+\r
+    \r
+    private final List<PathArgument> path;\r
+    private final Class<T> targetType;\r
+    \r
+    public InstanceIdentifier(Class<T> type) {\r
+        path = Collections.emptyList();\r
+        this.targetType = type;\r
+    }\r
+    \r
+    \r
+    public InstanceIdentifier(List<PathArgument> path,Class<T> type) {\r
+        this.path = Collections.<PathArgument>unmodifiableList(new ArrayList<>(path));\r
+        this.targetType = type;\r
+    }\r
+    \r
+\r
+    /**\r
+     * \r
+     * @return\r
+     */\r
+    public List<PathArgument> getPath() {\r
+        return this.path;\r
+    }\r
+    \r
+    public Class<T> getTargetType() {\r
+        return this.targetType;\r
+    }\r
+    \r
+    \r
+    /**\r
+     * Path argument of instance identifier.\r
+     * \r
+     * Interface which implementations are used as path components\r
+     * of the instance path.\r
+     * \r
+     * @author ttkacik\r
+     *\r
+     */\r
+    public static interface PathArgument {\r
+        \r
+    }\r
+    \r
+    public static class IdentifiableItem<I extends Identifiable<T>,T extends Identifier<I>>  implements PathArgument {\r
+           \r
+        private final T key;\r
+        private final Class<? extends I> type;\r
+\r
+        public IdentifiableItem(Class<I> type, T key) {\r
+            this.type = type;\r
+            this.key = key;\r
+        }\r
+        \r
+        T getKey() {\r
+            return this.key;\r
+        }\r
+        \r
+        Class<? extends I> getType() {\r
+            return this.type;\r
+        }\r
+    }\r
+}\r