diff --git a/tests/d3d12_streamout.c b/tests/d3d12_streamout.c index 60656b81..abd4fc86 100644 --- a/tests/d3d12_streamout.c +++ b/tests/d3d12_streamout.c @@ -22,6 +22,159 @@ #define VKD3D_DBG_CHANNEL VKD3D_DBG_CHANNEL_API #include "d3d12_crosstest.h" +void test_primitive_restart_list_topology_stream_output(void) +{ + D3D12_GRAPHICS_PIPELINE_STATE_DESC pso_desc; + ID3D12Resource *counter_buffer, *so_buffer; + ID3D12GraphicsCommandList *command_list; + D3D12_STREAM_OUTPUT_BUFFER_VIEW sobv; + struct test_context_desc desc; + ID3D12Resource *index_buffer; + struct resource_readback rb; + struct test_context context; + D3D12_INDEX_BUFFER_VIEW ibv; + ID3D12CommandQueue *queue; + const struct vec4 *data; + ID3D12Device *device; + uint32_t counter; + unsigned int i; + HRESULT hr; + + static const D3D12_SO_DECLARATION_ENTRY so_declaration[] = + { + {0, "SV_Position", 0, 0, 4, 0}, + }; + static const struct vec4 expected_output[] = + { + /* Strip */ + { 2000.0f, 2000.0f, 2000.0f, 2000.0f }, + { 3000.0f, 3000.0f, 3000.0f, 3000.0f }, + { 4000.0f, 4000.0f, 4000.0f, 4000.0f }, + + /* List */ + { 0.0f, 0.0f, 0.0f, 0.0f }, + { 1.0f, 1.0f, 1.0f, 1.0f }, + { -1.0f, -1.0f, -1.0f, -1.0f }, + { 9.0f, 9.0f, 9.0f, 9.0f }, + { -1.0f, -1.0f, -1.0f, -1.0f }, + { -1.0f, -1.0f, -1.0f, -1.0f }, + { 2000.0f, 2000.0f, 2000.0f, 2000.0f }, + { 3000.0f, 3000.0f, 3000.0f, 3000.0f }, + { 4000.0f, 4000.0f, 4000.0f, 4000.0f }, + + /* Strip */ + { 2000.0f, 2000.0f, 2000.0f, 2000.0f }, + { 3000.0f, 3000.0f, 3000.0f, 3000.0f }, + { 4000.0f, 4000.0f, 4000.0f, 4000.0f }, + }; + static const uint32_t index_data[] = { 0, 1, UINT32_MAX, 9, UINT32_MAX, UINT32_MAX, 2000, 3000, 4000 }; + static const UINT strides[] = { 16 }; + + static const DWORD vs_code[] = + { +#if 0 + float4 main(uint vid : SV_VertexID) : SV_Position + { + if (vid == ~0u) + return float4(-1, -1, -1, -1); + else + return float4(vid, vid, vid, vid); + } +#endif + 0x43425844, 0x59eaaf80, 0xf7ab5160, 0xf0ce6da4, 0x82ce289b, 0x00000001, 0x00000140, 0x00000003, + 0x0000002c, 0x00000060, 0x00000094, 0x4e475349, 0x0000002c, 0x00000001, 0x00000008, 0x00000020, + 0x00000000, 0x00000006, 0x00000001, 0x00000000, 0x00000101, 0x565f5653, 0x65747265, 0x00444978, + 0x4e47534f, 0x0000002c, 0x00000001, 0x00000008, 0x00000020, 0x00000000, 0x00000001, 0x00000003, + 0x00000000, 0x0000000f, 0x505f5653, 0x7469736f, 0x006e6f69, 0x58454853, 0x000000a4, 0x00010050, + 0x00000029, 0x0100086a, 0x04000060, 0x00101012, 0x00000000, 0x00000006, 0x04000067, 0x001020f2, + 0x00000000, 0x00000001, 0x02000068, 0x00000001, 0x07000020, 0x00100012, 0x00000000, 0x0010100a, + 0x00000000, 0x00004001, 0xffffffff, 0x0304001f, 0x0010000a, 0x00000000, 0x08000036, 0x001020f2, + 0x00000000, 0x00004002, 0xbf800000, 0xbf800000, 0xbf800000, 0xbf800000, 0x0100003e, 0x01000012, + 0x05000056, 0x001020f2, 0x00000000, 0x00101006, 0x00000000, 0x0100003e, 0x01000015, 0x0100003e, + }; + + static const D3D12_SHADER_BYTECODE vs = SHADER_BYTECODE(vs_code); + + memset(&desc, 0, sizeof(desc)); + desc.root_signature_flags = D3D12_ROOT_SIGNATURE_FLAG_ALLOW_STREAM_OUTPUT; + desc.no_pipeline = true; + if (!init_test_context(&context, &desc)) + return; + + device = context.device; + command_list = context.list; + queue = context.queue; + + init_pipeline_state_desc(&pso_desc, context.root_signature, 0, &vs, NULL, NULL); + pso_desc.StreamOutput.NumEntries = ARRAY_SIZE(so_declaration); + pso_desc.StreamOutput.pSODeclaration = so_declaration; + pso_desc.StreamOutput.pBufferStrides = strides; + pso_desc.StreamOutput.NumStrides = ARRAY_SIZE(strides); + pso_desc.StreamOutput.RasterizedStream = D3D12_SO_NO_RASTERIZED_STREAM; + pso_desc.IBStripCutValue = D3D12_INDEX_BUFFER_STRIP_CUT_VALUE_0xFFFFFFFF; + hr = ID3D12Device_CreateGraphicsPipelineState(device, &pso_desc, + &IID_ID3D12PipelineState, (void **)&context.pipeline_state); + ok(SUCCEEDED(hr), "Failed to create PSO, hr #%x.\n", hr); + + counter_buffer = create_default_buffer(device, 32, + D3D12_RESOURCE_FLAG_NONE, D3D12_RESOURCE_STATE_STREAM_OUT); + so_buffer = create_default_buffer(device, 4096, + D3D12_RESOURCE_FLAG_NONE, D3D12_RESOURCE_STATE_STREAM_OUT); + index_buffer = create_upload_buffer(device, sizeof(index_data), index_data); + sobv.BufferLocation = ID3D12Resource_GetGPUVirtualAddress(so_buffer); + sobv.SizeInBytes = 4096; + sobv.BufferFilledSizeLocation = ID3D12Resource_GetGPUVirtualAddress(counter_buffer); + + ibv.Format = DXGI_FORMAT_R32_UINT; + ibv.BufferLocation = ID3D12Resource_GetGPUVirtualAddress(index_buffer); + ibv.SizeInBytes = sizeof(index_data); + + ID3D12GraphicsCommandList_RSSetScissorRects(command_list, 1, &context.scissor_rect); + ID3D12GraphicsCommandList_RSSetViewports(command_list, 1, &context.viewport); + ID3D12GraphicsCommandList_SetGraphicsRootSignature(command_list, context.root_signature); + ID3D12GraphicsCommandList_SetPipelineState(command_list, context.pipeline_state); + ID3D12GraphicsCommandList_SOSetTargets(command_list, 0, 1, &sobv); + ID3D12GraphicsCommandList_IASetIndexBuffer(command_list, &ibv); + + /* Primitive restart state only applies to strip primitives. */ + ID3D12GraphicsCommandList_IASetPrimitiveTopology(command_list, D3D_PRIMITIVE_TOPOLOGY_TRIANGLESTRIP); + ID3D12GraphicsCommandList_DrawIndexedInstanced(command_list, ARRAY_SIZE(index_data), 1, + 0, 0, 0); + ID3D12GraphicsCommandList_IASetPrimitiveTopology(command_list, D3D_PRIMITIVE_TOPOLOGY_TRIANGLELIST); + ID3D12GraphicsCommandList_DrawIndexedInstanced(command_list, ARRAY_SIZE(index_data), 1, + 0, 0, 0); + ID3D12GraphicsCommandList_IASetPrimitiveTopology(command_list, D3D_PRIMITIVE_TOPOLOGY_TRIANGLESTRIP); + ID3D12GraphicsCommandList_DrawIndexedInstanced(command_list, ARRAY_SIZE(index_data), 1, + 0, 0, 0); + + transition_resource_state(command_list, counter_buffer, + D3D12_RESOURCE_STATE_STREAM_OUT, D3D12_RESOURCE_STATE_COPY_SOURCE); + transition_resource_state(command_list, so_buffer, + D3D12_RESOURCE_STATE_STREAM_OUT, D3D12_RESOURCE_STATE_COPY_SOURCE); + + get_buffer_readback_with_command_list(counter_buffer, DXGI_FORMAT_R32_UINT, &rb, queue, command_list); + counter = get_readback_uint(&rb, 0, 0, 0); + ok(counter == sizeof(expected_output), "Got unexpected counter %u, expected %u.\n", + counter, (unsigned int)sizeof(expected_output)); + release_resource_readback(&rb); + reset_command_list(command_list, context.allocator); + get_buffer_readback_with_command_list(so_buffer, DXGI_FORMAT_UNKNOWN, &rb, queue, command_list); + for (i = 0; i < ARRAY_SIZE(expected_output); ++i) + { + const struct vec4 *expected = &expected_output[i]; + data = get_readback_vec4(&rb, i, 0); + ok(compare_vec4(data, expected, 1), + "Got {%.8e, %.8e, %.8e, %.8e}, expected {%.8e, %.8e, %.8e, %.8e}.\n", + data->x, data->y, data->z, data->w, expected->x, expected->y, expected->z, expected->w); + } + release_resource_readback(&rb); + + ID3D12Resource_Release(index_buffer); + ID3D12Resource_Release(counter_buffer); + ID3D12Resource_Release(so_buffer); + destroy_test_context(&context); +} + static void test_vertex_shader_stream_output(bool use_dxil) { D3D12_GRAPHICS_PIPELINE_STATE_DESC pso_desc; diff --git a/tests/d3d12_tests.h b/tests/d3d12_tests.h index c76c7835..e42e8e3d 100644 --- a/tests/d3d12_tests.h +++ b/tests/d3d12_tests.h @@ -199,6 +199,7 @@ decl_test(test_primitive_restart); decl_test(test_index_buffer_edge_case_stream_output); decl_test(test_vertex_shader_stream_output_dxbc); decl_test(test_vertex_shader_stream_output_dxil); +decl_test(test_primitive_restart_list_topology_stream_output); decl_test(test_read_write_subresource); decl_test(test_queue_wait); decl_test(test_graphics_compute_queue_synchronization);