hayro-jbig2 0.4.0

A memory-safe, pure-Rust JBIG2 decoder.
Documentation 
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

//! Symbol ID (IAID) decoder (A.3).

use alloc::vec;
use alloc::vec::Vec;

use crate::arithmetic_decoder::{ArithmeticDecoder, ArithmeticDecoderContext};

pub(crate) struct SymbolIdDecoder {
    contexts: Vec<ArithmeticDecoderContext>,
    code_len: u32,
}

impl SymbolIdDecoder {
    pub(crate) fn new(code_len: u32) -> Self {
        // A.3: "The number of contexts required is 2^SBSYMCODELEN, which is less
        // than twice the maximum symbol ID."
        let num_contexts = 1_usize << code_len;

        Self {
            contexts: vec![ArithmeticDecoderContext::default(); num_contexts],
            code_len,
        }
    }

    #[inline(always)]
    pub(crate) fn decode(&mut self, decoder: &mut ArithmeticDecoder<'_>) -> u32 {
        let mut prev = 1_u32;

        for _ in 0..self.code_len {
            let d = decoder.read_bit(&mut self.contexts[prev as usize]);

            prev = (prev << 1) | d;
        }

        prev -= 1 << self.code_len;
        prev
    }
}