Dot output

Author: dibyendumajumdar

optvm/src/main/java/com/compilerprogramming/ezlang/compiler/BasicBlock.java

@@ -164,6 +164,28 @@ public static StringBuilder toStr(StringBuilder sb, BasicBlock bb, BitSet visite
         }
         return sb;
     }
+    private String escapeHtmlChars(String s) {
+        if (s.contains(">"))
+            s = s.replaceAll(">", " > ");
+        if (s.contains(">="))
+            s = s.replaceAll(">=", " ≥ ");
+        if (s.contains("<"))
+            s = s.replaceAll("<", " &lt; ");
+        if (s.contains("<="))
+            s = s.replaceAll("<=", " &le; ");
+        return s;
+    }
+    public StringBuilder toDot(StringBuilder sb, boolean verbose) {
+        sb.append("<TABLE BORDER=\"1\" CELLBORDER=\"0\">\n");
+        sb.append("<TR><TD><B>L").append(bid).append("</B></TD></TR>\n");
+        for (Instruction i: instructions) {
+            sb.append("<TR><TD>");
+            sb.append(escapeHtmlChars(i.toStr(new StringBuilder()).toString()));
+            sb.append("</TD></TR>\n");
+        }
+        sb.append("</TABLE>");
+        return sb;
+    }
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;

optvm/src/main/java/com/compilerprogramming/ezlang/compiler/CompiledFunction.java

@@ -14,7 +14,7 @@ public class CompiledFunction {
 
     public BasicBlock entry;
     public BasicBlock exit;
-    private int bid = 0;
+    private int BID = 0;
     public BasicBlock currentBlock;
     private BasicBlock currentBreakTarget;
     private BasicBlock currentContinueTarget;
@@ -41,7 +41,7 @@ public CompiledFunction(Symbol.FunctionTypeSymbol functionSymbol) {
         this.functionType = (Type.TypeFunction) functionSymbol.type;
         this.registerPool = new RegisterPool();
         setVirtualRegisters(funcDecl.scope);
-        this.bid = 0;
+        this.BID = 0;
         this.entry = this.currentBlock = createBlock();
         this.exit = createBlock();
         this.currentBreakTarget = null;
@@ -55,7 +55,7 @@ public CompiledFunction(Symbol.FunctionTypeSymbol functionSymbol) {
     public CompiledFunction(Type.TypeFunction functionType) {
         this.functionType = (Type.TypeFunction) functionType;
         this.registerPool = new RegisterPool();
-        this.bid = 0;
+        this.BID = 0;
         this.entry = this.currentBlock = createBlock();
         this.exit = createBlock();
         this.currentBreakTarget = null;
@@ -99,11 +99,11 @@ private void setVirtualRegisters(Scope scope) {
     }
 
     public BasicBlock createBlock() {
-        return new BasicBlock(bid++);
+        return new BasicBlock(BID++);
     }
 
     private BasicBlock createLoopHead() {
-        return new BasicBlock(bid++, true);
+        return new BasicBlock(BID++, true);
     }
 
     private void compileBlock(AST.BlockStmt block) {
@@ -574,4 +574,19 @@ public void livenessAnalysis() {
     public List<BasicBlock> getBlocks() {
         return BBHelper.findAllBlocks(entry);
     }
+
+    public StringBuilder toDot(StringBuilder sb, boolean verbose) {
+        sb.append("digraph CompiledFunction {\n");
+        List<BasicBlock> blocks = getBlocks();
+        for (BasicBlock block: blocks) {
+            sb.append("L").append(block.bid).append(" [shape=none, margin=0, label=<");
+            block.toDot(sb, verbose);
+            sb.append(">];\n");
+            for (BasicBlock s: block.successors) {
+                sb.append("L").append(block.bid).append(" -> ").append("L").append(s.bid).append("\n");
+            }
+        }
+        sb.append("}\n");
+        return sb;
+    }
 }