X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fsal%2Fyang-prototype%2Fcode-generator%2Fbinding-java-api-generator%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fsal%2Fjava%2Fapi%2Fgenerator%2FInterfaceGenerator.java;h=38d3bcdc070dd4522a7c9d28ef28e2721bd10561;hb=951731766152e865c55c8907f1f066aa581862c5;hp=bb5b91b97c2c5aad6d64c34b443ba45cda4b3eab;hpb=184e4f61d034f6c83da75de288d4f56ec48f60bf;p=controller.git

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 bb5b91b97c..38d3bcdc07 100644
--- 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
@@ -7,63 +7,103 @@
  */
 package org.opendaylight.controller.sal.java.api.generator;
 
-import static org.opendaylight.controller.sal.java.api.generator.Constants.NL;
-import static org.opendaylight.controller.sal.java.api.generator.Constants.RCB;
-import static org.opendaylight.controller.sal.java.api.generator.Constants.TAB;
+import static org.opendaylight.controller.sal.java.api.generator.Constants.*;
 
 import java.io.IOException;
 import java.io.StringWriter;
 import java.io.Writer;
+import java.util.LinkedHashMap;
 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.Constant;
 import org.opendaylight.controller.sal.binding.model.api.Enumeration;
+import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
 import org.opendaylight.controller.sal.binding.model.api.GeneratedType;
 import org.opendaylight.controller.sal.binding.model.api.MethodSignature;
 import org.opendaylight.controller.sal.binding.model.api.Type;
 
-public class InterfaceGenerator implements CodeGenerator {
-
-    public Writer generate(Type type) throws IOException {
-        Writer writer = new StringWriter();
-        if (type instanceof GeneratedType) {
-            GeneratedType genType = (GeneratedType) type;
-
-            final List<Constant> constants = genType.getConstantDefinitions();
-            final List<MethodSignature> methods = genType
-                    .getMethodDefinitions();
-            final List<Enumeration> enums = genType.getEnumDefintions();
-
-            writer.write(GeneratorUtil.createIfcDeclarationWithPkgName(genType,
-                    ""));
-            writer.write(NL);
-
-            if (constants != null) {
-                for (Constant c : constants) {
-                    writer.write(GeneratorUtil.createConstant(c, TAB) + NL);
-                }
-                writer.write(NL);
-            }
-
-            if (methods != null) {
-                for (MethodSignature m : methods) {
-                    writer.write(GeneratorUtil.createMethodDeclaration(m, TAB)
-                            + NL);
-                }
-                writer.write(NL);
-            }
-
-            if (enums != null) {
-                for (Enumeration e : enums) {
-                    writer.write(GeneratorUtil.createEnum(e, TAB) + NL);
-                }
-                writer.write(NL);
-            }
-
-            writer.write(RCB);
-        }
-        return writer;
-    }
+public final class InterfaceGenerator implements CodeGenerator {
+
+	private Map<String, LinkedHashMap<String, Integer>> imports;
+
+	private String generateEnums(List<Enumeration> enums) {
+		String result = "";
+		if (enums != null) {
+			EnumGenerator enumGenerator = new EnumGenerator();
+			for (Enumeration en : enums) {
+				try {
+					result = result
+							+ (enumGenerator.generateInnerEnumeration(en, TAB).toString() + NL);
+				} catch (IOException e) {
+					e.printStackTrace();
+				}
+			}
+		}
+		return result;
+	}
+
+	private String generateConstants(List<Constant> constants, String pkgName) {
+		String result = "";
+		if (constants != null) {
+			for (Constant c : constants) {
+				result = result
+						+ GeneratorUtil
+								.createConstant(c, TAB, imports, pkgName) + NL;
+			}
+			result.concat(NL);
+		}
+		return result;
+	}
+
+	public String generateMethods(List<MethodSignature> methods, String pkgName) {
+		String result = "";
+
+		if (methods != null) {
+			for (MethodSignature m : methods) {
+				result = result
+						+ GeneratorUtil.createMethodDeclaration(m, TAB,
+								imports, pkgName) + NL;
+			}
+			result = result + NL;
+		}
+		return result;
+	}
+
+	public Writer generate(Type type) throws IOException {
+		Writer writer = new StringWriter();
+		if (type instanceof GeneratedType
+				&& !(type instanceof GeneratedTransferObject)) {
+			GeneratedType genType = (GeneratedType) type;
+			imports = GeneratorUtil.createImports(genType);
+
+			final String currentPkg = genType.getPackageName();
+			final List<Constant> constants = genType.getConstantDefinitions();
+			final List<MethodSignature> methods = genType
+					.getMethodDefinitions();
+			final List<Enumeration> enums = genType.getEnumDefintions();
+
+			writer.write(GeneratorUtil.createPackageDeclaration(genType
+					.getPackageName()));
+			writer.write(NL);
+
+			List<String> importLines = GeneratorUtil.createImportLines(imports);
+			for (String line : importLines) {
+				writer.write(line + NL);
+			}
+			writer.write(NL);
+			writer.write(GeneratorUtil.createIfcDeclaration(genType, "",
+					imports));
+			writer.write(NL);
+
+			writer.write(generateEnums(enums));
+			writer.write(generateConstants(constants, currentPkg));
+			writer.write(generateMethods(methods, currentPkg));
+
+			writer.write(RCB);
+		}
+		return writer;
+	}
 
 }