Add multithreading rust shaders

Author: LegNeato

shaders/rust/Cargo.lock

@@ -51,6 +51,8 @@ members = [
     "instancing/planet",
     "instancing/starfield",
     "multisampling/mesh",
+    "multithreading/phong",
+    "multithreading/starsphere",
     "multiview/multiview",
     "multiview/viewdisplay",
     "negativeviewportheight/quad",

shaders/rust/Cargo.toml

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

shaders/rust/multithreading/phong.frag.spv

@@ -0,0 +1,63 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{mat3, vec3, vec4, Mat4, Vec3, Vec4}, num_traits::Float};
+
+#[repr(C)]
+#[derive(Copy, Clone)]
+pub struct PushConsts {
+    pub mvp: Mat4,
+    pub color: Vec3,
+}
+
+#[spirv(vertex)]
+pub fn main_vs(
+    in_pos: Vec3,
+    in_normal: Vec3,
+    in_color: Vec3,
+    #[spirv(push_constant)] push_consts: &PushConsts,
+    #[spirv(position)] out_position: &mut Vec4,
+    out_normal: &mut Vec3,
+    out_color: &mut Vec3,
+    out_view_vec: &mut Vec3,
+    out_light_vec: &mut Vec3,
+) {
+    *out_normal = in_normal;
+
+    // If color is red (1.0, 0.0, 0.0), use push constant color
+    if in_color.x == 1.0 && in_color.y == 0.0 && in_color.z == 0.0 {
+        *out_color = push_consts.color;
+    } else {
+        *out_color = in_color;
+    }
+    
+    *out_position = push_consts.mvp * vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+    
+    let pos = push_consts.mvp * vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+    let mvp_mat3 = mat3(
+        push_consts.mvp.x_axis.truncate(),
+        push_consts.mvp.y_axis.truncate(),
+        push_consts.mvp.z_axis.truncate(),
+    );
+    *out_normal = mvp_mat3 * in_normal;
+    let l_pos = vec3(0.0, 0.0, 0.0);
+    *out_light_vec = l_pos - pos.truncate();
+    *out_view_vec = -pos.truncate();
+}
+
+#[spirv(fragment)]
+pub fn main_fs(
+    in_normal: Vec3,
+    in_color: Vec3,
+    in_view_vec: Vec3,
+    in_light_vec: Vec3,
+    out_frag_color: &mut Vec4,
+) {
+    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(8.0) * vec3(0.75, 0.75, 0.75);
+    *out_frag_color = vec4(diffuse.x + specular.x, diffuse.y + specular.y, diffuse.z + specular.z, 1.0);
+}

shaders/rust/multithreading/phong.vert.spv

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

shaders/rust/multithreading/phong/Cargo.toml

@@ -0,0 +1,60 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{vec3, vec4, Mat4, Vec3, Vec4}, num_traits::Float};
+
+#[repr(C)]
+#[derive(Copy, Clone)]
+pub struct PushConsts {
+    pub mvp: Mat4,
+}
+
+#[spirv(vertex)]
+pub fn main_vs(
+    in_pos: Vec3,
+    #[spirv(push_constant)] push_consts: &PushConsts,
+    #[spirv(position)] out_position: &mut Vec4,
+    out_uvw: &mut Vec3,
+) {
+    *out_uvw = in_pos;
+    *out_position = push_consts.mvp * vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+}
+
+const HASHSCALE3: Vec3 = vec3(443.897, 441.423, 437.195);
+const STARFREQUENCY: f32 = 0.01;
+
+// Hash function by Dave Hoskins (https://www.shadertoy.com/view/4djSRW)
+fn hash33(p3_in: Vec3) -> f32 {
+    let mut p3 = p3_in.fract() * HASHSCALE3;
+    p3 += p3.dot(vec3(p3.y, p3.x, p3.z) + vec3(19.19, 19.19, 19.19));
+    ((p3.x + p3.y) * p3.z + (p3.x + p3.z) * p3.y + (p3.y + p3.z) * p3.x).fract()
+}
+
+fn star_field(pos: Vec3) -> Vec3 {
+    let mut color = vec3(0.0, 0.0, 0.0);
+    let threshold = 1.0 - STARFREQUENCY;
+    let rnd = hash33(pos);
+    if rnd >= threshold {
+        let star_col = ((rnd - threshold) / (1.0 - threshold)).powf(16.0);
+        color += vec3(star_col, star_col, star_col);
+    }
+    color
+}
+
+#[spirv(fragment)]
+pub fn main_fs(
+    in_uvw: Vec3,
+    out_frag_color: &mut Vec4,
+) {
+    // Fake atmosphere at the bottom
+    let atmosphere = vec3(0.1, 0.15, 0.4) * (in_uvw.y + 0.25);
+    let atmosphere = vec3(
+        atmosphere.x.clamp(0.0, 1.0),
+        atmosphere.y.clamp(0.0, 1.0),
+        atmosphere.z.clamp(0.0, 1.0)
+    );
+
+    let color = star_field(in_uvw) + atmosphere;
+    
+    *out_frag_color = vec4(color.x, color.y, color.z, 1.0);
+}