color nodes to indicate whether they'll be collected
[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 Interface.*;
19
20 public class Main {
21
22     /** Main method for the compiler.  */
23
24   public static void main(String args[]) throws Exception {
25       String ClassLibraryPrefix="./ClassLibrary/";
26       State state=new State();
27       
28       for(int i=0;i<args.length;i++) {
29           String option=args[i];
30           if (option.equals("-precise"))
31               IR.Flat.BuildCode.GENERATEPRECISEGC=true;
32           else if (option.equals("-dir"))
33               IR.Flat.BuildCode.PREFIX=args[++i]+"/";
34           else if (option.equals("-classlibrary"))
35               ClassLibraryPrefix=args[++i]+"/";
36           else if (option.equals("-mainclass"))
37               state.main=args[++i];
38           else if (option.equals("-struct"))
39               state.structfile=args[++i];
40           else if (option.equals("-conscheck"))
41               state.CONSCHECK=true;
42           else if (option.equals("-task"))
43               state.TASK=true;
44           else if (option.equals("-taskstate"))
45               state.TASKSTATE=true;
46           else if (option.equals("-thread"))
47               state.THREAD=true;
48           else if (option.equals("-webinterface"))
49               state.WEBINTERFACE=true;
50           else if (option.equals("-instructionfailures"))
51               state.INSTRUCTIONFAILURE=true;
52           else if (option.equals("-help")) {
53               System.out.println("-classlibrary classlibrarydirectory -- directory where classlibrary is located");
54               System.out.println("-dir outputdirectory -- output code in outputdirectory");
55               System.out.println("-struct structfile -- output structure declarations for repair tool");
56               System.out.println("-mainclass -- main function to call");
57               System.out.println("-precise -- use precise garbage collection");
58
59               System.out.println("-conscheck -- turn on consistency checking");
60               System.out.println("-task -- compiler for tasks");
61               System.out.println("-thread -- threads");
62               System.out.println("-instructionfailures -- insert code for instruction level failures");
63               System.out.println("-taskstate -- do task state analysis");
64               System.out.println("-webinterface -- enable web interface");
65               System.out.println("-help -- print out help");
66               System.exit(0);
67           } else {
68               readSourceFile(state, args[i]);
69           }
70       }
71       
72
73       readSourceFile(state, ClassLibraryPrefix+"System.java");
74       readSourceFile(state, ClassLibraryPrefix+"String.java");
75       readSourceFile(state, ClassLibraryPrefix+"HashSet.java");
76       readSourceFile(state, ClassLibraryPrefix+"HashMap.java");
77       readSourceFile(state, ClassLibraryPrefix+"HashMapIterator.java");
78       readSourceFile(state, ClassLibraryPrefix+"HashEntry.java");
79       readSourceFile(state, ClassLibraryPrefix+"Integer.java");
80       readSourceFile(state, ClassLibraryPrefix+"StringBuffer.java");
81       readSourceFile(state, ClassLibraryPrefix+"FileInputStream.java");
82       readSourceFile(state, ClassLibraryPrefix+"FileOutputStream.java");
83       readSourceFile(state, ClassLibraryPrefix+"File.java");
84       readSourceFile(state, ClassLibraryPrefix+"InetAddress.java");
85
86       if (state.THREAD) {
87           readSourceFile(state, ClassLibraryPrefix+"Thread.java");
88           readSourceFile(state, ClassLibraryPrefix+"ObjectJava.java");
89       } else {
90           readSourceFile(state, ClassLibraryPrefix+"Object.java");
91           readSourceFile(state, ClassLibraryPrefix+"TagDescriptor.java");
92       }
93
94       if (state.TASK) {
95           readSourceFile(state, ClassLibraryPrefix+"StartupObject.java");
96           readSourceFile(state, ClassLibraryPrefix+"Socket.java");
97           readSourceFile(state, ClassLibraryPrefix+"ServerSocket.java");
98       } else {
99           readSourceFile(state, ClassLibraryPrefix+"SocketJava.java");
100           readSourceFile(state, ClassLibraryPrefix+"ServerSocketJava.java");
101       }
102
103       BuildIR bir=new BuildIR(state);
104       bir.buildtree();
105       
106       TypeUtil tu=new TypeUtil(state);
107       
108       SemanticCheck sc=new SemanticCheck(state,tu);
109       sc.semanticCheck();
110       tu.createFullTable();
111
112       BuildFlat bf=new BuildFlat(state,tu);
113       bf.buildFlat();
114
115       if (state.TASKSTATE) {
116           CallGraph callgraph=new CallGraph(state);
117           TagAnalysis taganalysis=new TagAnalysis(state, callgraph);
118           TaskAnalysis ta=new TaskAnalysis(state, taganalysis);
119           ta.taskAnalysis();
120           TaskGraph tg=new TaskGraph(state, ta);
121           tg.createDOTfiles();
122           if (state.WEBINTERFACE) {
123               GarbageAnalysis ga=new GarbageAnalysis(state, ta);
124               WebInterface wi=new WebInterface(state, ta, tg, ga);
125               JhttpServer serve=new JhttpServer(8000,wi);
126               serve.run();
127           }
128
129
130       }
131
132       
133       
134       BuildCode bc=new BuildCode(state, bf.getMap(), tu);
135       bc.buildCode();
136       System.exit(0);
137   }
138
139     /** Reads in a source file and adds the parse tree to the state object. */
140     
141     private static void readSourceFile(State state, String sourcefile) throws Exception {
142         Reader fr = new BufferedReader(new FileReader(sourcefile));
143         Lex.Lexer l = new Lex.Lexer(fr);
144         java_cup.runtime.lr_parser g;
145         g = new Parse.Parser(l);
146         ParseNode p=null;
147         try {
148             p=(ParseNode) g./*debug_*/parse().value;
149         } catch (Exception e) {
150             System.err.println("Error parsing file:"+sourcefile);
151             e.printStackTrace();
152             System.exit(-1);
153         }
154         state.addParseNode(p);
155         if (l.numErrors()!=0) {
156             System.out.println("Error parsing "+sourcefile);
157             System.exit(l.numErrors());
158         }
159     }
160 }