Method Commit

trinityal/dx12/util/DescriptorStateCacheDx12.cpp:189–251

Apply state */

/** Apply state */
void DescriptorStateCache::Commit( ID3D12GraphicsCommandList* commandList, ID3D12DescriptorHeap* globalSrvUavHeap, ID3D12DescriptorHeap* globalSamplerHeap, const TrinityALImpl::Tr2RootSignatureAL* rootSignature )
{
	bool rootSignatureChanged = rootSignature->m_rootSignature != m_rootSignature;

	// If we are changing the root signature or rebinding heaps, we must rebind everything
	bool forceSet = rootSignatureChanged || m_heapsDirty || m_globalSrvUavHeap != globalSrvUavHeap || m_globalSamplerHeap != globalSamplerHeap;

	SetHeaps( commandList, globalSrvUavHeap, globalSamplerHeap );

	bool mustBindSrvUav = m_srvUavDirty || forceSet;
	bool mustBindSampler = m_samplerDirty || forceSet;

	auto setRootDescriptorTable = rootSignature->m_isCompute ? &ID3D12GraphicsCommandList::SetComputeRootDescriptorTable : &ID3D12GraphicsCommandList::SetGraphicsRootDescriptorTable;

	// Check if the SRV/UAV table *can* be bound, *must* be bound and isn't already bound
	if( mustBindSrvUav )
	{
		for( uint32_t i = 0; i < rootSignature->m_srvUavParameterCount; ++i )
		{
			uint32_t index = rootSignature->m_srvUavParameterOffset + i;
			auto handle = m_srvUav[index]->GetHandleGPU();
			if( forceSet || !m_parameterSlots[index].IsValidSRV( handle ) )
			{
				m_parameterSlots[index].SetSRV( handle );
				( commandList->*setRootDescriptorTable )( index, handle );
			}
		}
	}

	if( mustBindSampler )
	{
		for( uint32_t i = 0; i < rootSignature->m_samplerParameterCount; ++i )
		{
			uint32_t index = rootSignature->m_samplerParameterOffset + i;
			auto handle = m_sampler[index]->GetHandleGPU();
			if( forceSet || !m_parameterSlots[index].IsValidSampler( handle ) )
			{
				m_parameterSlots[index].SetSampler( handle );
				( commandList->*setRootDescriptorTable )( index, handle );
			}
		}
	}

	D3D12_GPU_VIRTUAL_ADDRESS nullCbv = m_primaryContext->m_nullCB.GetGpuView();
	auto setConstantBufferView = rootSignature->m_isCompute ? &ID3D12GraphicsCommandList::SetComputeRootConstantBufferView : &ID3D12GraphicsCommandList::SetGraphicsRootConstantBufferView;
	for( auto it = begin( rootSignature->m_cbRegisters ); it != end( rootSignature->m_cbRegisters ); ++it )
	{
		D3D12_GPU_VIRTUAL_ADDRESS address = m_cbv[it->stage][it->index];
		if( rootSignatureChanged || !m_parameterSlots[it->parameter].IsValidCBV( address ) )
		{
			m_parameterSlots[it->parameter].SetCBV( address );
			( commandList->*setConstantBufferView )( it->parameter, address != 0 ? address : nullCbv );
		}
	}

	// Clear dirty flags
	m_srvUavDirty = false;
	m_samplerDirty = false;