Function AssignRegisters

shadercompiler/ParserUtils.cpp:733–826

}

void AssignRegisters( ASTNode* root, InputStageType stage, Registers& registers, const std::vector<GlobalInputElement>& globalInput, std::vector<RegisterSpecifier>& globalInputRegisters )
{
	if( !root )
	{
		return;
	}
	switch( root->GetNodeType() )
	{
	case NT_CBUFFER:
		if( root->GetSymbol()->used )
		{
			bool assigned = false;
			for( auto& r : root->GetSymbol()->registerSpecifier )
			{
				if( r.second.explicitRegister && DoesProfileMatchStage( r.first, stage ) )
				{
					assigned = true;
					break;
				}
			}
			if( !assigned )
			{
				RegisterSpecifier r;
				r.shaderProfile = MakeInlineString( "" );
				r.registerType = 'b';
				r.registerNumber = AllocateRegister( registers.b );
				r.subComponent = -1;
				r.space = 0;
				r.explicitRegister = false;
				r.explicitSpace = false;
				root->GetSymbol()->registerSpecifier[r.shaderProfile] = r;
			}
		}
		break;
	case NT_NAME_DECLARATION:
		if( !root->GetSymbol()->used )
		{
			return;
		}
		if( auto registerType = GetRegisterType( root->GetSymbol()->type ) )
		{
			auto globalElement = std::find_if( begin( globalInput ), end( globalInput ), [symbol = root->GetSymbol()]( const auto& element ) { return SymbolMatchesGlobalInput( *symbol, element ); } );
			if( globalElement != end( globalInput ) )
			{
				auto& r = globalInputRegisters[globalElement - begin( globalInput )];
				root->GetSymbol()->registerSpecifier[r.shaderProfile] = r;
			}
			else
			{

				bool assigned = false;
				for( auto& r : root->GetSymbol()->registerSpecifier )
				{
					if( r.second.explicitRegister && DoesProfileMatchStage( r.first, stage ) )
					{
						assigned = true;
						break;
					}