polserver/pol-core/bscript/compiler/astbuilder/UserFunctionVisitor.cpp
Kevin Eady f381f46dc3
Add compiler support for function expressions (#671)
* initial poc of function expressions
- update grammar
- mock AST builder to return BBoolean(true) for a func expr
- update prettifier for skeleton implementation

* more skeleton work
create AST class FunctionExpression mimicking boolean value

* create ast UserFunction, add to workspace from functexpr

* can generate instructions

* reorg tests; add test for instructions

* can track FunctionDepth in Variable

* can get captures for funcexprs inside funcs.. tbd if this way of nesting works

* Implement create-functor instruction
- Move function depth from Variable to Variables
- Introduce stacking of `Variables` for function expresions via `FunctionVariableScope`
- Add `TOK_FUNCTOR` instruction for 'create-functor'
- Handle emitting a `FunctionExpression` AST node
- Update `emit.declare_variable` and `emit.access_variable` to account for function captures
- Remove `in_function` from instruction generator, as it is tracked via `UserFunction` stack
- Update tests

* Address Discord comments
- Swap pop param order

* Bubble up captured variables through nested functions

* Improve testing infrastructure; add some test cases

* Fix compilation error

* prepend captures to function parameters in funcref mth_call

* update tests

* move from function{} to @{}

* implementation fix

* some more tests

* update docs

* Some cleanup

* fix CI annotation warning
- 'argument': conversion from 'size_t' to 'VariableIndex', possible loss of data

* Address review comments
- Add `passert_always`
- Use better example in docs

* Some cleanup
- Remove unused functions
2024-07-29 22:30:52 +02:00

32 lines
1.1 KiB
C++

#include "UserFunctionVisitor.h"
#include "bscript/compiler/ast/UserFunction.h"
#include "bscript/compiler/astbuilder/BuilderWorkspace.h"
#include "bscript/compiler/model/CompilerWorkspace.h"
namespace Pol::Bscript::Compiler
{
UserFunctionVisitor::UserFunctionVisitor( const SourceFileIdentifier& sfi, BuilderWorkspace& ws )
: workspace( ws ), tree_builder( sfi, ws )
{
}
antlrcpp::Any UserFunctionVisitor::visitFunctionDeclaration(
EscriptGrammar::EscriptParser::FunctionDeclarationContext* ctx )
{
auto uf = tree_builder.function_declaration( ctx );
workspace.function_resolver.register_user_function( uf.get() );
workspace.compiler_workspace.user_functions.push_back( std::move( uf ) );
return antlrcpp::Any();
}
antlrcpp::Any UserFunctionVisitor::visitFunctionExpression(
EscriptGrammar::EscriptParser::FunctionExpressionContext* ctx )
{
auto uf = tree_builder.function_expression( ctx );
workspace.function_resolver.register_user_function( uf.get() );
workspace.compiler_workspace.user_functions.push_back( std::move( uf ) );
return antlrcpp::Any();
}
} // namespace Pol::Bscript::Compiler