Changed interface sal.binding.model.api.CodeGenerator

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

diff --git a/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/EnumGenerator.java b/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/EnumGenerator.java
index 0c54fe8842c053cf37b2bba84f3839e261f8e1e7..b32993b040436df70f8d40edecf85711be5e9e54 100644 (file)
--- a/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/EnumGenerator.java
+++ b/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/EnumGenerator.java
@@ -1,32 +1,36 @@
-/*
- * 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.yangtools.sal.java.api.generator;
-
-import java.io.IOException;
-import java.io.StringWriter;
-import java.io.Writer;
-
-import org.opendaylight.yangtools.sal.java.api.generator.EnumTemplate;
-import org.opendaylight.yangtools.sal.binding.model.api.CodeGenerator;
-import org.opendaylight.yangtools.sal.binding.model.api.Enumeration;
-import org.opendaylight.yangtools.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) {
-            final Enumeration enums = (Enumeration) type;
-            final EnumTemplate enumTemplate = new EnumTemplate(enums);
-            writer.write(enumTemplate.generate().toString());
-        }
-        return writer;
-    }
-
-}
+/*\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.yangtools.sal.java.api.generator;\r
+\r
+import org.opendaylight.yangtools.sal.binding.model.api.CodeGenerator;\r
+import org.opendaylight.yangtools.sal.binding.model.api.Enumeration;\r
+import org.opendaylight.yangtools.sal.binding.model.api.Type;\r
+\r
+public class EnumGenerator implements CodeGenerator {\r
+\r
+    @Override\r
+    public boolean isAcceptable(Type type) {\r
+        return type instanceof Enumeration;\r
+    }\r
+\r
+    @Override\r
+    public String generate(Type type) {\r
+        if (type instanceof Enumeration) {\r
+            final Enumeration enums = (Enumeration) type;\r
+            final EnumTemplate enumTemplate = new EnumTemplate(enums);\r
+            return enumTemplate.generate();\r
+        }\r
+        return "";\r
+    }\r
+\r
+    @Override\r
+    public String getUnitName(Type type) {\r
+        return type.getName();\r
+    }\r
+\r
+}\r