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
pub use self::encoder::PPMEncoder as PPMEncoder;
pub use self::decoder::PPMDecoder as PPMDecoder;
mod encoder;
mod decoder;
#[cfg(test)]
mod test {
use color::ColorType;
use image::{ImageDecoder, DecodingResult};
#[test]
fn test_roundtrip_ppm() {
let buf: [u8; 27] = [
0, 0, 0,
0, 0, 255,
0, 255, 0,
0, 255, 255,
255, 0, 0,
255, 0, 255,
255, 255, 0,
255, 255, 255,
255, 255, 255,
];
let mut stream = Vec::<u8>::new();
{
let mut encoder = super::PPMEncoder::new(&mut stream);
match encoder.encode(&buf, 3, 3, ColorType::RGB(8)) {
Ok(_) => {},
Err(_) => panic!("PPM encoder failed"),
};
}
let mut decoder = match super::PPMDecoder::new(&stream[..]) {
Ok(img) => img,
Err(e) => panic!("PPM decoder failed with {}", e),
};
match decoder.read_image() {
Ok(DecodingResult::U8(vec)) => {
assert_eq!(&buf[..], &vec[..]);
},
r => {
panic!("PPM: Got a strange image result {:?}", r);
}
}
}
#[test]
fn test_roundtrip_ppm_16bit() {
let buf: [u16; 27] = [
0, 0, 0,
0, 0, 65535,
0, 65535, 0,
0, 65535, 65535,
65535, 0, 0,
65535, 0, 65535,
65535, 65535, 0,
65535, 65535, 65535,
1000, 2000, 3000,
];
let mut bytebuf = [0 as u8; 54];
for (o, i) in bytebuf.chunks_mut(2).zip(buf.iter()) {
o[0] = (i >> 8) as u8;
o[1] = (i & 0xff) as u8;
}
let mut stream = Vec::<u8>::new();
{
let mut encoder = super::PPMEncoder::new(&mut stream);
match encoder.encode(&bytebuf, 3, 3, ColorType::RGB(16)) {
Ok(_) => {},
Err(_) => panic!("PPM encoder failed"),
};
}
let mut decoder = match super::PPMDecoder::new(&stream[..]) {
Ok(img) => img,
Err(e) => panic!("PPM decoder failed with {}", e),
};
match decoder.read_image() {
Ok(DecodingResult::U16(vec)) => {
assert_eq!(&buf[..], &vec[..]);
},
r => {
panic!("PPM: Got a strange image result {:?}", r);
}
}
}
}