frus-gpu 0.2.1

frus — GPU context and 2D rendering engine (Milestone 0)
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

// Image rendering: instanced quads sampling a texture — one pass per image, the
// texture being bound in group 1. Each instance carries its destination rectangle,
// its UV sub-region (the crop), its tint and its clip.

struct Viewport {
    size: vec2<f32>,
};

@group(0) @binding(0)
var<uniform> viewport: Viewport;

@group(1) @binding(0)
var tex: texture_2d<f32>;
@group(1) @binding(1)
var samp: sampler;

struct VertexInput {
    @location(0) unit_pos: vec2<f32>,
};

struct InstanceInput {
    @location(1) rect: vec4<f32>, // x, y, width, height (px)
    @location(2) uv: vec4<f32>,   // x, y, width, height (0..1)
    @location(3) tint: vec4<f32>, // sRGB, multiplied in
    @location(4) clip: vec4<f32>, // x, y, width, height (px)
};

struct VertexOutput {
    @builtin(position) clip_position: vec4<f32>,
    @location(0) uv: vec2<f32>,
    @location(1) @interpolate(flat) tint: vec4<f32>,
    @location(2) frag_px: vec2<f32>,
    @location(3) @interpolate(flat) clip: vec4<f32>,
};

@vertex
fn vs_main(vert: VertexInput, inst: InstanceInput) -> VertexOutput {
    let pos_px = inst.rect.xy + vert.unit_pos * inst.rect.zw;
    let ndc = vec2<f32>(
        pos_px.x / viewport.size.x * 2.0 - 1.0,
        1.0 - pos_px.y / viewport.size.y * 2.0,
    );
    var out: VertexOutput;
    out.clip_position = vec4<f32>(ndc, 0.0, 1.0);
    out.uv = inst.uv.xy + vert.unit_pos * inst.uv.zw;
    out.tint = inst.tint;
    out.frag_px = pos_px;
    out.clip = inst.clip;
    return out;
}

fn srgb_to_linear(c: vec3<f32>) -> vec3<f32> {
    let lower = c / 12.92;
    let higher = pow((c + vec3<f32>(0.055)) / 1.055, vec3<f32>(2.4));
    return select(higher, lower, c <= vec3<f32>(0.04045));
}

@fragment
fn fs_main(in: VertexOutput) -> @location(0) vec4<f32> {
    let inside_clip = f32(
        in.frag_px.x >= in.clip.x
        && in.frag_px.x <= in.clip.x + in.clip.z
        && in.frag_px.y >= in.clip.y
        && in.frag_px.y <= in.clip.y + in.clip.w
    );
    // The texture is in an sRGB format, so the sample is already linear. The tint,
    // an authored sRGB colour, is linearised before being multiplied in.
    let sample = textureSample(tex, samp, in.uv);
    let rgb = sample.rgb * srgb_to_linear(in.tint.rgb);
    let alpha = sample.a * in.tint.a * inside_clip;
    return vec4<f32>(rgb, alpha);
}