mirror of
https://github.com/polserver/polserver
synced 2026-08-13 08:23:08 -04:00
47 lines
1.3 KiB
C++
47 lines
1.3 KiB
C++
|
#include "ReferencedFunctionGatherer.h"
|
||
|
|
|
||
|
|
#include "compiler/ast/FunctionCall.h"
|
||
|
|
#include "compiler/model/FunctionLink.h"
|
||
|
|
|
||
|
|
namespace Pol::Bscript::Compiler
|
||
|
|
{
|
||
|
|
ReferencedFunctionGatherer::ReferencedFunctionGatherer(
|
||
|
|
std::vector<std::unique_ptr<ModuleFunctionDeclaration>>& all_module_function_declarations )
|
||
|
|
{
|
||
|
|
for ( auto& mfd : all_module_function_declarations )
|
||
|
|
{
|
||
|
|
unreferenced_module_function_declarations.insert( mfd.get() );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void ReferencedFunctionGatherer::visit_function_call( FunctionCall& fc )
|
||
|
|
{
|
||
|
|
visit_children( fc );
|
||
|
|
|
||
|
|
reference( *fc.function_link );
|
||
|
|
}
|
||
|
|
|
||
|
|
void ReferencedFunctionGatherer::reference( FunctionLink& link )
|
||
|
|
{
|
||
|
|
if ( auto mfd = link.module_function_declaration() )
|
||
|
|
reference( mfd );
|
||
|
|
}
|
||
|
|
|
||
|
|
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 );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<ModuleFunctionDeclaration*>
|
||
|
|
ReferencedFunctionGatherer::take_referenced_module_function_declarations()
|
||
|
|
{
|
||
|
|
return std::move( referenced_module_function_declarations );
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace Pol::Bscript::Compiler
|