From 052177994ca489078cf6057064ea2a97b35fe5d1 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sat, 28 Oct 2023 23:21:10 +0100 Subject: Introduce TypedFunctionBodyNode --- src/main/java/org/shedlang/compiler/stackir/StackCompiler.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/main/java/org/shedlang/compiler/stackir') diff --git a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java index 33d84149..2bd1229a 100644 --- a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java +++ b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java @@ -203,7 +203,7 @@ public class StackCompiler { )) .toList(); - context = compileBlockWithYield(node.body(), context); + context = compileBlockWithYield(node.body().block(), context); switch (node.body().behaviour()) { case TypedFunctionStatementBehaviour.BlockReturn blockReturn -> { @@ -221,7 +221,7 @@ public class StackCompiler { context = context.exitBlock(); var freeVariables = FreeVariables.builder(); - for (var shedFreeVariable : node.freeVariables()) { + for (var shedFreeVariable : node.body().freeVariables()) { var freeVariableLocalName = context.shedVariableToLocal(shedFreeVariable.name()); for (var freeVariableSource : shedFreeVariable.referenceSources()) { freeVariables = freeVariables.plus( -- cgit v1.2.3