more IO stuff
[IRC.git] / Robust / src / Main / Main.java
1 package Main;
2
3 import java.io.Reader;
4 import java.io.BufferedReader;
5 import java.io.FileReader;
6 import IR.Tree.ParseNode;
7 import IR.Tree.BuildIR;
8 import IR.Tree.SemanticCheck;
9 import IR.Flat.BuildFlat;
10 import IR.Flat.BuildCode;
11 import IR.State;
12 import IR.TypeUtil;
13 import Analysis.TaskStateAnalysis.TaskAnalysis;
14 import Analysis.TaskStateAnalysis.TaskGraph;
15 import Analysis.CallGraph.CallGraph;
16 import Analysis.TaskStateAnalysis.TagAnalysis;
17 import Analysis.TaskStateAnalysis.GarbageAnalysis;
18 import Analysis.TaskStateAnalysis.ExecutionGraph;
19 import Analysis.TaskStateAnalysis.SafetyAnalysis;
20 import Analysis.Locality.LocalityAnalysis;
21 import Analysis.Locality.GenerateConversions;
22 import Analysis.Prefetch.PrefetchAnalysis;
23 import Interface.*;
24
25 public class Main {
26
27     /** Main method for the compiler.  */
28
29   public static void main(String args[]) throws Exception {
30       String ClassLibraryPrefix="./ClassLibrary/";
31       State state=new State();
32       
33       for(int i=0;i<args.length;i++) {
34           String option=args[i];
35           if (option.equals("-precise"))
36               IR.Flat.BuildCode.GENERATEPRECISEGC=true;
37           else if (option.equals("-prefetch"))
38               state.PREFETCH=true;
39           else if (option.equals("-dir"))
40               IR.Flat.BuildCode.PREFIX=args[++i]+"/";
41           else if (option.equals("-classlibrary"))
42               ClassLibraryPrefix=args[++i]+"/";
43           else if (option.equals("-mainclass"))
44               state.main=args[++i];
45           else if (option.equals("-struct"))
46               state.structfile=args[++i];
47           else if (option.equals("-conscheck"))
48               state.CONSCHECK=true;
49           else if (option.equals("-task"))
50               state.TASK=true;
51           else if (option.equals("-taskstate"))
52               state.TASKSTATE=true;
53           else if (option.equals("-optional"))
54               state.OPTIONAL=true;
55           else if (option.equals("-thread"))
56               state.THREAD=true;
57           else if (option.equals("-dsm"))
58               state.DSM=true;
59           else if (option.equals("-webinterface"))
60               state.WEBINTERFACE=true;
61           else if (option.equals("-instructionfailures"))
62               state.INSTRUCTIONFAILURE=true;
63           else if (option.equals("-help")) {
64               System.out.println("-classlibrary classlibrarydirectory -- directory where classlibrary is located");
65               System.out.println("-dir outputdirectory -- output code in outputdirectory");
66               System.out.println("-struct structfile -- output structure declarations for repair tool");
67               System.out.println("-mainclass -- main function to call");
68               System.out.println("-dsm -- distributed shared memory support");
69               System.out.println("-precise -- use precise garbage collection");
70               System.out.println("-conscheck -- turn on consistency checking");
71               System.out.println("-task -- compiler for tasks");
72               System.out.println("-thread -- threads");
73               System.out.println("-instructionfailures -- insert code for instruction level failures");
74               System.out.println("-taskstate -- do task state analysis");
75               System.out.println("-optional -- enable optional arguments");
76               System.out.println("-webinterface -- enable web interface");
77               System.out.println("-help -- print out help");
78               System.exit(0);
79           } else {
80               readSourceFile(state, args[i]);
81           }
82       }
83       
84
85       readSourceFile(state, ClassLibraryPrefix+"System.java");
86       readSourceFile(state, ClassLibraryPrefix+"String.java");
87       readSourceFile(state, ClassLibraryPrefix+"HashSet.java");
88       readSourceFile(state, ClassLibraryPrefix+"HashMap.java");
89       readSourceFile(state, ClassLibraryPrefix+"HashMapIterator.java");
90       readSourceFile(state, ClassLibraryPrefix+"HashEntry.java");
91       readSourceFile(state, ClassLibraryPrefix+"Integer.java");
92       readSourceFile(state, ClassLibraryPrefix+"StringBuffer.java");
93       readSourceFile(state, ClassLibraryPrefix+"FileInputStream.java");
94       readSourceFile(state, ClassLibraryPrefix+"InputStream.java");
95       readSourceFile(state, ClassLibraryPrefix+"OutputStream.java");
96       readSourceFile(state, ClassLibraryPrefix+"FileOutputStream.java");
97       readSourceFile(state, ClassLibraryPrefix+"File.java");
98       readSourceFile(state, ClassLibraryPrefix+"InetAddress.java");
99
100
101
102       if (state.TASK) {
103           readSourceFile(state, ClassLibraryPrefix+"Object.java");
104           readSourceFile(state, ClassLibraryPrefix+"TagDescriptor.java");
105       } else if (state.DSM) {
106           readSourceFile(state, ClassLibraryPrefix+"ThreadDSM.java");
107           readSourceFile(state, ClassLibraryPrefix+"ObjectJavaDSM.java");
108       } else {
109           if (state.THREAD) {
110               readSourceFile(state, ClassLibraryPrefix+"Thread.java");
111               readSourceFile(state, ClassLibraryPrefix+"ObjectJava.java");
112           } else
113               readSourceFile(state, ClassLibraryPrefix+"ObjectJavaNT.java");
114       }
115
116       if (state.TASK) {
117           readSourceFile(state, ClassLibraryPrefix+"StartupObject.java");
118           readSourceFile(state, ClassLibraryPrefix+"Socket.java");
119           readSourceFile(state, ClassLibraryPrefix+"ServerSocket.java");
120       } else {
121           readSourceFile(state, ClassLibraryPrefix+"SocketJava.java");
122           readSourceFile(state, ClassLibraryPrefix+"ServerSocketJava.java");
123       }
124
125       BuildIR bir=new BuildIR(state);
126       bir.buildtree();
127       
128       TypeUtil tu=new TypeUtil(state);
129       
130       SemanticCheck sc=new SemanticCheck(state,tu);
131       sc.semanticCheck();
132       tu.createFullTable();
133
134       BuildFlat bf=new BuildFlat(state,tu);
135       bf.buildFlat();
136       SafetyAnalysis sa=null;
137
138       if (state.TASKSTATE) {
139           CallGraph callgraph=new CallGraph(state);
140           TagAnalysis taganalysis=new TagAnalysis(state, callgraph);
141           TaskAnalysis ta=new TaskAnalysis(state, taganalysis);
142           ta.taskAnalysis();
143           TaskGraph tg=new TaskGraph(state, ta);
144           tg.createDOTfiles();
145           
146           if (state.OPTIONAL) {
147               ExecutionGraph et=new ExecutionGraph(state, ta);
148               et.createExecutionGraph();
149               sa = new SafetyAnalysis(et.getExecutionGraph(), state, ta);
150               sa.doAnalysis();
151               state.storeAnalysisResult(sa.getResult());
152               state.storeOptionalTaskDescriptors(sa.getOptionalTaskDescriptors());
153           }
154           
155           if (state.WEBINTERFACE) {
156               GarbageAnalysis ga=new GarbageAnalysis(state, ta);
157               WebInterface wi=new WebInterface(state, ta, tg, ga, taganalysis);
158               JhttpServer serve=new JhttpServer(8000,wi);
159               serve.run();
160           }
161           
162           
163       }
164
165       if (state.DSM) {
166           CallGraph callgraph=new CallGraph(state);
167           if (state.PREFETCH) {
168               PrefetchAnalysis pa=new PrefetchAnalysis(state, callgraph, tu);
169           }
170           LocalityAnalysis la=new LocalityAnalysis(state, callgraph, tu);
171           GenerateConversions gc=new GenerateConversions(la, state);
172           BuildCode bc=new BuildCode(state, bf.getMap(), tu, la);
173           bc.buildCode();
174       } else {
175           BuildCode bc=new BuildCode(state, bf.getMap(), tu, sa);
176           bc.buildCode();
177       }
178       System.exit(0);
179   }
180     
181     /** Reads in a source file and adds the parse tree to the state object. */
182     
183     private static void readSourceFile(State state, String sourcefile) throws Exception {
184         Reader fr = new BufferedReader(new FileReader(sourcefile));
185         Lex.Lexer l = new Lex.Lexer(fr);
186         java_cup.runtime.lr_parser g;
187         g = new Parse.Parser(l);
188         ParseNode p=null;
189         try {
190             p=(ParseNode) g./*debug_*/parse().value;
191         } catch (Exception e) {
192             System.err.println("Error parsing file:"+sourcefile);
193             e.printStackTrace();
194             System.exit(-1);
195         }
196         state.addParseNode(p);
197         if (l.numErrors()!=0) {
198             System.out.println("Error parsing "+sourcefile);
199             System.exit(l.numErrors());
200         }
201     }
202 }