Add texturecubemap rust shaders

Author: LegNeato

shaders/rust/Cargo.lock

@@ -62,6 +62,8 @@ members = [
     "instancing/planet",
     "instancing/starfield",
     "texture3d",
+    "texturecubemap/reflect",
+    "texturecubemap/skybox",
 ]
 
 [workspace.dependencies]

shaders/rust/Cargo.toml

@@ -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"]:
+                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"]:
                     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/compileshaders.py

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

shaders/rust/texturecubemap/reflect.frag.spv

@@ -0,0 +1,67 @@
+#![no_std]
+
+use spirv_std::spirv;
+use spirv_std::{glam::{vec3, vec4, Mat3, Mat4, Vec3, Vec4}, image::{Cubemap, SampledImage}};
+use spirv_std::num_traits::Float;
+
+#[repr(C)]
+pub struct UBO {
+    pub projection: Mat4,
+    pub model: Mat4,
+    pub inv_model: Mat4,
+    pub lod_bias: f32,
+}
+
+#[spirv(vertex)]
+pub fn main_vs(
+    #[spirv(vertex_index)] _vertex_index: i32,
+    in_pos: Vec3,
+    in_normal: Vec3,
+    #[spirv(uniform, descriptor_set = 0, binding = 0)] ubo: &UBO,
+    #[spirv(position)] out_pos: &mut Vec4,
+    out_world_pos: &mut Vec3,
+    out_normal: &mut Vec3,
+    out_view_vec: &mut Vec3,
+    out_light_vec: &mut Vec3,
+) {
+    *out_pos = ubo.projection * ubo.model * vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+    
+    *out_world_pos = (ubo.model * vec4(in_pos.x, in_pos.y, in_pos.z, 1.0)).truncate();
+    *out_normal = Mat3::from_mat4(ubo.model) * in_normal;
+    
+    let light_pos = vec3(0.0, -5.0, 5.0);
+    *out_light_vec = light_pos - *out_world_pos;
+    *out_view_vec = -*out_world_pos;
+}
+
+#[spirv(fragment)]
+pub fn main_fs(
+    in_pos: Vec3,
+    in_normal: Vec3,
+    in_view_vec: Vec3,
+    in_light_vec: Vec3,
+    #[spirv(uniform, descriptor_set = 0, binding = 0)] ubo: &UBO,
+    #[spirv(descriptor_set = 0, binding = 1)] sampler_color: &SampledImage<Cubemap>,
+    out_frag_color: &mut Vec4,
+) {
+    let ci = in_pos.normalize();
+    let cr = ci.reflect(in_normal.normalize());
+    
+    let mut cr_transformed = (ubo.inv_model * vec4(cr.x, cr.y, cr.z, 0.0)).truncate();
+    cr_transformed.x *= -1.0;
+    cr_transformed.y *= -1.0;
+    
+    let color = sampler_color.sample_by_lod(cr_transformed, ubo.lod_bias);
+    
+    let n = in_normal.normalize();
+    let l = in_light_vec.normalize();
+    let v = in_view_vec.normalize();
+    let r = (-l).reflect(n);
+    
+    let ambient = vec3(0.5, 0.5, 0.5) * color.truncate();
+    let diffuse = n.dot(l).max(0.0) * vec3(1.0, 1.0, 1.0);
+    let specular = r.dot(v).max(0.0).powf(16.0) * vec3(0.5, 0.5, 0.5);
+    
+    let final_color = ambient + diffuse * color.truncate() + specular;
+    *out_frag_color = vec4(final_color.x, final_color.y, final_color.z, 1.0);
+}

shaders/rust/texturecubemap/reflect.vert.spv

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