Function AddShaderTableArguments

shadercompiler/EffectCompilerMetal.cpp:881–957

}

void AddShaderTableArguments( ParserState& state, const std::map<Symbol*, ASTNode*>& functions )
{
	auto traceRays = find_if( begin( state.GetDX9Functions() ), end( state.GetDX9Functions() ), []( auto& name ) { return strcmp( name.first, "TraceRay" ) == 0; } );
	if( traceRays == end( state.GetDX9Functions() ) )
	{
		return;
	}

	// Find TraceRay calls and map them to the payload type.
	std::map<Symbol*, Type> payloadTypes;
	// Maps function to the shader table argument symbol.
	std::map<Symbol*, Symbol*> patchedHeaders;
	patchedHeaders[traceRays->second] = nullptr;

	// Add shader table argument to functions that end up calling TraceRay.
	bool modified = true;
	while( modified )
	{
		modified = false;
		for( auto& func : functions )
		{
			func.second->Map( [&]( ASTNode* node ) {
				if( node->GetNodeType() == NT_FUNCTION_CALL )
				{
					Symbol* funcSymbol = node->GetSymbol();
					if( patchedHeaders.find( funcSymbol ) != end( patchedHeaders ) )
					{
						if( patchedHeaders.find( func.first ) == patchedHeaders.end() )
						{
							Type payloadType;
							if( funcSymbol == traceRays->second )
							{
								if( auto payloadArg = node->GetChildOrNull( 7 ) )
								{
									payloadType = payloadArg->GetType();
								}
							}
							else
							{
								payloadType = payloadTypes[funcSymbol];
							}
							auto shaderTable = CreateShaderTableArgument( state, payloadType, node->GetScope() );

							ASTNode* header = func.second->GetChild( 0 );
							header->InsertChild( 0, shaderTable );
							patchedHeaders[func.first] = shaderTable->GetSymbol();
							payloadTypes[func.first] = payloadType;

							modified = true;
						}
					}
				}
				return node;
			} );
		}
	}

	// Add shader table argument to function calls.