yum-mirror/slang

Making it easier to work with shaders

git clone https://git.yummers.dev/yum-mirror/slang

Gangzheng TongConvert gfx unit tests and examples to use slang-rhi (#7577)43d0c2100

master
branchesmasterall refs →
3.3 KiB103 linesraw
1#include "core/slang-basic.h"
2#include "gfx-test-util.h"
3#include "slang-rhi/shader-cursor.h"
4#include "unit-test/slang-unit-test.h"
5
6#include <slang-rhi.h>
7
8#if SLANG_WINDOWS_FAMILY
9#include <d3d12.h>
10#endif
11
12using namespace rhi;
13
14namespace gfx_test
15{
16void getBufferResourceHandleTestImpl(IDevice* device, UnitTestContext* context)
17{
18    const int numberCount = 1;
19    BufferDesc bufferDesc = {};
20    bufferDesc.size = numberCount * sizeof(float);
21    bufferDesc.format = rhi::Format::Undefined;
22    bufferDesc.elementSize = sizeof(float);
23    bufferDesc.defaultState = ResourceState::UnorderedAccess;
24    bufferDesc.memoryType = MemoryType::DeviceLocal;
25    bufferDesc.usage = BufferUsage::UnorderedAccess | BufferUsage::ShaderResource |
26                       BufferUsage::CopySource | BufferUsage::CopyDestination;
27
28    ComPtr<IBuffer> buffer;
29    GFX_CHECK_CALL_ABORT(device->createBuffer(bufferDesc, nullptr, buffer.writeRef()));
30
31    NativeHandle handle;
32    GFX_CHECK_CALL_ABORT(buffer->getNativeHandle(&handle));
33    if (device->getInfo().deviceType == rhi::DeviceType::Vulkan)
34    {
35        SLANG_CHECK(handle.value != 0);
36    }
37#if SLANG_WINDOWS_FAMILY
38    else
39    {
40        auto d3d12Handle = (ID3D12Resource*)handle.value;
41        Slang::ComPtr<IUnknown> testHandle1;
42        GFX_CHECK_CALL_ABORT(d3d12Handle->QueryInterface<IUnknown>(testHandle1.writeRef()));
43        Slang::ComPtr<ID3D12Resource> testHandle2;
44        GFX_CHECK_CALL_ABORT(testHandle1->QueryInterface<ID3D12Resource>(testHandle2.writeRef()));
45        SLANG_CHECK(d3d12Handle == testHandle2.get());
46    }
47#endif
48}
49
50void getBufferResourceHandleTestAPI(UnitTestContext* context, Slang::RenderApiFlag::Enum api)
51{
52    if (context->enableDebugLayers)
53        getRHI()->enableDebugLayers();
54    if ((api & context->enabledApis) == 0)
55    {
56        SLANG_IGNORE_TEST;
57    }
58    Slang::ComPtr<IDevice> device;
59    DeviceDesc deviceDesc = {};
60    switch (api)
61    {
62    case Slang::RenderApiFlag::D3D11:
63        deviceDesc.deviceType = rhi::DeviceType::D3D11;
64        break;
65    case Slang::RenderApiFlag::D3D12:
66        deviceDesc.deviceType = rhi::DeviceType::D3D12;
67        break;
68    case Slang::RenderApiFlag::Vulkan:
69        deviceDesc.deviceType = rhi::DeviceType::Vulkan;
70        break;
71    default:
72        SLANG_IGNORE_TEST;
73    }
74    deviceDesc.slang.slangGlobalSession = context->slangGlobalSession;
75    const char* searchPaths[] = {"", "../../tools/gfx-unit-test", "tools/gfx-unit-test"};
76    deviceDesc.slang.searchPathCount = (SlangInt)SLANG_COUNT_OF(searchPaths);
77    deviceDesc.slang.searchPaths = searchPaths;
78    auto createDeviceResult = getRHI()->createDevice(deviceDesc, device.writeRef());
79    if (SLANG_FAILED(createDeviceResult))
80    {
81        SLANG_IGNORE_TEST;
82    }
83    // Ignore this test on swiftshader. Swiftshader seems to have a bug that causes the test
84    // to crash.
85    if (Slang::String(device->getInfo().adapterName).toLower().contains("swiftshader"))
86    {
87        SLANG_IGNORE_TEST;
88    }
89
90    getBufferResourceHandleTestImpl(device, context);
91}
92
93SLANG_UNIT_TEST(getBufferResourceHandleD3D12)
94{
95    return getBufferResourceHandleTestAPI(unitTestContext, Slang::RenderApiFlag::D3D12);
96}
97
98SLANG_UNIT_TEST(getBufferResourceHandleVulkan)
99{
100    return getBufferResourceHandleTestAPI(unitTestContext, Slang::RenderApiFlag::Vulkan);
101}
102
103} // namespace gfx_test