Adding transparent geometry to ray tracing

src/Engine/Graphics/RenderPass/RayTracingPass.cpp

@@ -75,6 +75,24 @@ void RayTracingPass::render() {
                 }
             }
         }
+        for (const auto& transparentData : vertexData->getTransparentData()) {
+            PMaterial mat = transparentData.matInst->getBaseMaterial();
+
+            Gfx::ShaderPermutation permutation = graphics->getShaderCompiler()->getTemplate("RayTracing");
+            permutation.setMaterial(mat->getName());
+            permutation.setVertexData(vertexData->getTypeName());
+            const Gfx::ShaderCollection* collection = graphics->getShaderCompiler()->findShaders(Gfx::PermutationId(permutation));
+            assert(collection != nullptr);
+            Gfx::RayTracingHitGroup callableGroup = {
+                .closestHitShader = collection->callableShader,
+            };
+            callableGroup.parameters.resize(sizeof(VertexData::DrawCallOffsets));
+            std::memcpy(callableGroup.parameters.data(), &transparentData.offsets, sizeof(VertexData::DrawCallOffsets));
+            callableGroups.add(callableGroup);
+
+            instanceData.add(transparentData.instanceData);
+            accelerationStructures.add(transparentData.rayTracingScene);
+        }
     }
     pipeline = graphics->createRayTracingPipeline(Gfx::RayTracingPipelineCreateInfo{
         .pipelineLayout = pipelineLayout, .rayGenGroup = {.shader = rayGen}, .hitGroups = callableGroups, .missGroups = {{.shader = miss}},

src/Engine/Graphics/VertexData.cpp

@@ -20,6 +20,7 @@ void VertexData::resetMeshData() {
     instanceData.clear();
     instanceMeshData.clear();
     rayTracingScene.clear();
+    transparentData.clear();
     for (auto& mat : materialData) {
         for (auto& inst : mat.instances) {
             inst.instanceData.clear();
@@ -47,6 +48,8 @@ void VertexData::updateMesh(entt::entity id, uint32 meshIndex, PMesh mesh, Compo
         .inverseTransformMatrix = glm::inverse(transformMatrix),
     };
 
+    auto [instanceId, meshletOffset] = getCullingMapping(id, meshIndex, data.numMeshlets);
+
     referencedInstance->updateDescriptor();
     if (mat->hasTransparency()) {
         auto params = referencedInstance->getMaterialOffsets();
@@ -55,15 +58,16 @@ void VertexData::updateMesh(entt::entity id, uint32 meshIndex, PMesh mesh, Compo
             .vertexData = this,
             .offsets =
                 {
-                    .instanceOffset = static_cast<uint32>(instanceData.size()),
+                    .instanceOffset = 0,
                     .textureOffset = params.textureOffset,
                     .samplerOffset = params.samplerOffset,
                     .floatOffset = params.floatOffset,
                 },
             .worldPosition = Vector(inst.transformMatrix[3]),
+            .instanceData = inst,
+            .meshData = data,
+            .rayTracingScene = mesh->blas,
         });
-        instanceData.add(inst);
-        instanceMeshData.add(data);
         return;
     }
     if (materialData.size() <= mat->getId()) {
@@ -76,9 +80,6 @@ void VertexData::updateMesh(entt::entity id, uint32 meshIndex, PMesh mesh, Compo
     }
     BatchedDrawCall& matInstanceData = matData.instances[referencedInstance->getId()];
     matInstanceData.materialInstance = referencedInstance;
-    
-    auto [instanceId, meshletOffset] = getCullingMapping(id, meshIndex, data.numMeshlets);
-
     matInstanceData.rayTracingData.add(mesh->blas);
     matInstanceData.instanceData.add(inst);
     matInstanceData.instanceMeshData.add(data);
@@ -146,6 +147,13 @@ void VertexData::createDescriptors() {
             }
         }
     }
+    for (uint32 i = 0; i < transparentData.size(); ++i)
+    {
+        transparentData[i].offsets.instanceOffset = instanceData.size();
+        instanceData.add(transparentData[i].instanceData);
+        instanceMeshData.add(transparentData[i].meshData);
+        rayTracingScene.add(transparentData[i].rayTracingScene);
+    }
     cullingOffsetBuffer->rotateBuffer(cullingOffsets.size() * sizeof(uint32));
     cullingOffsetBuffer->updateContents(ShaderBufferCreateInfo{
         .sourceData =

src/Engine/Graphics/VertexData.h

@@ -49,6 +49,9 @@ class VertexData {
         VertexData* vertexData;
         DrawCallOffsets offsets;
         Vector worldPosition;
+        InstanceData instanceData;
+        MeshData meshData;
+        Gfx::PBottomLevelAS rayTracingScene;
     };
     void resetMeshData();
     void updateMesh(entt::entity id, uint32 meshIndex, PMesh mesh, Component::Transform& transform);

src/Engine/Graphics/Vulkan/RayTracing.cpp

@@ -151,7 +151,7 @@ TopLevelAS::TopLevelAS(PGraphics graphics, const Gfx::TopLevelASCreateInfo& crea
                              VmaAllocationCreateInfo{
                                  .usage = VMA_MEMORY_USAGE_AUTO,
                              },
-                             Gfx::QueueType::GRAPHICS);
+                             Gfx::QueueType::GRAPHICS, graphics->getAccelerationProperties().minAccelerationStructureScratchOffsetAlignment);
 
     VkAccelerationStructureBuildGeometryInfoKHR buildGeometry = {
         .sType = VK_STRUCTURE_TYPE_ACCELERATION_STRUCTURE_BUILD_GEOMETRY_INFO_KHR,