Robust/src/Analysis/FlatIRGraph/FlatIRGraph.java

   1 package Analysis.FlatIRGraph;
   2
   3 import IR.*;
   4 import IR.Flat.*;
   5 import java.util.*;
   6 import java.io.*;
   7
   8 public class FlatIRGraph {
   9
  10   private State state;
  11
  12   private BufferedWriter flatbw;
  13
  14   private HashSet<FlatNode> visited;
  15   private HashSet<FlatNode> toVisit;
  16
  17   private int labelindex;
  18   private Hashtable<FlatNode, Integer> flatnodetolabel;
  19
  20   public FlatIRGraph(State state, boolean tasks, boolean usermethods, boolean libmethods) throws java.io.IOException {
  21     this.state=state;
  22
  23     if( tasks )
  24       graphTasks();
  25
  26     if( usermethods || libmethods )
  27       graphMethods();
  28   }
  29
  30   private void graphTasks() throws java.io.IOException {
  31     for(Iterator it_tasks=state.getTaskSymbolTable().getDescriptorsIterator(); it_tasks.hasNext(); ) {
  32       TaskDescriptor td = (TaskDescriptor)it_tasks.next();
  33       FlatMethod fm = state.getMethodFlat(td);
  34       writeFlatIRGraph(fm,"task"+td.getSymbol());
  35     }
  36   }
  37
  38   private void graphMethods() throws java.io.IOException {
  39     for(Iterator it_classes=state.getClassSymbolTable().getDescriptorsIterator(); it_classes.hasNext(); ) {
  40       ClassDescriptor cd = (ClassDescriptor)it_classes.next();
  41       for(Iterator it_methods=cd.getMethods(); it_methods.hasNext(); ) {
  42         MethodDescriptor md = (MethodDescriptor)it_methods.next();
  43         FlatMethod fm = state.getMethodFlat(md);
  44         writeFlatIRGraph(fm,cd.getSymbol()+"."+md.getSymbol());
  45       }
  46     }
  47   }
  48
  49   private void writeFlatIRGraph(FlatMethod fm, String graphname) throws java.io.IOException {
  50     // give every node in the flat IR graph a unique label
  51     // so a human being can inspect the graph and verify
  52     // correctness
  53     flatnodetolabel=new Hashtable<FlatNode, Integer>();
  54     visited=new HashSet<FlatNode>();
  55     labelindex=0;
  56     labelFlatNodes(fm);
  57
  58     // take symbols out of graphname that cause dot to fail
  59     graphname = graphname.replaceAll("[\\W]", "");
  60
  61     flatbw=new BufferedWriter(new FileWriter(graphname+"_flatIRGraph.dot") );
  62     flatbw.write("digraph "+graphname+" {\n");
  63
  64     visited=new HashSet<FlatNode>();
  65     toVisit=new HashSet<FlatNode>();
  66     toVisit.add(fm);
  67
  68     while( !toVisit.isEmpty() ) {
  69       FlatNode fn=(FlatNode)toVisit.iterator().next();
  70       toVisit.remove(fn);
  71       visited.add(fn);
  72
  73       if( fn.kind() == FKind.FlatMethod ) {
  74         // FlatMethod does not have toString
  75         flatbw.write(makeDotNodeDec(graphname, flatnodetolabel.get(fn), fn.getClass().getName(), "FlatMethod") );
  76       } else {
  77         flatbw.write(makeDotNodeDec(graphname, flatnodetolabel.get(fn), fn.getClass().getName(), fn.toString() ) );
  78       }
  79
  80       for(int i=0; i<fn.numNext(); i++) {
  81         FlatNode nn=fn.getNext(i);
  82         flatbw.write("  node"+flatnodetolabel.get(fn)+" -> node"+flatnodetolabel.get(nn)+";\n");
  83
  84         if( !visited.contains(nn) ) {
  85           toVisit.add(nn);
  86         }
  87       }
  88     }
  89
  90     flatbw.write("}\n");
  91     flatbw.close();
  92   }
  93
  94   private void labelFlatNodes(FlatNode fn) {
  95     visited.add(fn);
  96     flatnodetolabel.put(fn,new Integer(labelindex++));
  97     for(int i=0; i<fn.numNext(); i++) {
  98       FlatNode nn=fn.getNext(i);
  99       if(!visited.contains(nn)) {
 100         labelFlatNodes(nn);
 101       }
 102     }
 103   }
 104
 105   private String makeNodeName(String graphname, Integer id, String type) {
 106     String s = String.format("%05d", id);
 107     return "FN"+s+"_"+type;
 108   }
 109
 110   private String makeDotNodeDec(String graphname, Integer id, String type, String details) {
 111     if( details == null ) {
 112       return "  node"+id+"[label=\""+makeNodeName(graphname,id,type)+"\"];\n";
 113     } else {
 114       return "  node"+id+"[label=\""+makeNodeName(graphname,id,type)+"\\n"+details+"\"];\n";
 115     }
 116   }
 117 }