Function PatchRtShader

shadercompiler/EffectCompilerMetal.cpp:3078–3367

}

ASTNode* PatchRtShader( RtShaderType shaderType, ASTNode* callNode, ParserState& state, std::vector<Symbol*>& rtConstantBuffers, const std::vector<GlobalInputElement>& globalInput, ASTNode* globalInputsStruct )
{
	ZoneScoped;

	Symbol* entryPointSymbol = callNode->GetSymbol();
	if( !entryPointSymbol || !entryPointSymbol->definition )
	{
		return nullptr;
	}

	ASTNode* functionHeader = entryPointSymbol->definition->GetChildOrNull( 0 );
	if( !functionHeader )
	{
		return nullptr;
	}

	ApplyPackedModifiersToConstantBuffers( entryPointSymbol->definition );

	ScannerToken shaderName = ScannerToken::ID( callNode->GetSymbol()->name );

	state.GetCurrentLocation().fileName = shaderName.stringValue;
	state.GetCurrentLocation().lineNumber = 1;

	auto header = state.NewNode( NT_FUNCTION_HEADER, shaderName );
	auto shaderSymbol = state.GetSymbolTable().AddSymbol( shaderName.stringValue, ALLOW_OVERRIDES );
	shaderSymbol->isFunction = true;
	header->SetSymbol( shaderSymbol );

	state.GetSymbolTable().EnterScope();

	auto shaderBody = state.NewNode( NT_BLOCK );

	ASTNode* payloadArg = nullptr;
	ASTNode* attributeArg = nullptr;

	auto sourceArguments = ClassifyRtShaderArguments( functionHeader );
	if( sourceArguments.payloadArg )
	{
		payloadArg = NewFunctionParameter( state, sourceArguments.payloadArg->GetType(), state.AllocateName( sourceArguments.payloadArg->GetSymbol()->name ) );
		payloadArg->SetToken( ScannerToken::FromTokenType( OP_INOUT ) );
		header->AddChild( payloadArg );
	}
	if( sourceArguments.attributeArg )
	{
		attributeArg = NewFunctionParameter( state, sourceArguments.attributeArg->GetType(), state.AllocateName( sourceArguments.attributeArg->GetSymbol()->name ) );
		header->AddChild( attributeArg );
	}

	ASTNode* localInputArg = nullptr;
	{
		auto t = state.GetSymbolTable().AddTypeSymbol( MakeInlineString( "device __RtLocalMaterial*" ) );
		localInputArg = NewFunctionParameter( state, TypeFromSymbol( t ), "__rtMaterials", RegisterSpecifier::Register( MetalRegister::SRV, 0 ) );
		header->AddChild( localInputArg );
	}

	ASTNode* globalInputArg = nullptr;
	{
		globalInputArg = NewFunctionParameter( state, TypeFromSymbol( globalInputsStruct->GetSymbol() ), "__rtGlobals", RegisterSpecifier::Register( MetalRegister::SRV, 1 ) );