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
65
66
67
68
69
70
71
72
|
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 {
//? this functions is broken
// TODO dont loose accuracy
Pixel {
r: (a.r >> 1) + (b.r >> 1),
g: (a.g >> 1) + (b.g >> 1),
b: (a.b >> 1) + (b.b >> 1),
}
}
#[inline]
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
}
|
