mirror of
https://github.com/polserver/polserver
synced 2026-08-13 08:23:08 -04:00
1463b76c7c
* all except pol * pol and includes under defines * something weird has happened * remove own folder from include searchpath * fixed remaining, adapted include style for external headers * missing include
90 lines
2.6 KiB
C++
90 lines
2.6 KiB
C++
#include "bscript/compiler/optimizer/ReferencedFunctionGatherer.h"
|
|
|
|
#include "bscript/compiler/ast/ClassParameterDeclaration.h"
|
|
#include "bscript/compiler/ast/FunctionCall.h"
|
|
#include "bscript/compiler/ast/FunctionExpression.h"
|
|
#include "bscript/compiler/ast/FunctionReference.h"
|
|
#include "bscript/compiler/ast/UserFunction.h"
|
|
#include "bscript/compiler/model/FunctionLink.h"
|
|
|
|
namespace Pol::Bscript::Compiler
|
|
{
|
|
ReferencedFunctionGatherer::ReferencedFunctionGatherer(
|
|
std::vector<std::unique_ptr<ModuleFunctionDeclaration>>& all_module_function_declarations,
|
|
std::vector<std::unique_ptr<UserFunction>> all_user_functions )
|
|
{
|
|
for ( auto& mfd : all_module_function_declarations )
|
|
{
|
|
unreferenced_module_function_declarations.insert( mfd.get() );
|
|
}
|
|
for ( auto& uf : all_user_functions )
|
|
{
|
|
UserFunction* p = uf.get();
|
|
unreferenced_user_functions[p] = std::move( uf );
|
|
}
|
|
}
|
|
|
|
void ReferencedFunctionGatherer::visit_function_call( FunctionCall& fc )
|
|
{
|
|
visit_children( fc );
|
|
|
|
reference( *fc.function_link );
|
|
}
|
|
|
|
void ReferencedFunctionGatherer::visit_function_reference( FunctionReference& fr )
|
|
{
|
|
reference( *fr.function_link );
|
|
}
|
|
|
|
void ReferencedFunctionGatherer::visit_function_expression( FunctionExpression& fr )
|
|
{
|
|
reference( *fr.function_link );
|
|
}
|
|
|
|
void ReferencedFunctionGatherer::visit_class_parameter_declaration( ClassParameterDeclaration& cpd )
|
|
{
|
|
reference( *cpd.constructor_link );
|
|
}
|
|
|
|
void ReferencedFunctionGatherer::reference( FunctionLink& link )
|
|
{
|
|
if ( auto mfd = link.module_function_declaration() )
|
|
reference( mfd );
|
|
else if ( auto uf = link.user_function() )
|
|
reference( uf );
|
|
}
|
|
|
|
void ReferencedFunctionGatherer::reference( ModuleFunctionDeclaration* mfd )
|
|
{
|
|
auto itr = unreferenced_module_function_declarations.find( mfd );
|
|
if ( itr != unreferenced_module_function_declarations.end() )
|
|
{
|
|
referenced_module_function_declarations.push_back( mfd );
|
|
unreferenced_module_function_declarations.erase( itr );
|
|
}
|
|
}
|
|
|
|
void ReferencedFunctionGatherer::reference( UserFunction* uf )
|
|
{
|
|
auto itr = unreferenced_user_functions.find( uf );
|
|
if ( itr != unreferenced_user_functions.end() )
|
|
{
|
|
referenced_user_functions.push_back( std::move( ( *itr ).second ) );
|
|
unreferenced_user_functions.erase( itr );
|
|
uf->accept( *this );
|
|
}
|
|
}
|
|
|
|
std::vector<std::unique_ptr<UserFunction>>
|
|
ReferencedFunctionGatherer::take_referenced_user_functions()
|
|
{
|
|
return std::move( referenced_user_functions );
|
|
}
|
|
|
|
std::vector<ModuleFunctionDeclaration*>
|
|
ReferencedFunctionGatherer::take_referenced_module_function_declarations()
|
|
{
|
|
return std::move( referenced_module_function_declarations );
|
|
}
|
|
|
|
} // namespace Pol::Bscript::Compiler
|