changes to allow abortting a transaction early
[IRC.git] / Robust / src / IR / State.java
1 package IR;
2 import IR.Tree.*;
3 import IR.Flat.*;
4 import IR.*;
5 import java.util.*;
6 import Analysis.TaskStateAnalysis.*;
7
8 public class State {
9   public State() {
10     this.classes=new SymbolTable();
11     this.tasks=new SymbolTable();
12     this.treemethodmap=new Hashtable();
13     this.flatmethodmap=new Hashtable();
14     this.parsetrees=new HashSet();
15     this.arraytypes=new HashSet();
16     this.arraytonumber=new Hashtable();
17     this.tagmap=new Hashtable();
18     this.selfloops=new HashSet();
19     this.excprefetch=new HashSet();
20   }
21
22   public void addParseNode(ParseNode parsetree) {
23     parsetrees.add(parsetree);
24   }
25
26   public void storeAnalysisResult(Hashtable<ClassDescriptor, Hashtable<FlagState, Set<OptionalTaskDescriptor>>> analysisresults) {
27     this.analysisresults=analysisresults;
28   }
29
30   public Hashtable<ClassDescriptor, Hashtable<FlagState, Set<OptionalTaskDescriptor>>> getAnalysisResult() {
31     return analysisresults;
32   }
33
34
35   public void storeOptionalTaskDescriptors(Hashtable<ClassDescriptor, Hashtable<OptionalTaskDescriptor, OptionalTaskDescriptor>> optionaltaskdescriptors) {
36     this.optionaltaskdescriptors=optionaltaskdescriptors;
37   }
38
39   public Hashtable<ClassDescriptor, Hashtable<OptionalTaskDescriptor, OptionalTaskDescriptor>> getOptionalTaskDescriptors() {
40     return optionaltaskdescriptors;
41   }
42
43
44   /** Boolean flag which indicates whether compiler is compiling a task-based
45    * program. */
46   public boolean WEBINTERFACE=false;
47   public boolean TASK=false;
48   public boolean FASTCHECK=false;
49   public boolean DSM=false;
50   public boolean PREFETCH=false;
51   public boolean TASKSTATE=false;
52   public boolean TAGSTATE=false;
53   public boolean FLATIRGRAPH=false;
54   public boolean FLATIRGRAPHTASKS=false;
55   public boolean FLATIRGRAPHUSERMETHODS=false;
56   public boolean FLATIRGRAPHLIBMETHODS=false;
57   public boolean MULTICORE=false;
58   public boolean OWNERSHIP=false;
59   public int OWNERSHIPALLOCDEPTH=3;
60   public boolean OWNERSHIPWRITEDOTS=false;
61   public boolean OWNERSHIPWRITEALL=false;
62   public String OWNERSHIPALIASFILE=null;
63   public boolean OPTIONAL=false;
64   public boolean ARRAYBOUNDARYCHECK=true;
65   public boolean RAW=false;
66   public boolean SCHEDULING=false;
67   public boolean USEPROFILE=false;
68   public boolean THREAD=false;
69   public boolean CONSCHECK=false;
70   public boolean INSTRUCTIONFAILURE=false;
71   public static double TRUEPROB=0.8;
72   public static boolean PRINTFLAT=false;
73   public static boolean PRINTSCHEDULING=false;
74   public static boolean PRINTSCHEDULESIM=false;
75   public static boolean ABORTREADERS=false;
76   public int CORENUM = 1;
77   public String structfile;
78   public String main;
79
80   public HashSet selfloops;
81   public HashSet excprefetch;
82   public SymbolTable classes;
83   public SymbolTable tasks;
84   public Set parsetrees;
85   public Hashtable treemethodmap;
86   public Hashtable flatmethodmap;
87   private HashSet arraytypes;
88   public Hashtable arraytonumber;
89   private int numclasses=0;
90   private int numtasks=0;
91   private int arraycount=0;
92
93
94
95   private Hashtable<ClassDescriptor, Hashtable<OptionalTaskDescriptor, OptionalTaskDescriptor>> optionaltaskdescriptors;
96   private Hashtable<ClassDescriptor, Hashtable<FlagState, Set<OptionalTaskDescriptor>>> analysisresults;
97
98   private Hashtable tagmap;
99   private int numtags=0;
100
101   public void addArrayType(TypeDescriptor td) {
102     if (!arraytypes.contains(td)) {
103       arraytypes.add(td);
104       arraytonumber.put(td,new Integer(arraycount++));
105     }
106   }
107
108   public Iterator getArrayIterator() {
109     return arraytypes.iterator();
110   }
111
112   public int getTagId(TagDescriptor tag) {
113     if (tagmap.containsKey(tag)) {
114       return ((Integer) tagmap.get(tag)).intValue();
115     } else {
116       tagmap.put(tag, new Integer(numtags));
117       return numtags++;
118     }
119   }
120
121   public int getArrayNumber(TypeDescriptor td) {
122     if (arraytonumber.containsKey(td))
123       return ((Integer)arraytonumber.get(td)).intValue();
124     else throw new Error("Could not find array type:" + td.toPrettyString());
125   }
126
127   public int numArrays() {
128     return arraytypes.size();
129   }
130
131   public static TypeDescriptor getTypeDescriptor(int t) {
132     TypeDescriptor td=new TypeDescriptor(t);
133     return td;
134   }
135
136   public static TypeDescriptor getTypeDescriptor(NameDescriptor n) {
137     TypeDescriptor td=new TypeDescriptor(n);
138     return td;
139   }
140
141   public void addClass(ClassDescriptor tdn) {
142     if (classes.contains(tdn.getSymbol()))
143       throw new Error("Class "+tdn.getSymbol()+" defined twice");
144     classes.add(tdn);
145     numclasses++;
146   }
147
148   public int numClasses() {
149     return numclasses;
150   }
151
152   public BlockNode getMethodBody(MethodDescriptor md) {
153     return (BlockNode)treemethodmap.get(md);
154   }
155
156   public BlockNode getMethodBody(TaskDescriptor td) {
157     return (BlockNode)treemethodmap.get(td);
158   }
159
160   public SymbolTable getClassSymbolTable() {
161     return classes;
162   }
163
164   public SymbolTable getTaskSymbolTable() {
165     return tasks;
166   }
167
168   /** Returns Flat IR representation of MethodDescriptor md. */
169
170   public FlatMethod getMethodFlat(MethodDescriptor md) {
171     return (FlatMethod)flatmethodmap.get(md);
172   }
173
174   /** Returns Flat IR representation of TaskDescriptor td. */
175
176   public FlatMethod getMethodFlat(TaskDescriptor td) {
177     return (FlatMethod)flatmethodmap.get(td);
178   }
179
180   public void addTreeCode(MethodDescriptor md, BlockNode bn) {
181     treemethodmap.put(md,bn);
182   }
183
184   public void addTreeCode(TaskDescriptor td, BlockNode bn) {
185     treemethodmap.put(td,bn);
186   }
187
188   public void addFlatCode(MethodDescriptor md, FlatMethod bn) {
189     flatmethodmap.put(md,bn);
190   }
191
192   public void addFlatCode(TaskDescriptor td, FlatMethod bn) {
193     flatmethodmap.put(td,bn);
194   }
195
196   public void addTask(TaskDescriptor td) {
197     if (tasks.contains(td.getSymbol()))
198       throw new Error("Task "+td.getSymbol()+" defined twice");
199     tasks.add(td);
200     numtasks++;
201   }
202 }