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