1 files changed, 83 insertions, 0 deletions
diff --git a/wgpu/src/shader/triangle_gradient.wgsl b/wgpu/src/shader/triangle_gradient.wgsl
new file mode 100644
index 00000000..cb35b61c
--- /dev/null
+++ b/wgpu/src/shader/triangle_gradient.wgsl
@@ -0,0 +1,83 @@
+// uniforms
+struct GradientUniforms {
+    transform: mat4x4<f32>,
+    @size(16) start: vec2<f32>,
+    @size(16) end: vec2<f32>,
+    @size(16) start_stop: i32,
+    @size(16) end_stop: i32,
+}
+
+struct Stop {
+    color: vec4<f32>,
+    offset: f32,
+};
+
+@group(0) @binding(0)
+var<uniform> gradient_uniforms: GradientUniforms;
+
+@group(0) @binding(1)
+var<storage, read> color_stops: array<Stop>;
+
+struct VertexOutput {
+    @builtin(position) position: vec4<f32>,
+    @location(0) raw_position: vec2<f32>
+}
+
+@vertex
+fn vs_main(@location(0) input: vec2<f32>) -> VertexOutput {
+    var output: VertexOutput;
+    output.position = gradient_uniforms.transform * vec4<f32>(input.xy, 0.0, 1.0);
+    output.raw_position = input;
+
+    return output;
+}
+
+@fragment
+fn fs_gradient(input: VertexOutput) -> @location(0) vec4<f32> {
+    let v1 = gradient_uniforms.end - gradient_uniforms.start;
+    let v2 = input.raw_position.xy - gradient_uniforms.start;
+    let unit = normalize(v1);
+    let offset = dot(unit, v2) / length(v1);
+
+    let min_stop = color_stops[gradient_uniforms.start_stop];
+    let max_stop = color_stops[gradient_uniforms.end_stop];
+
+    var color: vec4<f32>;
+
+    if (offset <= min_stop.offset) {
+        color = min_stop.color;
+    } else if (offset >= max_stop.offset) {
+        color = max_stop.color;
+    } else {
+        var min = min_stop;
+        var max = max_stop;
+        var min_index = gradient_uniforms.start_stop;
+        var max_index = gradient_uniforms.end_stop;
+
+        loop {
+            if (min_index >= max_index - 1) {
+                break;
+            }
+
+            let index = min_index + (max_index - min_index) / 2;
+
+            let stop = color_stops[index];
+
+            if (offset <= stop.offset) {
+                max = stop;
+                max_index = index;
+            } else {
+                min = stop;
+                min_index = index;
+            }
+        }
+
+        color = mix(min.color, max.color, smoothstep(
+            min.offset,
+            max.offset,
+            offset
+        ));
+    }
+
+    return color;
+}
+\ No newline at end of file