Add displacement rust shaders

Author: LegNeato

shaders/rust/Cargo.lock

@@ -28,6 +28,8 @@ members = [
     "descriptorbuffer/cube",
     "descriptorindexing",
     "descriptorsets/cube",
+    "displacement/base",
+    "displacement/displacement",
     "distancefieldfonts/bitmap",
     "distancefieldfonts/sdf",
     "dynamicrendering/texture",

shaders/rust/Cargo.toml

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

shaders/rust/displacement/base.frag.spv

@@ -0,0 +1,41 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{vec4, Vec2, Vec3, Vec4}};
+
+#[spirv(vertex)]
+pub fn main_vs(
+    in_pos: Vec3,
+    in_normal: Vec3,
+    in_uv: Vec2,
+    #[spirv(position)] out_position: &mut Vec4,
+    out_normal: &mut Vec3,
+    out_uv: &mut Vec2,
+) {
+    *out_position = vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+    *out_uv = in_uv;
+    *out_normal = in_normal;
+}
+
+#[spirv(fragment)]
+pub fn main_fs(
+    in_normal: Vec3,
+    in_uv: Vec2,
+    in_eye_pos: Vec3,
+    in_light_vec: Vec3,
+    #[spirv(descriptor_set = 0, binding = 1)] color_map: &spirv_std::image::SampledImage<spirv_std::Image!(2D, type=f32, sampled)>,
+    out_frag_color: &mut Vec4,
+) {
+    let _n = in_normal.normalize();
+    let _l = Vec3::new(1.0, 1.0, 1.0).normalize();
+    
+    let color = color_map.sample(in_uv);
+    
+    let _eye = (-in_eye_pos).normalize();
+    let _reflected = (-in_light_vec).reflect(in_normal).normalize();
+    
+    let i_ambient = vec4(0.0, 0.0, 0.0, 1.0);
+    let i_diffuse = vec4(1.0, 1.0, 1.0, 1.0) * in_normal.dot(in_light_vec).max(0.0);
+    
+    *out_frag_color = (i_ambient + i_diffuse) * vec4(color.x, color.y, color.z, 1.0);
+}

shaders/rust/displacement/base.vert.spv

@@ -0,0 +1,13 @@
+[package]
+name = "displacement-displacement"
+version = "0.1.0"
+edition.workspace = true
+
+[dependencies]
+spirv-std = { workspace = true }
+
+[lib]
+crate-type = ["dylib"]
+
+[package.metadata.rust-gpu.build]
+capabilities = ["Tessellation"]