Adding simple visibility pass

2024-06-04 21:34:14 +02:00
parent 157368a241
commit ad00e16cf9
10 changed files with 160 additions and 25 deletions
@@ -0,0 +1,42 @@
+import Common;
+import Scene;
+
+struct VisibilityCullingData
+{
+    Texture2D<uint> visibilityTexture;
+    RWStructuredBuffer<MeshletCullingInfo> cullingInfos;
+};
+ParameterBlock<VisibilityCullingData> pVisibilityParams;
+
+groupshared MeshletCullingInfo cullInfo;
+
+[numthreads(BLOCK_SIZE, 1, 1)]
+[shader("compute")]
+void computeMain(
+    uint threadID: SV_GroupIndex,
+    uint groupID: SV_GroupID,
+){
+    if (threadID < MAX_PRIMITIVES / 32)
+    {
+        cullInfo.visible[threadID] = 0;
+    }
+    GroupMemoryBarrierWithGroupSync();
+    for (uint y = 0; y < pViewParams.screenDimensions.y; y++)
+    {
+        for (uint x = threadID; x < pViewParams.screenDimensions.x; x += BLOCK_SIZE)
+        {
+            int3 texCoords = int3(x, y, 0);
+            uint encoded = pVisibilityParams.visibilityTexture.Load(texCoords).r;
+            uint2 decoded = decodePrimitive(encoded);
+            uint base = decoded.y == groupID ? 1 : 0;
+            uint arrIdx = decoded.x / 32;
+            uint bit = decoded.x % 32;
+            cullInfo.visible[arrIdx] |= (base << bit);
+        }
+    }
+    GroupMemoryBarrierWithGroupSync();
+    if (threadID < MAX_PRIMITIVES / 32)
+    {
+        pVisibilityParams.cullingInfos[groupID].visible[threadID] = cullInfo.visible[threadID];
+    }
+}