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/Return.java | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

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

diff --git a/submit/ast/Return.java b/submit/ast/Return.java
index 47a8c16..ebae15f 100644
--- a/submit/ast/Return.java
+++ b/submit/ast/Return.java
@@ -4,17 +4,18 @@
  */
 package submit.ast;
 
+import submit.MIPSResult;
+import submit.RegisterAllocator;
+import submit.SymbolTable;
+
 /**
  *
  * @author edwajohn
  */
 public class Return extends AbstractNode implements Statement {
-
   private final Expression expr;
 
-  public Return(Expression expr) {
-    this.expr = expr;
-  }
+  public Return(Expression expr) { this.expr = expr; }
 
   @Override
   public void toCminus(StringBuilder builder, String prefix) {
@@ -28,4 +29,19 @@ public class Return extends AbstractNode implements Statement {
     }
   }
 
+  @Override
+  public MIPSResult toMIPS(StringBuilder code, StringBuilder data,
+                           SymbolTable symbolTable,
+                           RegisterAllocator regAllocator) {
+    if (expr != null) {
+      String reg = regAllocator.getRegisterOrLoadIntoRegister(
+          expr.toMIPS(code, data, symbolTable, regAllocator), code);
+      code.append(
+          String.format("# symbol table size - %d\n", symbolTable.getOffset()));
+      code.append(
+          String.format("sw %s %d($sp)\n", reg, symbolTable.getOffset()));
+      code.append("jr $ra\n");
+    }
+    return MIPSResult.createVoidResult();
+  }
 }
-- 
cgit v1.2.3-70-g09d2