Function TEST_F

trinityal/tests/Compute.cpp:14–54

using namespace Tr2RenderContextEnum;

TEST_F( Compute, CanReadCSResult )
{
	ENSURE_GPU_OR_SKIP
	uint8_t vsBytecode[] = {
#include INCLUDE_SHADER_CODE( OutputVector.cs )
	};

	auto signature = Tr2ShaderSignatureAL()
						 .Add( Tr2ShaderRegisterAL::UAV_BUFFER, 0 )
						 .Add( Tr2ShaderThreadGroupSizeAL( 1, 1, 1 ) );
	Tr2ShaderAL cs;
	ASSERT_HRESULT_SUCCEEDED( cs.Create( COMPUTE_SHADER, vsBytecode, signature, "", *renderContext ) );

	Tr2ShaderAL shaders[] = { cs };
	Tr2ShaderProgramAL sp;
	ASSERT_HRESULT_SUCCEEDED( sp.Create( shaders, 1, *renderContext ) );

	Tr2BufferAL output;
	ASSERT_HRESULT_SUCCEEDED( output.Create( PIXEL_FORMAT_R32G32B32A32_FLOAT, 1, Tr2GpuUsage::UNORDERED_ACCESS, Tr2CpuUsage::READ, nullptr, *renderContext ) );

	Tr2ResourceSetDescriptionAL desc( sp );
	desc.SetUav( Tr2RenderContextEnum::COMPUTE_SHADER, 0, output );
	Tr2ResourceSetAL resourceSet;
	ASSERT_HRESULT_SUCCEEDED( resourceSet.Create( desc, sp, *renderContext ) );

	ASSERT_HRESULT_SUCCEEDED( renderContext->SetResourceSet( resourceSet ) );
	ASSERT_HRESULT_SUCCEEDED( renderContext->SetShaderProgram( sp ) );

	ASSERT_HRESULT_SUCCEEDED( renderContext->RunComputeShader( 1, 1, 1 ) );

	ASSERT_HRESULT_SUCCEEDED( renderContext->SetResourceSet( Tr2ResourceSetAL() ) );

	const float* data;
	ASSERT_HRESULT_SUCCEEDED( output.MapForReading( data, *renderContext ) );

	EXPECT_EQ( 1.0f, data[0] );
	EXPECT_EQ( 2.0f, data[1] );
	EXPECT_EQ( 3.0f, data[2] );
	EXPECT_EQ( 4.0f, data[3] );
	output.UnmapForReading( *renderContext );
}


TEST_F( Compute, DISABLED_CanAddInCS )