102 lines
3.7 KiB
C++
102 lines
3.7 KiB
C++
#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> Raygenspv;
|
|
typedef VulkanShader<"closesthit.spv", "main", VK_SHADER_STAGE_CLOSEST_HIT_BIT_KHR> Closesthitspv;
|
|
typedef VulkanShader<"miss.spv", "main", VK_SHADER_STAGE_MISS_BIT_KHR> Misspv;
|
|
typedef std::tuple<Raygenspv, Misspv, Closesthitspv> AllShaders;
|
|
typedef std::tuple<
|
|
ShaderGroup<0, VK_SHADER_UNUSED_KHR, VK_SHADER_UNUSED_KHR, VK_SHADER_UNUSED_KHR>,
|
|
ShaderGroup<1, VK_SHADER_UNUSED_KHR, VK_SHADER_UNUSED_KHR, VK_SHADER_UNUSED_KHR>,
|
|
ShaderGroup<VK_SHADER_UNUSED_KHR, 2, VK_SHADER_UNUSED_KHR, VK_SHADER_UNUSED_KHR>
|
|
> ShaderGroups;
|
|
typedef PipelineRTVulkan<AllShaders, ShaderGroups> Pipeline;
|
|
typedef DescriptorSetLayoutVulkan<2, {{
|
|
{
|
|
.binding = 0,
|
|
.descriptorType = VK_DESCRIPTOR_TYPE_ACCELERATION_STRUCTURE_KHR,
|
|
.descriptorCount = 1,
|
|
.stageFlags = VK_SHADER_STAGE_RAYGEN_BIT_KHR,
|
|
},
|
|
{
|
|
.binding = 1,
|
|
.descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
|
|
.descriptorCount = 1,
|
|
.stageFlags = VK_SHADER_STAGE_RAYGEN_BIT_KHR,
|
|
}
|
|
}}> descriptorSetLayout;
|
|
|
|
int main() {
|
|
VulkanDevice::CreateDevice();
|
|
WindowVulkan window(1280, 720, "HelloVulkan");
|
|
VkCommandBuffer cmd = window.StartInit();
|
|
|
|
Raygenspv::CreateShader();
|
|
Closesthitspv::CreateShader();
|
|
Misspv::CreateShader();
|
|
ShaderBindingTableVulkan<AllShaders>::Init();
|
|
|
|
descriptorSetLayout::Init();
|
|
std::array<VkDescriptorSetLayout, 1> layouts {{descriptorSetLayout::layout}};
|
|
|
|
DescriptorPool pool;
|
|
pool.sets.resize(1);
|
|
pool.BuildPool(DescriptorPool::GetPoolSizes<descriptorSetLayout>(), layouts);
|
|
|
|
Pipeline::Init(cmd, layouts);
|
|
|
|
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(1);
|
|
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);
|
|
|
|
window.SetPipelineRT<Pipeline>();
|
|
window.descriptorsRt = pool.sets;
|
|
|
|
window.FinishInit();
|
|
window.Render();
|
|
window.StartSync();
|
|
}
|