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
use std::io::Read;
use std::io::BufReader;
use color::{ColorType};
use image::{DecodingResult, ImageDecoder, ImageResult, ImageError};
extern crate byteorder;
use self::byteorder::{BigEndian, ByteOrder};
pub struct PPMDecoder<R> {
reader: BufReader<R>,
width: u32,
height: u32,
maxwhite: u32,
}
impl<R: Read> PPMDecoder<R> {
pub fn new(r: R) -> ImageResult<PPMDecoder<R>> {
let mut buf = BufReader::new(r);
try!(PPMDecoder::read_next_string(&mut buf));
let width = try!(PPMDecoder::read_next_u32(&mut buf));
let height = try!(PPMDecoder::read_next_u32(&mut buf));
let maxwhite = try!(PPMDecoder::read_next_u32(&mut buf));
Ok(PPMDecoder {
reader: buf,
width: width,
height: height,
maxwhite: maxwhite,
})
}
fn read_next_string(reader: &mut BufReader<R>) -> ImageResult<String> {
let mut bytes = Vec::new();
for byte in reader.bytes() {
match byte {
Ok(b'\n') | Ok(b' ') | Ok(b'\r') | Ok(b'\t') => {
if bytes.len() > 0 {
break
}
},
Ok(byte) => {
bytes.push(byte);
},
Err(_) => break,
}
}
match String::from_utf8(bytes) {
Ok(s) => Ok(s),
Err(_) => Err(ImageError::FormatError("Couldn't read preamble".to_string())),
}
}
fn read_next_u32(reader: &mut BufReader<R>) -> ImageResult<u32> {
let s = try!(PPMDecoder::read_next_string(reader));
match s.parse::<u32>() {
Ok(v) => Ok(v),
Err(_) => Err(ImageError::FormatError("Couldn't read preamble".to_string())),
}
}
}
impl<R: Read> ImageDecoder for PPMDecoder<R> {
fn dimensions(&mut self) -> ImageResult<(u32, u32)> {
Ok((self.width, self.height))
}
fn colortype(&mut self) -> ImageResult<ColorType> {
match self.bytewidth() {
1 => Ok(ColorType::RGB(8)),
2 => Ok(ColorType::RGB(16)),
_ => Err(ImageError::FormatError("Don't know how to decode PPM with more than 16 bits".to_string())),
}
}
fn row_len(&mut self) -> ImageResult<usize> {
Ok((self.width*3*self.bytewidth()) as usize)
}
fn read_scanline(&mut self, _buf: &mut [u8]) -> ImageResult<u32> {
unimplemented!();
}
fn read_image(&mut self) -> ImageResult<DecodingResult> {
let mut data = vec![0 as u8; (self.width*self.height*3*self.bytewidth()) as usize];
match self.reader.read_exact(&mut data) {
Ok(_) => {},
Err(e) => return Err(ImageError::IoError(e)),
};
if self.bytewidth() == 1 {
Ok(DecodingResult::U8(data))
} else {
let mut out = vec![0 as u16; (self.width*self.height*3) as usize];
for (o, i) in out.chunks_mut(1).zip(data.chunks(2)) {
o[0] = BigEndian::read_u16(i);
}
Ok(DecodingResult::U16(out))
}
}
}
impl<R: Read> PPMDecoder<R> {
fn bytewidth(&self) -> u32 {
if self.maxwhite < 256 { 1 } else { 2 }
}
}