Add renderheadless rust shader

Author: LegNeato

examples/renderheadless/renderheadless.cpp

@@ -99,6 +99,17 @@ class VulkanExample
 
 	std::string shaderDir = "glsl";
 
+	const char* getShaderEntryPoint(VkShaderStageFlagBits stage) const {
+		if (shaderDir == "rust") {
+			switch (stage) {
+				case VK_SHADER_STAGE_VERTEX_BIT: return "main_vs";
+				case VK_SHADER_STAGE_FRAGMENT_BIT: return "main_fs";
+				default: return "main";
+			}
+		}
+		return "main";
+	}
+
 	uint32_t getMemoryTypeIndex(uint32_t typeBits, VkMemoryPropertyFlags properties) {
 		VkPhysicalDeviceMemoryProperties deviceMemoryProperties;
 		vkGetPhysicalDeviceMemoryProperties(physicalDevice, &deviceMemoryProperties);
@@ -667,10 +678,10 @@ class VulkanExample
 
 			shaderStages[0].sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO;
 			shaderStages[0].stage = VK_SHADER_STAGE_VERTEX_BIT;
-			shaderStages[0].pName = "main";
+			shaderStages[0].pName = getShaderEntryPoint(VK_SHADER_STAGE_VERTEX_BIT);
 			shaderStages[1].sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO;
 			shaderStages[1].stage = VK_SHADER_STAGE_FRAGMENT_BIT;
-			shaderStages[1].pName = "main";
+			shaderStages[1].pName = getShaderEntryPoint(VK_SHADER_STAGE_FRAGMENT_BIT);
 #if defined(VK_USE_PLATFORM_ANDROID_KHR)
 			shaderStages[0].module = vks::tools::loadShader(androidapp->activity->assetManager, (shadersPath + "triangle.vert.spv").c_str(), device);
 			shaderStages[1].module = vks::tools::loadShader(androidapp->activity->assetManager, (shadersPath + "triangle.frag.spv").c_str(), device);

shaders/rust/Cargo.toml

@@ -64,6 +64,7 @@ members = [
     "texture3d",
     "texturecubemap/reflect",
     "texturecubemap/skybox",
+    "renderheadless/triangle",
 ]
 
 [workspace.dependencies]

shaders/rust/compileshaders.py

@@ -68,7 +68,7 @@ def compile_shader(shader_dir):
                 print(f"  Created {final_path.name} (entry point: {entry_point})")
 
                 # Special case: move to parent directory for nested shader structures
-                if shader_dir.parent.name in ["base", "descriptorsets", "dynamicuniformbuffer", "multisampling", "pipelines", "specializationconstants", "computeshader", "texturearray", "screenshot", "negativeviewportheight", "stencilbuffer", "parallaxmapping", "computecloth", "ssao", "shadowmapping", "deferred", "computenbody", "bloom", "hdr", "radialblur", "pbribl", "indirectdraw", "instancing", "texturecubemap"]:
+                if shader_dir.parent.name in ["base", "descriptorsets", "dynamicuniformbuffer", "multisampling", "pipelines", "specializationconstants", "computeshader", "texturearray", "screenshot", "negativeviewportheight", "stencilbuffer", "parallaxmapping", "computecloth", "ssao", "shadowmapping", "deferred", "computenbody", "bloom", "hdr", "radialblur", "pbribl", "indirectdraw", "instancing", "texturecubemap", "renderheadless"]:
                     parent_dir = shader_dir.parent
                     parent_final_path = parent_dir / f"{shader_name}.{shader_type}.spv"
                     shutil.move(str(final_path), str(parent_final_path))

shaders/rust/renderheadless/triangle.frag.spv

@@ -0,0 +1,10 @@
+[package]
+name = "renderheadless-triangle"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+crate-type = ["dylib"]
+
+[dependencies]
+spirv-std = { workspace = true }

shaders/rust/renderheadless/triangle.vert.spv

@@ -0,0 +1,30 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{vec4, Mat4, Vec3, Vec4}};
+
+#[repr(C)]
+#[derive(Copy, Clone)]
+pub struct PushConsts {
+    pub mvp: Mat4,
+}
+
+#[spirv(vertex)]
+pub fn main_vs(
+    in_pos: Vec3,
+    in_color: Vec3,
+    #[spirv(push_constant)] push_consts: &PushConsts,
+    #[spirv(position)] out_position: &mut Vec4,
+    out_color: &mut Vec3,
+) {
+    *out_color = in_color;
+    *out_position = push_consts.mvp * vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+}
+
+#[spirv(fragment)]
+pub fn main_fs(
+    in_color: Vec3,
+    out_frag_color: &mut Vec4,
+) {
+    *out_frag_color = vec4(in_color.x, in_color.y, in_color.z, 1.0);
+}