Preparing Makefiles, stub, skeleton, config files, etc. for porting LifxLightBulb...
[iot2.git] / iotjava / iotpolicy / tree / EnumDecl.java
1 package iotpolicy.tree;
2
3 import java.util.ArrayList;
4 import java.util.HashMap;
5 import java.util.List;
6 import java.util.Map;
7 import java.util.Set;
8
9
10 /** Class EnumDecl is a data structure for enumeration
11  *  declaration section in the policy file.
12  *
13  * @author      Rahmadi Trimananda <rahmadi.trimananda @ uci.edu>
14  * @version     1.0
15  * @since       2016-11-11
16  */
17 public class EnumDecl extends Declaration {
18
19         /**
20          * A "enum" declaration:
21          *              enum Enum {
22          *
23          *                      APPLE,
24          *                      ORANGE,
25          *                      GRAPE
26          *              } 
27          * In this data structure we will record its enum name, i.e. Enum
28          *              and its members.
29          */
30
31         /**
32          * Class properties
33          */
34         // Members of enum (can be more than 1 declaration
35         private Map<String, List<String>> mapEnumMembers;
36
37         /**
38          * Class constructors
39          */
40         public EnumDecl() {
41
42                 super();
43                 mapEnumMembers = new HashMap<String, List<String>>();
44         }
45
46
47         public EnumDecl(String _origInt) {
48
49                 super(_origInt);
50                 mapEnumMembers = new HashMap<String, List<String>>();
51         }
52
53
54         /**
55          * addNewMember() adds a new member into the list, e.g. MELON into the list of APPLE, ORANGE, and GRAPE
56          */
57         public void addNewMember(String enumType, String newMember) {
58
59                 if (mapEnumMembers.containsKey(enumType)) {
60                 // Existing enum declaration
61                         List<String> memberList = mapEnumMembers.get(enumType);
62                         memberList.add(newMember);
63                 } else {
64                 // New declaration
65                         List<String> newMemberList = new ArrayList<String>();
66                         newMemberList.add(newMember);
67                         mapEnumMembers.put(enumType, newMemberList);
68                 }
69         }
70
71
72         /**
73          * getEnumDeclarations() gets list of enum declarations
74          */
75         public Set<String> getEnumDeclarations() {
76
77                 return mapEnumMembers.keySet();
78         }
79
80
81         /**
82          * getMembers() gets list of enum members
83          */
84         public List<String> getMembers(String enumType) {
85
86                 return mapEnumMembers.get(enumType);
87         }
88 }