mousiki 0.2.1

use alloc::vec::Vec;

use crate::celt::{EcEnc, EcEncSnapshot, ec_tell};
use crate::silk::icdf::ICDFContext;

const EC_SYM_BITS: u32 = 8;
const EC_CODE_BITS: u32 = 32;
const EC_CODE_TOP: u32 = 1 << (EC_CODE_BITS - 1);
const EC_CODE_BOT: u32 = EC_CODE_TOP >> EC_SYM_BITS;
const EC_CODE_EXTRA: u32 = (EC_CODE_BITS - 2) % EC_SYM_BITS + 1;
const EC_UINT_BITS: u32 = 8;

/// See [section-4.1](https://datatracker.ietf.org/doc/html/rfc6716#section-4.1)
// Opus uses an entropy coder based on range coding [RANGE-CODING]
// [MARTIN79], which is itself a rediscovery of the FIFO arithmetic code
// introduced by [CODING-THESIS].  It is very similar to arithmetic
// encoding, except that encoding is done with digits in any base
// instead of with bits, so it is faster when using larger bases (i.e.,
// a byte).  All of the calculations in the range coder must use bit-
// exact integer arithmetic.
//
// Symbols may also be coded as "raw bits" packed directly into the
// bitstream, bypassing the range coder.  These are packed backwards
// starting at the end of the frame, as illustrated in Figure 12.  This
// reduces complexity and makes the stream more resilient to bit errors,
// as corruption in the raw bits will not desynchronize the decoding
// process, unlike corruption in the input to the range decoder.  Raw
// bits are only used in the CELT layer.
//
//	 0                   1                   2                   3
//	 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
//	+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
//	| Range coder data (packed MSB to LSB) ->                       :
//	+                                                               +
//	:                                                               :
//	+     +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
//	:     | <- Boundary occurs at an arbitrary bit position         :
//	+-+-+-+                                                         +
//	:                          <- Raw bits data (packed LSB to MSB) |
//	+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
//
//	Legend:
//
//	LSB = Least Significant Bit
//	MSB = Most Significant Bit
//
//	     Figure 12: Illustrative Example of Packing Range Coder
//	                        and Raw Bits Data
//
// Each symbol coded by the range coder is drawn from a finite alphabet
// and coded in a separate "context", which describes the size of the
// alphabet and the relative frequency of each symbol in that alphabet.
//
// Suppose there is a context with n symbols, identified with an index
// that ranges from 0 to n-1.  The parameters needed to encode or decode
// symbol k in this context are represented by a three-tuple
// `(fl[k], fh[k], ft)`, all 16-bit unsigned integers, with
// `0 <= fl[k] < fh[k] <= ft <= 65535`.  The values of this tuple are
// derived from the probability model for the symbol, represented by
// traditional "frequency counts".  Because Opus uses static contexts,
// those are not updated as symbols are decoded.  Let `f[i]` be the
// frequency of symbol i.  Then, the three-tuple corresponding to symbol
// k is given by the following:
//
//	        k-1                                   n-1
//	        __                                    __
//	fl[k] = \  f[i],  fh[k] = fl[k] + f[k],  ft = \  f[i]
//	        /_                                    /_
//	        i=0                                   i=0
//
// The range decoder extracts the symbols and integers encoded using the
// range encoder in Section 5.1.  The range decoder maintains an
// internal state vector composed of the two-tuple (val, rng), where val
// represents the difference between the high end of the current range
// and the actual coded value, minus one, and rng represents the size of
// the current range.  Both val and rng are 32-bit unsigned integer
// values.
#[derive(Debug, Clone)]
pub struct RangeDecoder<'a> {
    pub(crate) buf: &'a [u8],
    pub(crate) bits_read: usize,
    pub(crate) total_bits: i32,
    /// `rng`
    pub(crate) range_size: u32,
    /// `val`
    pub(crate) high_and_coded_difference: u32,
}

/// `MIN_RANGE_SIZE` is the minimum allowed size for rng.
/// It's equal to `2.pow(23)`.
const MIN_RANGE_SIZE: u32 = EC_CODE_BOT;

impl<'a> RangeDecoder<'a> {
    // To normalize the range, the decoder repeats the following process,
    // implemented by ec_dec_normalize() (entdec.c), until rng > 2**23.  If
    // rng is already greater than 2**23, the entire process is skipped.
    // First, it sets rng to (rng<<8).  Then, it reads the next byte of the
    // Opus frame and forms an 8-bit value sym, using the leftover bit
    // buffered from the previous byte as the high bit and the top 7 bits of
    // the byte just read as the other 7 bits of sym.  The remaining bit in
    // the byte just read is buffered for use in the next iteration.  If no
    // more input bytes remain, it uses zero bits instead.  See
    // Section 4.1.1 for the initialization used to process the first byte.
    // Then, it sets
    //
    // val = ((val<<8) + (255-sym)) & 0x7FFFFFFF
    //
    /// See [section-4.1.2.1](https://datatracker.ietf.org/doc/html/rfc6716#section-4.1.2.1)
    pub(crate) fn normalize(&mut self) {
        while self.range_size <= MIN_RANGE_SIZE {
            self.range_size <<= 8;
            self.total_bits += EC_SYM_BITS as i32;
            self.high_and_coded_difference =
                ((self.high_and_coded_difference << 8) + (255 - self.get_bits(8))) & 0x7FFFFFFF;
        }
    }

    #[allow(dead_code)]
    pub(crate) fn decode_bin(&self, bits: u32) -> u32 {
        let scale = self.range_size >> bits;
        let max = 1u32 << bits;
        let s = self.high_and_coded_difference / scale;
        max - (s + 1).min(max)
    }

    pub(crate) fn decode_icdf16(&mut self, icdf: &[u16], ftb: u32) -> usize {
        let mut range = self.range_size;
        let val = self.high_and_coded_difference;
        let scale = range >> ftb;

        let mut symbol = -1i32;
        let mut prev_range: u32;

        loop {
            symbol += 1;
            prev_range = range;
            let entry = u32::from(icdf[symbol as usize]);
            range = scale * entry;
            if val >= range {
                break;
            }
        }

        self.high_and_coded_difference = val - range;
        self.range_size = prev_range - range;
        self.normalize();

        symbol as usize
    }

    fn get_bit(&mut self) -> u32 {
        let index = self.bits_read / 8;
        let offset = self.bits_read % 8;

        if index >= self.buf.len() {
            return 0;
        }

        self.bits_read += 1;
        u32::from((self.buf[index] >> (7 - offset)) & 1)
    }

    fn get_bits(&mut self, n: usize) -> u32 {
        let mut bits = 0;

        for i in 0..n {
            if i != 0 {
                bits <<= 1;
            }

            bits |= self.get_bit();
        }

        bits
    }

    pub(crate) fn update(&mut self, scale: u32, low: u32, high: u32, total: u32) {
        self.high_and_coded_difference -= scale * (total - high);

        if low == 0 {
            self.range_size -= scale * (total - high);
        } else {
            self.range_size = scale * (high - low);
        }

        self.normalize();
    }

    /// See [section-4.1.3.2](https://datatracker.ietf.org/doc/html/rfc6716#section-4.1.3.2)
    pub fn decode_symbol_logp(&mut self, logp: usize) -> u32 {
        let scale = self.range_size >> logp;

        let k = if self.high_and_coded_difference >= scale {
            self.high_and_coded_difference -= scale;
            self.range_size -= scale;
            0
        } else {
            self.range_size = scale;
            1
        };

        self.normalize();

        k
    }

    /// decodes a single symbol
    /// with a table-based context of up to 8 bits.
    ///
    /// See [section-4.1.3.3](https://datatracker.ietf.org/doc/html/rfc6716#section-4.1.3.3)
    pub fn decode_symbol_with_icdf(&mut self, icdf_ctx: ICDFContext) -> u32 {
        let ICDFContext { total, dist_table } = icdf_ctx;

        let scale = self.range_size / total;
        let symbol = total - (self.high_and_coded_difference / scale + 1).min(total);
        let k = dist_table
            .iter()
            .position(|v| (*v) as u32 > symbol)
            .unwrap_or(0);
        let high = dist_table[k];
        let low = if k != 0 { dist_table[k - 1] } else { 0 };

        self.update(scale, low as u32, high as u32, total);

        k as u32
    }

    /// Decodes a symbol using the compact 8-bit cumulative distribution format
    /// employed by SILK's tables.
    pub fn decode_icdf(&mut self, icdf: &[u8], ftb: u32) -> usize {
        debug_assert!(!icdf.is_empty(), "icdf tables must not be empty");

        let mut s = self.range_size;
        let d = self.high_and_coded_difference;
        let r = s >> ftb;
        let mut index = 0usize;

        loop {
            let t = s;
            s = r * u32::from(icdf[index]);
            if d >= s {
                self.high_and_coded_difference = d - s;
                self.range_size = t - s;
                self.normalize();
                return index;
            }

            index += 1;
            debug_assert!(
                index < icdf.len(),
                "icdf table exhausted before decoding symbol"
            );
        }
    }

    /// Init sets the state of the Decoder
    /// Let b0 be an 8-bit unsigned integer containing first input byte (or
    /// containing zero if there are no bytes in this Opus frame).  The
    /// decoder initializes rng to 128 and initializes val to (127 - (b0>>1))
    /// , where (b0>>1) is the top 7 bits of the first input byte.
    ///
    /// It saves the remaining bit, (b0&1), for use in the renormalization
    /// procedure described in Section 4.1.2.1, which the decoder invokes
    /// immediately after initialization to read additional bits and
    /// establish the invariant that rng > 2**23.
    ///
    /// See [section-4.1.1](https://datatracker.ietf.org/doc/html/rfc6716#section-4.1.1)
    pub fn init(buf: &'a [u8]) -> RangeDecoder<'a> {
        let mut decoder = RangeDecoder {
            buf,
            bits_read: 0,
            total_bits: EC_CODE_BITS as i32 + 1
                - ((EC_CODE_BITS - EC_CODE_EXTRA) / EC_SYM_BITS) as i32 * EC_SYM_BITS as i32,
            range_size: 128,
            high_and_coded_difference: 0,
        };

        decoder.high_and_coded_difference = 127 - decoder.get_bits(7);
        decoder.normalize();

        decoder
    }

    /// Returns the number of whole bits consumed by the decoder so far.
    #[must_use]
    pub fn tell(&self) -> i32 {
        self.total_bits - ec_ilog(self.range_size)
    }

    /// Decodes a uniformly distributed integer in the range `[0, total)`.
    ///
    /// Mirrors `ec_dec_uint` for the small-`total` path used by the decoder to
    /// read redundancy byte counts during CELT/SILK transitions.
    pub fn decode_uint(&mut self, total: u32) -> u32 {
        debug_assert!(total > 1);
        let total_minus_one = total - 1;
        let bits = ec_ilog(total_minus_one);
        if bits > EC_UINT_BITS as i32 {
            // This branch is unreachable for the current redundancy use which
            // caps the decoded value at 256, but is kept for parity with the
            // reference helper.
            let shift = bits - EC_UINT_BITS as i32;
            let symbols = (total_minus_one >> shift) + 1;
            let symbol = self.decode_uniform(symbols);
            let tail = self.get_bits(shift as usize);
            self.total_bits += shift;
            (symbol << shift) | tail
        } else {
            self.decode_uniform(total)
        }
    }

    fn decode_uniform(&mut self, total: u32) -> u32 {
        let scale = self.range_size / total;
        let symbol = total - (self.high_and_coded_difference / scale + 1).min(total);
        self.high_and_coded_difference -= scale * (total - symbol - 1);
        if symbol > 0 {
            self.range_size = scale;
        } else {
            self.range_size -= scale * (total - 1);
        }
        self.normalize();
        symbol
    }
}

fn ec_ilog(value: u32) -> i32 {
    if value == 0 {
        0
    } else {
        32 - value.leading_zeros() as i32
    }
}

const RANGE_ENCODER_STORAGE_BYTES: usize = 1275;

#[derive(Debug)]
pub struct RangeEncoder {
    encoder: EcEnc<'static>,
}

impl RangeEncoder {
    pub fn new() -> Self {
        Self::with_capacity(RANGE_ENCODER_STORAGE_BYTES)
    }

    pub(crate) fn with_capacity(capacity: usize) -> Self {
        let encoder = EcEnc::with_capacity(capacity);
        Self { encoder }
    }

    fn from_snapshot(snapshot: &EcEncSnapshot) -> Self {
        let mut encoder = EcEnc::with_capacity(snapshot.buffer_len());
        snapshot.restore(&mut encoder);
        Self { encoder }
    }

    /// Returns the number of whole bits emitted so far.
    #[must_use]
    pub fn tell(&self) -> i32 {
        ec_tell(self.encoder.ctx()) as i32
    }

    /// Returns the final range-coder state (`rng`) used by Opus for diagnostics.
    #[must_use]
    pub fn range_final(&self) -> u32 {
        self.encoder.ctx().rng
    }

    pub fn encode_bin(&mut self, low: u32, high: u32, bits: u32) {
        self.encoder.encode_bin(low, high, bits);
    }

    pub fn encode_symbol_with_icdf(&mut self, symbol: usize, icdf_ctx: ICDFContext) {
        let ICDFContext { total, dist_table } = icdf_ctx;
        debug_assert!(symbol < dist_table.len(), "symbol index out of bounds");
        let high = dist_table[symbol] as u32;
        let low = if symbol > 0 {
            dist_table[symbol - 1] as u32
        } else {
            0
        };
        self.encoder.encode(low, high, total);
    }

    pub fn encode_icdf16(&mut self, symbol: usize, icdf: &[u16], ftb: u32) {
        debug_assert!(symbol < icdf.len(), "symbol index out of bounds");
        self.encoder.enc_icdf16(symbol, icdf, ftb);
    }

    /// Encodes a symbol using the compact 8-bit cumulative distribution format
    /// used by SILK's shell coder and related tables.
    pub fn encode_icdf(&mut self, symbol: usize, icdf: &[u8], ftb: u32) {
        debug_assert!(symbol < icdf.len(), "symbol index out of bounds");
        self.encoder.enc_icdf(symbol, icdf, ftb);
    }

    pub(crate) fn encode_bit_logp(&mut self, value: i32, logp: u32) {
        self.encoder.enc_bit_logp(value, logp);
    }

    pub(crate) fn encode_uint(&mut self, value: u32, total: u32) {
        self.encoder.enc_uint(value, total);
    }

    pub(crate) fn shrink(&mut self, size: usize) {
        self.encoder.enc_shrink(size as u32);
    }

    pub(crate) fn encoder_mut(&mut self) -> &mut EcEnc<'static> {
        &mut self.encoder
    }

    /// Patches bits at the start of the encoded stream.
    ///
    /// Mirrors `ec_enc_patch_initial_bits`, allowing callers to reserve space
    /// for header bits and fill them in once the final values are known.
    pub fn patch_initial_bits(&mut self, value: u32, nbits: u32) {
        if nbits == 0 {
            return;
        }
        self.encoder.enc_patch_initial_bits(value, nbits);
    }

    pub fn finish(mut self) -> Vec<u8> {
        self.encoder.enc_done();
        let size = self.encoder.ctx().offs + self.encoder.ctx().end_offs;
        if size < self.encoder.ctx().storage {
            self.encoder.enc_shrink(size);
        }
        let size = size as usize;
        let buffer = self.encoder.ctx().buffer();
        #[cfg(test)]
        range_done_trace::maybe_dump(&buffer[..size]);
        buffer[..size].to_vec()
    }

    pub(crate) fn finish_without_done(self) -> Vec<u8> {
        let size = self.encoder.ctx().storage as usize;
        let buffer = self.encoder.ctx().buffer();
        buffer[..size].to_vec()
    }
}

impl Default for RangeEncoder {
    fn default() -> Self {
        Self::new()
    }
}

impl Clone for RangeEncoder {
    fn clone(&self) -> Self {
        let snapshot = EcEncSnapshot::capture(&self.encoder);
        Self::from_snapshot(&snapshot)
    }
}

#[cfg(test)]
mod range_done_trace {
    extern crate std;

    use core::sync::atomic::{AtomicIsize, AtomicUsize, Ordering};
    use std::env;
    use std::sync::OnceLock;

    pub(crate) struct TraceConfig {
        frame: Option<usize>,
    }

    static TRACE_CONFIG: OnceLock<Option<TraceConfig>> = OnceLock::new();
    static FRAME_INDEX: AtomicUsize = AtomicUsize::new(0);
    static CURRENT_FRAME: AtomicIsize = AtomicIsize::new(-1);

    pub(crate) fn begin_frame() -> Option<usize> {
        if config().is_some() {
            Some(FRAME_INDEX.fetch_add(1, Ordering::Relaxed))
        } else {
            None
        }
    }

    pub(crate) fn set_frame(frame_idx: usize) {
        CURRENT_FRAME.store(frame_idx as isize, Ordering::Relaxed);
    }

    fn current_frame() -> Option<usize> {
        let value = CURRENT_FRAME.load(Ordering::Relaxed);
        if value >= 0 {
            Some(value as usize)
        } else {
            None
        }
    }

    fn config() -> Option<&'static TraceConfig> {
        TRACE_CONFIG
            .get_or_init(|| {
                let enabled = match env::var("OPUS_TRACE_RANGE_DONE") {
                    Ok(value) => !value.is_empty() && value != "0",
                    Err(_) => false,
                };
                if !enabled {
                    return None;
                }
                let frame = env::var("OPUS_TRACE_RANGE_DONE_FRAME")
                    .ok()
                    .and_then(|value| value.parse::<usize>().ok());
                Some(TraceConfig { frame })
            })
            .as_ref()
    }

    pub(crate) fn maybe_dump(buffer: &[u8]) {
        let cfg = match config() {
            Some(cfg) => cfg,
            None => return,
        };
        let frame_idx = match current_frame() {
            Some(frame_idx) => frame_idx,
            None => return,
        };
        if cfg.frame.map_or(true, |frame| frame == frame_idx) {
            crate::test_trace::trace_println!("opus_range_done[{frame_idx}].len={}", buffer.len());
            for (idx, value) in buffer.iter().enumerate() {
                crate::test_trace::trace_println!(
                    "opus_range_done[{frame_idx}].byte[{idx}]=0x{value:02x}"
                );
            }
        }
    }
}

#[cfg(test)]
pub(crate) fn begin_range_done_trace_frame() -> Option<usize> {
    range_done_trace::begin_frame()
}

#[cfg(test)]
pub(crate) fn set_range_done_trace_frame(frame_idx: usize) {
    range_done_trace::set_frame(frame_idx);
    crate::celt::set_enc_done_trace_frame(frame_idx);
}

// taken from <https://github.com/pion/opus/blob/e8536fe9e4ca2181db7d808e35d50b2c0400ceb1/internal/rangecoding/decoder_test.go>
#[cfg(test)]
mod tests {
    use super::*;
    use crate::celt::EcDec;
    use crate::icdf;
    use crate::silk::SilkRangeDecoder;
    use crate::silk::tables_other::SILK_UNIFORM4_ICDF;
    use crate::silk::tables_pulses_per_block::{
        SILK_SHELL_CODE_TABLE_OFFSETS, SILK_SHELL_CODE_TABLE0,
    };

    const SILK_FRAME_TYPE_INACTIVE: ICDFContext = icdf!(256; 26, 256);

    const SILK_GAIN_HIGH_BITS: [ICDFContext; 3] = [
        icdf!(256; 32, 144, 212, 241, 253, 254, 255, 256),
        icdf!(256; 2, 19, 64, 124, 186, 233, 252, 256),
        icdf!(256; 1, 4, 30, 101, 195, 245, 254, 256),
    ];

    const SILK_GAIN_LOW_BITS: ICDFContext = icdf!(256; 32, 64, 96, 128, 160, 192, 224, 256);

    const SILK_GAIN_DELTA: ICDFContext = icdf!(
        256; 6, 11, 22, 53, 185, 206, 214, 218, 221, 223, 225, 227, 228, 229, 230, 231, 232, 233,
        234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251,
        252, 253, 254, 255, 256
    );

    const SILK_LSF_S1: [[ICDFContext; 2]; 2] = [
        [
            icdf!(
                256; 44, 78, 108, 127, 148, 160, 171, 174, 177, 179, 195, 197, 199, 200, 205, 207,
                208, 211, 214, 215, 216, 218, 220, 222, 225, 226, 235, 244, 246, 253, 255, 256
            ),
            icdf!(
                256; 1, 11, 12, 20, 23, 31, 39, 53, 66, 80, 81, 95, 107, 120, 131, 142, 154, 165,
                175, 185, 196, 204, 213, 221, 228, 236, 237, 238, 244, 245, 251, 256
            ),
        ],
        [
            icdf!(
                256; 31, 52, 55, 72, 73, 81, 98, 102, 103, 121, 137, 141, 143, 146, 147, 157, 158,
                161, 177, 188, 204, 206, 208, 211, 213, 224, 225, 229, 238, 246, 253, 256
            ),
            icdf!(
                256; 1, 5, 21, 26, 44, 55, 60, 74, 89, 90, 93, 105, 118, 132, 146, 152, 166, 178,
                180, 186, 187, 199, 211, 222, 232, 235, 245, 250, 251, 252, 253, 256
            ),
        ],
    ];

    const SILK_LSF_S2: [ICDFContext; 16] = [
        icdf!(256; 1, 2, 3, 18, 242, 253, 254, 255, 256),
        icdf!(256; 1, 2, 4, 38, 221, 253, 254, 255, 256),
        icdf!(256; 1, 2, 6, 48, 197, 252, 254, 255, 256),
        icdf!(256; 1, 2, 10, 62, 185, 246, 254, 255, 256),
        icdf!(256; 1, 4, 20, 73, 174, 248, 254, 255, 256),
        icdf!(256; 1, 4, 21, 76, 166, 239, 254, 255, 256),
        icdf!(256; 1, 8, 32, 85, 159, 226, 252, 255, 256),
        icdf!(256; 1, 2, 20, 83, 161, 219, 249, 255, 256),
        icdf!(256; 1, 2, 3, 12, 244, 253, 254, 255, 256),
        icdf!(256; 1, 2, 4, 32, 218, 253, 254, 255, 256),
        icdf!(256; 1, 2, 5, 47, 199, 252, 254, 255, 256),
        icdf!(256; 1, 2, 12, 61, 187, 252, 254, 255, 256),
        icdf!(256; 1, 5, 24, 72, 172, 249, 254, 255, 256),
        icdf!(256; 1, 2, 16, 70, 170, 242, 254, 255, 256),
        icdf!(256; 1, 2, 17, 78, 165, 226, 251, 255, 256),
        icdf!(256; 1, 8, 29, 79, 156, 237, 254, 255, 256),
    ];

    const SILK_LSF_INTERPOLATION_OFFSET: ICDFContext = icdf!(256; 13, 35, 64, 75, 256);

    const SILK_LCG_SEED: ICDFContext = icdf!(256; 64, 128, 192, 256);

    const SILK_EXC_RATE: [ICDFContext; 2] = [
        icdf!(256; 15, 66, 78, 124, 169, 182, 215, 242, 256),
        icdf!(256; 33, 63, 99, 116, 150, 199, 217, 238, 256),
    ];

    const SILK_PULSE_COUNT: [ICDFContext; 11] = [
        icdf!(
            256; 131, 205, 230, 238, 241, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254,
            255, 256
        ),
        icdf!(
            256; 58, 151, 211, 234, 241, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254,
            255, 256
        ),
        icdf!(
            256; 43, 94, 140, 173, 197, 213, 224, 232, 238, 241, 244, 247, 249, 250, 251, 253, 254,
            256
        ),
        icdf!(
            256; 17, 69, 140, 197, 228, 240, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255,
            256
        ),
        icdf!(
            256; 6, 27, 68, 121, 170, 205, 226, 237, 243, 246, 248, 250, 251, 252, 253, 254, 255,
            256
        ),
        icdf!(
            256; 7, 21, 43, 71, 100, 128, 153, 173, 190, 203, 214, 223, 230, 235, 239, 243, 246,
            256
        ),
        icdf!(
            256; 2, 7, 21, 50, 92, 138, 179, 210, 229, 240, 246, 249, 251, 252, 253, 254, 255, 256
        ),
        icdf!(256; 1, 3, 7, 17, 36, 65, 100, 137, 171, 199, 219, 233, 241, 246, 250, 252, 254, 256),
        icdf!(256; 1, 3, 5, 10, 19, 33, 53, 77, 104, 132, 158, 181, 201, 216, 227, 235, 241, 256),
        icdf!(256; 1, 2, 3, 9, 36, 94, 150, 189, 214, 228, 238, 244, 247, 250, 252, 253, 254, 256),
        icdf!(
            256; 2, 3, 9, 36, 94, 150, 189, 214, 228, 238, 244, 247, 250, 252, 253, 254, 256, 256
        ),
    ];

    #[test]
    fn decoder() {
        let mut decoder = RangeDecoder::init(&[0x0b, 0xe4, 0xc1, 0x36, 0xec, 0xc5, 0x80]);

        assert_eq!(decoder.decode_symbol_logp(0x1), 0);
        assert_eq!(decoder.decode_symbol_logp(0x1), 0);

        assert_eq!(decoder.decode_symbol_with_icdf(SILK_FRAME_TYPE_INACTIVE), 1);

        assert_eq!(decoder.decode_symbol_with_icdf(SILK_GAIN_HIGH_BITS[0]), 0);
        assert_eq!(decoder.decode_symbol_with_icdf(SILK_GAIN_LOW_BITS), 6);

        assert_eq!(decoder.decode_symbol_with_icdf(SILK_GAIN_DELTA), 0);
        assert_eq!(decoder.decode_symbol_with_icdf(SILK_GAIN_DELTA), 3);
        assert_eq!(decoder.decode_symbol_with_icdf(SILK_GAIN_DELTA), 4);

        assert_eq!(decoder.decode_symbol_with_icdf(SILK_LSF_S1[1][0]), 9);
        assert_eq!(decoder.decode_symbol_with_icdf(SILK_LSF_S2[10]), 5);
        assert_eq!(decoder.decode_symbol_with_icdf(SILK_LSF_S2[9]), 4);

        for _i in 0..14 {
            assert_eq!(decoder.decode_symbol_with_icdf(SILK_LSF_S2[8]), 4);
        }

        assert_eq!(
            decoder.decode_symbol_with_icdf(SILK_LSF_INTERPOLATION_OFFSET),
            4
        );

        assert_eq!(decoder.decode_symbol_with_icdf(SILK_LCG_SEED), 2);

        assert_eq!(decoder.decode_symbol_with_icdf(SILK_EXC_RATE[0]), 0);

        for _i in 0..20 {
            assert_eq!(decoder.decode_symbol_with_icdf(SILK_PULSE_COUNT[0]), 0);
        }
    }

    #[test]
    fn encodes_symbols_with_icdf_context() {
        let mut encoder = RangeEncoder::new();
        encoder.encode_symbol_with_icdf(1, SILK_FRAME_TYPE_INACTIVE);
        encoder.encode_symbol_with_icdf(2, SILK_GAIN_HIGH_BITS[1]);
        encoder.encode_symbol_with_icdf(6, SILK_GAIN_LOW_BITS);
        encoder.encode_symbol_with_icdf(0, SILK_GAIN_DELTA);

        let mut storage = encoder.finish();
        let mut decoder = EcDec::new(storage.as_mut_slice());

        assert_eq!(decoder.decode_symbol_with_icdf(SILK_FRAME_TYPE_INACTIVE), 1);
        assert_eq!(decoder.decode_symbol_with_icdf(SILK_GAIN_HIGH_BITS[1]), 2);
        assert_eq!(decoder.decode_symbol_with_icdf(SILK_GAIN_LOW_BITS), 6);
        assert_eq!(decoder.decode_symbol_with_icdf(SILK_GAIN_DELTA), 0);
    }

    #[test]
    fn encode_decode_with_u8_icdf_roundtrip() {
        let mut encoder = RangeEncoder::new();
        encoder.encode_icdf(2, &SILK_UNIFORM4_ICDF, 8);
        encoder.encode_icdf(0, &SILK_UNIFORM4_ICDF, 8);

        let mut storage = encoder.finish();
        let mut decoder = EcDec::new(storage.as_mut_slice());

        assert_eq!(decoder.decode_icdf(&SILK_UNIFORM4_ICDF, 8), 2);
        assert_eq!(decoder.decode_icdf(&SILK_UNIFORM4_ICDF, 8), 0);
    }

    #[test]
    fn encode_decode_with_shell_table_slice() {
        let start = usize::from(SILK_SHELL_CODE_TABLE_OFFSETS[4]);
        let end = usize::from(SILK_SHELL_CODE_TABLE_OFFSETS[5]);
        let icdf = &SILK_SHELL_CODE_TABLE0[start..end];

        let mut encoder = RangeEncoder::new();
        encoder.encode_icdf(2, icdf, 8);
        encoder.encode_icdf(1, icdf, 8);

        let mut storage = encoder.finish();
        let mut decoder = EcDec::new(storage.as_mut_slice());

        assert_eq!(decoder.decode_icdf(icdf, 8), 2);
        assert_eq!(decoder.decode_icdf(icdf, 8), 1);
    }

    #[test]
    fn patch_initial_bits_round_trips_logp_flags() {
        for header_bits in 1u32..=8 {
            let value = (1u32 << header_bits).wrapping_sub(1) ^ (header_bits % 3);
            let mut encoder = RangeEncoder::new();
            let mut icdf = [0u8; 2];
            icdf[0] = (256u16 - (256u16 >> header_bits)) as u8;
            encoder.encode_icdf(0, &icdf, 8);
            encoder.patch_initial_bits(value, header_bits);

            let mut storage = encoder.finish();
            let mut decoder = EcDec::new(storage.as_mut_slice());
            let mut decoded = 0u32;
            for _ in 0..header_bits {
                decoded = (decoded << 1) | decoder.decode_symbol_logp(1);
            }
            let mask = (1u32 << header_bits) - 1;
            assert_eq!(decoded, value & mask);
        }
    }

    #[test]
    fn range_final_matches_after_roundtrip_decode() {
        let mut encoder = RangeEncoder::new();
        encoder.encode_icdf(2, &SILK_UNIFORM4_ICDF, 8);
        encoder.encode_symbol_with_icdf(1, SILK_FRAME_TYPE_INACTIVE);
        encoder.encode_icdf(0, &SILK_UNIFORM4_ICDF, 8);
        let range_final = encoder.range_final();

        let mut storage = encoder.finish();
        let mut decoder = EcDec::new(storage.as_mut_slice());
        assert_eq!(decoder.decode_icdf(&SILK_UNIFORM4_ICDF, 8), 2);
        assert_eq!(decoder.decode_symbol_with_icdf(SILK_FRAME_TYPE_INACTIVE), 1);
        assert_eq!(decoder.decode_icdf(&SILK_UNIFORM4_ICDF, 8), 0);

        assert_eq!(decoder.range_final(), range_final);
    }
}