Add imgui rust shaders

Author: LegNeato

shaders/rust/Cargo.lock

@@ -90,6 +90,8 @@ members = [
     "vertexattributes/scene",
     "multiview/multiview",
     "multiview/viewdisplay",
+    "imgui/ui",
+    "imgui/scene",
 ]
 
 [workspace.dependencies]

shaders/rust/Cargo.toml

@@ -0,0 +1,11 @@
+[package]
+name = "imgui-scene"
+version = "0.1.0"
+edition = "2021"
+authors = ["Sascha Willems <webmaster@saschawillems.de>"]
+
+[lib]
+crate-type = ["dylib"]
+
+[dependencies]
+spirv-std = { workspace = true }

shaders/rust/imgui/scene.frag.spv

@@ -0,0 +1,52 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{vec3, vec4, Mat3, Mat4, Vec3, Vec4, Vec4Swizzles}, num_traits::Float};
+
+#[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,
+) {
+    *out_color = in_color;
+    *out_position = ubo.projection * ubo.model * vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+    
+    let pos = ubo.model * vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+    *out_normal = Mat3::from_mat4(ubo.model) * in_normal;
+    let l_pos = Mat3::from_mat4(ubo.model) * ubo.light_pos.xyz();
+    *out_light_vec = l_pos - pos.xyz();
+    *out_view_vec = -pos.xyz();
+}
+
+#[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);
+    let specular = r.dot(v).max(0.0).powf(16.0) * vec3(0.75, 0.75, 0.75);
+    let color = diffuse * in_color + specular;
+    *out_frag_color = vec4(color.x, color.y, color.z, 1.0);
+}

shaders/rust/imgui/scene.vert.spv

@@ -0,0 +1,11 @@
+[package]
+name = "imgui-ui"
+version = "0.1.0"
+edition = "2021"
+authors = ["Sascha Willems <webmaster@saschawillems.de>"]
+
+[lib]
+crate-type = ["dylib"]
+
+[dependencies]
+spirv-std = { workspace = true }

shaders/rust/imgui/scene/Cargo.toml

@@ -0,0 +1,37 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{vec4, Vec2, Vec4}, Image, image::SampledImage};
+
+#[repr(C)]
+#[derive(Copy, Clone)]
+pub struct PushConstants {
+    pub scale: Vec2,
+    pub translate: Vec2,
+}
+
+#[spirv(vertex)]
+pub fn main_vs(
+    in_pos: Vec2,
+    in_uv: Vec2,
+    in_color: Vec4,
+    #[spirv(push_constant)] push_constants: &PushConstants,
+    #[spirv(position)] out_position: &mut Vec4,
+    out_uv: &mut Vec2,
+    out_color: &mut Vec4,
+) {
+    *out_uv = in_uv;
+    *out_color = in_color;
+    let xy = in_pos * push_constants.scale + push_constants.translate;
+    *out_position = vec4(xy.x, xy.y, 0.0, 1.0);
+}
+
+#[spirv(fragment)]
+pub fn main_fs(
+    in_uv: Vec2,
+    in_color: Vec4,
+    #[spirv(descriptor_set = 0, binding = 0)] font_sampler: &SampledImage<Image!(2D, type=f32, sampled)>,
+    out_color: &mut Vec4,
+) {
+    *out_color = in_color * font_sampler.sample(in_uv);
+}