Maping of union leaf a bits leaf YANG type to JAVA inner classes.

[controller.git] / opendaylight / sal / yang-prototype / code-generator / binding-java-api-generator / src / main / java / org / opendaylight / controller / sal / java / api / generator / InterfaceGenerator.java

diff --git a/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/InterfaceGenerator.java b/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/InterfaceGenerator.java
index fe0b24f890711969f4a6c446597919101caa2581..27570ca77d7ede88b8c72cb110dfdcbfced1a634 100644 (file)
--- a/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/InterfaceGenerator.java
+++ b/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/InterfaceGenerator.java
@@ -20,6 +20,7 @@ import org.opendaylight.controller.sal.binding.model.api.*;
 public final class InterfaceGenerator implements CodeGenerator {
 
     private Map<String, String> imports;
+    private Map<String, String> innerTypeImports;
 
     private String generateEnums(List<Enumeration> enums) {
         String result = "";
@@ -59,21 +60,40 @@ public final class InterfaceGenerator implements CodeGenerator {
         return result;
     }
 
+    public String generateInnerClasses(final List<GeneratedType> generatedTypes) throws IOException {
+        String result = "";
+
+        if (generatedTypes != null) {
+            ClassCodeGenerator classCodeGenerator = new ClassCodeGenerator();
+            for (GeneratedType genType : generatedTypes) {
+                if (genType instanceof GeneratedTransferObject) {
+                    result = result + classCodeGenerator.generateOnlyClass(genType, imports).toString();
+                    result = result + NL + NL;
+                }
+            }
+        }
+
+        return result;
+    }
+
     public Writer generate(Type type) throws IOException {
         Writer writer = new StringWriter();
         if (type instanceof GeneratedType && !(type instanceof GeneratedTransferObject)) {
             final GeneratedType genType = (GeneratedType) type;
             imports = GeneratorUtil.createImports(genType);
+            innerTypeImports = GeneratorUtil.createChildImports(genType);
 
             final String currentPkg = genType.getPackageName();
             final List<Constant> constants = genType.getConstantDefinitions();
             final List<MethodSignature> methods = genType.getMethodDefinitions();
             final List<Enumeration> enums = genType.getEnumerations();
+            final List<GeneratedType> enclosedGeneratedTypes = genType.getEnclosedTypes();
 
             writer.write(GeneratorUtil.createPackageDeclaration(genType.getPackageName()));
             writer.write(NL);
 
-            final List<String> importLines = GeneratorUtil.createImportLines(imports);
+            List<String> importLines = GeneratorUtil.createImportLines(imports, innerTypeImports);
+
             for (String line : importLines) {
                 writer.write(line + NL);
             }
@@ -81,6 +101,7 @@ public final class InterfaceGenerator implements CodeGenerator {
             writer.write(GeneratorUtil.createIfcDeclaration(genType, "", imports));
             writer.write(NL);
 
+            writer.write(generateInnerClasses(enclosedGeneratedTypes));
             writer.write(generateEnums(enums));
             writer.write(generateConstants(constants, currentPkg));
             writer.write(generateMethods(methods, currentPkg));