updates

[IRC.git] / Robust / src / IR / Tree / LoopNode.java

diff --git a/Robust/src/IR/Tree/LoopNode.java b/Robust/src/IR/Tree/LoopNode.java
index 2dcba8c5b11fe369f900667709eeb8a1c915a3cf..d604b1aae6565093e1a34079ece10499ebd46aba 100644 (file)
--- a/Robust/src/IR/Tree/LoopNode.java
+++ b/Robust/src/IR/Tree/LoopNode.java
@@ -1,11 +1,64 @@
 package IR.Tree;
 
-class LoopNode extends BlockStatementNode {
-    public SubBlockNode() {
+public class LoopNode extends BlockStatementNode {
+    BlockNode initializer;
+    ExpressionNode condition;
+    BlockNode update;
+    BlockNode body;
+    int type=0;
+    public static int FORLOOP=1;
+    public static int WHILELOOP=2;
+    public static int DOWHILELOOP=3;
+
+    public LoopNode(BlockNode initializer,ExpressionNode condition, BlockNode update, BlockNode body) {
+       this.initializer=initializer;
+       this.condition=condition;
+       this.update=update;
+       this.body=body;
+       initializer.setStyle(BlockNode.EXPRLIST);
+       update.setStyle(BlockNode.EXPRLIST);
+       type=FORLOOP;
+    }
+
+    public LoopNode(ExpressionNode condition, BlockNode body, int type) {
+       this.condition=condition;
+       this.body=body;
+       this.type=type;
     }
     
-    public String printNode() {
-       return null;
+    public BlockNode getInitializer() {
+       return initializer;
+    }
+
+    public ExpressionNode getCondition() {
+       return condition;
+    }
+
+    public BlockNode getUpdate() {
+       return update;
     }
 
+    public BlockNode getBody() {
+       return body;
+    }
+
+    public String printNode(int indent) {
+       if (type==FORLOOP) {
+           return "for("+initializer.printNode(0)+";"+condition.printNode(0)+
+               ";"+update.printNode(0)+") "+body.printNode(indent)+"\n";
+       } else if (type==WHILELOOP) {
+           return "while("+condition.printNode(0)+") "+body.printNode(indent+INDENT)+"\n";
+       } else if (type==DOWHILELOOP) {
+           return "do "+ body.printNode(indent+INDENT)+
+               "while("+condition.printNode(0)+")\n";
+       } else throw new Error();
+    }
+
+    public int getType() {
+       return type;
+    }
+
+    public int kind() {
+       return Kind.LoopNode;
+    }
 }