commit 1025a17e3282b645dfde89ec3a6a0df57137f9c2
author Mahesh Poojary Huawei <mahesh.poojary@huawei.com> Tue Sep 27 16:06:07 2016 +0530
committer Mahesh Poojary Huawei <mahesh.poojary@huawei.com> Wed Oct 05 10:44:52 2016 +0000
tree 3ae657b5bd4e7fc30d11d2489af8856537bee4b2
parent 9fab16bcb3ff7d6b8de0f3165fbdc631c6aa0c2c

[ONOS-5255] Bits generated code modification

Change-Id: I6d4f22ea1663e3be060ce127e5f125347a5e2c4e

plugin/src/main/java/org/onosproject/yangutils/translator/tojava/JavaQualifiedTypeInfoTranslator.java

diff --git a/plugin/src/main/java/org/onosproject/yangutils/translator/tojava/JavaQualifiedTypeInfoTranslator.java b/plugin/src/main/java/org/onosproject/yangutils/translator/tojava/JavaQualifiedTypeInfoTranslator.java
index 25535e2..8570b6c 100644
--- a/plugin/src/main/java/org/onosproject/yangutils/translator/tojava/JavaQualifiedTypeInfoTranslator.java
+++ b/plugin/src/main/java/org/onosproject/yangutils/translator/tojava/JavaQualifiedTypeInfoTranslator.java
@@ -30,14 +30,17 @@
 
 import com.google.common.base.MoreObjects;
 
+import static org.onosproject.yangutils.utils.UtilConstants.BASE64;
 import static org.onosproject.yangutils.datamodel.utils.builtindatatype
         .YangDataTypes.BINARY;
+import static org.onosproject.yangutils.datamodel.utils.builtindatatype.YangDataTypes.BITS;
 import static org.onosproject.yangutils.translator.tojava.javamodel
         .AttributesJavaDataType.getJavaImportClass;
 import static org.onosproject.yangutils.translator.tojava.javamodel
         .AttributesJavaDataType.getJavaImportPackage;
-import static org.onosproject.yangutils.utils.UtilConstants.BASE64;
 import static org.onosproject.yangutils.utils.UtilConstants.COLLECTION_IMPORTS;
+import static org.onosproject.yangutils.utils.UtilConstants.REGEX_IMPORTS;
+import static org.onosproject.yangutils.utils.UtilConstants.PATTERN;
 
 /**
  * Represents the information about individual imports in the generated file.
@@ -215,6 +218,9 @@
         if (referredTypesAttrInfo.getAttributeType().getDataType() == BINARY) {
             qualifiedInfoOfFromString.setClassInfo(BASE64);
             qualifiedInfoOfFromString.setPkgInfo(COLLECTION_IMPORTS);
+        } else if (referredTypesAttrInfo.getAttributeType().getDataType() == BITS) {
+            qualifiedInfoOfFromString.setClassInfo(PATTERN);
+            qualifiedInfoOfFromString.setPkgInfo(REGEX_IMPORTS);
         } else {
             qualifiedInfoOfFromString.setClassInfo(
                     getJavaImportClass(referredTypesAttrInfo.getAttributeType(),