InitHEADmain

1 files changed, 33 insertions, 0 deletions

diff --git a/Homework/cs5300/p4-formatter/submit/ast/IfStatement.java b/Homework/cs5300/p4-formatter/submit/ast/IfStatement.java
new file mode 100644
index 0000000..1f9b577
--- /dev/null
+++ b/Homework/cs5300/p4-formatter/submit/ast/IfStatement.java
@@ -0,0 +1,33 @@
+package submit.ast;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class IfStatement implements Statement {
+    private ArrayList<Statement> statements;
+    private Expression simpleExpression;
+
+    public IfStatement(Expression simpleExpression, List<Statement> statements) {
+        this.simpleExpression = simpleExpression;
+        this.statements = new ArrayList<>(statements);
+    }
+
+    @Override
+    public void toCminus(StringBuilder builder, String prefix) {
+        builder.append(prefix);
+        builder.append("if").append(" (");
+        simpleExpression.toCminus(builder, "");
+        builder.append(")\n");
+
+        Statement ifStmt = statements.get(0);
+        boolean isIfCompound = ifStmt.getClass() == CompoundStatement.class;
+        ifStmt.toCminus(builder, prefix + (isIfCompound ? "" : "\t"));
+        if (statements.size() == 2) {
+            Statement elseStmt = statements.get(1);
+            boolean isElseCompound = elseStmt.getClass() == CompoundStatement.class;
+
+            builder.append(prefix).append("else\n");
+            elseStmt.toCminus(builder, prefix + (isElseCompound ? "" : "\t"));
+        }
+    }
+}