X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=nihav-core%2Fsrc%2Fio%2Fcodebook.rs;h=f533cb263abe8abf76dfe67143f4100b979470b4;hb=79fa5fbff6700b9443a58e0bd71bfef9a1e5a073;hp=553f2f9c84e77415580404ae38fb58c092bd465c;hpb=cec9f863d6272d15d51045c3cf16f2db9d779eff;p=nihav.git diff --git a/nihav-core/src/io/codebook.rs b/nihav-core/src/io/codebook.rs index 553f2f9..f533cb2 100644 --- a/nihav-core/src/io/codebook.rs +++ b/nihav-core/src/io/codebook.rs @@ -113,6 +113,7 @@ pub struct ShortCodebookDesc { /// /// [`ShortCodebookDescReader`]: ./struct.ShortCodebookDescReader.html /// [`TableCodebookDescReader`]: ./struct.TableCodebookDescReader.html +#[allow(clippy::len_without_is_empty)] pub trait CodebookDescReader { /// Returns the codeword length for the provided index. fn bits(&mut self, idx: usize) -> u8; @@ -143,7 +144,7 @@ pub trait CodebookReader { pub const TABLE_FILL_VALUE: u32 = 0x7F; const MAX_LUT_BITS: u8 = 10; -fn fill_lut_msb(table: &mut Vec, off: usize, +fn fill_lut_msb(table: &mut [u32], off: usize, code: u32, bits: u8, lut_bits: u8, symidx: u32, esc: bool) -> CodebookResult<()> { if !esc { let fill_len = lut_bits - bits; @@ -163,7 +164,7 @@ fn fill_lut_msb(table: &mut Vec, off: usize, Ok(()) } -fn fill_lut_lsb(table: &mut Vec, off: usize, +fn fill_lut_lsb(table: &mut [u32], off: usize, code: u32, bits: u8, lut_bits: u8, symidx: u32, esc: bool) -> CodebookResult<()> { if !esc { let fill_len = lut_bits - bits; @@ -183,7 +184,7 @@ fn fill_lut_lsb(table: &mut Vec, off: usize, Ok(()) } -fn fill_lut(table: &mut Vec, mode: CodebookMode, +fn fill_lut(table: &mut [u32], mode: CodebookMode, off: usize, code: u32, bits: u8, lut_bits: u8, symidx: u32, esc: bool) -> CodebookResult { match mode { CodebookMode::MSB => fill_lut_msb(table, off, code, bits, lut_bits, symidx, esc)?, @@ -270,7 +271,7 @@ fn build_esc_lut(table: &mut Vec, let cur_offset = bucket.offset; for (ckey, sec_bucket) in &mut escape_list { - let key = *ckey as u32; + let key = *ckey; let maxlen = min(sec_bucket.maxlen, MAX_LUT_BITS); let new_off = resize_table(table, maxlen)?; fill_lut(table, mode, cur_offset, key, maxlen, @@ -288,7 +289,7 @@ fn build_esc_lut(table: &mut Vec, impl Codebook { /// Constructs a new `Codebook` instance using provided codebook description and mode. - pub fn new(cb: &mut CodebookDescReader, mode: CodebookMode) -> CodebookResult { + pub fn new(cb: &mut dyn CodebookDescReader, mode: CodebookMode) -> CodebookResult { let mut maxbits = 0; let mut nnz = 0; let mut escape_list: EscapeCodes = HashMap::new(); @@ -345,7 +346,7 @@ impl Codebook { } for bucket in escape_list.values() { - build_esc_lut(&mut table, mode, &bucket)?; + build_esc_lut(&mut table, mode, bucket)?; } for i in 0..cb.len() { @@ -365,16 +366,16 @@ impl<'a, S: Copy> CodebookReader for BitReader<'a> { let mut idx = 0; let mut lut_bits = cb.lut_bits; while esc { - let lut_idx = (self.peek(lut_bits) as usize) + (idx as usize); + let lut_idx = (self.peek(lut_bits) as usize) + idx; if cb.table[lut_idx] == TABLE_FILL_VALUE { return Err(CodebookError::InvalidCode); } let bits = cb.table[lut_idx] & 0x7F; esc = (cb.table[lut_idx] & 0x80) != 0; idx = (cb.table[lut_idx] >> 8) as usize; - if (bits as isize) > self.left() { + let skip_bits = if esc { u32::from(lut_bits) } else { bits }; + if (skip_bits as isize) > self.left() { return Err(CodebookError::InvalidCode); } - let skip_bits = if esc { u32::from(lut_bits) } else { bits }; - self.skip(skip_bits as u32).unwrap(); + self.skip(skip_bits).unwrap(); lut_bits = bits as u8; } Ok(cb.syms[idx])