X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=nihav-core%2Fsrc%2Fio%2Fcodebook.rs;h=6ea3b1ee5ea4ff65df5700d02368a2d021d0f786;hb=812348ae811aa57ae10c155a50f176861207a386;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..6ea3b1e 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)?, @@ -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() { @@ -370,10 +371,10 @@ impl<'a, S: Copy> CodebookReader for BitReader<'a> { 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(); lut_bits = bits as u8; }