From eceb071ef3be7f0b33212810ceea4ae24934f24e Mon Sep 17 00:00:00 2001
From: jjenista <jjenista>
Date: Fri, 9 Nov 2007 20:53:03 +0000
Subject: [PATCH] FlatIRGraph for creating dot files of Flat IR.

---
 .../src/Analysis/FlatIRGraph/FlatIRGraph.java | 114 ++++++++++++++++++
 Robust/src/IR/State.java                      |   4 +
 Robust/src/Main/Main.java                     |  22 ++++
 Robust/src/Makefile                           |   4 +-
 4 files changed, 142 insertions(+), 2 deletions(-)
 create mode 100644 Robust/src/Analysis/FlatIRGraph/FlatIRGraph.java

diff --git a/Robust/src/Analysis/FlatIRGraph/FlatIRGraph.java b/Robust/src/Analysis/FlatIRGraph/FlatIRGraph.java
new file mode 100644
index 00000000..fac5e59b
--- /dev/null
+++ b/Robust/src/Analysis/FlatIRGraph/FlatIRGraph.java
@@ -0,0 +1,114 @@
+package Analysis.FlatIRGraph;
+
+import IR.*;
+import IR.Flat.*;
+import java.util.*;
+import java.io.*;
+
+public class FlatIRGraph {
+
+    private State state;
+
+    private BufferedWriter flatbw;
+
+    private HashSet<FlatNode> visited;
+    private HashSet<FlatNode> toVisit;
+    
+    private int labelindex;
+    private Hashtable<FlatNode, Integer> flatnodetolabel;
+
+    public FlatIRGraph(State state, boolean tasks, boolean usermethods, boolean libmethods) throws java.io.IOException {
+	this.state=state;
+
+	if( tasks )
+	    graphTasks();
+
+	if( usermethods || libmethods )
+	    graphMethods();
+    }
+
+    private void graphTasks() throws java.io.IOException {
+	for(Iterator it_tasks=state.getTaskSymbolTable().getDescriptorsIterator();it_tasks.hasNext();) {
+	    TaskDescriptor td = (TaskDescriptor)it_tasks.next();
+	    FlatMethod fm = state.getMethodFlat(td);
+	    writeFlatIRGraph(fm,"task"+td.getSymbol());
+	}	
+    }
+
+    private void graphMethods() throws java.io.IOException {
+	for(Iterator it_classes=state.getClassSymbolTable().getDescriptorsIterator();it_classes.hasNext();) {
+	    ClassDescriptor cd = (ClassDescriptor)it_classes.next();
+	    for(Iterator it_methods=cd.getMethods();it_methods.hasNext();) {
+		MethodDescriptor md = (MethodDescriptor)it_methods.next();
+		FlatMethod fm = state.getMethodFlat(md);
+		writeFlatIRGraph(fm,cd.getSymbol()+"."+md.getSymbol());
+	    }
+	}	
+    }
+
+    private void writeFlatIRGraph( FlatMethod fm, String graphname ) throws java.io.IOException {
+	// give every node in the flat IR graph a unique label
+	// so a human being can inspect the graph and verify
+	// correctness
+	flatnodetolabel=new Hashtable<FlatNode, Integer>();
+	visited=new HashSet<FlatNode>();
+	labelindex=0;
+	labelFlatNodes( fm );
+
+	flatbw=new BufferedWriter( new FileWriter( graphname+"_flatIRGraph.dot" ) );
+	flatbw.write("digraph "+graphname+" {\n");
+
+	visited=new HashSet<FlatNode>();
+	toVisit=new HashSet<FlatNode>();
+	toVisit.add(fm);
+
+	while( !toVisit.isEmpty() ) {
+	    FlatNode fn=(FlatNode)toVisit.iterator().next();
+	    toVisit.remove(fn);
+	    visited.add(fn);
+
+	    if( fn.kind() == FKind.FlatMethod ) {
+		// FlatMethod does not have toString
+		flatbw.write( makeDotNodeDec( graphname, flatnodetolabel.get(fn), fn.getClass().getName(), "FlatMethod" ) );
+	    } else {
+		flatbw.write( makeDotNodeDec( graphname, flatnodetolabel.get(fn), fn.getClass().getName(), fn.toString() ) );
+	    }
+	    
+	    for(int i=0;i<fn.numNext();i++) {
+		FlatNode nn=fn.getNext(i);
+		flatbw.write( "  node"+flatnodetolabel.get(fn)+" -> node"+flatnodetolabel.get(nn)+";\n" );
+
+		if( !visited.contains( nn ) ) {
+		    toVisit.add( nn );
+		}
+	    }
+	}
+
+	flatbw.write( "}\n" );
+	flatbw.close();
+    }
+
+    private void labelFlatNodes(FlatNode fn) {
+	visited.add(fn);
+	flatnodetolabel.put(fn,new Integer(labelindex++));
+	for(int i=0;i<fn.numNext();i++) {
+	    FlatNode nn=fn.getNext(i);
+	    if(!visited.contains(nn)) {
+		labelFlatNodes(nn);
+	    }
+	}
+    }
+
+    private String makeNodeName( String graphname, Integer id, String type ) {
+	String s = String.format( "%05d", id );
+	return "FN"+s+"_"+type;
+    }
+
+    private String makeDotNodeDec( String graphname, Integer id, String type, String details ) {
+	if( details == null ) {
+	    return "  node"+id+"[label=\""+makeNodeName(graphname,id,type)+"\"];\n";
+	} else {
+	    return "  node"+id+"[label=\""+makeNodeName(graphname,id,type)+"\\n"+details+"\"];\n";
+	}
+    }
+}
diff --git a/Robust/src/IR/State.java b/Robust/src/IR/State.java
index 0b803a97..89fd4727 100644
--- a/Robust/src/IR/State.java
+++ b/Robust/src/IR/State.java
@@ -45,6 +45,10 @@ public class State {
     public boolean DSM=false;
     public boolean PREFETCH=false;
     public boolean TASKSTATE=false;
+    public boolean FLATIRGRAPH=false;
+    public boolean FLATIRGRAPHTASKS=false;
+    public boolean FLATIRGRAPHUSERMETHODS=false;
+    public boolean FLATIRGRAPHLIBMETHODS=false;
     public boolean OPTIONAL=false;
     public boolean THREAD=false;
     public boolean CONSCHECK=false;
diff --git a/Robust/src/Main/Main.java b/Robust/src/Main/Main.java
index 4265e604..e61094c0 100644
--- a/Robust/src/Main/Main.java
+++ b/Robust/src/Main/Main.java
@@ -20,6 +20,7 @@ import Analysis.TaskStateAnalysis.SafetyAnalysis;
 import Analysis.Locality.LocalityAnalysis;
 import Analysis.Locality.GenerateConversions;
 import Analysis.Prefetch.PrefetchAnalysis;
+import Analysis.FlatIRGraph.FlatIRGraph;
 import Interface.*;
 
 public class Main {
@@ -50,6 +51,18 @@ public class Main {
 	      state.TASK=true;
 	  else if (option.equals("-taskstate"))
 	      state.TASKSTATE=true;
+	  else if (option.equals("-flatirtasks")) {
+	      state.FLATIRGRAPH=true;
+	      state.FLATIRGRAPHTASKS=true;
+	  }
+	  else if (option.equals("-flatirusermethods")) {
+	      state.FLATIRGRAPH=true;
+	      state.FLATIRGRAPHUSERMETHODS=true;
+	  }
+	  else if (option.equals("-flatirlibmethods")) {
+	      state.FLATIRGRAPH=true;
+	      state.FLATIRGRAPHLIBMETHODS=true;
+	  }
 	  else if (option.equals("-optional"))
 	      state.OPTIONAL=true;
 	  else if (option.equals("-thread"))
@@ -72,6 +85,7 @@ public class Main {
 	      System.out.println("-thread -- threads");
 	      System.out.println("-instructionfailures -- insert code for instruction level failures");
 	      System.out.println("-taskstate -- do task state analysis");
+	      System.out.println("-ownership -- do ownership analysis");
 	      System.out.println("-optional -- enable optional arguments");
 	      System.out.println("-webinterface -- enable web interface");
 	      System.out.println("-help -- print out help");
@@ -176,6 +190,14 @@ public class Main {
 	  BuildCode bc=new BuildCode(state, bf.getMap(), tu, sa);
 	  bc.buildCode();
       }
+
+      if (state.FLATIRGRAPH) {
+	  FlatIRGraph firg = new FlatIRGraph(state,
+					     state.FLATIRGRAPHTASKS,
+					     state.FLATIRGRAPHUSERMETHODS,
+					     state.FLATIRGRAPHLIBMETHODS);
+      }
+
       System.exit(0);
   }
     
diff --git a/Robust/src/Makefile b/Robust/src/Makefile
index bcb5fa2a..6f7ec5e2 100644
--- a/Robust/src/Makefile
+++ b/Robust/src/Makefile
@@ -71,6 +71,7 @@ Analysis/TaskStateAnalysis/TaskGraph.class				\
 Analysis/TaskStateAnalysis/TaskNode.class				\
 Analysis/TaskStateAnalysis/FlagComparator.class				\
 Analysis/TaskStateAnalysis/TaskNodeNamer.class Util/Edge.class		\
+Analysis/FlatIRGraph/FlatIRGraph.class                                  \
 Util/GraphNode.class Util/Namer.class Util/Relation.class		\
 Interface/HTTPHeader.class Interface/HTTPResponse.class			\
 Interface/HTTPServices.class Interface/HashStrings.class		\
@@ -80,7 +81,6 @@ Interface/WebInterface.class Analysis/Prefetch/PrefetchAnalysis.class
 
 
 
-
 all: Parse/Sym.class Parse/Parser.class $(CLASSFILES) javadoc
 
 wc:
@@ -101,7 +101,7 @@ javadoc:
 	javadoc -classpath ../cup:.:$(CLASSPATH) -sourcepath . -private -d javadoc Lex Util IR IR.Tree IR.Flat Analysis Analysis.CallGraph Analysis.Flag Analysis.TaskStateAnalysis Analysis.Locality Analysis.Prefetch Main 
 
 clean:
-	rm IR/*.class IR/Tree/*.class Main/*.class Lex/*.class Parse/*.class Parse/Sym.java Parse/Parser.java IR/Flat/*.class classdefs.h methodheaders.h methods.c structdefs.h virtualtable.h task.h taskdefs.c taskdefs.h Analysis/*.class Analysis/Flag/*.class Analysis/CallGraph/*.class  Analysis/TaskStateAnalysis/*.class Interface/*.class Util/*.class Analysis/Locality/*.class Analysis/Prefetch/*.class
+	rm -f IR/*.class IR/Tree/*.class Main/*.class Lex/*.class Parse/*.class Parse/Sym.java Parse/Parser.java IR/Flat/*.class classdefs.h methodheaders.h methods.c structdefs.h virtualtable.h task.h taskdefs.c taskdefs.h Analysis/*.class Analysis/Flag/*.class Analysis/CallGraph/*.class  Analysis/TaskStateAnalysis/*.class Interface/*.class Util/*.class Analysis/Locality/*.class Analysis/Prefetch/*.class Analysis/FlatIRGraph/*.class
 
 cleandoc:
 	rm -rf javadoc
\ No newline at end of file
-- 
2.34.1