From cd244418efb005c27b1018458767c57dfa09e046 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 17 Jan 2016 18:02:27 +0000
Subject: Add transformation pass to C# backend

---
 src/main/java/org/zwobble/couscous/ast/WhileNode.java | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/main/java/org/zwobble/couscous/ast/WhileNode.java')

diff --git a/src/main/java/org/zwobble/couscous/ast/WhileNode.java b/src/main/java/org/zwobble/couscous/ast/WhileNode.java
index 6462821..34d1883 100644
--- a/src/main/java/org/zwobble/couscous/ast/WhileNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/WhileNode.java
@@ -1,6 +1,7 @@
 package org.zwobble.couscous.ast;
 
 import com.google.common.base.Objects;
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
 import org.zwobble.couscous.ast.visitors.StatementNodeMapper;
 
 import java.util.List;
@@ -41,6 +42,12 @@ public class WhileNode implements StatementNode {
             eagerMap(body, statement -> statement.replaceExpressions(replace)));
     }
 
+    public StatementNode transform(NodeTransformer transformer) {
+        return new WhileNode(
+            transformer.visit(condition),
+            transformer.transformStatements(body));
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
-- 
cgit v1.2.3