Upload spectral-render/src/shaders.rs

spectral-render/src/shaders.rs

@@ -0,0 +1,146 @@
+pub const BG_VERTEX: &str = r#"
+struct BgInstance {
+    col: u32, row: u32,
+    bg_r: u32, bg_g: u32, bg_b: u32, bg_a: u32,
+};
+@group(0) @binding(0)
+var<storage, read> instances: array<BgInstance>;
+
+struct Uniforms {
+    cell_width: f32, cell_height: f32,
+    screen_cols: u32, screen_rows: u32,
+};
+@group(0) @binding(1)
+var<uniform> uniforms: Uniforms;
+
+struct VertexOutput {
+    @builtin(position) position: vec4<f32>,
+    @location(0) color: vec4<f32>,
+};
+
+@vertex
+fn main(@builtin(vertex_index) vertex_idx: u32, @builtin(instance_index) instance_idx: u32) -> VertexOutput {
+    let cell = instances[instance_idx];
+    var corners = array<vec2<f32>, 6>(
+        vec2<f32>(0.0, 0.0), vec2<f32>(1.0, 0.0), vec2<f32>(0.0, 1.0),
+        vec2<f32>(0.0, 1.0), vec2<f32>(1.0, 0.0), vec2<f32>(1.0, 1.0),
+    );
+    let corner = corners[vertex_idx];
+    let x = (f32(cell.col) + corner.x) * uniforms.cell_width;
+    let y = (f32(cell.row) + corner.y) * uniforms.cell_height;
+    let screen_w = f32(uniforms.screen_cols) * uniforms.cell_width;
+    let screen_h = f32(uniforms.screen_rows) * uniforms.cell_height;
+    let ndc_x = (x / screen_w) * 2.0 - 1.0;
+    let ndc_y = -((y / screen_h) * 2.0 - 1.0);
+    var out: VertexOutput;
+    out.position = vec4<f32>(ndc_x, ndc_y, 0.0, 1.0);
+    out.color = vec4<f32>(f32(cell.bg_r) / 255.0, f32(cell.bg_g) / 255.0, f32(cell.bg_b) / 255.0, f32(cell.bg_a) / 255.0);
+    return out;
+}
+"#;
+
+pub const BG_FRAGMENT: &str = r#"
+struct VertexOutput {
+    @builtin(position) position: vec4<f32>,
+    @location(0) color: vec4<f32>,
+};
+@fragment
+fn main(in: VertexOutput) -> @location(0) vec4<f32> {
+    return in.color;
+}
+"#;
+
+pub const TEXT_VERTEX: &str = r#"
+struct TextInstance {
+    col: f32, row: f32,
+    bearing_x: f32, bearing_y: f32,
+    glyph_w: f32, glyph_h: f32,
+    atlas_x: f32, atlas_y: f32,
+    atlas_w: f32, atlas_h: f32,
+    fg_r: u32, fg_g: u32, fg_b: u32,
+    flags: u32, thickening: f32,
+};
+@group(0) @binding(0)
+var<storage, read> instances: array<TextInstance>;
+
+struct Uniforms {
+    cell_width: f32, cell_height: f32,
+    atlas_size: f32,
+    screen_cols: u32, screen_rows: u32,
+};
+@group(0) @binding(1)
+var<uniform> uniforms: Uniforms;
+
+struct VertexOutput {
+    @builtin(position) position: vec4<f32>,
+    @location(0) uv: vec2<f32>,
+    @location(1) fg_color: vec4<f32>,
+    @location(2) atlas_uv: vec4<f32>,
+    @location(3) thickening: f32,
+    @location(4) flags: u32,
+};
+
+@vertex
+fn main(@builtin(vertex_index) vertex_idx: u32, @builtin(instance_index) instance_idx: u32) -> VertexOutput {
+    let cell = instances[instance_idx];
+    var corners = array<vec2<f32>, 6>(
+        vec2<f32>(0.0, 0.0), vec2<f32>(1.0, 0.0), vec2<f32>(0.0, 1.0),
+        vec2<f32>(0.0, 1.0), vec2<f32>(1.0, 0.0), vec2<f32>(1.0, 1.0),
+    );
+    let corner = corners[vertex_idx];
+    let x = (cell.col * uniforms.cell_width) + cell.bearing_x + corner.x * cell.glyph_w;
+    let y = (cell.row * uniforms.cell_height) + cell.bearing_y + corner.y * cell.glyph_h;
+    let screen_w = f32(uniforms.screen_cols) * uniforms.cell_width;
+    let screen_h = f32(uniforms.screen_rows) * uniforms.cell_height;
+    let ndc_x = (x / screen_w) * 2.0 - 1.0;
+    let ndc_y = -((y / screen_h) * 2.0 - 1.0);
+    var out: VertexOutput;
+    out.position = vec4<f32>(ndc_x, ndc_y, 0.0, 1.0);
+    out.uv = corner;
+    out.fg_color = vec4<f32>(f32(cell.fg_r) / 255.0, f32(cell.fg_g) / 255.0, f32(cell.fg_b) / 255.0, 1.0);
+    let norm = uniforms.atlas_size;
+    out.atlas_uv = vec4<f32>(cell.atlas_x / norm, cell.atlas_y / norm, cell.atlas_w / norm, cell.atlas_h / norm);
+    out.thickening = cell.thickening;
+    out.flags = cell.flags;
+    return out;
+}
+"#;
+
+pub const TEXT_FRAGMENT: &str = r#"
+@group(0) @binding(2)
+var atlas_sampler: sampler;
+@group(0) @binding(3)
+var atlas_texture: texture_2d<f32>;
+
+struct VertexOutput {
+    @builtin(position) position: vec4<f32>,
+    @location(0) uv: vec2<f32>,
+    @location(1) fg_color: vec4<f32>,
+    @location(2) atlas_uv: vec4<f32>,
+    @location(3) thickening: f32,
+    @location(4) flags: u32,
+};
+
+fn median(r: f32, g: f32, b: f32) -> f32 {
+    return max(min(r, g), min(max(r, g), b));
+}
+
+@fragment
+fn main(in: VertexOutput) -> @location(0) vec4<f32> {
+    let atlas_pos = vec2<f32>(
+        in.atlas_uv.x + in.uv.x * in.atlas_uv.z,
+        in.atlas_uv.y + in.uv.y * in.atlas_uv.w,
+    );
+    let sample = textureSample(atlas_texture, atlas_sampler, atlas_pos);
+    let sig_dist = median(sample.r, sample.g, sample.b);
+    let threshold = 0.5 - in.thickening * 0.15;
+    let smoothing = 0.05;
+    let opacity = smoothstep(threshold - smoothing, threshold + smoothing, sig_dist);
+    let is_colored = (in.flags & 0x0040u) != 0u;
+    if is_colored {
+        return vec4<f32>(sample.rgb, opacity);
+    } else {
+        return vec4<f32>(in.fg_color.rgb, opacity);
+    }
+}
+"#;