X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fsal%2Fyang-prototype%2Fcode-generator%2Fbinding-java-api-generator%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fsal%2Fjava%2Fapi%2Fgenerator%2FClassCodeGenerator.java;h=0dcaa71c139be25d46bdaa250d07e9270dc13dd5;hp=942174fdcede1485b1c931133400d945f70787c6;hb=37ff82351675cc5c279dfe88c6daf10cbbf9f48b;hpb=8398f3adb544427642694be13abe9c3bc1a4e192;ds=sidebyside

diff --git a/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/ClassCodeGenerator.java b/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/ClassCodeGenerator.java
index 942174fdce..0dcaa71c13 100644
--- a/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/ClassCodeGenerator.java
+++ b/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/ClassCodeGenerator.java
@@ -17,11 +17,12 @@ import java.util.List;
 import java.util.Map;
 
 import org.opendaylight.controller.sal.binding.model.api.CodeGenerator;
+import org.opendaylight.controller.sal.binding.model.api.Enumeration;
 import org.opendaylight.controller.sal.binding.model.api.GeneratedProperty;
 import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
 import org.opendaylight.controller.sal.binding.model.api.Type;
 
-public class ClassCodeGenerator implements CodeGenerator {
+public final class ClassCodeGenerator implements CodeGenerator {
 
     private Map<String, LinkedHashMap<String, Integer>> imports;
 
@@ -29,11 +30,12 @@ public class ClassCodeGenerator implements CodeGenerator {
     public Writer generate(Type type) throws IOException {
         final Writer writer = new StringWriter();
         if (type instanceof GeneratedTransferObject) {
-            GeneratedTransferObject genTO = (GeneratedTransferObject) type;
+            GeneratedTransferObject genTO = (GeneratedTransferObject) type;            
             imports = GeneratorUtil.createImports(genTO);
-
+            
             final String currentPkg = genTO.getPackageName();
             final List<GeneratedProperty> fields = genTO.getProperties();
+            final List<Enumeration> enums = genTO.getEnumDefintions();
 
             writer.write(GeneratorUtil.createPackageDeclaration(currentPkg));
             writer.write(NL);
@@ -48,6 +50,14 @@ public class ClassCodeGenerator implements CodeGenerator {
                     imports));
             writer.write(NL);
             writer.write(NL);
+            
+            if (enums != null) {           
+               	EnumGenerator enumGenerator = new EnumGenerator();
+            	for ( Enumeration e : enums ) {            		
+            		writer.write(enumGenerator.generateInnerEnumeration(e, TAB).toString());
+            		writer.write(NL);
+            	}
+            }
 
             if (fields != null) {
                 for (GeneratedProperty field : fields) {