init

2025-09-04 10:54:47 +08:00
commit 6bc8f61b18
1808 changed files with 208268 additions and 0 deletions
@@ -0,0 +1,123 @@
+/* Copyright (c) 2021-2025, Arm Limited and Contributors
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Licensed under the Apache License, Version 2.0 the "License";
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#pragma once
+
+#include "rendering/render_pipeline.h"
+#include "rendering/subpasses/forward_subpass.h"
+#include "scene_graph/components/camera.h"
+#include "timer.h"
+#include "vulkan_sample.h"
+
+/**
+ * @brief Using multiple queues to achieve more parallelism on the GPU
+ */
+class AsyncComputeSample : public vkb::VulkanSampleC
+{
+  public:
+	AsyncComputeSample();
+
+	virtual ~AsyncComputeSample() = default;
+
+	virtual void request_gpu_features(vkb::PhysicalDevice &gpu) override;
+
+	virtual bool prepare(const vkb::ApplicationOptions &options) override;
+
+	virtual void update(float delta_time) override;
+
+	virtual void finish() override;
+
+  private:
+	vkb::sg::Camera *camera{nullptr};
+	vkb::sg::Camera *shadow_camera{nullptr};
+
+	virtual void draw_gui() override;
+
+	std::chrono::system_clock::time_point start_time;
+
+	void        render_shadow_pass();
+	VkSemaphore render_forward_offscreen_pass(VkSemaphore hdr_wait_semaphore);
+	VkSemaphore render_compute_post(VkSemaphore wait_graphics_semaphore, VkSemaphore wait_present_semaphore);
+	VkSemaphore render_swapchain(VkSemaphore post_semaphore);
+	void        setup_queues();
+
+	void                                               prepare_render_targets();
+	std::unique_ptr<vkb::RenderTarget>                 forward_render_targets[2];
+	std::unique_ptr<vkb::RenderTarget>                 shadow_render_target;
+	vkb::RenderPipeline                                shadow_render_pipeline;
+	vkb::RenderPipeline                                forward_render_pipeline;
+	std::unique_ptr<vkb::core::Sampler>                comparison_sampler;
+	std::unique_ptr<vkb::core::Sampler>                linear_sampler;
+	std::vector<std::unique_ptr<vkb::core::Image>>     blur_chain;
+	std::vector<std::unique_ptr<vkb::core::ImageView>> blur_chain_views;
+
+	vkb::PipelineLayout *threshold_pipeline{nullptr};
+	vkb::PipelineLayout *blur_down_pipeline{nullptr};
+	vkb::PipelineLayout *blur_up_pipeline{nullptr};
+
+	const vkb::Queue *early_graphics_queue{nullptr};
+	const vkb::Queue *present_graphics_queue{nullptr};
+	const vkb::Queue *post_compute_queue{nullptr};
+
+	VkSemaphore hdr_wait_semaphores[2]{};
+	VkSemaphore compute_post_semaphore{};
+	bool        async_enabled{false};
+	bool        rotate_shadows{true};
+	bool        last_async_enabled{false};
+	bool        double_buffer_hdr_frames{false};
+	unsigned    forward_render_target_index{};
+
+	struct DepthMapSubpass : vkb::ForwardSubpass
+	{
+		DepthMapSubpass(vkb::RenderContext &render_context,
+		                vkb::ShaderSource &&vertex_shader, vkb::ShaderSource &&fragment_shader,
+		                vkb::sg::Scene &scene, vkb::sg::Camera &camera);
+		virtual void draw(vkb::core::CommandBufferC &command_buffer) override;
+	};
+
+	struct ShadowMapForwardSubpass : vkb::ForwardSubpass
+	{
+		ShadowMapForwardSubpass(vkb::RenderContext &render_context,
+		                        vkb::ShaderSource &&vertex_shader, vkb::ShaderSource &&fragment_shader,
+		                        vkb::sg::Scene &scene, vkb::sg::Camera &camera, vkb::sg::Camera &shadow_camera);
+		void         set_shadow_map(const vkb::core::ImageView *view, const vkb::core::Sampler *sampler);
+		virtual void draw(vkb::core::CommandBufferC &command_buffer) override;
+
+		const vkb::core::ImageView *shadow_view{nullptr};
+		const vkb::core::Sampler   *shadow_sampler{nullptr};
+		vkb::sg::Camera            &shadow_camera;
+	};
+
+	struct CompositeSubpass : vkb::rendering::SubpassC
+	{
+		CompositeSubpass(vkb::RenderContext &render_context,
+		                 vkb::ShaderSource &&vertex_shader, vkb::ShaderSource &&fragment_shader);
+		void         set_texture(const vkb::core::ImageView *hdr_view, const vkb::core::ImageView *bloom_view,
+		                         const vkb::core::Sampler *sampler);
+		virtual void draw(vkb::core::CommandBufferC &command_buffer) override;
+		virtual void prepare() override;
+
+		const vkb::core::ImageView *hdr_view{nullptr};
+		const vkb::core::ImageView *bloom_view{nullptr};
+		const vkb::core::Sampler   *sampler{nullptr};
+		vkb::PipelineLayout        *layout{nullptr};
+	};
+
+	vkb::RenderTarget &get_current_forward_render_target();
+};
+
+std::unique_ptr<vkb::VulkanSampleC> create_async_compute();