Function SearchFunctionsForGlobals

shadercompiler/EffectCompilerMetal.cpp:276–346

}

bool SearchFunctionsForGlobals( ASTNode* node, const std::set<Symbol*>& structMembers, const std::map<Symbol*, ASTNode*>& functions, std::map<Symbol*, ASTNode*>& functionsWithGlobals, std::map<Symbol*, ASTNode*>& functionsWithoutGlobals )
{
	for( size_t i = 0, n = node->GetChildrenCount(); i < n; ++i )
	{
		ASTNode* childNode = node->GetChild( i );
		if( !childNode )
		{
			continue;
		}

		if( childNode->GetNodeType() == NT_VAR_IDENTIFIER )
		{
			Symbol* symbol = childNode->GetSymbol();

			auto it = structMembers.find( symbol );
			if( it != structMembers.end() )
			{
				return true;
			}
		}
		else if( childNode->GetNodeType() == NT_FUNCTION_CALL )
		{
			Symbol* symbol = childNode->GetSymbol();
			if( symbol )
			{
				if( functionsWithGlobals.find( symbol ) != functionsWithGlobals.end() )
				{
					// The function was aready marked as using globals.
					return true;
				}
				else if( functionsWithoutGlobals.find( symbol ) == functionsWithoutGlobals.end() )
				{
					// If the function wasn't found in both lists it means either it was not checked yet
					// or it's a standard library function.
					auto it = functions.find( symbol );
					if( it != functions.end() )
					{
						ASTNode* definition = it->second;
						assert( definition && definition->GetNodeType() == NT_FUNCTION_DEFINITION );

						[[maybe_unused]] ASTNode* header = definition->GetChild( 0 );
						assert( header->GetNodeType() == NT_FUNCTION_HEADER );
						assert( header->GetSymbol() == symbol );

						ASTNode* body = definition->GetChild( 1 );
						assert( body->GetNodeType() == NT_BLOCK );

						bool found = SearchFunctionsForGlobals( body, structMembers, functions, functionsWithGlobals, functionsWithoutGlobals );
						if( found )
						{
							functionsWithGlobals.insert( { it->first, it->second } );
							return true;
						}
						else
						{
							functionsWithoutGlobals.insert( { it->first, it->second } );
						}
					}