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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
use bitstream_io::{BigEndian, BitRead, BitReader, Endianness, LittleEndian};
#[cfg(not(feature = "std"))]
use core2::io::{self, Cursor, SeekFrom};
#[cfg(feature = "std")]
use std::io::{self, Cursor, SeekFrom};
#[test]
fn test_reader_pos_be() -> io::Result<()> {
test_reader_pos::<BigEndian>()
}
#[test]
fn test_reader_pos_le() -> io::Result<()> {
test_reader_pos::<LittleEndian>()
}
fn test_reader_pos<E: Endianness>() -> io::Result<()> {
let actual_data: [u8; 7] = [
0b1000_1110,
0b1000_1110,
0b1000_1110,
0b1000_1110,
0b1000_1110,
0b1000_1110,
0b1000_1110,
];
let mut r: BitReader<_, E> = BitReader::new(Cursor::new(&actual_data));
assert_eq!(r.position_in_bits()?, 0);
r.read_bit()?;
r.read_bit()?;
assert_eq!(r.position_in_bits()?, 2);
let _: u32 = r.read_var(5)?;
assert_eq!(r.position_in_bits()?, 7);
let _: u32 = r.read_var(4)?;
assert_eq!(r.position_in_bits()?, 11);
let mut buf = [0u8; 2];
r.read_bytes(&mut buf)?;
assert_eq!(r.position_in_bits()?, 27);
r.read_bit()?;
r.read_bit()?;
r.read_bit()?;
r.read_bit()?;
r.read_bit()?;
r.read_bit()?;
r.read_bit()?;
let _: i32 = r.read_signed_var(9)?;
assert_eq!(r.position_in_bits()?, 43);
let _: i32 = r.read_signed_var(5)?;
assert_eq!(r.position_in_bits()?, 48);
Ok(())
}
#[test]
pub fn test_reader_seek_start() -> io::Result<()> {
let actual_data: [u8; 4] = [0xB1, 0xED, 0x3B, 0xC1];
let mut r = BitReader::endian(Cursor::new(&actual_data), BigEndian);
r.seek_bits(SeekFrom::Start(0))?;
assert_eq!(r.position_in_bits()?, 0);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.position_in_bits()?, 8);
r.seek_bits(SeekFrom::Start(2))?;
assert_eq!(r.position_in_bits()?, 2);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.position_in_bits()?, 8);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.position_in_bits()?, 10);
r.seek_bits(SeekFrom::Start(7))?;
assert_eq!(r.position_in_bits()?, 7);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, false);
Ok(())
}
#[test]
pub fn test_reader_seek_current() -> io::Result<()> {
let actual_data: [u8; 4] = [0xB1, 0xED, 0x3B, 0xC1];
let mut r = BitReader::endian(Cursor::new(&actual_data), BigEndian);
r.seek_bits(SeekFrom::Current(2))?;
assert_eq!(r.position_in_bits()?, 2);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, false);
let _: i32 = r.read_signed_var(11)?;
assert_eq!(r.position_in_bits()?, 17);
r.seek_bits(SeekFrom::Current(-3))?;
assert_eq!(r.position_in_bits()?, 14);
r.skip(10)?;
assert_eq!(r.position_in_bits()?, 24);
r.seek_bits(SeekFrom::Current(0))?;
assert_eq!(r.position_in_bits()?, 24);
Ok(())
}
#[test]
pub fn test_reader_seek_end() -> io::Result<()> {
let actual_data: [u8; 4] = [0xB1, 0xED, 0x3B, 0xC1];
let mut r = BitReader::endian(Cursor::new(&actual_data), BigEndian);
r.seek_bits(SeekFrom::End(7))?;
assert_eq!(r.position_in_bits()?, 25);
assert_eq!(r.read_bit()?, true);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.read_bit()?, false);
assert_eq!(r.position_in_bits()?, 29);
r.seek_bits(SeekFrom::End(0))?;
assert_eq!(r.position_in_bits()?, 32);
Ok(())
}
|
