undither 1.0.8

Smart filter to remove Floyd-Steinberg dithering from paletted images
Documentation 
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

use crate::acc::Similarity;
use rgb::RGB8;
use std::cell::{RefCell, RefMut};
use std::cmp::PartialEq;
use std::ops::Index;

pub struct Pal {
    pal: Vec<RGB8>,
    similarity: RefCell<Similarity>,
}

impl Pal {
    #[must_use]
    pub fn new(palette_bytes: &[u8]) -> Self {
        let pal = to_rgb(palette_bytes);
        Pal {
            similarity: RefCell::new(Similarity::new(pal.clone())),
            pal,
        }
    }

    pub fn similarity(&self) -> RefMut<'_, Similarity> {
        self.similarity.borrow_mut()
    }
}

impl Index<usize> for Pal {
    type Output = RGB8;
    fn index(&self, i: usize) -> &Self::Output {
        &self.pal[i]
    }
}

impl PartialEq for Pal {
    fn eq(&self, other: &Pal) -> bool {
        self.pal.eq(&other.pal)
    }
}

fn to_rgb(palette_bytes: &[u8]) -> Vec<RGB8> {
    palette_bytes.chunks(3).map(|byte| RGB8{r:byte[0], g:byte[1], b:byte[2]}).collect()
}

#[test]
fn paltest() {
    let a = vec![1, 2, 3];
    let b = vec![1, 2, 3];
    let c = vec![1, 1, 1];

    assert!(Pal::new(&a) == Pal::new(&a));
    assert!(Pal::new(&b) == Pal::new(&a));
    assert!(&Pal::new(&a) == &Pal::new(&b));

    assert!(Pal::new(&a) != Pal::new(&c));
}