76 lines
1.5 KiB
Plaintext
76 lines
1.5 KiB
Plaintext
import Bounding;
|
|
|
|
struct MeshletDescription
|
|
{
|
|
AABB bounding;
|
|
uint32_t vertexCount;
|
|
uint32_t primitiveCount;
|
|
uint32_t vertexOffset;
|
|
uint32_t primitiveOffset;
|
|
float3 color;
|
|
uint32_t indicesOffset;
|
|
};
|
|
|
|
struct MeshData
|
|
{
|
|
AABB bounding;
|
|
uint32_t numMeshlets;
|
|
uint32_t meshletOffset;
|
|
uint32_t firstIndex;
|
|
uint32_t numIndices;
|
|
};
|
|
|
|
static const uint MAX_VERTICES = 256;
|
|
static const uint MAX_PRIMITIVES = 256;
|
|
static const uint TASK_GROUP_SIZE = 128;
|
|
static const uint MESH_GROUP_SIZE = 32;
|
|
|
|
struct InstanceData
|
|
{
|
|
float4x4 transformMatrix;
|
|
float4x4 inverseTransformMatrix;
|
|
};
|
|
|
|
struct MeshletCullingInfo
|
|
{
|
|
uint64_t cull[MAX_PRIMITIVES / 64];
|
|
// lookup if a specific triangle is visible
|
|
bool triangleVisible(uint32_t primIndex)
|
|
{
|
|
uint32_t arrIdx = primIndex / 64;
|
|
uint32_t cullIdx = primIndex % 64;
|
|
return (cull[arrIdx] & (1 << cullIdx)) != 0;
|
|
}
|
|
bool anyVisible()
|
|
{
|
|
return (cull[0] | cull[1] | cull[2] | cull[3]) != 0;
|
|
}
|
|
};
|
|
|
|
struct DrawCallOffsets
|
|
{
|
|
uint32_t instanceOffset;
|
|
};
|
|
layout(push_constant)
|
|
ConstantBuffer<DrawCallOffsets> pOffsets;
|
|
|
|
struct Scene
|
|
{
|
|
StructuredBuffer<InstanceData> instances;
|
|
StructuredBuffer<MeshData> meshData;
|
|
StructuredBuffer<MeshletDescription> meshletInfos;
|
|
StructuredBuffer<uint8_t> primitiveIndices;
|
|
StructuredBuffer<uint32_t> vertexIndices;
|
|
StructuredBuffer<MeshletCullingInfo> culledMeshlets;
|
|
StructuredBuffer<uint32_t> cullingOffsets;
|
|
};
|
|
layout(set=2)
|
|
ParameterBlock<Scene> pScene;
|
|
|
|
struct MeshPayload
|
|
{
|
|
uint instanceId;
|
|
uint culledMeshlets[2048];
|
|
};
|
|
|