From 1b3835a38ed38d377ce337836a4b32ee89ca0289 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Thu, 8 Dec 2022 20:13:27 +0100
Subject: rename modes

---
 evc/src/codec/encode/advanced.rs | 109 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 evc/src/codec/encode/advanced.rs

(limited to 'evc/src/codec/encode/advanced.rs')

diff --git a/evc/src/codec/encode/advanced.rs b/evc/src/codec/encode/advanced.rs
new file mode 100644
index 0000000..e86a1b8
--- /dev/null
+++ b/evc/src/codec/encode/advanced.rs
@@ -0,0 +1,109 @@
+use super::EncodeConfig;
+use crate::{
+    block::{AdvancedReference, Block},
+    frame::Frame,
+    refsampler::Sampler,
+    view::View,
+};
+
+pub fn default(
+    view: &View<&Frame>,
+    prev: &View<&Frame>,
+    config: &EncodeConfig,
+    max_diff: f64,
+) -> (f64, Block) {
+    let mut pm = AdvancedReference::default();
+    let sampler = Sampler::from_refblock(prev.clone(), &pm);
+    let mut diff = View::diff_sampler(&view, &sampler);
+    if diff < max_diff {
+        (diff, Block::REFZERO)
+    } else {
+        loop {
+            let (mut d, mut p) = (diff, pm.clone());
+
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 4);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 4);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 4);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 4);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 2);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 2);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 2);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 2);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 1);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 1);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 1);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 1);
+            if config.do_value_scale {
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.value_scale += 1);
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.value_scale -= 1);
+            }
+            if config.do_matrix_transform {
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.a -= 1);
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.a += 1);
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.b -= 1);
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.b += 1);
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.c -= 1);
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.c += 1);
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.d -= 1);
+                pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.d += 1);
+            }
+
+            if d >= diff {
+                break (diff, Block::AdvancedReference(pm));
+            }
+
+            diff = d;
+            pm = p;
+        }
+    }
+}
+
+pub fn partial(
+    view: &View<&Frame>,
+    prev: &View<&Frame>,
+    _config: &EncodeConfig,
+    max_diff: f64,
+) -> (f64, Block) {
+    let mut pm = AdvancedReference::default();
+    let sampler = Sampler::from_refblock(prev.clone(), &pm);
+    let mut diff = View::diff_sampler(&view, &sampler);
+    if diff < max_diff {
+        (diff, Block::REFZERO)
+    } else {
+        loop {
+            let (mut d, mut p) = (diff, pm.clone());
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 2);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 2);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 2);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 2);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 1);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 1);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 1);
+            pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 1);
+            if d >= diff {
+                break (diff, Block::AdvancedReference(pm));
+            }
+            diff = d;
+            pm = p;
+        }
+    }
+}
+
+#[inline]
+fn pk<F: FnMut(&mut AdvancedReference) -> ()>(
+    view: &View<&Frame>,
+    prev: &View<&Frame>,
+    diff: &mut f64,
+    params: &mut AdvancedReference,
+    initial_params: &AdvancedReference,
+    mut f: F,
+) {
+    let mut p = initial_params.clone();
+    f(&mut p);
+    let sampler = Sampler::from_refblock(prev.clone(), &p);
+    let d = View::diff_sampler(view, &sampler);
+    if d < *diff {
+        *diff = d;
+        *params = p;
+    }
+}
-- 
cgit v1.2.3-70-g09d2