]>
Commit | Line | Data |
---|---|---|
a4bd7707 KS |
1 | //! Codebook support for bitstream reader. |
2 | //! | |
3 | //! Codebook is a set of unique bit strings and values assigned to them. | |
4 | //! Since there are many ways to define codebook, this implementation employs [`CodebookDescReader`] trait to provide codebook generator with the codes. | |
5 | //! Please also pay attention to the codebook creation mode: if bitstream reader reads bits starting from most significant bit first then you should use [`MSB`] mode and [`LSB`] mode otherwise. | |
6 | //! | |
7 | //! # Examples | |
8 | //! | |
9 | //! Create a codebook from arrays with codeword descriptions: | |
10 | //! ``` | |
11 | //! use nihav_core::io::codebook::{ShortCodebookDesc, ShortCodebookDescReader, Codebook, CodebookMode}; | |
12 | //! | |
13 | //! let cb_desc: Vec<ShortCodebookDesc> = vec!( | |
14 | //! ShortCodebookDesc { code: 0b00, bits: 2 }, | |
15 | //! ShortCodebookDesc { code: 0, bits: 0 }, | |
16 | //! ShortCodebookDesc { code: 0b01, bits: 2 }, | |
17 | //! ShortCodebookDesc { code: 0b1, bits: 1 }); | |
18 | //! let mut cr = ShortCodebookDescReader::new(cb_desc); | |
19 | //! let cb = Codebook::new(&mut cr, CodebookMode::LSB).unwrap(); | |
20 | //! ``` | |
21 | //! | |
22 | //! Create a codebook using more flexible [`TableCodebookDescReader`] approach. | |
23 | //! This will create a codebook for the following set: `1` -> -2, `01` -> -1, `001` -> 0, `0001` -> 1, `00001` -> 2. | |
24 | //! ``` | |
25 | //! use nihav_core::io::codebook::{TableCodebookDescReader, Codebook, CodebookMode}; | |
26 | //! | |
27 | //! fn map_cb_index(index: usize) -> i16 { (index as i16) - 2 } | |
28 | //! const CB_BITS: [u8; 5] = [ 1, 2, 3, 4, 5 ]; | |
29 | //! const CB_CODES: [u8; 5] = [ 1, 1, 1, 1, 1 ]; | |
30 | //! | |
31 | //! let mut tcr = TableCodebookDescReader::new(&CB_CODES, &CB_BITS, map_cb_index); | |
32 | //! let cb = Codebook::new(&mut tcr, CodebookMode::MSB).unwrap(); | |
33 | //! ``` | |
34 | //! | |
35 | //! Read value using a codebook: | |
36 | //! ```no_run | |
37 | //! use nihav_core::io::bitreader::BitReader; | |
38 | //! use nihav_core::io::codebook::{Codebook, CodebookReader, CodebookMode}; | |
39 | //! # use nihav_core::io::codebook::{ShortCodebookDesc, ShortCodebookDescReader, CodebookDescReader, CodebookResult}; | |
40 | //! | |
41 | //! # fn foo(br: &mut BitReader) -> CodebookResult<()> { | |
42 | //! # let mut cr = ShortCodebookDescReader::new(vec![ShortCodebookDesc { code: 0b00, bits: 2 }]); | |
43 | //! let cb = Codebook::new(&mut cr, CodebookMode::MSB).unwrap(); | |
44 | //! let value = br.read_cb(&cb)?; | |
45 | //! # Ok(()) | |
46 | //! # } | |
47 | //! ``` | |
48 | //! | |
49 | //! [`MSB`]: ./enum.CodebookMode.html#variant.MSB | |
50 | //! [`LSB`]: ./enum.CodebookMode.html#variant.LSB | |
51 | //! [`CodebookDescReader`]: ./trait.CodebookDescReader.html | |
52 | //! [`TableCodebookDescReader`]: ./struct.TableCodebookDescReader.html | |
53 | ||
0ca65ffd KS |
54 | use std::collections::HashMap; |
55 | use std::cmp::{max, min}; | |
aca89041 | 56 | use super::bitreader::BitReader; |
4667915a | 57 | |
a4bd7707 | 58 | /// A list specifying general codebook operations errors. |
4667915a KS |
59 | #[derive(Debug)] |
60 | pub enum CodebookError { | |
a4bd7707 | 61 | /// Codebook description contains errors. |
4667915a | 62 | InvalidCodebook, |
a4bd7707 | 63 | /// Could not allocate memory for codebook. |
4667915a | 64 | MemoryError, |
a4bd7707 | 65 | /// Bitstream contains a sequence not present in codebook. |
4667915a KS |
66 | InvalidCode, |
67 | } | |
68 | ||
a4bd7707 | 69 | /// Codebook operation modes. |
0ca65ffd KS |
70 | #[derive(Debug, Copy, Clone)] |
71 | pub enum CodebookMode { | |
a4bd7707 | 72 | /// Codes in the codebook should be read most significant bit first. |
0ca65ffd | 73 | MSB, |
a4bd7707 | 74 | /// Codes in the codebook should be read least significant bit first. |
0ca65ffd KS |
75 | LSB, |
76 | } | |
77 | ||
a4bd7707 KS |
78 | /// A specialised `Result` type for codebook operations. |
79 | pub type CodebookResult<T> = Result<T, CodebookError>; | |
4667915a | 80 | |
a4bd7707 KS |
81 | /// Codebook description for `(code bits, code length, code value)` triplet. |
82 | /// | |
83 | /// This should be used to create a list of codeword definitions for [`FullCodebookDescReader`]. | |
84 | /// | |
85 | /// [`FullCodebookDescReader`]: ./struct.FullCodebookDescReader.html | |
4667915a | 86 | pub struct FullCodebookDesc<S> { |
a4bd7707 | 87 | /// Codeword bits. |
9a62b981 | 88 | pub code: u32, |
a4bd7707 | 89 | /// Codeword length. |
9a62b981 | 90 | pub bits: u8, |
a4bd7707 | 91 | /// Codeword value (symbol). |
9a62b981 | 92 | pub sym: S, |
4667915a KS |
93 | } |
94 | ||
a4bd7707 KS |
95 | /// Codebook description for `(code bits, code length)` pair with array index being used as codeword value. |
96 | /// | |
97 | /// This should be used to create a list of codeword definitions for [`ShortCodebookDescReader`]. | |
98 | /// | |
99 | /// [`ShortCodebookDescReader`]: ./struct.ShortCodebookDescReader.html | |
4667915a | 100 | pub struct ShortCodebookDesc { |
a4bd7707 | 101 | /// Codeword bits. |
9a62b981 | 102 | pub code: u32, |
a4bd7707 | 103 | /// Codeword length. |
9a62b981 | 104 | pub bits: u8, |
4667915a KS |
105 | } |
106 | ||
a4bd7707 KS |
107 | /// The interface for providing a list of codeword definitions to the codebook creator. |
108 | /// | |
109 | /// The structure implementing this trait should be able to provide the total number of defined codewords and their bits and values. [`ShortCodebookDescReader`] or [`TableCodebookDescReader`] are some examples of such implementation. | |
110 | /// Codeword definitions with zero length are ignored (those may be used to create sparse codebook definitions though). | |
111 | /// | |
112 | /// [`ShortCodebookDescReader`]: ./struct.ShortCodebookDescReader.html | |
113 | /// [`TableCodebookDescReader`]: ./struct.TableCodebookDescReader.html | |
4667915a | 114 | pub trait CodebookDescReader<S> { |
a4bd7707 | 115 | /// Returns the codeword length for the provided index. |
4667915a | 116 | fn bits(&mut self, idx: usize) -> u8; |
a4bd7707 | 117 | /// Returns the codeword bits for the provided index. |
4667915a | 118 | fn code(&mut self, idx: usize) -> u32; |
a4bd7707 | 119 | /// Returns the codeword value (aka codeword symbol) for the provided index. |
4667915a | 120 | fn sym (&mut self, idx: usize) -> S; |
a4bd7707 | 121 | /// Returns the total number of defined codewords. |
4667915a KS |
122 | fn len (&mut self) -> usize; |
123 | } | |
124 | ||
a4bd7707 | 125 | /// The codebook structure for code reading. |
4667915a KS |
126 | #[allow(dead_code)] |
127 | pub struct Codebook<S> { | |
128 | table: Vec<u32>, | |
129 | syms: Vec<S>, | |
130 | lut_bits: u8, | |
131 | } | |
132 | ||
a4bd7707 | 133 | /// Trait allowing bitreader to use codebook for decoding bit sequences. |
4667915a | 134 | pub trait CodebookReader<S> { |
a4bd7707 KS |
135 | /// Reads the codeword from the bitstream and returns its value (or [`InvalidCode`] on error). |
136 | /// | |
137 | /// [`InvalidCode`]: ./enum.CodebookError.html#variant.InvalidCode | |
4667915a KS |
138 | fn read_cb(&mut self, cb: &Codebook<S>) -> CodebookResult<S>; |
139 | } | |
140 | ||
0ca65ffd KS |
141 | const TABLE_FILL_VALUE: u32 = 0x7F; |
142 | const MAX_LUT_BITS: u8 = 10; | |
143 | ||
144 | fn fill_lut_msb(table: &mut Vec<u32>, off: usize, | |
145 | code: u32, bits: u8, lut_bits: u8, symidx: u32, esc: bool) { | |
146 | if !esc { | |
147 | let fill_len = lut_bits - bits; | |
148 | let fill_size = 1 << fill_len; | |
149 | let fill_code = code << (lut_bits - bits); | |
e243ceb4 | 150 | let lut_value = (symidx << 8) | u32::from(bits); |
0ca65ffd KS |
151 | for j in 0..fill_size { |
152 | let idx = (fill_code + j) as usize; | |
153 | table[idx + off] = lut_value; | |
154 | } | |
155 | } else { | |
156 | let idx = (code as usize) + off; | |
e243ceb4 | 157 | table[idx] = (symidx << 8) | 0x80 | u32::from(bits); |
0ca65ffd KS |
158 | } |
159 | } | |
160 | ||
161 | fn fill_lut_lsb(table: &mut Vec<u32>, off: usize, | |
162 | code: u32, bits: u8, lut_bits: u8, symidx: u32, esc: bool) { | |
163 | if !esc { | |
164 | let fill_len = lut_bits - bits; | |
165 | let fill_size = 1 << fill_len; | |
166 | let fill_code = code; | |
167 | let step = lut_bits - fill_len; | |
168 | for j in 0..fill_size { | |
169 | let idx = (fill_code + (j << step)) as usize; | |
e243ceb4 | 170 | table[idx + off] = (symidx << 8) | u32::from(bits); |
0ca65ffd KS |
171 | } |
172 | } else { | |
173 | let idx = (code as usize) + off; | |
e243ceb4 | 174 | table[idx] = (symidx << 8) | 0x80 | u32::from(bits); |
0ca65ffd KS |
175 | } |
176 | } | |
177 | ||
178 | fn fill_lut(table: &mut Vec<u32>, mode: CodebookMode, | |
179 | off: usize, code: u32, bits: u8, lut_bits: u8, symidx: u32, esc: bool) -> bool { | |
180 | match mode { | |
181 | CodebookMode::MSB => fill_lut_msb(table, off, code, bits, lut_bits, symidx, esc), | |
182 | CodebookMode::LSB => fill_lut_lsb(table, off, code, bits, lut_bits, symidx, esc), | |
183 | }; | |
184 | bits > lut_bits | |
185 | } | |
186 | ||
187 | fn resize_table(table: &mut Vec<u32>, bits: u8) -> CodebookResult<u32> { | |
188 | let add_size = (1 << bits) as usize; | |
189 | table.reserve(add_size); | |
190 | let cur_off = table.len() as u32; | |
191 | let new_size = table.len() + add_size; | |
192 | if table.capacity() < new_size { return Err(CodebookError::MemoryError); } | |
193 | table.resize(new_size, TABLE_FILL_VALUE); | |
194 | Ok(cur_off) | |
195 | } | |
196 | ||
197 | ||
198 | fn extract_lut_part(code: u32, bits: u8, lut_bits: u8, mode: CodebookMode) -> u32 { | |
199 | match mode { | |
200 | CodebookMode::MSB => code >> (bits - lut_bits), | |
201 | CodebookMode::LSB => code & ((1 << lut_bits) - 1), | |
202 | } | |
203 | } | |
204 | ||
205 | fn extract_esc_part(code: u32, bits: u8, lut_bits: u8, mode: CodebookMode) -> u32 { | |
206 | match mode { | |
207 | CodebookMode::MSB => code & ((1 << (bits - lut_bits)) - 1), | |
208 | CodebookMode::LSB => code >> lut_bits, | |
209 | } | |
210 | } | |
211 | ||
212 | #[derive(Clone,Copy)] | |
213 | struct Code { | |
214 | code: u32, | |
215 | bits: u8, | |
216 | idx: usize, | |
217 | } | |
218 | ||
219 | struct CodeBucket { | |
220 | maxlen: u8, | |
221 | offset: usize, | |
222 | codes: Vec<Code>, | |
223 | } | |
224 | ||
225 | impl CodeBucket { | |
226 | fn new() -> Self { | |
227 | CodeBucket { maxlen: 0, offset: 0, codes: Vec::new() } | |
228 | } | |
229 | fn add_code(&mut self, c: Code) { | |
230 | if c.bits > self.maxlen { self.maxlen = c.bits; } | |
231 | self.codes.push(c); | |
232 | } | |
233 | } | |
234 | ||
235 | type EscapeCodes = HashMap<u32, CodeBucket>; | |
236 | ||
237 | fn add_esc_code(cc: &mut EscapeCodes, key: u32, code: u32, bits: u8, idx: usize) { | |
e243ceb4 | 238 | cc.entry(key).or_insert_with(CodeBucket::new); |
0ca65ffd KS |
239 | let b = cc.get_mut(&key); |
240 | if let Some(bucket) = b { | |
e243ceb4 | 241 | bucket.add_code(Code {code, bits, idx }); |
0ca65ffd KS |
242 | } else { panic!("no bucket when expected!"); } |
243 | } | |
244 | ||
245 | fn build_esc_lut(table: &mut Vec<u32>, | |
246 | mode: CodebookMode, | |
247 | bucket: &CodeBucket) -> CodebookResult<()> { | |
248 | let mut escape_list: EscapeCodes = HashMap::new(); | |
249 | let maxlen = if bucket.maxlen > MAX_LUT_BITS { MAX_LUT_BITS } else { bucket.maxlen }; | |
250 | ||
251 | for code in &bucket.codes { | |
252 | let bits = code.bits; | |
26af5ca8 | 253 | if code.bits <= MAX_LUT_BITS { |
0ca65ffd KS |
254 | fill_lut(table, mode, bucket.offset, code.code, bits, |
255 | maxlen, code.idx as u32, false); | |
256 | } else { | |
257 | let ckey = extract_lut_part(code.code, bits, MAX_LUT_BITS, mode); | |
258 | let cval = extract_esc_part(code.code, bits, MAX_LUT_BITS, mode); | |
259 | add_esc_code(&mut escape_list, ckey, cval, bits - MAX_LUT_BITS, code.idx); | |
260 | } | |
261 | } | |
262 | ||
263 | let cur_offset = bucket.offset; | |
264 | for (ckey, sec_bucket) in &mut escape_list { | |
265 | let key = *ckey as u32; | |
266 | let maxlen = min(sec_bucket.maxlen, MAX_LUT_BITS); | |
267 | let new_off = resize_table(table, maxlen)?; | |
268 | fill_lut(table, mode, cur_offset, key, maxlen, | |
269 | MAX_LUT_BITS, new_off, true); | |
270 | sec_bucket.offset = new_off as usize; | |
271 | } | |
272 | ||
e243ceb4 | 273 | for sec_bucket in escape_list.values() { |
0ca65ffd KS |
274 | build_esc_lut(table, mode, sec_bucket)?; |
275 | } | |
1a151e53 | 276 | |
0ca65ffd KS |
277 | Ok(()) |
278 | } | |
279 | ||
4667915a | 280 | impl<S: Copy> Codebook<S> { |
0ca65ffd | 281 | |
a4bd7707 | 282 | /// Constructs a new `Codebook` instance using provided codebook description and mode. |
0ca65ffd | 283 | pub fn new(cb: &mut CodebookDescReader<S>, mode: CodebookMode) -> CodebookResult<Self> { |
4667915a KS |
284 | let mut maxbits = 0; |
285 | let mut nnz = 0; | |
0ca65ffd KS |
286 | let mut escape_list: EscapeCodes = HashMap::new(); |
287 | ||
288 | let mut symidx: usize = 0; | |
4667915a KS |
289 | for i in 0..cb.len() { |
290 | let bits = cb.bits(i); | |
26488721 | 291 | if bits > 0 { |
e243ceb4 | 292 | nnz += 1; |
26488721 KS |
293 | if cb.code(i) >= (1 << bits) { |
294 | return Err(CodebookError::InvalidCodebook); | |
295 | } | |
296 | } | |
0ca65ffd KS |
297 | maxbits = max(bits, maxbits); |
298 | if bits > MAX_LUT_BITS { | |
299 | let code = cb.code(i); | |
300 | let ckey = extract_lut_part(code, bits, MAX_LUT_BITS, mode); | |
301 | let cval = extract_esc_part(code, bits, MAX_LUT_BITS, mode); | |
302 | add_esc_code(&mut escape_list, ckey, cval, bits - MAX_LUT_BITS, symidx); | |
4667915a | 303 | } |
e243ceb4 | 304 | if bits > 0 { symidx += 1; } |
4667915a KS |
305 | } |
306 | if maxbits == 0 { return Err(CodebookError::InvalidCodebook); } | |
307 | ||
0ca65ffd KS |
308 | if maxbits > MAX_LUT_BITS { maxbits = MAX_LUT_BITS; } |
309 | ||
4667915a | 310 | let tab_len = 1 << maxbits; |
0ca65ffd KS |
311 | let mut table: Vec<u32> = Vec::with_capacity(tab_len); |
312 | let mut syms: Vec<S> = Vec::with_capacity(nnz); | |
4667915a | 313 | if table.capacity() < tab_len { return Err(CodebookError::MemoryError); } |
0ca65ffd KS |
314 | if syms.capacity() < nnz { return Err(CodebookError::MemoryError); } |
315 | table.resize(tab_len, TABLE_FILL_VALUE); | |
4667915a KS |
316 | |
317 | let mut symidx: u32 = 0; | |
318 | for i in 0..cb.len() { | |
319 | let bits = cb.bits(i); | |
0ca65ffd | 320 | let code = cb.code(i); |
4667915a | 321 | if bits == 0 { continue; } |
0ca65ffd KS |
322 | if bits <= MAX_LUT_BITS { |
323 | fill_lut(&mut table, mode, 0, code, bits, maxbits, symidx, false); | |
324 | } else { | |
325 | let ckey = extract_lut_part(code, bits, MAX_LUT_BITS, mode) as usize; | |
326 | if table[ckey] == TABLE_FILL_VALUE { | |
327 | let key = ckey as u32; | |
328 | if let Some(bucket) = escape_list.get_mut(&key) { | |
329 | let maxlen = min(bucket.maxlen, MAX_LUT_BITS); | |
330 | let new_off = resize_table(&mut table, maxlen)?; | |
331 | fill_lut(&mut table, mode, 0, key, maxlen, MAX_LUT_BITS, new_off, true); | |
332 | bucket.offset = new_off as usize; | |
333 | } | |
334 | } | |
4667915a | 335 | } |
e243ceb4 | 336 | symidx += 1; |
4667915a KS |
337 | } |
338 | ||
e243ceb4 | 339 | for bucket in escape_list.values() { |
0ca65ffd KS |
340 | build_esc_lut(&mut table, mode, &bucket)?; |
341 | } | |
342 | ||
4667915a KS |
343 | for i in 0..cb.len() { |
344 | if cb.bits(i) > 0 { | |
345 | syms.push(cb.sym(i)); | |
346 | } | |
347 | } | |
348 | ||
e243ceb4 | 349 | Ok(Codebook { table, syms, lut_bits: maxbits }) |
4667915a KS |
350 | } |
351 | } | |
352 | ||
353 | impl<'a, S: Copy> CodebookReader<S> for BitReader<'a> { | |
354 | #[allow(unused_variables)] | |
355 | fn read_cb(&mut self, cb: &Codebook<S>) -> CodebookResult<S> { | |
0ca65ffd KS |
356 | let mut esc = true; |
357 | let mut idx = 0; | |
358 | let mut lut_bits = cb.lut_bits; | |
359 | while esc { | |
360 | let lut_idx = (self.peek(lut_bits) as usize) + (idx as usize); | |
361 | if cb.table[lut_idx] == TABLE_FILL_VALUE { return Err(CodebookError::InvalidCode); } | |
362 | let bits = cb.table[lut_idx] & 0x7F; | |
363 | esc = (cb.table[lut_idx] & 0x80) != 0; | |
364 | idx = (cb.table[lut_idx] >> 8) as usize; | |
365 | if (bits as isize) > self.left() { | |
366 | return Err(CodebookError::InvalidCode); | |
367 | } | |
e243ceb4 KS |
368 | let skip_bits = if esc { u32::from(lut_bits) } else { bits }; |
369 | self.skip(skip_bits as u32).unwrap(); | |
0ca65ffd | 370 | lut_bits = bits as u8; |
4667915a | 371 | } |
0ca65ffd | 372 | Ok(cb.syms[idx]) |
4667915a KS |
373 | } |
374 | } | |
375 | ||
a4bd7707 | 376 | /// Codebook description that stores a list of codewords and their values. |
4667915a KS |
377 | pub struct FullCodebookDescReader<S> { |
378 | data: Vec<FullCodebookDesc<S>>, | |
379 | } | |
380 | ||
381 | impl<S> FullCodebookDescReader<S> { | |
a4bd7707 | 382 | /// Constructs a new `FullCodebookDescReader` instance. |
4667915a | 383 | pub fn new(data: Vec<FullCodebookDesc<S>>) -> Self { |
e243ceb4 | 384 | FullCodebookDescReader { data } |
4667915a KS |
385 | } |
386 | } | |
387 | ||
388 | impl<S: Copy> CodebookDescReader<S> for FullCodebookDescReader<S> { | |
389 | fn bits(&mut self, idx: usize) -> u8 { self.data[idx].bits } | |
390 | fn code(&mut self, idx: usize) -> u32 { self.data[idx].code } | |
391 | fn sym (&mut self, idx: usize) -> S { self.data[idx].sym } | |
392 | fn len(&mut self) -> usize { self.data.len() } | |
393 | } | |
394 | ||
a4bd7707 | 395 | /// Codebook description that stores a list of codewords and their value is equal to the index. |
4667915a KS |
396 | pub struct ShortCodebookDescReader { |
397 | data: Vec<ShortCodebookDesc>, | |
398 | } | |
399 | ||
400 | impl ShortCodebookDescReader { | |
a4bd7707 | 401 | /// Constructs a new `ShortCodebookDescReader` instance. |
4667915a | 402 | pub fn new(data: Vec<ShortCodebookDesc<>>) -> Self { |
e243ceb4 | 403 | ShortCodebookDescReader { data } |
4667915a KS |
404 | } |
405 | } | |
406 | ||
407 | impl CodebookDescReader<u32> for ShortCodebookDescReader { | |
408 | fn bits(&mut self, idx: usize) -> u8 { self.data[idx].bits } | |
409 | fn code(&mut self, idx: usize) -> u32 { self.data[idx].code } | |
410 | fn sym (&mut self, idx: usize) -> u32 { idx as u32 } | |
411 | fn len(&mut self) -> usize { self.data.len() } | |
412 | } | |
413 | ||
a4bd7707 | 414 | /// Flexible codebook description that uses two separate arrays for codeword bits and lengths and a function that maps codeword index into its symbol. |
db63f876 KS |
415 | pub struct TableCodebookDescReader<'a, CodeType:'static, IndexType:'static> { |
416 | bits: &'a [u8], | |
417 | codes: &'a [CodeType], | |
6465a946 KS |
418 | idx_map: fn(usize) -> IndexType, |
419 | } | |
420 | ||
db63f876 | 421 | impl<'a, CodeType, IndexType> TableCodebookDescReader<'a, CodeType, IndexType> { |
a4bd7707 | 422 | /// Constructs a new `TableCodebookDescReader` instance. |
db63f876 | 423 | pub fn new(codes: &'a [CodeType], bits: &'a [u8], idx_map: fn(usize) -> IndexType) -> Self { |
6465a946 KS |
424 | Self { bits, codes, idx_map } |
425 | } | |
426 | } | |
db63f876 | 427 | impl<'a, CodeType: Copy+Into<u32>, IndexType> CodebookDescReader<IndexType> for TableCodebookDescReader<'a, CodeType, IndexType> |
6465a946 KS |
428 | { |
429 | fn bits(&mut self, idx: usize) -> u8 { self.bits[idx] } | |
430 | fn code(&mut self, idx: usize) -> u32 { self.codes[idx].into() } | |
431 | fn sym (&mut self, idx: usize) -> IndexType { (self.idx_map)(idx) } | |
432 | fn len(&mut self) -> usize { self.bits.len() } | |
433 | } | |
434 | ||
4667915a KS |
435 | #[cfg(test)] |
436 | mod test { | |
437 | use super::*; | |
aca89041 | 438 | use crate::io::bitreader::*; |
4667915a KS |
439 | |
440 | #[test] | |
441 | fn test_cb() { | |
442 | const BITS: [u8; 2] = [0b01011011, 0b10111100]; | |
443 | let cb_desc: Vec<FullCodebookDesc<i8>> = vec!( | |
444 | FullCodebookDesc { code: 0b0, bits: 1, sym: 16 }, | |
445 | FullCodebookDesc { code: 0b10, bits: 2, sym: -3 }, | |
446 | FullCodebookDesc { code: 0b110, bits: 3, sym: 42 }, | |
447 | FullCodebookDesc { code: 0b1110, bits: 4, sym: -42 } | |
448 | ); | |
449 | let buf = &BITS; | |
fa90ccfb | 450 | let mut br = BitReader::new(buf, BitReaderMode::BE); |
4667915a | 451 | let mut cfr = FullCodebookDescReader::new(cb_desc); |
0ca65ffd | 452 | let cb = Codebook::new(&mut cfr, CodebookMode::MSB).unwrap(); |
4667915a KS |
453 | assert_eq!(br.read_cb(&cb).unwrap(), 16); |
454 | assert_eq!(br.read_cb(&cb).unwrap(), -3); | |
455 | assert_eq!(br.read_cb(&cb).unwrap(), 42); | |
456 | assert_eq!(br.read_cb(&cb).unwrap(), -42); | |
457 | let ret = br.read_cb(&cb); | |
458 | if let Err(e) = ret { | |
459 | assert_eq!(e as i32, CodebookError::InvalidCode as i32); | |
460 | } else { | |
461 | assert_eq!(0, 1); | |
462 | } | |
463 | ||
464 | let scb_desc: Vec<ShortCodebookDesc> = vec!( | |
465 | ShortCodebookDesc { code: 0b0, bits: 1 }, | |
466 | ShortCodebookDesc { code: 0, bits: 0 }, | |
467 | ShortCodebookDesc { code: 0b10, bits: 2 }, | |
468 | ShortCodebookDesc { code: 0, bits: 0 }, | |
469 | ShortCodebookDesc { code: 0, bits: 0 }, | |
470 | ShortCodebookDesc { code: 0b110, bits: 3 }, | |
471 | ShortCodebookDesc { code: 0, bits: 0 }, | |
0ca65ffd KS |
472 | ShortCodebookDesc { code: 0b11100, bits: 5 }, |
473 | ShortCodebookDesc { code: 0b11101, bits: 5 }, | |
474 | ShortCodebookDesc { code: 0b1111010, bits: 7 }, | |
475 | ShortCodebookDesc { code: 0b1111011, bits: 7 }, | |
476 | ShortCodebookDesc { code: 0b1111110, bits: 7 }, | |
477 | ShortCodebookDesc { code: 0b11111111, bits: 8 } | |
4667915a | 478 | ); |
fa90ccfb | 479 | let mut br2 = BitReader::new(buf, BitReaderMode::BE); |
4667915a | 480 | let mut cfr = ShortCodebookDescReader::new(scb_desc); |
0ca65ffd | 481 | let cb = Codebook::new(&mut cfr, CodebookMode::MSB).unwrap(); |
4667915a KS |
482 | assert_eq!(br2.read_cb(&cb).unwrap(), 0); |
483 | assert_eq!(br2.read_cb(&cb).unwrap(), 2); | |
484 | assert_eq!(br2.read_cb(&cb).unwrap(), 5); | |
0ca65ffd KS |
485 | assert_eq!(br2.read_cb(&cb).unwrap(), 8); |
486 | ||
06fd8c88 | 487 | assert_eq!(reverse_bits(0b0000_0101_1011_1011_1101_1111_0111_1111, 32), |
0ca65ffd KS |
488 | 0b1111_1110_1111_1011_1101_1101_1010_0000); |
489 | ||
490 | const BITS_LE: [u8; 3] = [0b11101111, 0b01110010, 0b01]; | |
491 | let buf = &BITS_LE; | |
492 | let scble_desc: Vec<ShortCodebookDesc> = vec!( | |
493 | ShortCodebookDesc { code: 0b00, bits: 2 }, | |
494 | ShortCodebookDesc { code: 0, bits: 0 }, | |
495 | ShortCodebookDesc { code: 0b01, bits: 2 }, | |
496 | ShortCodebookDesc { code: 0, bits: 0 }, | |
497 | ShortCodebookDesc { code: 0, bits: 0 }, | |
498 | ShortCodebookDesc { code: 0b011, bits: 3 }, | |
499 | ShortCodebookDesc { code: 0, bits: 0 }, | |
500 | ShortCodebookDesc { code: 0b10111, bits: 5 }, | |
501 | ShortCodebookDesc { code: 0b00111, bits: 5 }, | |
502 | ShortCodebookDesc { code: 0b0101111, bits: 7 }, | |
503 | ShortCodebookDesc { code: 0b0111111, bits: 7 }, | |
504 | ShortCodebookDesc { code: 0b1011101111, bits: 10 } | |
505 | ); | |
fa90ccfb | 506 | let mut brl = BitReader::new(buf, BitReaderMode::LE); |
0ca65ffd KS |
507 | let mut cfr = ShortCodebookDescReader::new(scble_desc); |
508 | let cb = Codebook::new(&mut cfr, CodebookMode::LSB).unwrap(); | |
509 | assert_eq!(brl.read_cb(&cb).unwrap(), 11); | |
510 | assert_eq!(brl.read_cb(&cb).unwrap(), 0); | |
511 | assert_eq!(brl.read_cb(&cb).unwrap(), 7); | |
512 | assert_eq!(brl.read_cb(&cb).unwrap(), 0); | |
4667915a KS |
513 | } |
514 | } |