repose-render-wgpu 0.11.0

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

struct VSOut {
  @builtin(position) pos: vec4<f32>,
  @location(0) uv: vec2<f32>,
  @location(1) tint: vec4<f32>,
  @location(2) full_range: f32,
};

@vertex
fn vs_main(
  @location(0) xywh: vec4<f32>,      // NDC rect
  @location(1) uv_rect: vec4<f32>,   // u0,v0,u1,v1
  @location(2) tint: vec4<f32>,
  @location(3) full_range: f32,
  @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];
  var o: VSOut;
  o.pos = vec4(xywh.xy + p * xywh.zw, 0.0, 1.0);
  o.uv = mix(uv_rect.xy, uv_rect.zw, uv_lerp);
  o.tint = tint;
  o.full_range = full_range;
  return o;
}

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

fn yuv_to_rgb(y: f32, u: f32, v: f32, full_range: bool) -> vec3<f32> {
  let uu = u - 0.5;
  let vv = v - 0.5;

  // TODO: BT.709/601 + range handling later.
  // If limited range: Y typically [16/255..235/255]; UV [16/255..240/255]
  var yy = y;
  if (!full_range) {
    yy = clamp((y - (16.0/255.0)) * (255.0/219.0), 0.0, 1.0);
  }

  let r = yy + 1.5748 * vv;
  let g = yy - 0.1873 * uu - 0.4681 * vv;
  let b = yy + 1.8556 * uu;
  return clamp(vec3(r,g,b), vec3(0.0), vec3(1.0));
}

@fragment
fn fs_main(i: VSOut) -> @location(0) vec4<f32> {
  let y = textureSample(tex_y, samp, i.uv).r;
  let uv = textureSample(tex_uv, samp, i.uv).rg;
  let rgb = yuv_to_rgb(y, uv.r, uv.g, i.full_range > 0.5);

  let a = i.tint.a;
  let out_rgb = rgb * i.tint.rgb * a;
  return vec4(out_rgb, a);
}