1 files changed, 44 insertions, 0 deletions
diff --git a/submit/ast/If.java b/submit/ast/If.java
new file mode 100644
index 0000000..a516c94
--- /dev/null
+++ b/submit/ast/If.java
@@ -0,0 +1,44 @@
+/*
+ * Code formatter project
+ * CS 4481
+ */
+package submit.ast;
+
+/**
+ *
+ * @author edwajohn
+ */
+public class If extends AbstractNode implements Statement {
+
+  private final Expression expression;
+  private final Statement trueStatement;
+  private final Statement falseStatement;
+
+  public If(Expression expression, Statement trueStatement, Statement falseStatement) {
+    this.expression = expression;
+    this.trueStatement = trueStatement;
+    this.falseStatement = falseStatement;
+  }
+
+  @Override
+  public void toCminus(StringBuilder builder, String prefix) {
+    builder.append(prefix).append("if (");
+    expression.toCminus(builder, prefix);
+    builder.append(")\n");
+    if (trueStatement instanceof CompoundStatement) {
+      trueStatement.toCminus(builder, prefix);
+    } else {
+      trueStatement.toCminus(builder, prefix + " ");
+    }
+    if (falseStatement != null) {
+      builder.append(prefix).append("else\n");
+      // falseStatement.toCminus(builder, prefix);
+      if (falseStatement instanceof CompoundStatement) {
+        falseStatement.toCminus(builder, prefix);
+      } else {
+        falseStatement.toCminus(builder, prefix + " ");
+      }
+    }
+    // builder.append(prefix).append("}");
+  }
+}