spritely 0.1.0

Sprite-related utils for the Image crate.
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use std::{
    collections::BTreeSet,
    hash::Hash,
    ops::{Deref, DerefMut},
};

use image::{ImageBuffer, Pixel};

use crate::color::Color;
use crate::palette::Palette;

pub trait ImageColors<P: Pixel> {
    fn colors(&self) -> Vec<Color<P>>;
}

pub trait ImageRecolorMut<P: Pixel> {
    fn recolor(&mut self, palette: &Palette<P>);
    fn into_recolored(self, palette: &Palette<P>) -> Self;
}

pub trait ImageRecolorClone<P: Pixel> {
    fn clone_recolored(&self, palette: &Palette<P>) -> Self;
}

impl<P, Container> ImageColors<P> for ImageBuffer<P, Container>
where
    P: Pixel,
    Container: Deref<Target = [P::Subpixel]>,
{
    fn colors(&self) -> Vec<Color<P>> {
        let mut set = BTreeSet::new();

        for pixel in self.pixels() {
            set.insert(Color::from(*pixel));
        }

        set.into_iter().collect()
    }
}

impl<P, Container> ImageRecolorMut<P> for ImageBuffer<P, Container>
where
    P: Pixel,
    P::Subpixel: Hash,
    Container: DerefMut<Target = [P::Subpixel]>,
{
    fn recolor(&mut self, palette: &Palette<P>) {
        for pixel in self.pixels_mut() {
            if let Some(new) = palette.get(&Color::from(*pixel)) {
                pixel.apply2(&new.value(), |_, other| other);
            }
        }
    }

    fn into_recolored(mut self, palette: &Palette<P>) -> Self {
        self.recolor(palette);
        self
    }
}

impl<P, C> ImageRecolorClone<P> for ImageBuffer<P, C>
where
    P: Pixel,
    P::Subpixel: Hash,
    C: Deref<Target = [P::Subpixel]> + DerefMut<Target = [P::Subpixel]> + Clone,
{
    fn clone_recolored(&self, palette: &Palette<P>) -> Self {
        let mut result = self.clone();
        result.recolor(palette);
        result
    }
}