InitHEADmain

1 files changed, 22 insertions, 0 deletions

diff --git a/Homework/cs5300/p4-formatter/submit/ast/WhileStatement.java b/Homework/cs5300/p4-formatter/submit/ast/WhileStatement.java
new file mode 100644
index 0000000..cd99d51
--- /dev/null
+++ b/Homework/cs5300/p4-formatter/submit/ast/WhileStatement.java
@@ -0,0 +1,22 @@
+package submit.ast;
+
+public class WhileStatement implements Statement {
+    Expression simpleExpression;
+    Statement statement;
+
+    public WhileStatement(Expression simpleExpression, Statement statement) {
+        this.simpleExpression = simpleExpression;
+        this.statement = statement;
+    }
+
+    public void toCminus(StringBuilder builder, String prefix) {
+        builder.append(prefix);
+
+        builder.append("while (");
+        simpleExpression.toCminus(builder, "");
+        builder.append(")\n");
+
+        boolean isBodyCompound = statement.getClass() == CompoundStatement.class;
+        statement.toCminus(builder, prefix + (isBodyCompound ? "" : "\t"));
+    }
+}