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 --- .../org/zwobble/couscous/ast/LocalVariableDeclarationNode.java | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java') diff --git a/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java b/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java index bb272fc..3499e26 100644 --- a/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java +++ b/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java @@ -2,6 +2,8 @@ package org.zwobble.couscous.ast; import org.zwobble.couscous.ast.visitors.StatementNodeMapper; +import java.util.function.Function; + import static org.zwobble.couscous.ast.VariableDeclaration.var; public class LocalVariableDeclarationNode implements VariableNode, StatementNode { @@ -50,6 +52,11 @@ public class LocalVariableDeclarationNode implements VariableNode, StatementNode return visitor.visit(this); } + @Override + public StatementNode replaceExpressions(Function replace) { + return new LocalVariableDeclarationNode(declaration, replace.apply(initialValue)); + } + @Override public String toString() { return "LocalVariableDeclarationNode(declaration=" + declaration -- cgit v1.2.3