Add inputattachments rust shaders

Author: LegNeato

shaders/rust/Cargo.lock

@@ -52,6 +52,8 @@ members = [
     "instancing/instancing",
     "instancing/planet",
     "instancing/starfield",
+    "inputattachments/attachmentwrite",
+    "inputattachments/attachmentread",
     "multisampling/mesh",
     "multithreading/phong",
     "multithreading/starsphere",

shaders/rust/Cargo.toml

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

shaders/rust/inputattachments/attachmentread.frag.spv

@@ -0,0 +1,53 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{vec4, Vec2, Vec3, Vec4}};
+
+#[repr(C)]
+#[derive(Copy, Clone)]
+pub struct Ubo {
+    pub brightness_contrast: Vec2,
+    pub range: Vec2,
+    pub attachment_index: i32,
+}
+
+#[spirv(vertex)]
+pub fn main_vs(
+    #[spirv(vertex_index)] vert_idx: i32,
+    #[spirv(position)] out_position: &mut Vec4,
+) {
+    let x = ((vert_idx << 1) & 2) as f32;
+    let y = (vert_idx & 2) as f32;
+    *out_position = vec4(x * 2.0 - 1.0, y * 2.0 - 1.0, 0.0, 1.0);
+}
+
+fn brightness_contrast(color: Vec3, brightness: f32, contrast: f32) -> Vec3 {
+    (color - 0.5) * contrast + 0.5 + brightness
+}
+
+#[spirv(fragment)]
+pub fn main_fs(
+    #[spirv(frag_coord)] frag_coord: Vec4,
+    #[spirv(input_attachment_index = 0, descriptor_set = 0, binding = 0)] input_color: &spirv_std::Image!(subpass, type=f32, sampled=false),
+    #[spirv(input_attachment_index = 1, descriptor_set = 0, binding = 1)] input_depth: &spirv_std::Image!(subpass, type=f32, sampled=false),
+    #[spirv(uniform, descriptor_set = 0, binding = 2)] ubo: &Ubo,
+    out_color: &mut Vec4,
+) {
+    let coord = spirv_std::glam::IVec2::new(frag_coord.x as i32, frag_coord.y as i32);
+    
+    // Apply brightness and contrast filter to color input
+    if ubo.attachment_index == 0 {
+        // Read color from previous color input attachment
+        let color = input_color.read_subpass(coord).truncate();
+        let adjusted = brightness_contrast(color, ubo.brightness_contrast.x, ubo.brightness_contrast.y);
+        *out_color = vec4(adjusted.x, adjusted.y, adjusted.z, 1.0);
+    }
+    
+    // Visualize depth input range
+    if ubo.attachment_index == 1 {
+        // Read depth from previous depth input attachment
+        let depth = input_depth.read_subpass(coord).x;
+        let normalized = (depth - ubo.range.x) * (1.0 / (ubo.range.y - ubo.range.x));
+        *out_color = vec4(normalized, normalized, normalized, 1.0);
+    }
+}

shaders/rust/inputattachments/attachmentread.vert.spv

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

shaders/rust/inputattachments/attachmentread/Cargo.toml

@@ -0,0 +1,65 @@
+#![cfg_attr(target_arch = "spirv", no_std)]
+#![allow(clippy::missing_safety_doc)]
+
+use spirv_std::{spirv, glam::{vec3, vec4, Mat4, Vec3, Vec4}};
+
+#[repr(C)]
+#[derive(Copy, Clone)]
+pub struct Ubo {
+    pub projection: Mat4,
+    pub model: Mat4,
+    pub view: Mat4,
+}
+
+#[spirv(vertex)]
+pub fn main_vs(
+    in_pos: Vec3,
+    in_color: Vec3,
+    in_normal: Vec3,
+    #[spirv(uniform, descriptor_set = 0, binding = 0)] ubo: &Ubo,
+    #[spirv(position)] out_position: &mut Vec4,
+    out_color: &mut Vec3,
+    out_normal: &mut Vec3,
+    out_view_vec: &mut Vec3,
+    out_light_vec: &mut Vec3,
+) {
+    *out_position = ubo.projection * ubo.view * ubo.model * vec4(in_pos.x, in_pos.y, in_pos.z, 1.0);
+    *out_color = in_color;
+    *out_normal = in_normal;
+    *out_light_vec = vec3(0.0, 5.0, 15.0) - in_pos;
+    *out_view_vec = -in_pos;
+}
+
+#[spirv(fragment)]
+pub fn main_fs(
+    in_color: Vec3,
+    in_normal: Vec3,
+    _in_view_vec: Vec3,
+    in_light_vec: Vec3,
+    out_color: &mut Vec4,
+) {
+    // Toon shading color attachment output
+    let intensity = in_normal.normalize().dot(in_light_vec.normalize());
+    let mut shade = 1.0;
+    if intensity < 0.5 {
+        shade = 0.75;
+    }
+    if intensity < 0.35 {
+        shade = 0.6;
+    }
+    if intensity < 0.25 {
+        shade = 0.5;
+    }
+    if intensity < 0.1 {
+        shade = 0.25;
+    }
+    
+    *out_color = vec4(
+        in_color.x * 3.0 * shade,
+        in_color.y * 3.0 * shade,
+        in_color.z * 3.0 * shade,
+        1.0
+    );
+    
+    // Depth attachment does not need to be explicitly written
+}