aboutsummaryrefslogtreecommitdiff
path: root/evc/src/helpers/pixel.rs
diff options
context:
space:
mode:
authormetamuffin <metamuffin@disroot.org>2022-12-07 17:55:27 +0100
committermetamuffin <metamuffin@disroot.org>2022-12-07 17:55:27 +0100
commit8ca219c6b0d5448fd4529713ccd093e89de4e252 (patch)
tree8fe02fd35ae337766d6f8e7533da170830162d14 /evc/src/helpers/pixel.rs
parenta99674b911cb9b2fa398ccf61830d5933ccaf931 (diff)
downloadvideo-codec-experiments-8ca219c6b0d5448fd4529713ccd093e89de4e252.tar
video-codec-experiments-8ca219c6b0d5448fd4529713ccd093e89de4e252.tar.bz2
video-codec-experiments-8ca219c6b0d5448fd4529713ccd093e89de4e252.tar.zst
refactor
Diffstat (limited to 'evc/src/helpers/pixel.rs')
-rw-r--r--evc/src/helpers/pixel.rs67
1 files changed, 67 insertions, 0 deletions
diff --git a/evc/src/helpers/pixel.rs b/evc/src/helpers/pixel.rs
new file mode 100644
index 0000000..7b7c8d5
--- /dev/null
+++ b/evc/src/helpers/pixel.rs
@@ -0,0 +1,67 @@
+use crate::format::ser::{Ser, Sink, Source};
+
+#[derive(Copy, Clone, Debug, Default)]
+pub struct Pixel {
+ pub r: u8,
+ pub g: u8,
+ pub b: u8,
+}
+
+impl Ser for Pixel {
+ fn write(&self, sink: &mut impl std::io::Write) -> anyhow::Result<()> {
+ sink.put((self.r, self.g, self.b))
+ }
+
+ fn read(source: &mut impl std::io::Read) -> anyhow::Result<Self> {
+ let (r, g, b) = source.get()?;
+ Ok(Self { r, g, b })
+ }
+}
+
+impl Pixel {
+ pub const BLACK: Pixel = Pixel { r: 0, g: 0, b: 0 };
+ #[inline]
+ pub fn distance(a: Pixel, b: Pixel) -> usize {
+ let (rd, gd, bd) = (
+ a.r.abs_diff(b.r) as usize,
+ a.r.abs_diff(b.r) as usize,
+ a.r.abs_diff(b.r) as usize,
+ );
+ SQRT[rd + gd + bd]
+ }
+ #[inline]
+ pub fn average(a: Pixel, b: Pixel) -> Pixel {
+ Pixel {
+ r: (a.r >> 2) + (b.r >> 2),
+ g: (a.g >> 2) + (b.g >> 2),
+ b: (a.b >> 2) + (b.b >> 2),
+ }
+ }
+ pub fn scale(&self, factor: f32) -> Pixel {
+ Pixel {
+ r: ((self.r as f32) * factor).clamp(0.0, 255.0) as u8,
+ g: ((self.g as f32) * factor).clamp(0.0, 255.0) as u8,
+ b: ((self.b as f32) * factor).clamp(0.0, 255.0) as u8,
+ }
+ }
+}
+
+const SQRT: [usize; 256 * 3] = gen_sqrt_lookup();
+
+const fn gen_sqrt_lookup<const N: usize>() -> [usize; N] {
+ let mut arr = [0; N];
+ let mut i = 0;
+ while i < N {
+ arr[i] = sqrt(i as f32) as usize;
+ i += 1;
+ }
+ arr
+}
+
+const fn sqrt(x: f32) -> f32 {
+ let a = 1.0;
+ let a = (a + x / a) * 0.5;
+ let a = (a + x / a) * 0.5;
+ let a = (a + x / a) * 0.5;
+ a
+}