fixed descriptors

2026-01-29 23:31:56 +01:00 · 2026-01-29 23:31:56 +01:00 · 5e3a7738ed
commit 5e3a7738ed
parent 83bb8ebd61
14 changed files with 334 additions and 167 deletions
--- a/examples/VulkanAnimation/main.cpp
+++ b/examples/VulkanAnimation/main.cpp
@ -0,0 +1,128 @@
+#include <vulkan/vulkan.h>
+
+import Crafter.Graphics;
+using namespace Crafter;
+import std;
+import Crafter.Event;
+import Crafter.Math;
+
+typedef VulkanShader<"raygen.spv", "main", VK_SHADER_STAGE_RAYGEN_BIT_KHR, 2, {{{VK_DESCRIPTOR_TYPE_ACCELERATION_STRUCTURE_KHR, 0}, {VK_DESCRIPTOR_TYPE_STORAGE_IMAGE, 1}}}> Raygenspv;
+typedef VulkanShader<"closesthit.spv", "main", VK_SHADER_STAGE_CLOSEST_HIT_BIT_KHR, 0, {{}}> Closesthitspv;
+typedef VulkanShader<"miss.spv", "main", VK_SHADER_STAGE_MISS_BIT_KHR, 0, {{}}> Misspv;
+
+int main() {
+    /*
+        This sets up all necessary things and creates the vulkan device.
+        This must be called before any vulkan related things.
+        Things like VkDevice are static members of the VulkanDevice class.
+    */
+    VulkanDevice::CreateDevice();
+    Raygenspv::CreateShader();
+    Closesthitspv::CreateShader();
+    Misspv::CreateShader();
+    DescriptorLayoutVulkan<Raygenspv, Closesthitspv, Misspv>::Init();
+    PipelineRTVulkan<Raygenspv, Closesthitspv, Misspv, Raygenspv, Closesthitspv, Misspv>::Init();
+    DescriptorPool<1, Raygenspv, Closesthitspv, Misspv> pool;
+    pool.setsCount = 2;
+    pool.BuildPool(0);
+
+    WindowVulkan window(1280, 720, "HelloVulkan");
+
+    /*
+        StartInit gives you a VkCommandBuffer to use before the event loop starts
+        Use this for inititializing things like textures.
+    */
+    VkCommandBuffer cmd = window.StartInit();
+
+    Mesh triangleMesh;
+    std::array<Vertex, 3> verts {{{-150, -150, 100}, {0, 150, 100}, {150, -150, 100}}};
+    std::array<std::uint32_t, 3> index {{2,1,0}};
+    triangleMesh.Build(verts, index, cmd);
+    RenderingElement3DVulkan& el = RenderingElement3DVulkan::elements.emplace_back(triangleMesh);
+    MatrixRowMajor<float, 4, 3, 1> transform = MatrixRowMajor<float, 4, 3, 1>::Identity();
+    std::memcpy(el.instance.transform.matrix, transform.m, sizeof(transform.m));
+    RenderingElement3DVulkan::tlases.resize(2);
+
+    window.SetPipelineRT<Raygenspv, Closesthitspv, Misspv, Raygenspv, Closesthitspv, Misspv>();
+    window.descriptorsRt = pool.sets;
+
+    RenderingElement3DVulkan::BuildTLAS(cmd, 0);
+    VkDescriptorImageInfo imageInfo = {
+        .imageView = window.imageViews[0],
+        .imageLayout = VK_IMAGE_LAYOUT_GENERAL
+    };
+
+    VkWriteDescriptorSetAccelerationStructureKHR writeDescriptorSetAccelerationStructure {
+        .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET_ACCELERATION_STRUCTURE_KHR,
+        .accelerationStructureCount = 1,
+        .pAccelerationStructures = &RenderingElement3DVulkan::tlases[0].accelerationStructure
+    };
+
+    VkWriteDescriptorSet write[2] = {
+        {
+            .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
+            .pNext = &writeDescriptorSetAccelerationStructure,
+            .dstSet = pool.sets[0],
+            .dstBinding = 0,
+            .dstArrayElement = 0,
+            .descriptorCount = 1,
+            .descriptorType = VK_DESCRIPTOR_TYPE_ACCELERATION_STRUCTURE_KHR,
+        },
+        {
+            .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
+            .dstSet = pool.sets[0],
+            .dstBinding = 1,
+            .dstArrayElement = 0,
+            .descriptorCount = 1,
+            .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
+            .pImageInfo = &imageInfo
+        }
+    };
+    vkUpdateDescriptorSets(VulkanDevice::device, 2, write, 0, nullptr);
+
+    /*
+        FinishInit executes all commands recorded to StartInit.
+        This must be called before the the event loops starts if you called StartInit before.
+    */
+    window.FinishInit();
+
+    EventListener<VkCommandBuffer> updateListener(&window.onRender, [&](VkCommandBuffer cmd){
+        RenderingElement3DVulkan::BuildTLAS(cmd, window.currentBuffer);
+        VkDescriptorImageInfo imageInfo = {
+            .imageView = window.imageViews[window.currentBuffer],
+            .imageLayout = VK_IMAGE_LAYOUT_GENERAL
+        };
+
+        VkWriteDescriptorSetAccelerationStructureKHR writeDescriptorSetAccelerationStructure {
+            .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET_ACCELERATION_STRUCTURE_KHR,
+            .accelerationStructureCount = 1,
+            .pAccelerationStructures = &RenderingElement3DVulkan::tlases[window.currentBuffer].accelerationStructure
+        };
+
+        VkWriteDescriptorSet write[2] = {
+            {
+                .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
+                .pNext = &writeDescriptorSetAccelerationStructure,
+                .dstSet = pool.sets[window.currentBuffer],
+                .dstBinding = 0,
+                .dstArrayElement = 0,
+                .descriptorCount = 1,
+                .descriptorType = VK_DESCRIPTOR_TYPE_ACCELERATION_STRUCTURE_KHR,
+            },
+            {
+                .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
+                .dstSet = pool.sets[window.currentBuffer],
+                .dstBinding = 1,
+                .dstArrayElement = 0,
+                .descriptorCount = 1,
+                .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
+                .pImageInfo = &imageInfo
+            }
+        };
+        vkUpdateDescriptorSets(VulkanDevice::device, 2, write, 0, nullptr);
+    });
+    
+    window.Render();
+	window.StartUpdate();
+    window.StartSync();
+}