1 files changed, 65 insertions, 0 deletions
diff --git a/Homework/cs5300/compiler/p5-compiler/submit/ast/FunDeclaration.java b/Homework/cs5300/compiler/p5-compiler/submit/ast/FunDeclaration.java
new file mode 100644
index 0000000..f60840b
--- /dev/null
+++ b/Homework/cs5300/compiler/p5-compiler/submit/ast/FunDeclaration.java
@@ -0,0 +1,65 @@
+/*
+ * Code formatter project
+ * CS 4481
+ */
+package submit.ast;
+
+import java.util.ArrayList;
+import java.util.List;
+import submit.MIPSResult;
+import submit.RegisterAllocator;
+import submit.SymbolTable;
+
+/**
+ *
+ * @author edwajohn
+ */
+public class FunDeclaration extends AbstractNode implements Declaration {
+
+  private final VarType returnType;
+  private final String id;
+  private final ArrayList<Param> params;
+  private final Statement statement;
+
+  public FunDeclaration(VarType returnType, String id, List<Param> params,
+                        Statement statement) {
+    this.returnType = returnType;
+    this.id = id;
+    this.params = new ArrayList<>(params);
+    this.statement = statement;
+  }
+
+  public void toCminus(StringBuilder builder, final String prefix) {
+    String rt = (returnType != null) ? returnType.toString() : "void";
+    builder.append("\n").append(rt).append(" ");
+    builder.append(id);
+    builder.append("(");
+
+    for (Param param : params) {
+      param.toCminus(builder, prefix);
+      builder.append(", ");
+    }
+    if (!params.isEmpty()) {
+      builder.delete(builder.length() - 2, builder.length());
+    }
+    builder.append(")\n");
+    statement.toCminus(builder, prefix);
+  }
+
+  @Override
+  public MIPSResult toMIPS(StringBuilder code, StringBuilder data,
+                           SymbolTable symbolTable,
+                           RegisterAllocator registerAllocator) {
+    code.append(id).append(":\n");
+
+    statement.toMIPS(code, data, symbolTable, registerAllocator);
+
+    if (id.equals("main"))
+      code.append("li $v0 10\nsyscall\n");
+    else
+      code.append("jr $ra\n");
+    registerAllocator.clearAll();
+
+    return MIPSResult.createVoidResult();
+  }
+}