diff options
author | metamuffin <metamuffin@disroot.org> | 2022-12-07 17:55:27 +0100 |
---|---|---|
committer | metamuffin <metamuffin@disroot.org> | 2022-12-07 17:55:27 +0100 |
commit | 8ca219c6b0d5448fd4529713ccd093e89de4e252 (patch) | |
tree | 8fe02fd35ae337766d6f8e7533da170830162d14 /evc/src/helpers/pixel.rs | |
parent | a99674b911cb9b2fa398ccf61830d5933ccaf931 (diff) | |
download | video-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.rs | 67 |
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 +} |