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

use std::ops::*;
use std::iter::FromIterator;
use super::pixel::*;
use super::rgb::RGB;
use super::rgba::RGBA;

impl<T: Clone + Add> Add for RGB<T>
    where RGB<T>: FromIterator<<T as Add>::Output> {
    type Output = RGB<T>;
    fn add(self, other: RGB<T>) -> Self::Output {
        self.iter().zip(other.iter()).map(|(l,r)| l+r).collect()
    }
}

impl<T: Clone + Add> Add<RGBA<T>> for RGBA<T>
    where RGBA<T>: FromIterator<<T as Add>::Output>,
        T: From<<T as Add>::Output> {
    type Output = RGBA<T>;
    fn add(self, other: RGBA<T>) -> Self::Output {
        self.iter().zip(other.iter()).map(|(l,r)| l+r).collect()
    }
}

impl<T: Clone + Sub> Sub for RGB<T>
    where RGB<T>: FromIterator<<T as Sub>::Output> {
    type Output = RGB<T>;
    fn sub(self, other: RGB<T>) -> Self::Output {
        self.iter().zip(other.iter()).map(|(l,r)| l-r).collect()
    }
}

impl<T: Clone + Sub> Sub<RGBA<T>> for RGBA<T>
    where RGBA<T>: FromIterator<<T as Sub>::Output>,
        T: From<<T as Sub>::Output> {
    type Output = RGBA<T>;
    fn sub(self, other: RGBA<T>) -> Self::Output {
        self.iter().zip(other.iter()).map(|(l,r)| l-r).collect()
    }
}

impl<T: Clone + Copy + Add> Add<T> for RGB<T>
    where T: Add<Output=T> {
    type Output = RGB<T>;
    fn add(self, r: T) -> Self::Output {
        self.map(|l|l+r)
    }
}

impl<T: Clone + Copy + Add> Add<T> for RGBA<T>
    where T: Add<Output=T> {
    type Output = RGBA<T>;
    fn add(self, r: T) -> Self::Output {
        self.map(|l|l+r)
    }
}

impl<T: Clone + Copy + Mul> Mul<T> for RGB<T>
    where T: Mul<Output=T> {
    type Output = RGB<T>;
    fn mul(self, r: T) -> Self::Output {
        self.map(|l|l*r)
    }
}

impl<T: Clone + Copy + Mul> Mul<T> for RGBA<T>
    where T: Mul<Output=T> {
    type Output = RGBA<T>;
    fn mul(self, r: T) -> Self::Output {
        self.map(|l|l*r)
    }
}

#[test]
fn test_math() {
    assert_eq!(RGB::new(2,4,6), RGB::new(1,2,3) + RGB{r:1,g:2,b:3});
    assert_eq!(RGB::new(2.,4.,6.), RGB::new(1.,3.,5.) + 1.);
    assert_eq!(RGB::new(0.5,1.5,2.5), RGB::new(1.,3.,5.) * 0.5);

    assert_eq!(RGBA::new(2,4,6,8), RGBA::new(1,2,3,4) + RGBA{r:1,g:2,b:3,a:4});
    assert_eq!(RGBA::new(2i16,4,6,8), RGBA::new(1,3,5,7) + 1);
    assert_eq!(RGBA::new(2,4,6,8), RGBA::new(1,2,3,4) * 2);
}