Add if branch token and node

1 files changed, 42 insertions, 0 deletions

diff --git a/main/core/parser/ast/BranchNode.java b/main/core/parser/ast/BranchNode.java
new file mode 100644
index 0000000..d6fb4cf
--- /dev/null
+++ b/main/core/parser/ast/BranchNode.java
@@ -0,0 +1,42 @@
+package com.jantuomi.interpreter.main.core.parser.ast;
+
+import com.jantuomi.interpreter.main.core.parser.datatype.BooleanDataContainer;
+import com.jantuomi.interpreter.main.core.parser.datatype.DataContainer;
+import com.jantuomi.interpreter.main.core.tokenizer.token.Token;
+import com.jantuomi.interpreter.main.exception.InterpreterException;
+
+import java.util.Arrays;
+import java.util.List;
+
+/**
+ * Created by jan on 17.6.2016.
+ */
+public class BranchNode extends ASTNode {
+
+    private ASTNode expression;
+    private BlockBodyNode branch;
+
+    public BranchNode(Token token, ASTNode expression, BlockBodyNode branch) {
+        super(token);
+        this.expression = expression;
+        this.branch = branch;
+    }
+
+    @Override
+    public DataContainer evaluate() throws InterpreterException {
+        DataContainer ev = expression.evaluate();
+        boolean returnValue = false;
+        if (ev instanceof BooleanDataContainer) {
+            returnValue = ((BooleanDataContainer) ev).getData();
+            if (returnValue) {
+                branch.evaluate();
+            }
+        }
+        return new BooleanDataContainer(returnValue);
+    }
+
+    @Override
+    List<ASTNode> getChildren() {
+        return Arrays.asList(expression, branch);
+    }
+}