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 java.io.FileInputStream;
   7 import java.util.Iterator;
   8 import java.util.Vector;
   9
  10 import IR.Tree.ParseNode;
  11 import IR.Tree.BuildIR;
  12 import IR.Tree.SemanticCheck;
  13 import IR.Flat.BuildFlat;
  14 import IR.Flat.BuildCode;
  15 import IR.ClassDescriptor;
  16 import IR.State;
  17 import IR.TaskDescriptor;
  18 import IR.TypeUtil;
  19 import Analysis.Scheduling.ScheduleAnalysis;
  20 import Analysis.Scheduling.ScheduleEdge;
  21 import Analysis.Scheduling.ScheduleNode;
  22 import Analysis.TaskStateAnalysis.TaskAnalysis;
  23 import Analysis.TaskStateAnalysis.TaskTagAnalysis;
  24 import Analysis.TaskStateAnalysis.TaskGraph;
  25 import Analysis.CallGraph.CallGraph;
  26 import Analysis.TaskStateAnalysis.FEdge;
  27 import Analysis.TaskStateAnalysis.FlagState;
  28 import Analysis.TaskStateAnalysis.TagAnalysis;
  29 import Analysis.TaskStateAnalysis.GarbageAnalysis;
  30 import Analysis.TaskStateAnalysis.ExecutionGraph;
  31 import Analysis.TaskStateAnalysis.SafetyAnalysis;
  32 import Analysis.Locality.LocalityAnalysis;
  33 import Analysis.Locality.GenerateConversions;
  34 import Analysis.Prefetch.PrefetchAnalysis;
  35 import Analysis.FlatIRGraph.FlatIRGraph;
  36 import Analysis.OwnershipAnalysis.OwnershipAnalysis;
  37 import Interface.*;
  38
  39 public class Main {
  40
  41     /** Main method for the compiler.  */
  42
  43   public static void main(String args[]) throws Exception {
  44       String ClassLibraryPrefix="./ClassLibrary/";
  45       State state=new State();
  46
  47       for(int i=0;i<args.length;i++) {
  48           String option=args[i];
  49           if (option.equals("-precise"))
  50               IR.Flat.BuildCode.GENERATEPRECISEGC=true;
  51           else if (option.equals("-prefetch"))
  52               state.PREFETCH=true;
  53           else if (option.equals("-dir"))
  54               IR.Flat.BuildCode.PREFIX=args[++i]+"/";
  55           else if (option.equals("-selfloop"))
  56               state.selfloops.add(args[++i]);
  57           else if (option.equals("-classlibrary"))
  58               ClassLibraryPrefix=args[++i]+"/";
  59           else if (option.equals("-mainclass"))
  60               state.main=args[++i];
  61           else if (option.equals("-struct"))
  62               state.structfile=args[++i];
  63           else if (option.equals("-conscheck"))
  64               state.CONSCHECK=true;
  65           else if (option.equals("-task"))
  66               state.TASK=true;
  67           else if (option.equals("-taskstate"))
  68               state.TASKSTATE=true;
  69           else if (option.equals("-tagstate"))
  70               state.TAGSTATE=true;
  71           else if (option.equals("-flatirtasks")) {
  72               state.FLATIRGRAPH=true;
  73               state.FLATIRGRAPHTASKS=true;
  74           }
  75           else if (option.equals("-flatirusermethods")) {
  76               state.FLATIRGRAPH=true;
  77               state.FLATIRGRAPHUSERMETHODS=true;
  78           }
  79           else if (option.equals("-flatirlibmethods")) {
  80               state.FLATIRGRAPH=true;
  81               state.FLATIRGRAPHLIBMETHODS=true;
  82           }
  83           else if (option.equals("-ownership"))
  84               state.OWNERSHIP=true;
  85           else if (option.equals("-optional"))
  86               state.OPTIONAL=true;
  87           else if (option.equals("-scheduling"))
  88                   state.SCHEDULING=true;
  89           else if (option.equals("-thread"))
  90               state.THREAD=true;
  91           else if (option.equals("-dsm"))
  92               state.DSM=true;
  93           else if (option.equals("-webinterface"))
  94               state.WEBINTERFACE=true;
  95           else if (option.equals("-instructionfailures"))
  96               state.INSTRUCTIONFAILURE=true;
  97           else if (option.equals("-help")) {
  98               System.out.println("-classlibrary classlibrarydirectory -- directory where classlibrary is located");
  99               System.out.println("-selfloop task -- this task doesn't self loop its parameters forever");
 100               System.out.println("-dir outputdirectory -- output code in outputdirectory");
 101               System.out.println("-struct structfile -- output structure declarations for repair tool");
 102               System.out.println("-mainclass -- main function to call");
 103               System.out.println("-dsm -- distributed shared memory support");
 104               System.out.println("-precise -- use precise garbage collection");
 105               System.out.println("-conscheck -- turn on consistency checking");
 106               System.out.println("-task -- compiler for tasks");
 107               System.out.println("-thread -- threads");
 108               System.out.println("-instructionfailures -- insert code for instruction level failures");
 109               System.out.println("-taskstate -- do task state analysis");
 110               System.out.println("-flatirtasks -- create dot files for flat IR graphs of tasks");
 111               System.out.println("-flatirusermethods -- create dot files for flat IR graphs of user methods");
 112               System.out.println("-flatirlibmethods -- create dot files for flat IR graphs of library class methods");
 113               System.out.println("  note: -flatirusermethods or -flatirlibmethods currently generate all class method flat IR graphs");
 114               System.out.println("-ownership -- do ownership analysis");
 115               System.out.println("-optional -- enable optional arguments");
 116               System.out.println("-webinterface -- enable web interface");
 117               System.out.println("-help -- print out help");
 118               System.exit(0);
 119           } else {
 120               readSourceFile(state, args[i]);
 121           }
 122       }
 123
 124
 125       readSourceFile(state, ClassLibraryPrefix+"System.java");
 126       readSourceFile(state, ClassLibraryPrefix+"String.java");
 127       readSourceFile(state, ClassLibraryPrefix+"HashSet.java");
 128       readSourceFile(state, ClassLibraryPrefix+"HashMap.java");
 129       readSourceFile(state, ClassLibraryPrefix+"HashMapIterator.java");
 130       readSourceFile(state, ClassLibraryPrefix+"HashEntry.java");
 131       readSourceFile(state, ClassLibraryPrefix+"Integer.java");
 132       readSourceFile(state, ClassLibraryPrefix+"StringBuffer.java");
 133       readSourceFile(state, ClassLibraryPrefix+"FileInputStream.java");
 134       readSourceFile(state, ClassLibraryPrefix+"InputStream.java");
 135       readSourceFile(state, ClassLibraryPrefix+"OutputStream.java");
 136       readSourceFile(state, ClassLibraryPrefix+"FileOutputStream.java");
 137       readSourceFile(state, ClassLibraryPrefix+"File.java");
 138       readSourceFile(state, ClassLibraryPrefix+"Math.java");
 139       readSourceFile(state, ClassLibraryPrefix+"InetAddress.java");
 140       readSourceFile(state, ClassLibraryPrefix+"SocketInputStream.java");
 141       readSourceFile(state, ClassLibraryPrefix+"SocketOutputStream.java");
 142       readSourceFile(state, ClassLibraryPrefix+"gnu/Random.java");
 143
 144
 145       if (state.TASK) {
 146           readSourceFile(state, ClassLibraryPrefix+"Object.java");
 147           readSourceFile(state, ClassLibraryPrefix+"TagDescriptor.java");
 148       } else if (state.DSM) {
 149           readSourceFile(state, ClassLibraryPrefix+"ThreadDSM.java");
 150           readSourceFile(state, ClassLibraryPrefix+"ObjectJavaDSM.java");
 151       } else {
 152           if (state.THREAD) {
 153               readSourceFile(state, ClassLibraryPrefix+"Thread.java");
 154               readSourceFile(state, ClassLibraryPrefix+"ObjectJava.java");
 155           } else
 156               readSourceFile(state, ClassLibraryPrefix+"ObjectJavaNT.java");
 157       }
 158
 159       if (state.TASK) {
 160           readSourceFile(state, ClassLibraryPrefix+"StartupObject.java");
 161           readSourceFile(state, ClassLibraryPrefix+"Socket.java");
 162           readSourceFile(state, ClassLibraryPrefix+"ServerSocket.java");
 163       } else {
 164           readSourceFile(state, ClassLibraryPrefix+"SocketJava.java");
 165           readSourceFile(state, ClassLibraryPrefix+"ServerSocketJava.java");
 166       }
 167
 168       BuildIR bir=new BuildIR(state);
 169       bir.buildtree();
 170
 171       TypeUtil tu=new TypeUtil(state);
 172
 173       SemanticCheck sc=new SemanticCheck(state,tu);
 174       sc.semanticCheck();
 175       tu.createFullTable();
 176
 177       BuildFlat bf=new BuildFlat(state,tu);
 178       bf.buildFlat();
 179       SafetyAnalysis sa=null;
 180
 181       if (state.TAGSTATE) {
 182           CallGraph callgraph=new CallGraph(state);
 183           TagAnalysis taganalysis=new TagAnalysis(state, callgraph);
 184           TaskTagAnalysis tta=new TaskTagAnalysis(state, taganalysis);
 185       }
 186
 187       if (state.TASKSTATE) {
 188           CallGraph callgraph=new CallGraph(state);
 189           TagAnalysis taganalysis=new TagAnalysis(state, callgraph);
 190           TaskAnalysis ta=new TaskAnalysis(state, taganalysis);
 191           ta.taskAnalysis();
 192           TaskGraph tg=new TaskGraph(state, ta);
 193           tg.createDOTfiles();
 194
 195           if (state.OPTIONAL) {
 196               ExecutionGraph et=new ExecutionGraph(state, ta);
 197               et.createExecutionGraph();
 198               sa = new SafetyAnalysis(et.getExecutionGraph(), state, ta);
 199               sa.doAnalysis();
 200               state.storeAnalysisResult(sa.getResult());
 201               state.storeOptionalTaskDescriptors(sa.getOptionalTaskDescriptors());
 202           }
 203
 204           if (state.WEBINTERFACE) {
 205               GarbageAnalysis ga=new GarbageAnalysis(state, ta);
 206               WebInterface wi=new WebInterface(state, ta, tg, ga, taganalysis);
 207               JhttpServer serve=new JhttpServer(8000,wi);
 208               serve.run();
 209           }
 210
 211           if (state.SCHEDULING) {
 212               // for test
 213               // Randomly set the newRate and probability of FEdges
 214               java.util.Random r=new java.util.Random();
 215               int tint = 0;
 216               for(Iterator it_classes=state.getClassSymbolTable().getDescriptorsIterator();it_classes.hasNext();) {
 217                   ClassDescriptor cd=(ClassDescriptor) it_classes.next();
 218                   if(cd.hasFlags()){
 219                       Vector rootnodes=ta.getRootNodes(cd);
 220                       if(rootnodes!=null)
 221                           for(Iterator it_rootnodes=rootnodes.iterator();it_rootnodes.hasNext();){
 222                               FlagState root=(FlagState)it_rootnodes.next();
 223                               Vector allocatingTasks = root.getAllocatingTasks();
 224                               if(allocatingTasks != null) {
 225                                   for(int k = 0; k < allocatingTasks.size(); k++) {
 226                                       TaskDescriptor td = (TaskDescriptor)allocatingTasks.elementAt(k);
 227                                       Vector<FEdge> fev = (Vector<FEdge>)ta.getFEdgesFromTD(td);
 228                                       int numEdges = fev.size();
 229                                       for(int j = 0; j < numEdges; j++) {
 230                                           FEdge pfe = fev.elementAt(j);
 231                                           do {
 232                                               tint = r.nextInt()%10;
 233                                           } while(tint <= 0);
 234                                           //int newRate = tint;
 235                                           int newRate = (j+1)%2+1;
 236                                           /*do {
 237                                               tint = r.nextInt()%100;
 238                                           } while(tint <= 0);
 239                                           int probability = tint;*/
 240                                           int probability = 100;
 241                                           pfe.addNewObjInfo(cd, newRate, probability);
 242                                       }
 243                                   }
 244                               }
 245                           }
 246
 247                       Iterator it_flags = ta.getFlagStates(cd).iterator();
 248                       while(it_flags.hasNext()) {
 249                           FlagState fs = (FlagState)it_flags.next();
 250                           Iterator it_edges = fs.edges();
 251                           while(it_edges.hasNext()) {
 252                               do {
 253                                   tint = r.nextInt()%10;
 254                               } while(tint <= 0);
 255                               ((FEdge)it_edges.next()).setExeTime(tint);
 256                           }
 257                       }
 258                   }
 259               }
 260
 261               ScheduleAnalysis scheduleAnalysis = new ScheduleAnalysis(state, ta);
 262               scheduleAnalysis.preSchedule();
 263
 264               // Randomly set the newRate and probability of ScheduleEdges
 265               /*Vector<ScheduleEdge> sedges = scheduleAnalysis.getSEdges4Test();
 266                 java.util.Random r=new java.util.Random();
 267                 for(int i = 0; i < sedges.size(); i++) {
 268                 ScheduleEdge temp = sedges.elementAt(i);
 269                 int tint = 0;
 270                 do {
 271                     tint = r.nextInt()%100;
 272                 }while(tint <= 0);
 273                 temp.setProbability(tint);
 274                 do {
 275                     tint = r.nextInt()%10;
 276                 } while(tint <= 0);
 277                 temp.setNewRate(tint);
 278                 //temp.setNewRate((i+1)%2+1);
 279                 }
 280                 //sedges.elementAt(3).setNewRate(2);*/
 281                 scheduleAnalysis.scheduleAnalysis();
 282                 scheduleAnalysis.setCoreNum(scheduleAnalysis.getSEdges4Test().size() - 1);
 283                 scheduleAnalysis.schedule();
 284           }
 285
 286       }
 287
 288       if (state.DSM) {
 289           CallGraph callgraph=new CallGraph(state);
 290           if (state.PREFETCH) {
 291               PrefetchAnalysis pa=new PrefetchAnalysis(state, callgraph, tu);
 292           }
 293           LocalityAnalysis la=new LocalityAnalysis(state, callgraph, tu);
 294           GenerateConversions gc=new GenerateConversions(la, state);
 295           BuildCode bc=new BuildCode(state, bf.getMap(), tu, la);
 296           bc.buildCode();
 297       } else {
 298           BuildCode bc=new BuildCode(state, bf.getMap(), tu, sa);
 299           bc.buildCode();
 300       }
 301
 302       if (state.FLATIRGRAPH) {
 303           FlatIRGraph firg = new FlatIRGraph(state,
 304                                              state.FLATIRGRAPHTASKS,
 305                                              state.FLATIRGRAPHUSERMETHODS,
 306                                              state.FLATIRGRAPHLIBMETHODS);
 307       }
 308
 309       if (state.OWNERSHIP) {
 310           //      OwnershipAnalysis oa = new OwnershipAnalysis(state);
 311       }
 312
 313       System.exit(0);
 314   }
 315
 316     /** Reads in a source file and adds the parse tree to the state object. */
 317
 318     private static void readSourceFile(State state, String sourcefile) throws Exception {
 319         Reader fr = new BufferedReader(new FileReader(sourcefile));
 320         Lex.Lexer l = new Lex.Lexer(fr);
 321         java_cup.runtime.lr_parser g;
 322         g = new Parse.Parser(l);
 323         ParseNode p=null;
 324         try {
 325             p=(ParseNode) g./*debug_*/parse().value;
 326         } catch (Exception e) {
 327             System.err.println("Error parsing file:"+sourcefile);
 328             e.printStackTrace();
 329             System.exit(-1);
 330         }
 331         state.addParseNode(p);
 332         if (l.numErrors()!=0) {
 333             System.out.println("Error parsing "+sourcefile);
 334             System.exit(l.numErrors());
 335         }
 336     }
 337 }