Code refactoring

[yangtools.git] / code-generator / binding-java-api-generator / src / main / java / org / opendaylight / yangtools / sal / java / api / generator / ClassTemplate.xtend

diff --git a/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/ClassTemplate.xtend b/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/ClassTemplate.xtend
index c58a781761b3a5bf595570b8afd841dd0f6eb4db..44263a526e7746ea3106a5a1783298d05620ff4c 100644 (file)
--- a/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/ClassTemplate.xtend
+++ b/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/ClassTemplate.xtend
@@ -9,6 +9,7 @@ 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.Type
 import org.opendaylight.yangtools.binding.generator.util.Types
+import org.opendaylight.yangtools.sal.binding.model.api.GeneratedType
 
 /**
  * Template for generating JAVA class. 
@@ -40,6 +41,11 @@ class ClassTemplate {
      */
     val List<Constant> consts
     
+    /**
+     * List of generated types which are enclosed inside <code>genType</code>
+     */
+    val List<GeneratedType> enclosedGeneratedTypes;
+        
     /**
      * Creates instance of this class with concrete <code>genTO</code>.
      * 
@@ -55,6 +61,7 @@ class ClassTemplate {
         this.fields = genTO.properties
         this.enums = genTO.enumerations
         this.consts = genTO.constantDefinitions
+        this.enclosedGeneratedTypes = genTO.enclosedTypes
     }
     
     /**
@@ -86,6 +93,7 @@ class ClassTemplate {
     def private generateBody(boolean isInnerClass) '''
         «genTO.comment.generateComment»
         «generateClassDeclaration(isInnerClass)» {
+               «generateInnerClasses»
 
             «generateEnums»
         
@@ -112,6 +120,24 @@ class ClassTemplate {
         }
     '''
     
+    
+    /**
+     * Template method which generates inner classes inside this interface.
+     * 
+     * @return string with the source code for inner classes in JAVA format
+     */
+    def private generateInnerClasses() '''
+        «IF !enclosedGeneratedTypes.empty»
+            «FOR innerClass : enclosedGeneratedTypes SEPARATOR "\n"»
+                «IF (innerClass instanceof GeneratedTransferObject)»
+                    «val classTemplate = new ClassTemplate(innerClass as GeneratedTransferObject)»
+                    «classTemplate.generateAsInnerClass»
+                    
+                «ENDIF»
+            «ENDFOR»
+        «ENDIF»
+    '''
+        
     /**
      * Template method which generates JAVA comments.
      *