Correctly call Vapor visitor

1 files changed, 8 insertions, 3 deletions

diff --git a/V2VM.java b/V2VM.java
index 0f745f0..155c239 100644
--- a/V2VM.java
+++ b/V2VM.java
@@ -3,6 +3,7 @@ import cs132.util.ProblemException;
 import cs132.vapor.parser.VaporParser;
 import cs132.vapor.ast.VaporProgram;
 import cs132.vapor.ast.VBuiltIn.Op;
+import cs132.vapor.ast.VFunction;
 
 import java.io.InputStreamReader;
 import java.io.IOException;
@@ -16,10 +17,14 @@ public class V2VM {
 
     public static void main(String[] args) {
         try {
-            VaporProgram pgrm = parseVapor(System.in, System.out);
-            System.out.println(pgrm);
+            VFunction[] funts = parseVapor(System.in, System.out).functions;
 
-            ControlFlowGraph vp = new ControlFlowGraph<Void, Void>();
+            ControlFlowGraph cfg = new ControlFlowGraph<Void, Void>();
+
+            TotalSpill<String, Void> ts = new TotalSpill<String, Void>();
+            for (VFunction f : funts) {
+                f.body[0].accept("", ts);
+            }
 
         } catch (IOException e) {
             System.out.println(e.toString());