multiFileYangTranslator
Change-Id: I2adfef3acaec4bd74ba2c487404d2c655b800988
diff --git a/src/main/java/org/onosproject/yangutils/translator/tojava/javamodel/YangJavaChoice.java b/src/main/java/org/onosproject/yangutils/translator/tojava/javamodel/YangJavaChoice.java
index 0299e6c..cc12782 100644
--- a/src/main/java/org/onosproject/yangutils/translator/tojava/javamodel/YangJavaChoice.java
+++ b/src/main/java/org/onosproject/yangutils/translator/tojava/javamodel/YangJavaChoice.java
@@ -16,21 +16,23 @@
package org.onosproject.yangutils.translator.tojava.javamodel;
import java.io.IOException;
+
import org.onosproject.yangutils.datamodel.YangChoice;
import org.onosproject.yangutils.translator.exception.TranslatorException;
import org.onosproject.yangutils.translator.tojava.JavaCodeGenerator;
import org.onosproject.yangutils.translator.tojava.JavaFileInfo;
-import org.onosproject.yangutils.translator.tojava.JavaImportData;
import org.onosproject.yangutils.translator.tojava.TempJavaCodeFragmentFiles;
+import org.onosproject.yangutils.translator.tojava.utils.YangJavaModelUtils;
import org.onosproject.yangutils.translator.tojava.utils.YangPluginConfig;
import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.INTERFACE_MASK;
-import static org.onosproject.yangutils.translator.tojava.utils.YangJavaModelUtils.generateCodeOfNode;
/**
* Represents choice information extended to support java code generation.
*/
-public class YangJavaChoice extends YangChoice implements JavaCodeGeneratorInfo, JavaCodeGenerator {
+public class YangJavaChoice
+ extends YangChoice
+ implements JavaCodeGeneratorInfo, JavaCodeGenerator {
/**
* Contains the information of the java file being generated.
@@ -38,12 +40,6 @@
private JavaFileInfo javaFileInfo;
/**
- * Contains information of the imports to be inserted in the java file
- * generated.
- */
- private JavaImportData javaImportData;
-
- /**
* File handle to maintain temporary java code fragments as per the code
* snippet types.
*/
@@ -55,7 +51,6 @@
public YangJavaChoice() {
super();
setJavaFileInfo(new JavaFileInfo());
- setJavaImportData(new JavaImportData());
getJavaFileInfo().setGeneratedFileTypes(INTERFACE_MASK);
}
@@ -83,27 +78,6 @@
}
/**
- * Returns the data of java imports to be included in generated file.
- *
- * @return data of java imports to be included in generated file
- */
- @Override
- public JavaImportData getJavaImportData() {
- return javaImportData;
- }
-
- /**
- * Sets the data of java imports to be included in generated file.
- *
- * @param javaImportData data of java imports to be included in generated
- * file
- */
- @Override
- public void setJavaImportData(JavaImportData javaImportData) {
- this.javaImportData = javaImportData;
- }
-
- /**
* Returns the temporary file handle.
*
* @return temporary file handle
@@ -131,15 +105,17 @@
* @throws IOException IO operation fail
*/
@Override
- public void generateCodeEntry(YangPluginConfig yangPlugin) throws IOException {
- generateCodeOfNode(this, yangPlugin, false);
+ public void generateCodeEntry(YangPluginConfig yangPlugin)
+ throws IOException {
+ YangJavaModelUtils.generateCodeAndUpdateInParent(this, yangPlugin, false);
}
/**
* Creates a java file using the YANG choice info.
*/
@Override
- public void generateCodeExit() throws IOException {
+ public void generateCodeExit()
+ throws IOException {
getTempJavaCodeFragmentFiles().generateJavaFile(INTERFACE_MASK, this);
}
}