Generation of java files from templates

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

diff --git a/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/EnumGenerator.java b/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/EnumGenerator.java
index 33d2917339a389c9d411af9e9b5ce5a90bd48796..f2acefd169fcc16588f8fb65fac9cd6d80e00ddd 100644 (file)
--- a/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/EnumGenerator.java
+++ b/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/EnumGenerator.java
@@ -7,8 +7,6 @@
  */
 package org.opendaylight.controller.sal.java.api.generator;
 
-import static org.opendaylight.controller.sal.java.api.generator.Constants.*;
-
 import java.io.IOException;
 import java.io.StringWriter;
 import java.io.Writer;
@@ -19,30 +17,15 @@ import org.opendaylight.controller.sal.binding.model.api.Type;
 
 public class EnumGenerator implements CodeGenerator {
 
-       @Override
-       public Writer generate(Type type) throws IOException {
-               final Writer writer = new StringWriter();
-
-               if (type instanceof Enumeration) {
-                       Enumeration enums = (Enumeration) type;
-                       writer.write(GeneratorUtil.createPackageDeclaration(enums
-                                       .getPackageName()));
-                       writer.write(NL + NL);
-                       writer.write(GeneratorUtil.createEnum(enums, ""));
-               }
-
-               return writer;
-       }
-
-       public Writer generateInnerEnumeration(Type type, String indent) throws IOException {
-               final Writer writer = new StringWriter();
-
-               if (type instanceof Enumeration) {
-                       Enumeration enums = (Enumeration) type;
-                       writer.write(GeneratorUtil.createEnum(enums, indent));
-               }
-
-               return writer;
-       }
+    @Override
+    public Writer generate(Type type) throws IOException {
+        final Writer writer = new StringWriter();
+        if (type instanceof Enumeration) {
+            final Enumeration enums = (Enumeration) type;
+            final EnumTemplate enumTemplate = new EnumTemplate(enums);
+            writer.write(enumTemplate.generate().toString());
+        }
+        return writer;
+    }
 
 }