Fun with GraphViz

1 files changed, 9 insertions, 4 deletions

diff --git a/vaporize/library/ControlFlowGraph.java b/vaporize/library/ControlFlowGraph.java
index c0eff10..364cec1 100644
--- a/vaporize/library/ControlFlowGraph.java
+++ b/vaporize/library/ControlFlowGraph.java
@@ -37,12 +37,17 @@ public class ControlFlowGraph {
         this.nodes.add(node);
     }
 
-    protected void addEdge(CFGNode source, CFGNode dest) {
-        MinimalLogger.info(String.format("Edge %s -> %s",
-                                         source.getInstruction().sourcePos.line,
-                                         dest.getInstruction().sourcePos.line));
+    protected String addEdge(CFGNode source, CFGNode dest) {
+        String ret = String.format("%d -> %d",
+                                   source.getInstruction().sourcePos.line,
+                                   dest.getInstruction().sourcePos.line);
+        MinimalLogger.info(String.format("Edge %s",
+                                         ret));
+
         source.addDest(dest);
         dest.addSource(source);
+
+        return ret +";";
     }
 
     protected void setStart(CFGNode start) {