squash all the thingsHEADmain

1 files changed, 27 insertions, 0 deletions

diff --git a/submit/ast/While.java b/submit/ast/While.java
index a37195e..41c3963 100644
--- a/submit/ast/While.java
+++ b/submit/ast/While.java
@@ -4,6 +4,10 @@
  */
 package submit.ast;
 
+import submit.MIPSResult;
+import submit.RegisterAllocator;
+import submit.SymbolTable;
+
 /**
  *
  * @author edwajohn
@@ -28,6 +32,29 @@ public class While extends AbstractNode implements Statement {
     } else {
       statement.toCminus(builder, prefix + " ");
     }
+  }
+
+  @Override
+  public MIPSResult toMIPS(StringBuilder code, StringBuilder data,
+                           SymbolTable symbolTable,
+                           RegisterAllocator regAllocator) {
+    String loopLabel = "while_truthy_" + SymbolTable.nextId();
+    String finishedLabel = "finished_while_" + SymbolTable.nextId();
+
+    code.append(String.format("%s:\n", loopLabel));
+
+    MIPSResult expressionTruthiness =
+        expression.toMIPS(code, data, symbolTable, regAllocator);
+
+    code.append(String.format("bne %s $zero %s\n",
+                              expressionTruthiness.getRegister(),
+                              finishedLabel));
+
+    MIPSResult inside = statement.toMIPS(code, data, symbolTable, regAllocator);
+    regAllocator.clear(inside.getRegister());
+    code.append(String.format("j %s\n", loopLabel))
+        .append(String.format("%s:\n", finishedLabel));
 
+    return MIPSResult.createVoidResult();
   }
 }