From 51e8ff721cc7079212119ba44fcc4bf0f9818d1e Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 6 Dec 2015 11:20:51 +0000 Subject: Replace captured variable references in method bodies --- .../java/org/zwobble/couscous/ast/ExpressionStatementNode.java | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'src/main/java/org/zwobble/couscous/ast/ExpressionStatementNode.java') diff --git a/src/main/java/org/zwobble/couscous/ast/ExpressionStatementNode.java b/src/main/java/org/zwobble/couscous/ast/ExpressionStatementNode.java index b814076..0ac7381 100644 --- a/src/main/java/org/zwobble/couscous/ast/ExpressionStatementNode.java +++ b/src/main/java/org/zwobble/couscous/ast/ExpressionStatementNode.java @@ -2,6 +2,8 @@ package org.zwobble.couscous.ast; import org.zwobble.couscous.ast.visitors.StatementNodeMapper; +import java.util.function.Function; + public class ExpressionStatementNode implements StatementNode { public static ExpressionStatementNode expressionStatement(ExpressionNode expression) { return new ExpressionStatementNode(expression); @@ -22,6 +24,11 @@ public class ExpressionStatementNode implements StatementNode { return visitor.visit(this); } + @Override + public StatementNode replaceExpressions(Function replace) { + return new ExpressionStatementNode(replace.apply(expression)); + } + @Override public String toString() { return "ExpressionStatementNode(expression=" + expression + ")"; -- cgit v1.2.3