Merge "Cache Sync for HostTracker"
[controller.git] / opendaylight / sal / yang-prototype / code-generator / binding-java-api-generator / src / main / java / org / opendaylight / controller / sal / java / api / generator / InterfaceGenerator.java
index 8a0054bd8afd1f244d970d1c7643fcf4cb748dab..fe0b24f890711969f4a6c446597919101caa2581 100644 (file)
@@ -1,61 +1,92 @@
-/*\r
- * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.\r
- *\r
- * This program and the accompanying materials are made available under the\r
- * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
- * and is available at http://www.eclipse.org/legal/epl-v10.html\r
- */\r
-package org.opendaylight.controller.sal.java.api.generator;\r
-\r
-import static org.opendaylight.controller.sal.java.api.generator.Constants.*;\r
-\r
-import java.io.IOException;\r
-import java.io.StringWriter;\r
-import java.io.Writer;\r
-import java.util.List;\r
-\r
-import org.opendaylight.controller.sal.binding.model.api.CodeGenerator;\r
-import org.opendaylight.controller.sal.binding.model.api.Constant;\r
-import org.opendaylight.controller.sal.binding.model.api.Enumeration;\r
-import org.opendaylight.controller.sal.binding.model.api.GeneratedType;\r
-import org.opendaylight.controller.sal.binding.model.api.MethodSignature;\r
-\r
-public class InterfaceGenerator implements CodeGenerator {\r
-\r
-    public Writer generate(GeneratedType type) throws IOException {\r
-        Writer writer = new StringWriter();\r
-        final List<Constant> constants = type.getConstantDefinitions();\r
-        final List<MethodSignature> methods = type.getMethodDefinitions();\r
-        final List<Enumeration> enums = type.getEnumDefintions();\r
-\r
-        writer.write(GeneratorUtil.createIfcDeclarationWithPkgName(\r
-                type.getPackageName(), type.getName(), ""));\r
-        writer.write(NL);\r
-\r
-        if (constants != null) {\r
-            for (Constant c : constants) {\r
-                writer.write(GeneratorUtil.createConstant(c, TAB) + NL);\r
-            }\r
-            writer.write(NL);\r
-        }\r
-\r
-        if (methods != null) {\r
-            for (MethodSignature m : methods) {\r
-                writer.write(GeneratorUtil.createMethodDeclaration(m, TAB) + NL);\r
-            }\r
-            writer.write(NL);\r
-        }\r
-\r
-        if (enums != null) {\r
-            for (Enumeration e : enums) {\r
-                writer.write(GeneratorUtil.createEnum(e, TAB) + NL);\r
-            }\r
-            writer.write(NL);\r
-        }\r
-\r
-        writer.write(RCB);\r
-\r
-        return writer;\r
-    }\r
-\r
-}\r
+/*
+ * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+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;
+import java.util.List;
+import java.util.Map;
+
+import org.opendaylight.controller.sal.binding.model.api.*;
+
+public final class InterfaceGenerator implements CodeGenerator {
+
+    private Map<String, String> 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)) {
+            final 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.getEnumerations();
+
+            writer.write(GeneratorUtil.createPackageDeclaration(genType.getPackageName()));
+            writer.write(NL);
+
+            final 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;
+    }
+}