Adding struct support for Java in compiler

[iot2.git] / iotjava / iotpolicy / tree / StructDecl.java

diff --git a/iotjava/iotpolicy/tree/StructDecl.java b/iotjava/iotpolicy/tree/StructDecl.java
index 56a6fe47542617595d0cd7cb1a176178ee503951..75ef1d1fa13d2a4d4397f7f898811d8ebcdbd96a 100644 (file)
--- a/iotjava/iotpolicy/tree/StructDecl.java
+++ b/iotjava/iotpolicy/tree/StructDecl.java
@@ -67,6 +67,7 @@ public class StructDecl extends Declaration {
                        memberList.add(newMember);
                } else {
                // New declaration
+                       listStructs.add(structType);
                        List<String> newMemberTypeList = new ArrayList<String>();
                        newMemberTypeList.add(newMemberType);
                        listMemberTypes.add(newMemberTypeList);
@@ -104,4 +105,14 @@ public class StructDecl extends Declaration {
                int index = listStructs.indexOf(structType);
                return listMembers.get(index);
        }
+
+
+       /**
+        * getNumOfMembers() gets number of members
+        */
+       public int getNumOfMembers(String structType) {
+
+               int index = listStructs.indexOf(structType);
+               return listMembers.get(index).size();
+       }
 }