init

2025-09-04 10:54:47 +08:00
commit 6bc8f61b18
1808 changed files with 208268 additions and 0 deletions
@@ -0,0 +1,120 @@
+/* Copyright (c) 2023-2025, NVIDIA CORPORATION. All rights reserved.
+ *
+ * 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 "core/shader_module.h"
+#include <vulkan/vulkan.hpp>
+
+namespace vkb
+{
+namespace core
+{
+template <vkb::BindingType bindingType>
+class Device;
+using DeviceCpp = Device<vkb::BindingType::Cpp>;
+using DeviceC   = Device<vkb::BindingType::C>;
+
+/**
+ * @brief facade class around vkb::ShaderModule, providing a vulkan.hpp-based interface
+ *
+ * See vkb::ShaderModule for documentation
+ */
+
+/// Types of shader resources
+enum class HPPShaderResourceType
+{
+	Input,
+	InputAttachment,
+	Output,
+	Image,
+	ImageSampler,
+	ImageStorage,
+	Sampler,
+	BufferUniform,
+	BufferStorage,
+	PushConstant,
+	SpecializationConstant,
+	All
+};
+
+/// This determines the type and method of how descriptor set should be created and bound
+enum class HPPShaderResourceMode
+{
+	Static,
+	Dynamic,
+	UpdateAfterBind
+};
+
+/// Store shader resource data.
+/// Used by the shader module.
+struct HPPShaderResource
+{
+	vk::ShaderStageFlags  stages;
+	HPPShaderResourceType type;
+	HPPShaderResourceMode mode;
+	uint32_t              set;
+	uint32_t              binding;
+	uint32_t              location;
+	uint32_t              input_attachment_index;
+	uint32_t              vec_size;
+	uint32_t              columns;
+	uint32_t              array_size;
+	uint32_t              offset;
+	uint32_t              size;
+	uint32_t              constant_id;
+	uint32_t              qualifiers;
+	std::string           name;
+};
+
+class HPPShaderSource : private vkb::ShaderSource
+{
+  public:
+	HPPShaderSource(const std::string &filename) :
+	    vkb::ShaderSource(filename)
+	{}
+};
+
+class HPPShaderVariant : private vkb::ShaderVariant
+{};
+
+class HPPShaderModule : private vkb::ShaderModule
+{
+  public:
+	using vkb::ShaderModule::get_id;
+
+  public:
+	HPPShaderModule(vkb::core::DeviceCpp              &device,
+	                vk::ShaderStageFlagBits            stage,
+	                const vkb::core::HPPShaderSource  &glsl_source,
+	                const std::string                 &entry_point,
+	                const vkb::core::HPPShaderVariant &shader_variant) :
+	    vkb::ShaderModule(reinterpret_cast<vkb::core::DeviceC &>(device),
+	                      static_cast<VkShaderStageFlagBits>(stage),
+	                      reinterpret_cast<vkb::ShaderSource const &>(glsl_source),
+	                      entry_point,
+	                      reinterpret_cast<vkb::ShaderVariant const &>(shader_variant))
+	{}
+
+	const std::vector<vkb::core::HPPShaderResource> &get_resources() const
+	{
+		return reinterpret_cast<std::vector<vkb::core::HPPShaderResource> const &>(vkb::ShaderModule::get_resources());
+	}
+};
+
+}        // namespace core
+}        // namespace vkb