Robust/src/Main/Main.java

   1 package Main;
   2
   3 import java.io.FileOutputStream;
   4 import java.io.InputStream;
   5 import java.io.PrintStream;
   6 import java.io.Reader;
   7 import java.io.BufferedReader;
   8 import java.io.FileReader;
   9 import java.io.FileInputStream;
  10 import java.util.Iterator;
  11 import java.util.Vector;
  12
  13 import IR.Tree.ParseNode;
  14 import IR.Tree.BuildIR;
  15 import IR.Tree.SemanticCheck;
  16 import IR.Flat.BuildCodeMultiCore;
  17 import IR.Flat.BuildFlat;
  18 import IR.Flat.BuildCode;
  19 import IR.ClassDescriptor;
  20 import IR.State;
  21 import IR.TaskDescriptor;
  22 import IR.TypeUtil;
  23 import Analysis.Scheduling.Schedule;
  24 import Analysis.Scheduling.ScheduleAnalysis;
  25 import Analysis.Scheduling.ScheduleEdge;
  26 import Analysis.Scheduling.ScheduleNode;
  27 import Analysis.Scheduling.ScheduleSimulator;
  28 import Analysis.TaskStateAnalysis.TaskAnalysis;
  29 import Analysis.TaskStateAnalysis.TaskTagAnalysis;
  30 import Analysis.TaskStateAnalysis.TaskGraph;
  31 import Analysis.CallGraph.CallGraph;
  32 import Analysis.TaskStateAnalysis.FEdge;
  33 import Analysis.TaskStateAnalysis.FlagState;
  34 import Analysis.TaskStateAnalysis.TagAnalysis;
  35 import Analysis.TaskStateAnalysis.GarbageAnalysis;
  36 import Analysis.TaskStateAnalysis.ExecutionGraph;
  37 import Analysis.TaskStateAnalysis.SafetyAnalysis;
  38 import Analysis.Locality.LocalityAnalysis;
  39 import Analysis.Locality.GenerateConversions;
  40 import Analysis.Prefetch.PrefetchAnalysis;
  41 import Analysis.FlatIRGraph.FlatIRGraph;
  42 import Analysis.OwnershipAnalysis.OwnershipAnalysis;
  43 import Interface.*;
  44
  45 public class Main {
  46
  47     /** Main method for the compiler.  */
  48
  49   public static void main(String args[]) throws Exception {
  50       String ClassLibraryPrefix="./ClassLibrary/";
  51       State state=new State();
  52
  53       for(int i=0;i<args.length;i++) {
  54           String option=args[i];
  55           if (option.equals("-precise"))
  56               IR.Flat.BuildCode.GENERATEPRECISEGC=true;
  57           else if (option.equals("-prefetch"))
  58               state.PREFETCH=true;
  59           else if (option.equals("-dir"))
  60               IR.Flat.BuildCode.PREFIX=args[++i]+"/";
  61           else if (option.equals("-selfloop"))
  62               state.selfloops.add(args[++i]);
  63           else if (option.equals("-excprefetch"))
  64               state.excprefetch.add(args[++i]);
  65           else if (option.equals("-classlibrary"))
  66               ClassLibraryPrefix=args[++i]+"/";
  67           else if(option.equals("-numcore")) {
  68               ++i;
  69               state.CORENUM = Integer.parseInt(args[i]);
  70           }
  71           else if (option.equals("-mainclass"))
  72               state.main=args[++i];
  73           else if (option.equals("-trueprob")) {
  74               state.TRUEPROB=Double.parseDouble(args[++i]);
  75           } else if (option.equals("-printflat"))
  76               State.PRINTFLAT=true;
  77           else if (option.equals("-struct"))
  78               state.structfile=args[++i];
  79           else if (option.equals("-conscheck"))
  80               state.CONSCHECK=true;
  81           else if (option.equals("-task"))
  82               state.TASK=true;
  83           else if (option.equals("-taskstate"))
  84               state.TASKSTATE=true;
  85           else if (option.equals("-tagstate"))
  86               state.TAGSTATE=true;
  87           else if (option.equals("-flatirtasks")) {
  88               state.FLATIRGRAPH=true;
  89               state.FLATIRGRAPHTASKS=true;
  90           }
  91           else if (option.equals("-flatirusermethods")) {
  92               state.FLATIRGRAPH=true;
  93               state.FLATIRGRAPHUSERMETHODS=true;
  94           }
  95           else if (option.equals("-flatirlibmethods")) {
  96               state.FLATIRGRAPH=true;
  97               state.FLATIRGRAPHLIBMETHODS=true;
  98           }
  99           else if (option.equals("-multicore"))
 100                   state.MULTICORE=true;
 101           else if (option.equals("-ownership"))
 102               state.OWNERSHIP=true;
 103           else if (option.equals("-optional"))
 104               state.OPTIONAL=true;
 105           else if (option.equals("-raw"))
 106                   state.RAW=true;
 107           else if (option.equals("-scheduling"))
 108                   state.SCHEDULING=true;
 109           else if (option.equals("-thread"))
 110               state.THREAD=true;
 111           else if (option.equals("-dsm"))
 112               state.DSM=true;
 113           else if (option.equals("-webinterface"))
 114               state.WEBINTERFACE=true;
 115           else if (option.equals("-instructionfailures"))
 116               state.INSTRUCTIONFAILURE=true;
 117           else if (option.equals("-help")) {
 118               System.out.println("-classlibrary classlibrarydirectory -- directory where classlibrary is located");
 119               System.out.println("-selfloop task -- this task doesn't self loop its parameters forever");
 120               System.out.println("-dir outputdirectory -- output code in outputdirectory");
 121               System.out.println("-struct structfile -- output structure declarations for repair tool");
 122               System.out.println("-mainclass -- main function to call");
 123               System.out.println("-dsm -- distributed shared memory support");
 124               System.out.println("-precise -- use precise garbage collection");
 125               System.out.println("-conscheck -- turn on consistency checking");
 126               System.out.println("-task -- compiler for tasks");
 127               System.out.println("-thread -- threads");
 128               System.out.println("-trueprob <d> -- probability of true branch");
 129               System.out.println("-printflat -- print out flat representation");
 130               System.out.println("-instructionfailures -- insert code for instruction level failures");
 131               System.out.println("-taskstate -- do task state analysis");
 132               System.out.println("-flatirtasks -- create dot files for flat IR graphs of tasks");
 133               System.out.println("-flatirusermethods -- create dot files for flat IR graphs of user methods");
 134               System.out.println("-flatirlibmethods -- create dot files for flat IR graphs of library class methods");
 135               System.out.println("  note: -flatirusermethods or -flatirlibmethods currently generate all class method flat IR graphs");
 136               System.out.println("-ownership -- do ownership analysis");
 137               System.out.println("-optional -- enable optional arguments");
 138               System.out.println("-webinterface -- enable web interface");
 139               System.out.println("-help -- print out help");
 140               System.exit(0);
 141           } else {
 142               readSourceFile(state, args[i]);
 143           }
 144       }
 145
 146
 147       readSourceFile(state, ClassLibraryPrefix+"System.java");
 148       readSourceFile(state, ClassLibraryPrefix+"String.java");
 149       readSourceFile(state, ClassLibraryPrefix+"HashSet.java");
 150       readSourceFile(state, ClassLibraryPrefix+"HashMap.java");
 151       readSourceFile(state, ClassLibraryPrefix+"HashMapIterator.java");
 152       readSourceFile(state, ClassLibraryPrefix+"HashEntry.java");
 153       readSourceFile(state, ClassLibraryPrefix+"Integer.java");
 154       readSourceFile(state, ClassLibraryPrefix+"StringBuffer.java");
 155       //if(!state.RAW) {
 156       readSourceFile(state, ClassLibraryPrefix+"FileInputStream.java");
 157       readSourceFile(state, ClassLibraryPrefix+"InputStream.java");
 158       readSourceFile(state, ClassLibraryPrefix+"OutputStream.java");
 159       readSourceFile(state, ClassLibraryPrefix+"FileOutputStream.java");
 160       readSourceFile(state, ClassLibraryPrefix+"File.java");
 161       readSourceFile(state, ClassLibraryPrefix+"InetAddress.java");
 162       readSourceFile(state, ClassLibraryPrefix+"SocketInputStream.java");
 163       readSourceFile(state, ClassLibraryPrefix+"SocketOutputStream.java");
 164       //}
 165       readSourceFile(state, ClassLibraryPrefix+"Math.java");
 166       readSourceFile(state, ClassLibraryPrefix+"gnu/Random.java");
 167           readSourceFile(state, ClassLibraryPrefix+"Vector.java");
 168           readSourceFile(state, ClassLibraryPrefix+"Enumeration.java");
 169
 170       if (state.TASK) {
 171           readSourceFile(state, ClassLibraryPrefix+"Object.java");
 172           readSourceFile(state, ClassLibraryPrefix+"TagDescriptor.java");
 173       } else if (state.DSM) {
 174           readSourceFile(state, ClassLibraryPrefix+"ThreadDSM.java");
 175           readSourceFile(state, ClassLibraryPrefix+"ObjectJavaDSM.java");
 176           readSourceFile(state, ClassLibraryPrefix+"Barrier.java");
 177       } else {
 178           if (state.THREAD) {
 179               readSourceFile(state, ClassLibraryPrefix+"Thread.java");
 180               readSourceFile(state, ClassLibraryPrefix+"ObjectJava.java");
 181           } else
 182               readSourceFile(state, ClassLibraryPrefix+"ObjectJavaNT.java");
 183       }
 184
 185       if (state.TASK) {
 186           readSourceFile(state, ClassLibraryPrefix+"StartupObject.java");
 187           readSourceFile(state, ClassLibraryPrefix+"Socket.java");
 188           readSourceFile(state, ClassLibraryPrefix+"ServerSocket.java");
 189       } else {
 190           readSourceFile(state, ClassLibraryPrefix+"SocketJava.java");
 191           readSourceFile(state, ClassLibraryPrefix+"ServerSocketJava.java");
 192       }
 193
 194       BuildIR bir=new BuildIR(state);
 195       bir.buildtree();
 196
 197       TypeUtil tu=new TypeUtil(state);
 198
 199       SemanticCheck sc=new SemanticCheck(state,tu);
 200       sc.semanticCheck();
 201       tu.createFullTable();
 202
 203       BuildFlat bf=new BuildFlat(state,tu);
 204       bf.buildFlat();
 205       SafetyAnalysis sa=null;
 206       PrefetchAnalysis pa=null;
 207
 208       if (state.TAGSTATE) {
 209           CallGraph callgraph=new CallGraph(state);
 210           TagAnalysis taganalysis=new TagAnalysis(state, callgraph);
 211           TaskTagAnalysis tta=new TaskTagAnalysis(state, taganalysis);
 212       }
 213
 214       if (state.TASKSTATE) {
 215           CallGraph callgraph=new CallGraph(state);
 216           TagAnalysis taganalysis=new TagAnalysis(state, callgraph);
 217           TaskAnalysis ta=new TaskAnalysis(state, taganalysis);
 218           ta.taskAnalysis();
 219           TaskGraph tg=new TaskGraph(state, ta);
 220           tg.createDOTfiles();
 221
 222           if (state.OPTIONAL) {
 223               ExecutionGraph et=new ExecutionGraph(state, ta);
 224               et.createExecutionGraph();
 225               sa = new SafetyAnalysis(et.getExecutionGraph(), state, ta);
 226               sa.doAnalysis();
 227               state.storeAnalysisResult(sa.getResult());
 228               state.storeOptionalTaskDescriptors(sa.getOptionalTaskDescriptors());
 229           }
 230
 231           if (state.WEBINTERFACE) {
 232               GarbageAnalysis ga=new GarbageAnalysis(state, ta);
 233               WebInterface wi=new WebInterface(state, ta, tg, ga, taganalysis);
 234               JhttpServer serve=new JhttpServer(8000,wi);
 235               serve.run();
 236           }
 237
 238           if (state.SCHEDULING) {
 239               // Save the current standard input, output, and error streams
 240               // for later restoration.
 241               PrintStream       origOut = System.out;
 242
 243               // Create a new output stream for the standard output.
 244               PrintStream       stdout  = null;
 245               try {
 246                   stdout = new PrintStream (new FileOutputStream("SimulatorResult.out"));
 247               } catch (Exception e) {
 248                   // Sigh.  Couldn't open the file.
 249                   System.out.println ("Redirect:  Unable to open output file!");
 250                   System.exit (1);
 251               }
 252
 253               // Print stuff to the original output and error streams.
 254               // On most systems all of this will end up on your console when you
 255               // run this application.
 256               //origOut.println ("\nRedirect:  Round #1");
 257               //System.out.println ("Test output via 'System.out'.");
 258               //origOut.println ("Test output via 'origOut' reference.");
 259
 260               // Set the System out and err streams to use our replacements.
 261               System.setOut(stdout);
 262
 263               // Print stuff to the original output and error streams.
 264               // The stuff printed through the 'origOut' and 'origErr' references
 265               // should go to the console on most systems while the messages
 266               // printed through the 'System.out' and 'System.err' will end up in
 267               // the files we created for them.
 268               //origOut.println ("\nRedirect:  Round #2");
 269               //System.out.println ("Test output via 'SimulatorResult.out'.");
 270               //origOut.println ("Test output via 'origOut' reference.");
 271
 272               // for test
 273               // Randomly set the newRate and probability of FEdges
 274               java.util.Random r=new java.util.Random();
 275               int tint = 0;
 276               for(Iterator it_classes=state.getClassSymbolTable().getDescriptorsIterator();it_classes.hasNext();) {
 277                   ClassDescriptor cd=(ClassDescriptor) it_classes.next();
 278                   if(cd.hasFlags()){
 279                       Vector rootnodes=ta.getRootNodes(cd);
 280                       if(rootnodes!=null)
 281                           for(Iterator it_rootnodes=rootnodes.iterator();it_rootnodes.hasNext();){
 282                               FlagState root=(FlagState)it_rootnodes.next();
 283                               Vector allocatingTasks = root.getAllocatingTasks();
 284                               if(allocatingTasks != null) {
 285                                   for(int k = 0; k < allocatingTasks.size(); k++) {
 286                                       TaskDescriptor td = (TaskDescriptor)allocatingTasks.elementAt(k);
 287                                       Vector<FEdge> fev = (Vector<FEdge>)ta.getFEdgesFromTD(td);
 288                                       int numEdges = fev.size();
 289                                       int total = 100;
 290                                       for(int j = 0; j < numEdges; j++) {
 291                                           FEdge pfe = fev.elementAt(j);
 292                                           if(numEdges - j == 1) {
 293                                               pfe.setProbability(total);
 294                                           } else {
 295                                               if((total != 0) && (total != 1)){
 296                                                   do {
 297                                                       tint = r.nextInt()%total;
 298                                                   } while(tint <= 0);
 299                                               }
 300                                               pfe.setProbability(tint);
 301                                               total -= tint;
 302                                           }
 303                                           /*do {
 304                                               tint = r.nextInt()%10;
 305                                           } while(tint <= 0);*/
 306                                           //int newRate = tint;
 307                                           //int newRate = (j+1)%2+1;
 308                                           int newRate = 1;
 309                                           String cdname = cd.getSymbol();
 310                                           if(cdname.equals("SeriesRunner")) {
 311                                               newRate = 16;
 312                                           } else if(cdname.equals("MapWorker")) {
 313                                               newRate = 6;
 314                                           } else if(cdname.equals("ReduceWorker")) {
 315                                               newRate = 3;
 316                                           }
 317                                           /*do {
 318                                               tint = r.nextInt()%100;
 319                                           } while(tint <= 0);
 320                                           int probability = tint;*/
 321                                           int probability = 100;
 322                                           pfe.addNewObjInfo(cd, newRate, probability);
 323                                       }
 324                                   }
 325                               }
 326                           }
 327
 328                       Iterator it_flags = ta.getFlagStates(cd).iterator();
 329                       while(it_flags.hasNext()) {
 330                           FlagState fs = (FlagState)it_flags.next();
 331                           Iterator it_edges = fs.edges();
 332                           while(it_edges.hasNext()) {
 333                               /*do {
 334                                   tint = r.nextInt()%10;
 335                               } while(tint <= 0);*/
 336                               tint = 3;
 337                               ((FEdge)it_edges.next()).setExeTime(tint);
 338                           }
 339                       }
 340                   }
 341               }
 342
 343               // generate multiple schedulings
 344               ScheduleAnalysis scheduleAnalysis = new ScheduleAnalysis(state, ta);
 345               scheduleAnalysis.preSchedule();
 346               scheduleAnalysis.scheduleAnalysis();
 347               //scheduleAnalysis.setCoreNum(scheduleAnalysis.getSEdges4Test().size());
 348               scheduleAnalysis.setCoreNum(state.CORENUM);
 349               scheduleAnalysis.schedule();
 350
 351               //simulate these schedulings
 352               ScheduleSimulator scheduleSimulator = new ScheduleSimulator(scheduleAnalysis.getCoreNum(), state, ta);
 353               Iterator it_scheduling = scheduleAnalysis.getSchedulingsIter();
 354               int index = 0;
 355               Vector<Integer> selectedScheduling = new Vector<Integer>();
 356               int processTime = Integer.MAX_VALUE;
 357               while(it_scheduling.hasNext()) {
 358                   Vector<Schedule> scheduling = (Vector<Schedule>)it_scheduling.next();
 359                   scheduleSimulator.setScheduling(scheduling);
 360                   int tmpTime = scheduleSimulator.process();
 361                   if(tmpTime < processTime) {
 362                       selectedScheduling.clear();
 363                       selectedScheduling.add(index);
 364                       processTime = tmpTime;
 365                   } else if(tmpTime == processTime) {
 366                       selectedScheduling.add(index);
 367                   }
 368                   index++;
 369               }
 370               System.out.print("Selected schedulings with least exectution time " + processTime + ": \n\t");
 371               for(int i = 0; i < selectedScheduling.size(); i++) {
 372                   System.out.print((selectedScheduling.elementAt(i) + 1) + ", ");
 373               }
 374               System.out.println();
 375
 376               /*ScheduleSimulator scheduleSimulator = new ScheduleSimulator(4, state, ta);
 377               Vector<Schedule> scheduling = new Vector<Schedule>();
 378               for(int i = 0; i < 4; i++) {
 379                   Schedule schedule = new Schedule(i);
 380                   scheduling.add(schedule);
 381               }
 382               Iterator it_tasks = state.getTaskSymbolTable().getAllDescriptorsIterator();
 383               while(it_tasks.hasNext()) {
 384                   TaskDescriptor td = (TaskDescriptor)it_tasks.next();
 385                   if(td.getSymbol().equals("t10")) {
 386                       scheduling.elementAt(1).addTask(td);
 387                   } else {
 388                       scheduling.elementAt(0).addTask(td);
 389                   }
 390               }
 391               ClassDescriptor cd = (ClassDescriptor)state.getClassSymbolTable().get("E");
 392               scheduling.elementAt(0).addTargetCore(cd, 1);
 393               scheduleSimulator.setScheduling(scheduling);
 394               scheduleSimulator.process();
 395
 396               Vector<Schedule> scheduling1 = new Vector<Schedule>();
 397               for(int i = 0; i < 4; i++) {
 398                   Schedule schedule = new Schedule(i);
 399                   scheduling1.add(schedule);
 400               }
 401               Iterator it_tasks1 = state.getTaskSymbolTable().getAllDescriptorsIterator();
 402               while(it_tasks1.hasNext()) {
 403                   TaskDescriptor td = (TaskDescriptor)it_tasks1.next();
 404                   scheduling1.elementAt(0).addTask(td);
 405               }
 406               scheduleSimulator.setScheduling(scheduling1);
 407               scheduleSimulator.process();*/
 408
 409               // Close the streams.
 410               try {
 411                   stdout.close ();
 412                   System.setOut(origOut);
 413               } catch (Exception e) {
 414                   origOut.println ("Redirect:  Unable to close files!");
 415               }
 416
 417               if(state.MULTICORE) {
 418                   //it_scheduling = scheduleAnalysis.getSchedulingsIter();
 419                   //Vector<Schedule> scheduling = (Vector<Schedule>)it_scheduling.next();
 420                   Vector<Schedule> scheduling = scheduleAnalysis.getSchedulings().elementAt(selectedScheduling.elementAt(0));
 421                   BuildCodeMultiCore bcm=new BuildCodeMultiCore(state, bf.getMap(), tu, sa, scheduling, scheduleAnalysis.getCoreNum(), pa);
 422                   bcm.buildCode();
 423               }
 424           }
 425
 426       }
 427
 428       if(!state.MULTICORE) {
 429           if (state.DSM) {
 430               CallGraph callgraph=new CallGraph(state);
 431               if (state.PREFETCH) {
 432                   //speed up prefetch generation using locality analysis results
 433                   LocalityAnalysis la=new LocalityAnalysis(state, callgraph, tu);
 434                   pa=new PrefetchAnalysis(state, callgraph, tu, la);
 435               }
 436
 437               LocalityAnalysis la=new LocalityAnalysis(state, callgraph, tu);
 438               GenerateConversions gc=new GenerateConversions(la, state);
 439               BuildCode bc=new BuildCode(state, bf.getMap(), tu, la, pa);
 440               bc.buildCode();
 441           } else {
 442               BuildCode bc=new BuildCode(state, bf.getMap(), tu, sa, pa);
 443               bc.buildCode();
 444           }
 445       }
 446
 447       if (state.FLATIRGRAPH) {
 448           FlatIRGraph firg = new FlatIRGraph(state,
 449                                              state.FLATIRGRAPHTASKS,
 450                                              state.FLATIRGRAPHUSERMETHODS,
 451                                              state.FLATIRGRAPHLIBMETHODS);
 452       }
 453
 454       if (state.OWNERSHIP) {
 455           CallGraph callGraph  = new CallGraph( state );
 456           int allocationDepth  = 3;
 457           OwnershipAnalysis oa =
 458               new OwnershipAnalysis( state, callGraph, allocationDepth );
 459           //This was breaking the compile
 460           //      oa.writeAllAliases( "identifiedAliases.txt" );
 461       }
 462
 463       System.exit(0);
 464   }
 465
 466     /** Reads in a source file and adds the parse tree to the state object. */
 467
 468     private static void readSourceFile(State state, String sourcefile) throws Exception {
 469         Reader fr = new BufferedReader(new FileReader(sourcefile));
 470         Lex.Lexer l = new Lex.Lexer(fr);
 471         java_cup.runtime.lr_parser g;
 472         g = new Parse.Parser(l);
 473         ParseNode p=null;
 474         try {
 475             p=(ParseNode) g./*debug_*/parse().value;
 476         } catch (Exception e) {
 477             System.err.println("Error parsing file:"+sourcefile);
 478             e.printStackTrace();
 479             System.exit(-1);
 480         }
 481         state.addParseNode(p);
 482         if (l.numErrors()!=0) {
 483             System.out.println("Error parsing "+sourcefile);
 484             System.exit(l.numErrors());
 485         }
 486     }
 487 }