image-effects 0.1.0

An assortment of effects that can be applied to an image with the help of `image`.
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151

use image::{ImageBuffer, Rgb, DynamicImage, Rgba, Frame};

use crate::utils::{image::{RgbImageRepr, RgbPixelRepr, get_dimensions_of_matrix, RgbaImageRepr, RgbaPixelRepr}, numops::map_to_2d};

/// Defines an effect that can be applied onto `T`.
/// 
/// Implementing this auto-implements `Affectable<T, F>` for `T`,
/// where `F` is this effect.
pub trait Effect<T> {
    /// Affects `T` using `self`.
    fn affect(&self, item: T) -> T;
}

/// Defines something that can be affected.
/// 
/// This is auto-implemented for any `T` and `F` where `F`
/// is an effect that can be applied on `T`.
/// 
/// As a result it should almost never be implemented directly, like `Into<T>` in the
/// standard library.
pub trait Affectable<T, F: Effect<T>> {

    /// Applies an effect on `self`.
    fn apply(self, effect: &F) -> Self;
}

impl<T, F> Affectable<T, F> for T where F: Effect<T> {
    fn apply(self, effect: &F) -> Self {
        effect.affect(self)
    }
}

impl<F> Effect<RgbaPixelRepr> for F where F: Effect<RgbPixelRepr> {
    fn affect(&self, item: RgbaPixelRepr) -> RgbaPixelRepr {
        let [r, g, b, a] = item;
        let [r, g, b] = self.affect([r, g, b]);
        [r, g, b, a]
    }
}

impl<F> Effect<RgbImageRepr> for F where F: Effect<RgbPixelRepr> {
    fn affect(&self, mut item: RgbImageRepr) -> RgbImageRepr {
        for row in item.iter_mut() {
            for pixel in row.iter_mut() {
                *pixel = self.affect(*pixel);
            }
        }
        item
    }
}

impl<F> Effect<RgbaImageRepr> for F where F: Effect<RgbImageRepr> {
    fn affect(&self, item: RgbaImageRepr) -> RgbaImageRepr {
        let (xs, ys) = get_dimensions_of_matrix(&item);

        let mut rgb_repr = vec![vec![[0_u8; 3]; xs]; ys];
        let mut output = vec![vec![[0_u8; 4]; xs]; ys];

        for y in 0..ys {
            for x in 0..xs {
                let [r, g, b, _] = item[y][x];
                rgb_repr[y][x] = [r, g, b];
            }
        }

        let rgb_repr = self.affect(rgb_repr);

        for y in 0..ys {
            for x in 0..xs {
                let [r, g, b] = rgb_repr[y][x];
                output[y][x] = [r, g, b, item[y][x][3]];
            }
        }

        output
    }
}

impl<F> Effect<ImageBuffer<Rgb<u8>, Vec<u8>>> for F where F: Effect<RgbImageRepr> {
    fn affect(&self, item: ImageBuffer<Rgb<u8>, Vec<u8>>) -> ImageBuffer<Rgb<u8>, Vec<u8>> {
        let (xs, ys) = item.dimensions();
        let (xs, ys) = (xs as usize, ys as usize);
    
        let mut img_matrix = vec![vec![[0_u8; 3]; xs]; ys];
    
        for (i, pixel) in item.pixels().into_iter().enumerate() {
           let (x, y) = map_to_2d(i, xs);
           img_matrix[y][x] = pixel.0;
        }
    
        img_matrix = self.affect(img_matrix);

        let (xdim, ydim) = get_dimensions_of_matrix(&img_matrix);
    
        ImageBuffer::from_fn(xdim as u32, ydim as u32, |x, y| {
            image::Rgb(img_matrix[y as usize][x as usize])
        })
    }
}

impl<F> Effect<ImageBuffer<Rgba<u8>, Vec<u8>>> for F where F: Effect<RgbaImageRepr> {
    fn affect(&self, item: ImageBuffer<Rgba<u8>, Vec<u8>>) -> ImageBuffer<Rgba<u8>, Vec<u8>> {
        let (xs, ys) = item.dimensions();
        let (xs, ys) = (xs as usize, ys as usize);
    
        let mut img_matrix = vec![vec![[0_u8; 4]; xs]; ys];
    
        for (i, pixel) in item.pixels().into_iter().enumerate() {
           let (x, y) = map_to_2d(i, xs);
           img_matrix[y][x] = pixel.0;
        }
    
        img_matrix = self.affect(img_matrix);

        let (xdim, ydim) = get_dimensions_of_matrix(&img_matrix);
    
        ImageBuffer::from_fn(xdim as u32, ydim as u32, |x, y| {
            image::Rgba(img_matrix[y as usize][x as usize])
        })
    }
}

impl<F> Effect<DynamicImage> for F where F: 
    Effect<ImageBuffer<Rgb<u8>, Vec<u8>>> 
    + Effect<ImageBuffer<Rgba<u8>, Vec<u8>>>
{
    fn affect(&self, item: DynamicImage) -> DynamicImage {
        match item {
            DynamicImage::ImageRgb8(img) => {
                DynamicImage::from(self.affect(img))
            },
            DynamicImage::ImageRgba8(img) => {
                DynamicImage::from(self.affect(img))
            },
            _ => {
                DynamicImage::ImageRgb8(self.affect(item.into_rgb8()))
            }
        }
    }
} 

impl<F> Effect<Frame> for F where F: Effect<ImageBuffer<Rgba<u8>, Vec<u8>>> {
    fn affect(&self, item: Frame) -> Frame {
        let left = item.left();
        let top = item.top();
        let delay = item.delay();

        let new_buf = self.affect(item.into_buffer());
        Frame::from_parts(new_buf, left, top, delay)
    }
}