Add graphicspipelinelibrary rust shaders

Author: LegNeato

shaders/rust/Cargo.lock

@@ -36,6 +36,8 @@ members = [
     "gltfloading/mesh",
     "gltfscenerendering/scene",
     "gltfskinning/skinnedmodel",
+    "graphicspipelinelibrary/shared",
+    "graphicspipelinelibrary/uber",
     "hdr/bloom",
     "hdr/composition",
     "hdr/gbuffer",

shaders/rust/Cargo.toml

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

shaders/rust/graphicspipelinelibrary/shared.vert.spv

@@ -0,0 +1,46 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{mat3, vec4, Mat4, Vec3, Vec4}};
+
+#[repr(C)]
+#[derive(Copy, Clone)]
+pub struct Ubo {
+    pub projection: Mat4,
+    pub model: Mat4,
+    pub light_pos: Vec4,
+}
+
+#[spirv(vertex)]
+pub fn main_vs(
+    in_pos: Vec3,
+    in_normal: Vec3,
+    in_color: Vec3,
+    #[spirv(uniform, descriptor_set = 0, binding = 0)] ubo: &Ubo,
+    #[spirv(position)] out_position: &mut Vec4,
+    out_normal: &mut Vec3,
+    out_color: &mut Vec3,
+    out_view_vec: &mut Vec3,
+    out_light_vec: &mut Vec3,
+    #[spirv(flat)] out_flat_normal: &mut Vec3,
+) {
+    *out_normal = in_normal;
+    *out_color = in_color;
+    let pos = vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+
+    *out_position = ubo.projection * ubo.model * pos;
+    
+    let world_pos = ubo.model * pos;
+    let model_mat3 = mat3(
+        ubo.model.x_axis.truncate(),
+        ubo.model.y_axis.truncate(),
+        ubo.model.z_axis.truncate(),
+    );
+    *out_normal = model_mat3 * in_normal;
+    let l_pos = ubo.light_pos.truncate();
+    *out_light_vec = l_pos - world_pos.truncate();
+    *out_view_vec = -world_pos.truncate();
+    
+    // Flat shading normal is not interpolated
+    *out_flat_normal = *out_normal;
+}

shaders/rust/graphicspipelinelibrary/shared/Cargo.toml

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

shaders/rust/graphicspipelinelibrary/shared/src/lib.rs

@@ -0,0 +1,57 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{vec3, vec4, Vec3, Vec4}, num_traits::Float};
+
+#[spirv(fragment)]
+pub fn main_fs(
+    in_normal: Vec3,
+    in_color: Vec3,
+    in_view_vec: Vec3,
+    in_light_vec: Vec3,
+    #[spirv(flat)] _in_flat_normal: Vec3,
+    #[spirv(spec_constant(id = 0, default = 0))] lighting_model: u32,
+    out_frag_color: &mut Vec4,
+) {
+    let mut color = match lighting_model {
+        0 => { // Phong
+            let ambient = in_color * vec3(0.25, 0.25, 0.25);
+            let n = in_normal.normalize();
+            let l = in_light_vec.normalize();
+            let v = in_view_vec.normalize();
+            let r = (-l).reflect(n);
+            let diffuse = n.dot(l).max(0.0) * in_color;
+            let specular = r.dot(v).max(0.0).powf(32.0) * vec3(0.75, 0.75, 0.75);
+            ambient + diffuse * 1.75 + specular
+        }
+        1 => { // Toon
+            let n = in_normal.normalize();
+            let l = in_light_vec.normalize();
+            let intensity = n.dot(l);
+            if intensity > 0.98 {
+                in_color * 1.5
+            } else if intensity > 0.9 {
+                in_color * 1.0
+            } else if intensity > 0.5 {
+                in_color * 0.6
+            } else if intensity > 0.25 {
+                in_color * 0.4
+            } else {
+                in_color * 0.2
+            }
+        }
+        2 => { // No shading
+            in_color
+        }
+        3 => { // Greyscale
+            let grey = in_color.x * 0.299 + in_color.y * 0.587 + in_color.z * 0.114;
+            vec3(grey, grey, grey)
+        }
+        _ => in_color, // Default case
+    };
+
+    // Scene is dark, brighten up a bit
+    color *= 1.25;
+    
+    *out_frag_color = vec4(color.x, color.y, color.z, 1.0);
+}