Change the style of labels and goto statements inside a function

VipinindKumar · Aug 29, 2017 · 943f164 · 943f164
1 parent f6e67a1
commit 943f164
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/CodeWriter.java b/CodeWriter.java
@@ -668,13 +668,12 @@ public void writeFunction(String funcName, String nLocals) {
             out.newLine();
 
             // while(n > 0)
-            out.write("(LOOP" + i + ")");
-            out.newLine();
+            writeLabel(funcName + "$LOOP");
             out.write("@n");
             out.newLine();
             out.write("D = M");
             out.newLine();
-            out.write("@END_LOOP" + i);
+            out.write("@" + funcName + "$END_LOOP");
             out.newLine();
             out.write("D;JLE");
             out.newLine();
@@ -689,14 +688,10 @@ public void writeFunction(String funcName, String nLocals) {
             out.newLine();
 
             // goto LOOP
-            out.write("@LOOP" + i);
-            out.newLine();
-            out.write("0;JMP");
-            out.newLine();
+            writeGoto(funcName + "$LOOP");
 
             // (END_LOOPi)
-            out.write("(END_LOOP" + i + ")");
-            out.newLine();
+            writeLabel(funcName + "$END_LOOP");
 
             i++;
         }