repose-render-wgpu 0.31.2

Repose wgpu handler
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

struct Globals {
    ndc_to_px: vec2<f32>,
    _pad: vec2<f32>,
};
@group(0) @binding(0) var<uniform> G: Globals;

struct VSOut {
    @builtin(position) pos: vec4<f32>,
    @location(0) color: vec4<f32>,
    @location(1) uv: vec2<f32>,
    @location(2) blur_uv: vec2<f32>,
    @location(3) @interpolate(flat) edge_mode: u32,
};

@vertex
fn vs_main(
    @location(0) xywh: vec4<f32>,
    @location(1) uv_rect: vec4<f32>,
    @location(2) color: vec4<f32>,
    @location(3) blur_uv: vec2<f32>,
    @location(4) fwd_mat: vec4<f32>,
    @location(5) edge_mode: u32,
    @builtin(vertex_index) v: u32
) -> VSOut {
    var positions = array<vec2<f32>, 6>(
        vec2(0.0, 0.0), vec2(1.0, 0.0), vec2(1.0, 1.0),
        vec2(0.0, 0.0), vec2(1.0, 1.0), vec2(0.0, 1.0)
    );
    var uvs = array<vec2<f32>, 6>(
        vec2(0.0, 0.0), vec2(1.0, 0.0), vec2(1.0, 1.0),
        vec2(0.0, 0.0), vec2(1.0, 1.0), vec2(0.0, 1.0)
    );
    let p = positions[v];
    let uv_lerp = uvs[v];
    let half = 0.5 * xywh.zw;
    let corner = (p * 2.0 - 1.0) * half;
    let rotated = vec2(
        fwd_mat.x * corner.x + fwd_mat.y * corner.y,
        fwd_mat.z * corner.x + fwd_mat.w * corner.y
    );
    let pos_ndc = xywh.xy + rotated;

    var out: VSOut;
    out.pos = vec4(pos_ndc, 0.0, 1.0);
    out.uv = mix(uv_rect.xy, uv_rect.zw, uv_lerp);
    out.color = color;
    out.blur_uv = blur_uv;
    out.edge_mode = edge_mode;
    return out;
}

@group(1) @binding(0) var src_tex: texture_2d<f32>;
@group(1) @binding(1) var src_smp: sampler;

struct WeightedTap {
    color: vec4<f32>,
    weight: f32,
};

fn weighted_tap(uv: vec2<f32>, weight: f32, edge_mode: u32) -> WeightedTap {
    if (edge_mode != 0u && (any(uv < vec2<f32>(0.0)) || any(uv > vec2<f32>(1.0)))) {
        return WeightedTap(vec4<f32>(0.0), 0.0);
    }
    let sample = textureSample(src_tex, src_smp, clamp(uv, vec2<f32>(0.0), vec2<f32>(1.0)));
    return WeightedTap(vec4<f32>(sample.rgb * weight, sample.a * weight), weight);
}

@fragment
fn fs_main(in: VSOut) -> @location(0) vec4<f32> {
    let bu = in.blur_uv;
    let offsets = array<vec3<f32>, 9>(
        vec3<f32>(-1.0, -1.0, 1.0 / 16.0),
        vec3<f32>(0.0, -1.0, 2.0 / 16.0),
        vec3<f32>(1.0, -1.0, 1.0 / 16.0),
        vec3<f32>(-1.0, 0.0, 2.0 / 16.0),
        vec3<f32>(0.0, 0.0, 4.0 / 16.0),
        vec3<f32>(1.0, 0.0, 2.0 / 16.0),
        vec3<f32>(-1.0, 1.0, 1.0 / 16.0),
        vec3<f32>(0.0, 1.0, 2.0 / 16.0),
        vec3<f32>(1.0, 1.0, 1.0 / 16.0),
    );
    var blurred = vec4<f32>(0.0);
    var total_weight = 0.0;
    for (var i = 0u; i < 9u; i++) {
        let tap = weighted_tap(
            in.uv + offsets[i].xy * bu,
            offsets[i].z,
            in.edge_mode,
        );
        blurred += tap.color;
        total_weight += tap.weight;
    }
    let normalizer = max(total_weight, 1.0 / 16.0);
    return vec4<f32>(
        blurred.rgb / normalizer * in.color.a,
        blurred.a / normalizer * in.color.a,
    );
}