commit e2bc60df2fb2c9f151cd012153f730fd4eec957c
author Bharat saraswal <bharat.saraswal@huawei.com> Sat Apr 16 02:28:25 2016 +0530
committer Gerrit Code Review <gerrit@onlab.us> Mon Apr 18 17:09:28 2016 +0000
tree 20784b174b1ccafd82f99888e132c033690166dc
parent 02f115f8c05930dc6930b947f0bf75da8b49eef8

[ONOS-4286],[ONOS-3911] YANG typedef and YANG augment
                        translator implementation.

Change-Id: I3e21d1cb52bcb90b935b672eee42b836c21f448b

src/main/java/org/onosproject/yangutils/translator/tojava/JavaImportData.java

diff --git a/src/main/java/org/onosproject/yangutils/translator/tojava/JavaImportData.java b/src/main/java/org/onosproject/yangutils/translator/tojava/JavaImportData.java
index 2c2ce05..7d8f4e0 100644
--- a/src/main/java/org/onosproject/yangutils/translator/tojava/JavaImportData.java
+++ b/src/main/java/org/onosproject/yangutils/translator/tojava/JavaImportData.java
@@ -24,10 +24,15 @@
 import org.onosproject.yangutils.datamodel.YangNode;
 import org.onosproject.yangutils.translator.exception.TranslatorException;
 
+import static org.onosproject.yangutils.utils.UtilConstants.ARRAY_LIST;
+import static org.onosproject.yangutils.utils.UtilConstants.AUGMENTED_INFO_CLASS_IMPORT_CLASS;
+import static org.onosproject.yangutils.utils.UtilConstants.AUGMENTED_INFO_CLASS_IMPORT_PKG;
 import static org.onosproject.yangutils.utils.UtilConstants.COLLECTION_IMPORTS;
 import static org.onosproject.yangutils.utils.UtilConstants.EMPTY_STRING;
 import static org.onosproject.yangutils.utils.UtilConstants.GOOGLE_MORE_OBJECT_IMPORT_CLASS;
 import static org.onosproject.yangutils.utils.UtilConstants.GOOGLE_MORE_OBJECT_IMPORT_PKG;
+import static org.onosproject.yangutils.utils.UtilConstants.HAS_AUGMENTATION_CLASS_IMPORT_CLASS;
+import static org.onosproject.yangutils.utils.UtilConstants.HAS_AUGMENTATION_CLASS_IMPORT_PKG;
 import static org.onosproject.yangutils.utils.UtilConstants.IMPORT;
 import static org.onosproject.yangutils.utils.UtilConstants.JAVA_LANG;
 import static org.onosproject.yangutils.utils.UtilConstants.JAVA_UTIL_OBJECTS_IMPORT_CLASS;
@@ -88,7 +93,7 @@
     }
 
     /**
-     * Assign the set containing the imported class/interface info.
+     * Assigns the set containing the imported class/interface info.
      *
      * @param importSet the set containing the imported class/interface info
      */
@@ -97,7 +102,7 @@
     }
 
     /**
-     * Add an imported class/interface info if it is not already part of the
+     * Adds an imported class/interface info if it is not already part of the
      * collection.
      *
      * If already part of the collection, check if the packages are same, if so
@@ -177,10 +182,36 @@
     /**
      * Returns import for list attribute.
      *
-     * @return import for for list attribute
+     * @return import for list attribute
      */
-
-    private static String getImportForList() {
+    public static String getImportForList() {
         return IMPORT + COLLECTION_IMPORTS + PERIOD + LIST + SEMI_COLAN + NEW_LINE;
     }
+
+    /**
+     * Returns import for array list attribute.
+     *
+     * @return import for array list attribute
+     */
+    public static String getImportForArrayList() {
+        return IMPORT + COLLECTION_IMPORTS + PERIOD + ARRAY_LIST + SEMI_COLAN + NEW_LINE;
+    }
+
+    /**
+     * Returns import string for HasAugmentation class.
+     *
+     * @return import string for HasAugmentation class
+     */
+    public static String getHasAugmentationImport() {
+        return IMPORT + HAS_AUGMENTATION_CLASS_IMPORT_PKG + PERIOD + HAS_AUGMENTATION_CLASS_IMPORT_CLASS;
+    }
+
+    /**
+     * Returns import string for AugmentedInfo class.
+     *
+     * @return import string for AugmentedInfo class
+     */
+    public static String getAugmentedInfoImport() {
+        return IMPORT + AUGMENTED_INFO_CLASS_IMPORT_PKG + PERIOD + AUGMENTED_INFO_CLASS_IMPORT_CLASS;
+    }
 }