Create Pipeline struct

Signed-off-by: Slendi <slendi@socopon.com>
2025-12-14 03:49:51 +02:00 · 2025-12-11 23:04:09 +02:00
parent 054a196b13
commit c7283c8faf
5 changed files with 188 additions and 87 deletions
--- a/meson.build
+++ b/meson.build
@@ -120,6 +120,7 @@ exe = executable('vr-compositor',
 		'src/DescriptorAllocator.cpp',
 		'src/GraphicsPipelineBuilder.cpp',
 		'src/DescriptorAllocatorGrowable.cpp',
+		'src/Pipeline.cpp',
 		'src/Loader.cpp',
 		'src/DescriptorWriter.cpp',
 		'src/VulkanRenderer.cpp',
--- a/src/Pipeline.cpp
+++ b/src/Pipeline.cpp
@@ -0,0 +1,79 @@
+#include "Pipeline.h"
+
+#include "Util.h"
+
+namespace Lunar {
+
+Pipeline::Builder::Builder(vk::Device device, Logger &logger)
+    : m_device(device)
+    , m_logger(logger)
+{
+}
+
+auto Pipeline::Builder::set_descriptor_set_layouts(
+    std::span<vk::DescriptorSetLayout const> layouts) -> Builder &
+{
+	m_set_layouts.assign(layouts.begin(), layouts.end());
+	return *this;
+}
+
+auto Pipeline::Builder::set_push_constant_ranges(
+    std::span<vk::PushConstantRange const> ranges) -> Builder &
+{
+	m_push_constant_ranges.assign(ranges.begin(), ranges.end());
+	return *this;
+}
+
+auto Pipeline::Builder::build_compute(
+    vk::PipelineShaderStageCreateInfo const &stage) -> Pipeline
+{
+	auto pipeline_layout = build_layout();
+
+	vk::ComputePipelineCreateInfo pipeline_ci {};
+	pipeline_ci.layout = pipeline_layout.get();
+	pipeline_ci.stage = stage;
+
+	auto pipeline_ret = m_device.createComputePipelineUnique({}, pipeline_ci);
+	VK_CHECK(m_logger, pipeline_ret.result);
+
+	return Pipeline {
+		.pipeline = std::move(pipeline_ret.value),
+		.layout = std::move(pipeline_layout),
+	};
+}
+
+auto Pipeline::Builder::build_graphics(
+    std::function<GraphicsPipelineBuilder &(GraphicsPipelineBuilder &)> const
+        &configure) -> Pipeline
+{
+	auto pipeline_layout = build_layout();
+
+	auto builder = GraphicsPipelineBuilder { m_logger };
+	builder.set_pipeline_layout(
+	    static_cast<VkPipelineLayout>(pipeline_layout.get()));
+	configure(builder);
+
+	auto pipeline_handle = builder.build(static_cast<VkDevice>(m_device));
+	vk::UniquePipeline pipeline_unique(pipeline_handle,
+	    vk::detail::ObjectDestroy<vk::Device,
+	        VULKAN_HPP_DEFAULT_DISPATCHER_TYPE>(m_device));
+
+	return Pipeline {
+		.pipeline = std::move(pipeline_unique),
+		.layout = std::move(pipeline_layout),
+	};
+}
+
+auto Pipeline::Builder::build_layout() -> vk::UniquePipelineLayout
+{
+	vk::PipelineLayoutCreateInfo layout_ci {};
+	layout_ci.setLayoutCount = static_cast<uint32_t>(m_set_layouts.size());
+	layout_ci.pSetLayouts = m_set_layouts.data();
+	layout_ci.pushConstantRangeCount
+	    = static_cast<uint32_t>(m_push_constant_ranges.size());
+	layout_ci.pPushConstantRanges = m_push_constant_ranges.data();
+
+	return m_device.createPipelineLayoutUnique(layout_ci);
+}
+
+} // namespace Lunar
--- a/src/Pipeline.h
+++ b/src/Pipeline.h
@@ -0,0 +1,49 @@
+#pragma once
+
+#include <functional>
+#include <span>
+#include <vector>
+
+#include <vulkan/vulkan.hpp>
+
+#include "GraphicsPipelineBuilder.h"
+#include "Logger.h"
+
+namespace Lunar {
+
+struct Pipeline {
+	vk::UniquePipeline pipeline {};
+	vk::UniquePipelineLayout layout {};
+
+	auto get() const -> vk::Pipeline { return pipeline.get(); }
+	auto get_layout() const -> vk::PipelineLayout { return layout.get(); }
+	auto reset() -> void
+	{
+		pipeline.reset();
+		layout.reset();
+	}
+
+	struct Builder {
+		Builder(vk::Device device, Logger &logger);
+
+		auto set_descriptor_set_layouts(
+		    std::span<vk::DescriptorSetLayout const> layouts) -> Builder &;
+		auto set_push_constant_ranges(
+		    std::span<vk::PushConstantRange const> ranges) -> Builder &;
+
+		auto build_compute(vk::PipelineShaderStageCreateInfo const &stage)
+		    -> Pipeline;
+		auto build_graphics(std::function<GraphicsPipelineBuilder &(
+		        GraphicsPipelineBuilder &)> const &configure) -> Pipeline;
+
+	private:
+		auto build_layout() -> vk::UniquePipelineLayout;
+
+		vk::Device m_device {};
+		Logger &m_logger;
+		std::vector<vk::DescriptorSetLayout> m_set_layouts {};
+		std::vector<vk::PushConstantRange> m_push_constant_ranges {};
+	};
+};
+
+} // namespace Lunar
--- a/src/VulkanRenderer.cpp
+++ b/src/VulkanRenderer.cpp
@@ -59,17 +59,8 @@ VulkanRenderer::~VulkanRenderer()
 	m_vk.imm_command_pool.reset();
 	m_vk.imm_fence.reset();
 	m_vk.gradient_pipeline.reset();
-	m_vk.gradient_pipeline_layout.reset();
-	if (m_vk.triangle_pipeline) {
-		m_device.destroyPipeline(m_vk.triangle_pipeline);
-		m_vk.triangle_pipeline = vk::Pipeline {};
-	}
-	m_vk.triangle_pipeline_layout.reset();
-	if (m_vk.mesh_pipeline) {
-		m_device.destroyPipeline(m_vk.mesh_pipeline);
-		m_vk.mesh_pipeline = vk::Pipeline {};
-	}
-	m_vk.mesh_pipeline_layout.reset();
+	m_vk.triangle_pipeline.reset();
+	m_vk.mesh_pipeline.reset();
 	m_vk.default_sampler_linear.reset();
 	m_vk.default_sampler_nearest.reset();

@@ -375,13 +366,9 @@ auto VulkanRenderer::pipelines_init() -> void

 auto VulkanRenderer::background_pipelines_init() -> void
 {
-	vk::PipelineLayoutCreateInfo layout_ci {};
-	auto draw_layout_handle = m_vk.draw_image_descriptor_layout;
-	layout_ci.pSetLayouts = &draw_layout_handle;
-	layout_ci.setLayoutCount = 1;
-
-	m_vk.gradient_pipeline_layout
-	    = m_device.createPipelineLayoutUnique(layout_ci);
+	Pipeline::Builder builder { m_device, m_logger };
+	std::array layout_handles { m_vk.draw_image_descriptor_layout };
+	builder.set_descriptor_set_layouts(layout_handles);

 	uint8_t compute_draw_shader_data[] {
 #embed "gradient_comp.spv"
@@ -397,18 +384,13 @@ auto VulkanRenderer::background_pipelines_init() -> void
 	auto stage_ci { vkinit::pipeline_shader_stage(
 		vk::ShaderStageFlagBits::eCompute, compute_draw_shader.get()) };

-	vk::ComputePipelineCreateInfo compute_pip_ci {};
-	compute_pip_ci.layout = m_vk.gradient_pipeline_layout.get();
-	compute_pip_ci.stage = stage_ci;
-
-	auto pipeline_ret
-	    = m_device.createComputePipelineUnique({}, compute_pip_ci);
-	VK_CHECK(m_logger, pipeline_ret.result);
-	m_vk.gradient_pipeline = std::move(pipeline_ret.value);
+	m_vk.gradient_pipeline = builder.build_compute(stage_ci);
 }

 auto VulkanRenderer::triangle_pipeline_init() -> void
 {
+	Pipeline::Builder builder { m_device, m_logger };
+
 	uint8_t triangle_vert_shader_data[] {
 #embed "triangle_vert.spv"
 	};
@@ -431,13 +413,10 @@ auto VulkanRenderer::triangle_pipeline_init() -> void
 		m_logger.err("Failed to load triangle frag shader");
 	}

-	vk::PipelineLayoutCreateInfo layout_ci {};
-	m_vk.triangle_pipeline_layout
-	    = m_device.createPipelineLayoutUnique(layout_ci);
-
-	auto pip
-	    = GraphicsPipelineBuilder { m_logger }
-	          .set_pipeline_layout(m_vk.triangle_pipeline_layout.get())
+	m_vk.triangle_pipeline
+	    = builder.build_graphics([&](GraphicsPipelineBuilder &pipeline_builder)
+	                                 -> GraphicsPipelineBuilder & {
+		      return pipeline_builder
 		          .set_shaders(
 		              triangle_vert_shader.get(), triangle_frag_shader.get())
 		          .set_input_topology(VK_PRIMITIVE_TOPOLOGY_TRIANGLE_LIST)
@@ -447,15 +426,15 @@ auto VulkanRenderer::triangle_pipeline_init() -> void
 		          .disable_depth_testing()
 		          .set_color_attachment_format(
 		              static_cast<VkFormat>(m_vk.draw_image.format))
-	          .set_depth_format(static_cast<VkFormat>(m_vk.depth_image.format))
-	          .build(m_vkb.dev.device);
-	m_vk.triangle_pipeline = vk::Pipeline { pip };
-	m_vk.deletion_queue.emplace(
-	    [this]() { m_device.destroyPipeline(m_vk.triangle_pipeline); });
+		          .set_depth_format(
+		              static_cast<VkFormat>(m_vk.depth_image.format));
+	      });
 }

 auto VulkanRenderer::mesh_pipeline_init() -> void
 {
+	Pipeline::Builder builder { m_device, m_logger };
+
 	uint8_t triangle_vert_shader_data[] {
 #embed "triangle_mesh_vert.spv"
 	};
@@ -483,18 +462,15 @@ auto VulkanRenderer::mesh_pipeline_init() -> void
 	push_constant_range.offset = 0;
 	push_constant_range.size = sizeof(GPUDrawPushConstants);

-	vk::PipelineLayoutCreateInfo layout_ci {};
-	layout_ci.pushConstantRangeCount = 1;
-	layout_ci.pPushConstantRanges = &push_constant_range;
-	auto single_layout = m_vk.single_image_descriptor_layout;
-	layout_ci.setLayoutCount = 1;
-	layout_ci.pSetLayouts = &single_layout;
+	std::array push_constant_ranges { push_constant_range };
+	builder.set_push_constant_ranges(push_constant_ranges);
+	std::array descriptor_set_layouts { m_vk.single_image_descriptor_layout };
+	builder.set_descriptor_set_layouts(descriptor_set_layouts);

-	m_vk.mesh_pipeline_layout = m_device.createPipelineLayoutUnique(layout_ci);
-
-	auto pip
-	    = GraphicsPipelineBuilder { m_logger }
-	          .set_pipeline_layout(m_vk.mesh_pipeline_layout.get())
+	m_vk.mesh_pipeline
+	    = builder.build_graphics([&](GraphicsPipelineBuilder &pipeline_builder)
+	                                 -> GraphicsPipelineBuilder & {
+		      return pipeline_builder
 		          .set_shaders(
 		              triangle_vert_shader.get(), triangle_frag_shader.get())
 		          .set_input_topology(VK_PRIMITIVE_TOPOLOGY_TRIANGLE_LIST)
@@ -505,11 +481,9 @@ auto VulkanRenderer::mesh_pipeline_init() -> void
 		          .enable_depth_testing()
 		          .set_color_attachment_format(
 		              static_cast<VkFormat>(m_vk.draw_image.format))
-	          .set_depth_format(static_cast<VkFormat>(m_vk.depth_image.format))
-	          .build(m_vkb.dev.device);
-	m_vk.mesh_pipeline = vk::Pipeline { pip };
-	m_vk.deletion_queue.emplace(
-	    [this]() { m_device.destroyPipeline(m_vk.mesh_pipeline); });
+		          .set_depth_format(
+		              static_cast<VkFormat>(m_vk.depth_image.format));
+	      });
 }

 auto VulkanRenderer::imgui_init() -> void
@@ -787,7 +761,7 @@ auto VulkanRenderer::draw_background(vk::CommandBuffer cmd) -> void
 	    vk::PipelineBindPoint::eCompute, m_vk.gradient_pipeline.get());
 	auto compute_set = vk::DescriptorSet { m_vk.draw_image_descriptors };
 	cmd.bindDescriptorSets(vk::PipelineBindPoint::eCompute,
-	    m_vk.gradient_pipeline_layout.get(), 0, compute_set, {});
+	    m_vk.gradient_pipeline.get_layout(), 0, compute_set, {});
 	cmd.dispatch(
 	    static_cast<uint32_t>(std::ceil(m_vk.draw_extent.width / 16.0)),
 	    static_cast<uint32_t>(std::ceil(m_vk.draw_extent.height / 16.0)), 1);
@@ -838,7 +812,8 @@ auto VulkanRenderer::draw_geometry(vk::CommandBuffer cmd) -> void

 	cmd.beginRendering(render_info);

-	cmd.bindPipeline(vk::PipelineBindPoint::eGraphics, m_vk.triangle_pipeline);
+	cmd.bindPipeline(
+	    vk::PipelineBindPoint::eGraphics, m_vk.triangle_pipeline.get());

 	vk::Viewport viewport {};
 	viewport.x = 0;
@@ -855,7 +830,8 @@ auto VulkanRenderer::draw_geometry(vk::CommandBuffer cmd) -> void
 	scissor.extent = m_vk.draw_extent;
 	cmd.setScissor(0, scissor);

-	cmd.bindPipeline(vk::PipelineBindPoint::eGraphics, m_vk.mesh_pipeline);
+	cmd.bindPipeline(
+	    vk::PipelineBindPoint::eGraphics, m_vk.mesh_pipeline.get());

 	auto const image_set { m_vk.get_current_frame().frame_descriptors.allocate(
 		m_logger, m_vkb.dev.device, m_vk.single_image_descriptor_layout) };
@@ -868,7 +844,7 @@ auto VulkanRenderer::draw_geometry(vk::CommandBuffer cmd) -> void

 	auto vk_image_set = vk::DescriptorSet { image_set };
 	cmd.bindDescriptorSets(vk::PipelineBindPoint::eGraphics,
-	    m_vk.mesh_pipeline_layout.get(), 0, vk_image_set, {});
+	    m_vk.mesh_pipeline.get_layout(), 0, vk_image_set, {});

 	auto view { smath::matrix_look_at(smath::Vec3 { 0.0f, 0.0f, 3.0f },
 		smath::Vec3 { 0.0f, 0.0f, 0.0f }, smath::Vec3 { 0.0f, 1.0f, 0.0f },
@@ -889,7 +865,7 @@ auto VulkanRenderer::draw_geometry(vk::CommandBuffer cmd) -> void
 	push_constants.world_matrix = view_projection * rect_model;
 	push_constants.vertex_buffer = m_vk.rectangle.vertex_buffer_address;

-	cmd.pushConstants(m_vk.mesh_pipeline_layout.get(),
+	cmd.pushConstants(m_vk.mesh_pipeline.get_layout(),
 	    vk::ShaderStageFlagBits::eVertex, 0, sizeof(push_constants),
 	    &push_constants);
 	cmd.bindIndexBuffer(
@@ -903,7 +879,7 @@ auto VulkanRenderer::draw_geometry(vk::CommandBuffer cmd) -> void
 	auto model { smath::Mat4::identity() };
 	push_constants.world_matrix = view_projection * model;

-	cmd.pushConstants(m_vk.mesh_pipeline_layout.get(),
+	cmd.pushConstants(m_vk.mesh_pipeline.get_layout(),
 	    vk::ShaderStageFlagBits::eVertex, 0, sizeof(push_constants),
 	    &push_constants);
 	cmd.bindIndexBuffer(m_vk.test_meshes[2]->mesh_buffers.index_buffer.buffer,
--- a/src/VulkanRenderer.h
+++ b/src/VulkanRenderer.h
@@ -13,6 +13,7 @@
 #include "DescriptorAllocator.h"
 #include "Loader.h"
 #include "Logger.h"
+#include "Pipeline.h"
 #include "Types.h"

 namespace Lunar {
@@ -119,14 +120,9 @@ private:

 		vk::DescriptorSetLayout single_image_descriptor_layout {};

-		vk::UniquePipeline gradient_pipeline;
-		vk::UniquePipelineLayout gradient_pipeline_layout;
-
-		vk::Pipeline triangle_pipeline {};
-		vk::UniquePipelineLayout triangle_pipeline_layout;
-
-		vk::Pipeline mesh_pipeline {};
-		vk::UniquePipelineLayout mesh_pipeline_layout;
+		Pipeline gradient_pipeline;
+		Pipeline triangle_pipeline;
+		Pipeline mesh_pipeline;

 		GPUMeshBuffers rectangle;