mirror of
https://github.com/polserver/polserver
synced 2026-08-13 08:23:08 -04:00
d7cdcff383
* Update grammar * Implement selfIs in BObjectImp * Add tests * Update grammar * Support obj.function as a MemberAccess [1/2] - AST generation changes * Support obj.function as a MemberAccess [2/2] - add funcref index to eprog class table ctor entry - allow check_mro on non-class inst (skip ctor calls) * Move address of funcref into funcref table * Add funcref.new() - Add class index to funcref table entry - Add MTH_NEW * Fix formatting * Address self-review comments * Some more tests * Address self-review comments * Update grammar * Remove ObjMember read_only, hidden No need to special-handle MBR_FUNCTION * Addres GitHub CI annotations - '=': conversion from 'size_t' to 'unsigned int', possible loss of data
103 lines
3.5 KiB
C++
103 lines
3.5 KiB
C++
#include "FunctionReferenceRegistrar.h"
|
|
|
|
#include <limits.h>
|
|
|
|
#include "bscript/compiler/ast/ClassDeclaration.h"
|
|
#include "bscript/compiler/ast/FunctionReference.h"
|
|
#include "bscript/compiler/ast/UserFunction.h"
|
|
#include "bscript/compiler/model/ClassLink.h"
|
|
#include "bscript/compiler/model/FlowControlLabel.h"
|
|
#include "bscript/compiler/model/FunctionLink.h"
|
|
#include "bscript/compiler/representation/FunctionReferenceDescriptor.h"
|
|
|
|
namespace Pol::Bscript::Compiler
|
|
{
|
|
FunctionReferenceRegistrar::FunctionReferenceRegistrar() = default;
|
|
|
|
void FunctionReferenceRegistrar::register_function_reference( const UserFunction& node,
|
|
FlowControlLabel& label )
|
|
{
|
|
auto function_name = node.scoped_name();
|
|
auto reference_itr = registered_references.find( function_name );
|
|
if ( reference_itr == registered_references.end() )
|
|
{
|
|
auto reference_index = static_cast<unsigned>( registered_references.size() );
|
|
|
|
registered_references[function_name] =
|
|
std::make_unique<EmittedReference>( reference_index, node, label );
|
|
reference_names_in_order.push_back( function_name );
|
|
}
|
|
}
|
|
|
|
std::vector<FunctionReferenceDescriptor> FunctionReferenceRegistrar::take_descriptors(
|
|
const std::map<std::string, size_t>& class_declaration_indexes )
|
|
{
|
|
std::vector<FunctionReferenceDescriptor> descriptors;
|
|
|
|
for ( const auto& k : reference_names_in_order )
|
|
{
|
|
const auto& v = registered_references[k];
|
|
if ( !v->label.has_address() )
|
|
{
|
|
v->user_function.internal_error(
|
|
"cannot take funcref descriptors as funcref has no address" );
|
|
}
|
|
|
|
unsigned class_index = std::numeric_limits<unsigned>::max();
|
|
|
|
if ( v->user_function.class_link )
|
|
{
|
|
if ( auto cd = v->user_function.class_link->class_declaration() )
|
|
{
|
|
auto cd_itr = class_declaration_indexes.find( cd->name );
|
|
|
|
if ( cd_itr == class_declaration_indexes.end() )
|
|
{
|
|
v->user_function.internal_error(
|
|
"cannot take funcref descriptors as funcref has no class declaration"
|
|
"index" );
|
|
}
|
|
|
|
class_index = static_cast<unsigned>( cd_itr->second );
|
|
}
|
|
}
|
|
|
|
descriptors.emplace_back( v->label.address(), v->user_function.parameter_count(),
|
|
v->user_function.capture_count(), v->user_function.is_variadic(),
|
|
class_index );
|
|
}
|
|
|
|
registered_references.clear();
|
|
reference_names_in_order.clear();
|
|
|
|
return descriptors;
|
|
}
|
|
|
|
void FunctionReferenceRegistrar::lookup_or_register_reference( const UserFunction& node,
|
|
FlowControlLabel& label,
|
|
unsigned& reference_index )
|
|
{
|
|
auto function_name = node.scoped_name();
|
|
|
|
auto reference_itr = registered_references.find( function_name );
|
|
if ( reference_itr != registered_references.end() )
|
|
{
|
|
reference_index = ( *reference_itr ).second->index;
|
|
return;
|
|
}
|
|
|
|
register_function_reference( node, label );
|
|
|
|
reference_index = registered_references[function_name]->index;
|
|
}
|
|
|
|
FunctionReferenceRegistrar::EmittedReference::EmittedReference( unsigned index,
|
|
const UserFunction& user_function,
|
|
FlowControlLabel& label )
|
|
: index( index ), user_function( user_function ), label( label )
|
|
{
|
|
}
|
|
|
|
FunctionReferenceRegistrar::EmittedReference::~EmittedReference() = default;
|
|
|
|
} // namespace Pol::Bscript::Compiler
|