aboutsummaryrefslogtreecommitdiff
path: root/flowy/src/motion/enc.wgsl
diff options
context:
space:
mode:
Diffstat (limited to 'flowy/src/motion/enc.wgsl')
-rw-r--r--flowy/src/motion/enc.wgsl121
1 files changed, 0 insertions, 121 deletions
diff --git a/flowy/src/motion/enc.wgsl b/flowy/src/motion/enc.wgsl
deleted file mode 100644
index dbc6410..0000000
--- a/flowy/src/motion/enc.wgsl
+++ /dev/null
@@ -1,121 +0,0 @@
-
-struct Params {
- block_size: vec2<i32>,
- offsets_stride: u32,
- search_radius: i32,
- skip_threshold: f32,
-}
-
-struct BlockOffset {
- score: f32,
- offset: vec2<i32>,
- tint: vec3<f32>,
-}
-
-@group(0) @binding(0) var<uniform> params: Params;
-@group(0) @binding(1) var<storage, read_write> offsets: array<BlockOffset>;
-@group(0) @binding(2) var next: texture_2d<f32>;
-@group(0) @binding(3) var prev: texture_2d<f32>;
-
-var<private> best_offset: vec2<i32> = vec2(0);
-var<private> best_error: f32 = 100000.;
-var<private> best_tint: vec3<f32> = vec3(0.);
-
-@compute @workgroup_size(1)
-fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
- let uv = vec2<i32>(global_id.xy) * params.block_size;
-
- loop {
- test_offset(uv, vec2(0, 0));
- if best_error < params.skip_threshold { break; }
- apply_tint(uv);
- if best_error < params.skip_threshold { break; }
- best_tint = vec3(0.);
- best_error = 10000000.;
-
- do_dist(uv, 64);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 32);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 24);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 16);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 12);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 8);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 6);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 4);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 3);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 2);
- if best_error < params.skip_threshold { break; }
- do_dist(uv, 1);
- if best_error < params.skip_threshold { break; }
-
- apply_tint(uv);
- break;
- }
-
- offsets[global_id.x + global_id.y * params.offsets_stride] = BlockOffset(best_error, best_offset, best_tint);
-}
-
-fn do_dist(uv: vec2<i32>, n: i32) {
- test_offset(uv, vec2(0, n));
- test_offset(uv, vec2(n, n));
- test_offset(uv, vec2(n, 0));
- test_offset(uv, vec2(n, -n));
- test_offset(uv, vec2(0, -n));
- test_offset(uv, vec2(-n, -n));
- test_offset(uv, vec2(-n, 0));
- test_offset(uv, vec2(-n, n));
-}
-
-fn apply_tint(uv: vec2<i32>) {
-
- var average_pcol = vec3(0.);
- var average_ncol = vec3(0.);
- for (var x = 0; x < params.block_size.x; x++) {
- for (var y = 0; y < params.block_size.y; y++) {
- let base = uv + vec2(x, y);
- average_pcol += textureLoad(prev, base + best_offset, 0).rgb;
- average_ncol += textureLoad(next, base, 0).rgb;
- }
- }
-
- let tint = (average_ncol - average_pcol) / f32(params.block_size.x * params.block_size.y);
-
- var err = 0.;
- for (var x = 0; x < params.block_size.x; x++) {
- for (var y = 0; y < params.block_size.y; y++) {
- let base = uv + vec2(x, y);
- let pcol = textureLoad(prev, base + best_offset, 0).rgb + tint;
- let ncol = textureLoad(next, base, 0).rgb;
- err += distance(pcol, ncol);
- }
- }
- if err < best_error {
- best_error = err;
- best_tint = tint;
- }
-}
-
-fn test_offset(uv: vec2<i32>, offset: vec2<i32>) {
- var err = 0.;
- for (var x = 0; x < params.block_size.x; x++) {
- for (var y = 0; y < params.block_size.y; y++) {
- let base = uv + vec2(x, y);
- let pcol = textureLoad(prev, base + offset, 0).rgb;
- let ncol = textureLoad(next, base, 0).rgb;
- err += distance(pcol, ncol);
- }
- }
- if err < best_error {
- best_error = err;
- best_offset = offset;
- }
-}
-