1use std::io::Read;
2use std::io::Result as Res;
3
4use crate::{BigEndian, Deserialize, Endianness, LittleEndian};
5
6pub trait ERead<E: Endianness>: Sized {
28 fn read <D: Deserialize<E, Self>>(&mut self) -> Res<D> { D::deserialize(self) }
34 fn read_be<D: Deserialize<BigEndian, Self>>(&mut self) -> Res<D> { D::deserialize(self) }
36 fn read_le<D: Deserialize<LittleEndian, Self>>(&mut self) -> Res<D> { D::deserialize(self) }
38}
39
40pub trait BERead: Sized {
48 fn read <D: Deserialize<BigEndian, Self>>(&mut self) -> Res<D> { D::deserialize(self) }
49 fn read_be<D: Deserialize<BigEndian, Self>>(&mut self) -> Res<D> { D::deserialize(self) }
50 fn read_le<D: Deserialize<LittleEndian, Self>>(&mut self) -> Res<D> { D::deserialize(self) }
51}
52
53pub trait LERead: Sized {
61 fn read <D: Deserialize<LittleEndian, Self>>(&mut self) -> Res<D> { D::deserialize(self) }
62 fn read_be<D: Deserialize<BigEndian, Self>>(&mut self) -> Res<D> { D::deserialize(self) }
63 fn read_le<D: Deserialize<LittleEndian, Self>>(&mut self) -> Res<D> { D::deserialize(self) }
64}
65
66impl<R: Read, E: Endianness> ERead<E> for R {}
67impl<R: Read> BERead for R {}
68impl<R: Read> LERead for R {}
69
70#[cfg(test)]
71mod tests {
72 const DATA: &[u8] = b"\xba\xad";
73
74 #[test]
75 fn read_be_forced() {
76 use super::LERead;
77 let mut reader = &DATA[..];
78 let val: u16 = reader.read_be().unwrap();
79 assert_eq!(val, 0xbaad);
80 }
81
82 #[test]
83 fn read_le_forced() {
84 use crate::BERead;
85 let mut reader = &DATA[..];
86 let val: u16 = reader.read_le().unwrap();
87 assert_eq!(val, 0xadba);
88 }
89}