aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java')
-rw-r--r--src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java b/src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java
index 9d17848..3ffcf63 100644
--- a/src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/TernaryConditionalNode.java
@@ -3,6 +3,7 @@ package org.zwobble.couscous.ast;
import org.zwobble.couscous.types.Type;
import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;
import org.zwobble.couscous.ast.visitors.NodeTransformer;
+import org.zwobble.couscous.util.ExtraIterables;
public class TernaryConditionalNode implements ExpressionNode {
public static TernaryConditionalNode ternaryConditional(
@@ -43,6 +44,11 @@ public class TernaryConditionalNode implements ExpressionNode {
}
@Override
+ public Iterable<? extends Node> childNodes() {
+ return ExtraIterables.of(condition, ifTrue, ifFalse);
+ }
+
+ @Override
public ExpressionNode transform(NodeTransformer transformer) {
return new TernaryConditionalNode(
transformer.transformExpression(condition),