From a1c15f046183373baf5deb66e77188e656806fb7 Mon Sep 17 00:00:00 2001
From: Elizabeth Hunt <elizabeth.hunt@simponic.xyz>
Date: Sun, 23 Apr 2023 00:24:42 -0600
Subject: squash all the things

---
 submit/ast/While.java | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'submit/ast/While.java')

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();
   }
 }
-- 
cgit v1.2.3-70-g09d2