blob: ea87e7c68c158443ccb07cb80fe7c0542f95cbda (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
|
// vk-util.cpp
#include "vk-util.h"
#include "core/slang-math.h"
#include <stdlib.h>
#include <stdio.h>
namespace gfx {
/* static */VkFormat VulkanUtil::getVkFormat(Format format)
{
switch (format)
{
case Format::R32G32B32A32_TYPELESS: return VK_FORMAT_R32G32B32A32_SFLOAT;
case Format::R32G32B32_TYPELESS: return VK_FORMAT_R32G32B32_SFLOAT;
case Format::R32G32_TYPELESS: return VK_FORMAT_R32G32_SFLOAT;
case Format::R32_TYPELESS: return VK_FORMAT_R32_SFLOAT;
case Format::R16G16B16A16_TYPELESS: return VK_FORMAT_R16G16B16A16_SFLOAT;
case Format::R16G16_TYPELESS: return VK_FORMAT_R16G16_SFLOAT;
case Format::R16_TYPELESS: return VK_FORMAT_R16_SFLOAT;
case Format::R8G8B8A8_TYPELESS: return VK_FORMAT_R8G8B8A8_UNORM;
case Format::R8G8_TYPELESS: return VK_FORMAT_R8G8_UNORM;
case Format::R8_TYPELESS: return VK_FORMAT_R8_UNORM;
case Format::B8G8R8A8_TYPELESS: return VK_FORMAT_B8G8R8A8_UNORM;
case Format::R32G32B32A32_FLOAT: return VK_FORMAT_R32G32B32A32_SFLOAT;
case Format::R32G32B32_FLOAT: return VK_FORMAT_R32G32B32_SFLOAT;
case Format::R32G32_FLOAT: return VK_FORMAT_R32G32_SFLOAT;
case Format::R32_FLOAT: return VK_FORMAT_R32_SFLOAT;
case Format::R16G16B16A16_FLOAT: return VK_FORMAT_R16G16B16A16_SFLOAT;
case Format::R16G16_FLOAT: return VK_FORMAT_R16G16_SFLOAT;
case Format::R16_FLOAT: return VK_FORMAT_R16_SFLOAT;
case Format::R32G32B32A32_UINT: return VK_FORMAT_R32G32B32A32_UINT;
case Format::R32G32B32_UINT: return VK_FORMAT_R32G32B32_UINT;
case Format::R32G32_UINT: return VK_FORMAT_R32G32_UINT;
case Format::R32_UINT: return VK_FORMAT_R32_UINT;
case Format::R16G16B16A16_UINT: return VK_FORMAT_R16G16B16A16_UINT;
case Format::R16G16_UINT: return VK_FORMAT_R16G16_UINT;
case Format::R16_UINT: return VK_FORMAT_R16_UINT;
case Format::R8G8B8A8_UINT: return VK_FORMAT_R8G8B8A8_UINT;
case Format::R8G8_UINT: return VK_FORMAT_R8G8_UINT;
case Format::R8_UINT: return VK_FORMAT_R8_UINT;
case Format::R32G32B32A32_SINT: return VK_FORMAT_R32G32B32A32_SINT;
case Format::R32G32B32_SINT: return VK_FORMAT_R32G32B32_SINT;
case Format::R32G32_SINT: return VK_FORMAT_R32G32_SINT;
case Format::R32_SINT: return VK_FORMAT_R32_SINT;
case Format::R16G16B16A16_SINT: return VK_FORMAT_R16G16B16A16_SINT;
case Format::R16G16_SINT: return VK_FORMAT_R16G16_SINT;
case Format::R16_SINT: return VK_FORMAT_R16_SINT;
case Format::R8G8B8A8_SINT: return VK_FORMAT_R8G8B8A8_SINT;
case Format::R8G8_SINT: return VK_FORMAT_R8G8_SINT;
case Format::R8_SINT: return VK_FORMAT_R8_SINT;
case Format::R16G16B16A16_UNORM: return VK_FORMAT_R16G16B16A16_UNORM;
case Format::R16G16_UNORM: return VK_FORMAT_R16G16_UNORM;
case Format::R16_UNORM: return VK_FORMAT_R16_UNORM;
case Format::R8G8B8A8_UNORM: return VK_FORMAT_R8G8B8A8_UNORM;
case Format::R8G8B8A8_UNORM_SRGB: return VK_FORMAT_R8G8B8A8_SRGB;
case Format::R8G8_UNORM: return VK_FORMAT_R8G8_UNORM;
case Format::R8_UNORM: return VK_FORMAT_R8_UNORM;
case Format::B8G8R8A8_UNORM: return VK_FORMAT_B8G8R8A8_UNORM;
case Format::B8G8R8A8_UNORM_SRGB: return VK_FORMAT_B8G8R8A8_SRGB;
case Format::B8G8R8X8_UNORM: return VK_FORMAT_B8G8R8A8_UNORM;
case Format::B8G8R8X8_UNORM_SRGB: return VK_FORMAT_B8G8R8A8_SRGB;
case Format::R16G16B16A16_SNORM: return VK_FORMAT_R16G16B16A16_SNORM;
case Format::R16G16_SNORM: return VK_FORMAT_R16G16_SNORM;
case Format::R16_SNORM: return VK_FORMAT_R16_SNORM;
case Format::R8G8B8A8_SNORM: return VK_FORMAT_R8G8B8A8_SNORM;
case Format::R8G8_SNORM: return VK_FORMAT_R8G8_SNORM;
case Format::R8_SNORM: return VK_FORMAT_R8_SNORM;
case Format::D32_FLOAT: return VK_FORMAT_D32_SFLOAT;
case Format::D16_UNORM: return VK_FORMAT_D16_UNORM;
case Format::B4G4R4A4_UNORM: return VK_FORMAT_A4R4G4B4_UNORM_PACK16_EXT;
case Format::B5G6R5_UNORM: return VK_FORMAT_R5G6B5_UNORM_PACK16;
case Format::B5G5R5A1_UNORM: return VK_FORMAT_A1R5G5B5_UNORM_PACK16;
case Format::R9G9B9E5_SHAREDEXP: return VK_FORMAT_E5B9G9R9_UFLOAT_PACK32;
case Format::R10G10B10A2_TYPELESS: return VK_FORMAT_A2B10G10R10_UINT_PACK32;
case Format::R10G10B10A2_UINT: return VK_FORMAT_A2B10G10R10_UINT_PACK32;
case Format::R10G10B10A2_UNORM: return VK_FORMAT_A2B10G10R10_UNORM_PACK32;
case Format::R11G11B10_FLOAT: return VK_FORMAT_B10G11R11_UFLOAT_PACK32;
case Format::BC1_UNORM: return VK_FORMAT_BC1_RGBA_UNORM_BLOCK;
case Format::BC1_UNORM_SRGB: return VK_FORMAT_BC1_RGBA_SRGB_BLOCK;
case Format::BC2_UNORM: return VK_FORMAT_BC2_UNORM_BLOCK;
case Format::BC2_UNORM_SRGB: return VK_FORMAT_BC2_SRGB_BLOCK;
case Format::BC3_UNORM: return VK_FORMAT_BC3_UNORM_BLOCK;
case Format::BC3_UNORM_SRGB: return VK_FORMAT_BC3_SRGB_BLOCK;
case Format::BC4_UNORM: return VK_FORMAT_BC4_UNORM_BLOCK;
case Format::BC4_SNORM: return VK_FORMAT_BC4_SNORM_BLOCK;
case Format::BC5_UNORM: return VK_FORMAT_BC5_UNORM_BLOCK;
case Format::BC5_SNORM: return VK_FORMAT_BC5_SNORM_BLOCK;
case Format::BC6H_UF16: return VK_FORMAT_BC6H_UFLOAT_BLOCK;
case Format::BC6H_SF16: return VK_FORMAT_BC6H_SFLOAT_BLOCK;
case Format::BC7_UNORM: return VK_FORMAT_BC7_UNORM_BLOCK;
case Format::BC7_UNORM_SRGB: return VK_FORMAT_BC7_SRGB_BLOCK;
default: return VK_FORMAT_UNDEFINED;
}
}
/* static */SlangResult VulkanUtil::toSlangResult(VkResult res)
{
return (res == VK_SUCCESS) ? SLANG_OK : SLANG_FAIL;
}
VkShaderStageFlags VulkanUtil::getShaderStage(SlangStage stage)
{
switch (stage)
{
case SLANG_STAGE_ANY_HIT:
return VK_SHADER_STAGE_ANY_HIT_BIT_KHR;
case SLANG_STAGE_CALLABLE:
return VK_SHADER_STAGE_CALLABLE_BIT_KHR;
case SLANG_STAGE_CLOSEST_HIT:
return VK_SHADER_STAGE_CLOSEST_HIT_BIT_KHR;
case SLANG_STAGE_COMPUTE:
return VK_SHADER_STAGE_COMPUTE_BIT;
case SLANG_STAGE_DOMAIN:
return VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT;
case SLANG_STAGE_FRAGMENT:
return VK_SHADER_STAGE_FRAGMENT_BIT;
case SLANG_STAGE_GEOMETRY:
return VK_SHADER_STAGE_GEOMETRY_BIT;
case SLANG_STAGE_HULL:
return VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT;
case SLANG_STAGE_INTERSECTION:
return VK_SHADER_STAGE_INTERSECTION_BIT_KHR;
case SLANG_STAGE_RAY_GENERATION:
return VK_SHADER_STAGE_RAYGEN_BIT_KHR;
case SLANG_STAGE_VERTEX:
return VK_SHADER_STAGE_VERTEX_BIT;
default:
assert(!"unsupported stage.");
return VkShaderStageFlags(-1);
}
}
VkImageLayout VulkanUtil::getImageLayoutFromState(ResourceState state)
{
switch (state)
{
case ResourceState::ShaderResource:
return VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;
case ResourceState::UnorderedAccess:
return VK_IMAGE_LAYOUT_GENERAL;
case ResourceState::Present:
return VK_IMAGE_LAYOUT_PRESENT_SRC_KHR;
case ResourceState::CopySource:
return VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL;
case ResourceState::CopyDestination:
return VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL;
case ResourceState::RenderTarget:
return VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL;
case ResourceState::DepthWrite:
return VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL;
case ResourceState::DepthRead:
return VK_IMAGE_LAYOUT_DEPTH_STENCIL_READ_ONLY_OPTIMAL;
case ResourceState::ResolveSource:
return VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL;
case ResourceState::ResolveDestination:
return VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL;
default:
return VK_IMAGE_LAYOUT_UNDEFINED;
}
return VkImageLayout();
}
/* static */Slang::Result VulkanUtil::handleFail(VkResult res)
{
if (res != VK_SUCCESS)
{
assert(!"Vulkan returned a failure");
}
return toSlangResult(res);
}
/* static */void VulkanUtil::checkFail(VkResult res)
{
assert(res != VK_SUCCESS);
assert(!"Vulkan check failed");
}
/* static */VkPrimitiveTopology VulkanUtil::getVkPrimitiveTopology(PrimitiveTopology topology)
{
switch (topology)
{
case PrimitiveTopology::TriangleList: return VK_PRIMITIVE_TOPOLOGY_TRIANGLE_LIST;
default: break;
}
assert(!"Unknown topology");
return VK_PRIMITIVE_TOPOLOGY_MAX_ENUM;
}
VkImageLayout VulkanUtil::mapResourceStateToLayout(ResourceState state)
{
switch (state)
{
case ResourceState::Undefined:
return VK_IMAGE_LAYOUT_UNDEFINED;
case ResourceState::ShaderResource:
return VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;
case ResourceState::UnorderedAccess:
return VK_IMAGE_LAYOUT_GENERAL;
case ResourceState::RenderTarget:
return VK_IMAGE_LAYOUT_COLOR_ATTACHMENT_OPTIMAL;
case ResourceState::DepthRead:
return VK_IMAGE_LAYOUT_DEPTH_STENCIL_READ_ONLY_OPTIMAL;
case ResourceState::DepthWrite:
return VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL;
case ResourceState::Present:
return VK_IMAGE_LAYOUT_PRESENT_SRC_KHR;
case ResourceState::CopySource:
return VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL;
case ResourceState::CopyDestination:
return VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL;
case ResourceState::ResolveSource:
return VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL;
case ResourceState::ResolveDestination:
return VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL;
default:
return VK_IMAGE_LAYOUT_UNDEFINED;
}
}
Result AccelerationStructureBuildGeometryInfoBuilder::build(
const IAccelerationStructure::BuildInputs& buildInputs,
IDebugCallback* debugCallback)
{
buildInfo.dstAccelerationStructure = VK_NULL_HANDLE;
switch (buildInputs.kind)
{
case IAccelerationStructure::Kind::BottomLevel:
buildInfo.type = VK_ACCELERATION_STRUCTURE_TYPE_BOTTOM_LEVEL_KHR;
break;
case IAccelerationStructure::Kind::TopLevel:
buildInfo.type = VK_ACCELERATION_STRUCTURE_TYPE_TOP_LEVEL_KHR;
break;
default:
debugCallback->handleMessage(
DebugMessageType::Error,
DebugMessageSource::Layer,
"invalid value of IAccelerationStructure::Kind encountered in buildInputs.kind");
return SLANG_E_INVALID_ARG;
}
if (buildInputs.flags & IAccelerationStructure::BuildFlags::Enum::PerformUpdate)
{
buildInfo.mode = VK_BUILD_ACCELERATION_STRUCTURE_MODE_UPDATE_KHR;
}
else
{
buildInfo.mode = VK_BUILD_ACCELERATION_STRUCTURE_MODE_BUILD_KHR;
}
if (buildInputs.flags & IAccelerationStructure::BuildFlags::Enum::AllowCompaction)
{
buildInfo.flags |= VK_BUILD_ACCELERATION_STRUCTURE_ALLOW_COMPACTION_BIT_KHR;
}
if (buildInputs.flags & IAccelerationStructure::BuildFlags::Enum::AllowUpdate)
{
buildInfo.flags |= VK_BUILD_ACCELERATION_STRUCTURE_ALLOW_UPDATE_BIT_KHR;
}
if (buildInputs.flags & IAccelerationStructure::BuildFlags::Enum::MinimizeMemory)
{
buildInfo.flags |= VK_BUILD_ACCELERATION_STRUCTURE_LOW_MEMORY_BIT_KHR;
}
if (buildInputs.flags & IAccelerationStructure::BuildFlags::Enum::PreferFastBuild)
{
buildInfo.flags |= VK_BUILD_ACCELERATION_STRUCTURE_PREFER_FAST_BUILD_BIT_KHR;
}
if (buildInputs.flags & IAccelerationStructure::BuildFlags::Enum::PreferFastTrace)
{
buildInfo.flags |= VK_BUILD_ACCELERATION_STRUCTURE_PREFER_FAST_TRACE_BIT_KHR;
}
if (buildInputs.kind == IAccelerationStructure::Kind::BottomLevel)
{
m_geometryInfos.setCount(buildInputs.descCount);
primitiveCounts.setCount(buildInputs.descCount);
memset(
m_geometryInfos.getBuffer(),
0,
sizeof(VkAccelerationStructureGeometryKHR) * buildInputs.descCount);
for (int i = 0; i < buildInputs.descCount; i++)
{
auto& geomDesc = buildInputs.geometryDescs[i];
m_geometryInfos[i].sType = VK_STRUCTURE_TYPE_ACCELERATION_STRUCTURE_GEOMETRY_KHR;
if (geomDesc.flags & IAccelerationStructure::GeometryFlags::NoDuplicateAnyHitInvocation)
{
m_geometryInfos[i].flags |= VK_GEOMETRY_NO_DUPLICATE_ANY_HIT_INVOCATION_BIT_KHR;
}
else if (geomDesc.flags & IAccelerationStructure::GeometryFlags::Opaque)
{
m_geometryInfos[i].flags |= VK_GEOMETRY_OPAQUE_BIT_KHR;
}
auto& vkGeomData = m_geometryInfos[i].geometry;
switch (geomDesc.type)
{
case IAccelerationStructure::GeometryType::Triangles:
m_geometryInfos[i].geometryType = VK_GEOMETRY_TYPE_TRIANGLES_KHR;
vkGeomData.triangles.sType =
VK_STRUCTURE_TYPE_ACCELERATION_STRUCTURE_GEOMETRY_TRIANGLES_DATA_KHR;
vkGeomData.triangles.vertexFormat =
VulkanUtil::getVkFormat(geomDesc.content.triangles.vertexFormat);
vkGeomData.triangles.vertexData.deviceAddress =
geomDesc.content.triangles.vertexData;
vkGeomData.triangles.vertexStride = geomDesc.content.triangles.vertexStride;
vkGeomData.triangles.maxVertex = geomDesc.content.triangles.vertexCount - 1;
switch (geomDesc.content.triangles.indexFormat)
{
case Format::R32_UINT:
vkGeomData.triangles.indexType = VK_INDEX_TYPE_UINT32;
break;
case Format::R16_UINT:
vkGeomData.triangles.indexType = VK_INDEX_TYPE_UINT16;
break;
case Format::Unknown:
vkGeomData.triangles.indexType = VK_INDEX_TYPE_NONE_KHR;
break;
default:
debugCallback->handleMessage(
DebugMessageType::Error,
DebugMessageSource::Layer,
"unsupported value of Format encountered in "
"GeometryDesc::content.triangles.indexFormat");
return SLANG_E_INVALID_ARG;
}
vkGeomData.triangles.indexData.deviceAddress = geomDesc.content.triangles.indexData;
vkGeomData.triangles.transformData.deviceAddress =
geomDesc.content.triangles.transform3x4;
primitiveCounts[i] = Slang::Math::Max(
geomDesc.content.triangles.vertexCount,
geomDesc.content.triangles.indexCount) /
3;
break;
case IAccelerationStructure::GeometryType::ProcedurePrimitives:
m_geometryInfos[i].geometryType = VK_GEOMETRY_TYPE_AABBS_KHR;
vkGeomData.aabbs.sType =
VK_STRUCTURE_TYPE_ACCELERATION_STRUCTURE_GEOMETRY_AABBS_DATA_KHR;
vkGeomData.aabbs.data.deviceAddress = geomDesc.content.proceduralAABBs.data;
vkGeomData.aabbs.stride = geomDesc.content.proceduralAABBs.stride;
primitiveCounts[i] =
(uint32_t)buildInputs.geometryDescs[i].content.proceduralAABBs.count;
break;
default:
debugCallback->handleMessage(
DebugMessageType::Error,
DebugMessageSource::Layer,
"invalid value of IAccelerationStructure::GeometryType encountered in "
"buildInputs.geometryDescs");
return SLANG_E_INVALID_ARG;
}
}
buildInfo.geometryCount = buildInputs.descCount;
buildInfo.pGeometries = m_geometryInfos.getBuffer();
}
else
{
m_vkInstanceInfo.geometryType = VK_GEOMETRY_TYPE_INSTANCES_KHR;
m_vkInstanceInfo.geometry.instances.sType =
VK_STRUCTURE_TYPE_ACCELERATION_STRUCTURE_GEOMETRY_INSTANCES_DATA_KHR;
m_vkInstanceInfo.geometry.instances.arrayOfPointers = 0;
m_vkInstanceInfo.geometry.instances.data.deviceAddress = buildInputs.instanceDescs;
buildInfo.pGeometries = &m_vkInstanceInfo;
buildInfo.geometryCount = 1;
primitiveCounts.setCount(1);
primitiveCounts[0] = buildInputs.descCount;
}
return SLANG_OK;
}
} // namespace gfx
|
