diff options
Diffstat (limited to 'jlox/src/com/craftinginterpreters/lox/LoxFunction.java')
-rw-r--r-- | jlox/src/com/craftinginterpreters/lox/LoxFunction.java | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/jlox/src/com/craftinginterpreters/lox/LoxFunction.java b/jlox/src/com/craftinginterpreters/lox/LoxFunction.java new file mode 100644 index 0000000..07dd727 --- /dev/null +++ b/jlox/src/com/craftinginterpreters/lox/LoxFunction.java @@ -0,0 +1,52 @@ +package com.craftinginterpreters.lox; + +import java.util.List; + +class LoxFunction implements LoxCallable { + private final Stmt.Function declaration; + private final Environment closure; + private final boolean isInitializer; + + LoxFunction(Stmt.Function declaration, Environment closure, boolean isInitializer) { + this.isInitializer = isInitializer; + this.closure = closure; + this.declaration = declaration; + } + + LoxFunction bind(LoxInstance instance) { + Environment environment = new Environment(closure); + environment.define("this", instance); + return new LoxFunction(declaration, environment, isInitializer); + } + + @Override + public int arity() { + return declaration.params.size(); + } + + @Override + public Object call(Interpreter interpreter, List<Object> arguments) { + Environment environment = new Environment(closure); + + for (int i = 0; i < declaration.params.size(); i++) { + environment.define(declaration.params.get(i).lexeme, arguments.get(i)); + } + + try { + interpreter.executeBlock(declaration.body, environment); + } catch (Return returnValue) { + if (isInitializer) + return closure.getAt(0, "this"); + return returnValue.value; + } + + if (isInitializer) + return closure.getAt(0, "this"); + return null; + } + + @Override + public String toString() { + return "<fn " + declaration.name.lexeme + ">"; + } +} |