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

use std::vec::Vec;
use std::marker::PhantomData;
use {Pixel, Image, ImageVal};

/// TODO: Struct documentation
#[derive(Clone)]
pub struct ImageBuffer<PixelP>
    where PixelP: Pixel
{
    width: u32,
    height: u32,
    pitch: u32,
    raw_data: Vec<u8>,
    _marker: PhantomData<PixelP>,
}

// Abbreviation
pub type ImageBufferVal<PixelP> where PixelP: Pixel = ImageVal<ImageBuffer<PixelP>>;

/// TODO: Impl documentation
impl<PixelP> ImageBufferVal<PixelP>
    where PixelP: Pixel
{
    pub fn new_with_size(width: u32, height: u32) -> ImageBufferVal<PixelP> {
        Self::new_with_size_and_pitch(width,
                                      height,
                                      PixelP::calc_minimum_pitch(width, height) as u32)
    }

    pub fn new_with_size_and_pitch(width: u32, height: u32, pitch: u32) -> ImageBufferVal<PixelP> {
        let size_in_bytes = PixelP::calc_size_in_bytes(width, height, pitch)
            .expect(&format!("Invalid combindation of width ({:}), height ({:}) and pitch ({:}) \
                             for this pixel type",
                             width,
                             height,
                             pitch));

        let data = vec![0; size_in_bytes];
        ImageVal(ImageBuffer {
            width: width,
            height: height,
            pitch: pitch,
            raw_data: data,
            _marker: PhantomData,
        })
    }
}

// Implement Image trait for ImageBuffer
/// TODO: Impl documentation
impl<PixelP> Image for ImageBuffer<PixelP>
    where PixelP: Pixel
{
    type PixelT = PixelP;

    fn get_size_in_bytes(&self) -> usize {
        PixelP::calc_size_in_bytes(self.width, self.height, self.pitch)
            .expect(&format!("Invalid combindation of width ({:}), height ({:}) and pitch ({:}) \
                             for this pixel type - for an image ALREADY CREATED",
                             self.width,
                             self.height,
                             self.pitch))
    }
    fn load_from_raw_buffer(&mut self, buffer: &[u8]) {
        assert_eq!(self.get_size_in_bytes(), buffer.len());
        self.raw_data = Vec::from(buffer);
    }
    fn write_into_raw_buffer(&self, buffer: &mut [u8]) {
        assert_eq!(self.get_size_in_bytes(), buffer.len());
        buffer.clone_from_slice(&self.raw_data[..]);
    }
    fn width(&self) -> u32 {
        self.width
    }
    fn height(&self) -> u32 {
        self.height
    }
    fn pitch(&self) -> u32 {
        self.pitch
    }
    fn get_pixel(&self, x: u32, y: u32) -> Option<PixelP> {
        if x < self.width || y < self.height {
            Some(PixelP::load_from_raw_buffer(x, y, self.pitch, &self.raw_data))
        } else {
            None
        }
    }
    fn set_pixel(&mut self, x: u32, y: u32, value: PixelP) {
        assert!(x < self.width || y < self.height);
        value.write_into_raw_buffer(x, y, self.pitch, &mut self.raw_data)
    }
}