leyoda/iced-yoda
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

use single sdf for all corners

Browse source
This commit is contained in:
Daniel 2025-05-29 22:16:33 +10:00
parent cd60385b1e
commit 74665d696d
3 changed files with 33 additions and 102 deletions
Download patch file Download diff file Expand all files Collapse all files

32
wgpu/src/shader/quad.wgsl
View file

@ -5,31 +5,9 @@ struct Globals {
@group(0) @binding(0) var<uniform> globals: Globals; @group(0) @binding(0) var<uniform> globals: Globals;
fn distance_alg( fn rounded_box_sdf(p: vec2<f32>, size: vec2<f32>, corners: vec4<f32>) -> f32 {
frag_coord: vec2<f32>, var box_half = select(corners.yz, corners.xw, p.x > 0.0);
position: vec2<f32>, var corner = select(box_half.y, box_half.x, p.y > 0.0);
size: vec2<f32>, var q = abs(p) - size + corner;
radius: f32 return min(max(q.x, q.y), 0.0) + length(max(q, vec2(0.0))) - corner;
) -> f32 {
var inner_half_size: vec2<f32> = (size - vec2<f32>(radius, radius) * 2.0) / 2.0;
var top_left: vec2<f32> = position + vec2<f32>(radius, radius);
return rounded_box_sdf(frag_coord - top_left - inner_half_size, inner_half_size, 0.0);
}
// Given a vector from a point to the center of a rounded rectangle of the given `size` and
// border `radius`, determines the point's distance from the nearest edge of the rounded rectangle
fn rounded_box_sdf(to_center: vec2<f32>, size: vec2<f32>, radius: f32) -> f32 {
return length(max(abs(to_center) - size + vec2<f32>(radius, radius), vec2<f32>(0.0, 0.0))) - radius;
}
// Based on the fragment position and the center of the quad, select one of the 4 radii.
// Order matches CSS border radius attribute:
// radii.x = top-left, radii.y = top-right, radii.z = bottom-right, radii.w = bottom-left
fn select_border_radius(radii: vec4<f32>, position: vec2<f32>, center: vec2<f32>) -> f32 {
var rx = radii.x;
var ry = radii.y;
rx = select(radii.x, radii.y, position.x > center.x);
ry = select(radii.w, radii.z, position.x > center.x);
rx = select(rx, ry, position.y > center.y);
return rx;
} }

43
wgpu/src/shader/quad/gradient.wgsl
View file

@ -157,42 +157,19 @@ fn gradient_fs_main(input: GradientVertexOutput) -> @location(0) vec4<f32> {
let pos = input.position_and_scale.xy; let pos = input.position_and_scale.xy;
let scale = input.position_and_scale.zw; let scale = input.position_and_scale.zw;
var border_radius = select_border_radius( var dist: f32 = rounded_box_sdf(
input.border_radius, -(input.position.xy - pos - scale / 2.0) * 2.0,
input.position.xy, scale,
(pos + scale * 0.5).xy input.border_radius * 2.0
); ) / 2.0;
if (input.border_width > 0.0) { if (input.border_width > 0.0) {
var internal_border: f32 = max(border_radius - input.border_width, 0.0); mixed_color = mix(
mixed_color,
var internal_distance: f32 = distance_alg( input.border_color,
input.position.xy, clamp(0.5 + dist + input.border_width, 0.0, 1.0)
pos + vec2<f32>(input.border_width, input.border_width),
scale - vec2<f32>(input.border_width * 2.0, input.border_width * 2.0),
internal_border
); );
var border_mix: f32 = smoothstep(
max(internal_border - 0.5, 0.0),
internal_border + 0.5,
internal_distance
);
mixed_color = mix(mixed_color, input.border_color, border_mix);
} }
var dist: f32 = distance_alg( return mixed_color * clamp(0.5-dist, 0.0, 1.0);
input.position.xy,
pos,
scale,
border_radius
);
var radius_alpha: f32 = 1.0 - smoothstep(
max(border_radius - 0.5, 0.0),
border_radius + 0.5,
dist);
return mixed_color * radius_alpha;
} }

60
wgpu/src/shader/quad/solid.wgsl
View file

@ -75,57 +75,33 @@ fn solid_fs_main(
) -> @location(0) vec4<f32> { ) -> @location(0) vec4<f32> {
var mixed_color: vec4<f32> = input.color; var mixed_color: vec4<f32> = input.color;
var border_radius = select_border_radius( var dist = rounded_box_sdf(
input.border_radius, -(input.position.xy - input.pos - input.scale * 0.5) * 2.0,
input.position.xy, input.scale,
(input.pos + input.scale * 0.5).xy input.border_radius * 2.0
); ) / 2.0;
if (input.border_width > 0.0) { if (input.border_width > 0.0) {
var internal_border: f32 = max(border_radius - input.border_width, 0.0); mixed_color = mix(
input.color,
var internal_distance: f32 = distance_alg( input.border_color,
input.position.xy, clamp(0.5 + dist + input.border_width, 0.0, 1.0)
input.pos + vec2<f32>(input.border_width, input.border_width),
input.scale - vec2<f32>(input.border_width * 2.0, input.border_width * 2.0),
internal_border
); );
var border_mix: f32 = smoothstep(
max(internal_border - 0.5, 0.0),
internal_border + 0.5,
internal_distance
);
mixed_color = mix(input.color, input.border_color, border_mix);
} }
var dist: f32 = distance_alg( var quad_alpha: f32 = clamp(0.5-dist, 0.0, 1.0);
vec2<f32>(input.position.x, input.position.y),
input.pos,
input.scale,
border_radius
);
var radius_alpha: f32 = 1.0 - smoothstep( let quad_color = mixed_color * quad_alpha;
max(border_radius - 0.5, 0.0),
border_radius + 0.5,
dist
);
let quad_color = mixed_color * radius_alpha;
if input.shadow_color.a > 0.0 { if input.shadow_color.a > 0.0 {
let shadow_radius = select_border_radius( var shadow_dist: f32 = rounded_box_sdf(
input.border_radius, -(input.position.xy - input.pos - input.shadow_offset - input.scale/2.0) * 2.0,
input.position.xy - input.shadow_offset, input.scale,
(input.pos + input.scale * 0.5).xy input.border_radius * 2.0
); ) / 2.0;
let shadow_alpha = 1.0 - smoothstep(-input.shadow_blur_radius, input.shadow_blur_radius, max(shadow_dist, 0.0));
let shadow_distance = max(rounded_box_sdf(input.position.xy - input.pos - input.shadow_offset - (input.scale / 2.0), input.scale / 2.0, shadow_radius), 0.); return mix(quad_color, input.shadow_color, (1.0 - quad_alpha) * shadow_alpha);
let shadow_alpha = 1.0 - smoothstep(-input.shadow_blur_radius, input.shadow_blur_radius, shadow_distance);
return mix(quad_color, input.shadow_color, (1.0 - radius_alpha) * shadow_alpha);
} else { } else {
return quad_color; return quad_color;
} }