[nihav.git] / nihav-core / src / io / intcode.rs

//! Some universal integer codes support for bitstream reader.
use crate::io::bitreader::{BitReader, BitReaderError, BitReaderResult};
use crate::io::bitwriter::BitWriter;

/// Unsigned integer code types.
#[derive(Debug)]
pub enum UintCodeType {
    /// Code where number is represented as run of ones with terminating zero.
    UnaryOnes,
    /// Code where number is represented as run of zeroes with terminating one.
    UnaryZeroes,
    /// Code for 0, 1 and 2 coded as `0`, `10` and `11`
    Unary012,
    /// Code for 0, 1 and 2 coded as `11`, `10` and `0`
    Unary210,
    /// General limited unary code with defined run and terminating bit.
    LimitedUnary(u32, u32),
    /// Limited run of zeroes with terminating one (unless the code has maximum length).
    ///
    /// [`Unary012`] is essentially an alias for `LimitedZeroes(2)`.
    ///
    /// [`Unary012`]: #variant.Unary012
    LimitedZeroes(u32),
    /// Limited run of one with terminating zero (unless the code has maximum length).
    LimitedOnes(u32),
    /// Golomb code.
    Golomb(u8),
    /// Rice code.
    Rice(u8),
    /// Elias Gamma code (interleaved).
    Gamma,
    /// Elias Gamma' code (sometimes incorrectly called exp-Golomb).
    GammaP,
}

/// Signed integer code types.
pub enum IntCodeType {
    /// Golomb code. Last bit represents the sign.
    Golomb(u8),
    /// Golomb code. Last bit represents the sign.
    Rice(u8),
    /// Elias Gamma code. Unsigned values are remapped as 0, 1, -1, 2, -2, ...
    Gamma,
    /// Elias Gamma' code. Unsigned values are remapped as 0, 1, -1, 2, -2, ...
    GammaP,
}

/// Universal integer code reader trait for bitstream reader.
///
/// # Examples
///
/// Read an unsigned Golomb code:
/// ````
/// use nihav_core::io::bitreader::*;
/// use nihav_core::io::intcode::{IntCodeReader,UintCodeType};
///
/// # fn foo() -> BitReaderResult<()> {
/// let mem: [u8; 4] = [ 0, 1, 2, 3];
/// let mut br = BitReader::new(&mem, BitReaderMode::BE);
/// let val = br.read_code(UintCodeType::Golomb(3))?;
/// # Ok(())
/// # }
/// ````
///
/// Read signed Elias code:
/// ````
/// use nihav_core::io::bitreader::*;
/// use nihav_core::io::intcode::{IntCodeReader,IntCodeType};
///
/// # fn foo() -> BitReaderResult<()> {
/// let mem: [u8; 4] = [ 0, 1, 2, 3];
/// let mut br = BitReader::new(&mem, BitReaderMode::BE);
/// let val = br.read_code_signed(IntCodeType::Gamma)?;
/// # Ok(())
/// # }
/// ````
pub trait IntCodeReader {
    /// Reads an unsigned integer code of requested type.
    fn read_code(&mut self, t: UintCodeType) -> BitReaderResult<u32>;
    /// Reads signed integer code of requested type.
    fn read_code_signed(&mut self, t: IntCodeType) -> BitReaderResult<i32>;
}

fn read_unary(br: &mut BitReader, terminator: u32) -> BitReaderResult<u32> {
    let mut res: u32 = 0;
    loop {
        if br.read(1)? == terminator { return Ok(res); }
        res += 1;
    }
}

fn read_unary_lim(br: &mut BitReader, len: u32, terminator: u32) -> BitReaderResult<u32> {
    let mut res: u32 = 0;
    loop {
        if br.read(1)? == terminator { return Ok(res); }
        res += 1;
        if res == len { return Ok(res); }
    }
}

fn read_unary210(br: &mut BitReader) -> BitReaderResult<u32> {
    let val = read_unary_lim(br, 2, 0)?;
    Ok(2 - val)
}

fn read_golomb(br: &mut BitReader, m: u8) -> BitReaderResult<u32> {
    if m == 0 { return Err(BitReaderError::InvalidValue); }
    let nbits = (8 - m.leading_zeros()) as u8;
    if (m & (m - 1)) == 0 { return read_rice(br, nbits); }
    let cutoff = u32::from((1 << nbits) - m);
    let pfx = read_unary(br, 0)?;
    let tail = br.read(nbits - 1)?;
    if tail < cutoff {
        let res = pfx * u32::from(m) + tail;
        Ok (res)
    } else {
        let add = br.read(1)?;
        let res = pfx * u32::from(m) + (tail - cutoff) * 2 + add + cutoff;
        Ok (res)
    }
}

fn read_rice(br: &mut BitReader, k: u8) -> BitReaderResult<u32> {
    let pfx = read_unary(br, 1)?;
    let ret = (pfx << k) + br.read(k)?;
    Ok(ret)
}

fn read_gamma(br: &mut BitReader) -> BitReaderResult<u32> {
    let mut ret = 1;
    while br.read(1)? != 1 {
        ret = (ret << 1) | br.read(1)?;
    }
    Ok(ret - 1)
}

fn read_gammap(br: &mut BitReader) -> BitReaderResult<u32> {
    let pfx = read_unary(br, 1)?;
    if pfx > 32 { return Err(BitReaderError::InvalidValue); }
    let ret = (1 << pfx) + br.read(pfx as u8)?;
    Ok(ret)
}

fn uval_to_sval0mp(uval: u32) -> i32 {
    if (uval & 1) != 0 { -(((uval + 1) >> 1) as i32) }
    else               { (uval >> 1) as i32 }
}

fn uval_to_sval0pm(uval: u32) -> i32 {
    if (uval & 1) != 0 { ((uval + 1) >> 1) as i32 }
    else               { -((uval >> 1) as i32) }
}

impl<'a> IntCodeReader for BitReader<'a> {
    #[inline(always)]
    fn read_code(&mut self, t: UintCodeType) -> BitReaderResult<u32> {
        match t {
            UintCodeType::UnaryOnes               => read_unary(self, 0),
            UintCodeType::UnaryZeroes             => read_unary(self, 1),
            UintCodeType::LimitedZeroes(len)      => read_unary_lim(self, len, 1),
            UintCodeType::LimitedOnes(len)        => read_unary_lim(self, len, 0),
            UintCodeType::LimitedUnary(len, term) => read_unary_lim(self, len, term),
            UintCodeType::Unary012                => read_unary_lim(self, 2, 0),
            UintCodeType::Unary210                => read_unary210(self),
            UintCodeType::Golomb(m)               => read_golomb(self, m),
            UintCodeType::Rice(k)                 => read_rice(self, k),
            UintCodeType::Gamma                   => read_gamma(self),
            UintCodeType::GammaP                  => read_gammap(self),
        }
    }
    fn read_code_signed(&mut self, t: IntCodeType) -> BitReaderResult<i32> {
        let uval =
            match t {
                IntCodeType::Golomb(m)               => read_golomb(self, m)?,
                IntCodeType::Rice(k)                 => read_rice(self, k)?,
                IntCodeType::Gamma                   => read_gamma(self)?,
                IntCodeType::GammaP                  => read_gammap(self)?,
            };
        match t {
            IntCodeType::Golomb(_)               => Ok(uval_to_sval0mp(uval)),
            IntCodeType::Rice(_)                 => Ok(uval_to_sval0mp(uval)),
            IntCodeType::Gamma                   => Ok(uval_to_sval0pm(uval)),
            IntCodeType::GammaP                  => Ok(uval_to_sval0pm(uval)),
        }
    }
}

/// Universal integer code writer trait for bitstream writer.
///
/// # Examples
///
/// Write an unsigned Golomb code:
/// ````
/// use nihav_core::io::bitwriter::*;
/// use nihav_core::io::intcode::{IntCodeWriter,UintCodeType};
///
/// let mut bw = BitWriter::new(Vec::new(), BitWriterMode::BE);
/// bw.write_code(UintCodeType::Golomb(3), 42);
/// ````
///
/// Write signed Elias code:
/// ````
/// use nihav_core::io::bitwriter::*;
/// use nihav_core::io::intcode::{IntCodeWriter,IntCodeType};
///
/// let mut bw = BitWriter::new(Vec::new(), BitWriterMode::BE);
/// bw.write_code_signed(IntCodeType::Gamma, 42);
/// ````
pub trait IntCodeWriter {
    /// Writes an unsigned integer code of requested type.
    fn write_code(&mut self, t: UintCodeType, val: u32);
    /// Writes signed integer code of requested type.
    fn write_code_signed(&mut self, t: IntCodeType, val: i32);
}

impl IntCodeWriter for BitWriter {
    #[inline(always)]
    fn write_code(&mut self, t: UintCodeType, val: u32) {
        match t {
            UintCodeType::UnaryOnes               => write_unary(self, val, 0),
            UintCodeType::UnaryZeroes             => write_unary(self, val, 1),
            UintCodeType::LimitedZeroes(len)      => write_unary_lim(self, val, len, 1),
            UintCodeType::LimitedOnes(len)        => write_unary_lim(self, val, len, 0),
            UintCodeType::LimitedUnary(len, term) => write_unary_lim(self, val, len, term),
            UintCodeType::Unary012                => write_unary_lim(self, val, 2, 0),
            UintCodeType::Unary210                => write_unary210(self, val),
            UintCodeType::Golomb(m)               => write_golomb(self, val, m),
            UintCodeType::Rice(k)                 => write_rice(self, val, k),
            UintCodeType::Gamma                   => write_gamma(self, val),
            UintCodeType::GammaP                  => write_gammap(self, val),
        };
    }
    fn write_code_signed(&mut self, t: IntCodeType, val: i32) {
        match t {
            IntCodeType::Golomb(m)  => write_golomb(self, sval0mp_to_uval(val), m),
            IntCodeType::Rice(k)    => write_rice(self, sval0mp_to_uval(val), k),
            IntCodeType::Gamma      => write_gamma(self, sval0pm_to_uval(val)),
            IntCodeType::GammaP     => write_gammap(self, sval0pm_to_uval(val)),
        };
    }
}

fn sval0mp_to_uval(val: i32) -> u32 {
    if val < 0  { (-val as u32) * 2 - 1 }
    else        { (val as u32) * 2 }
}

fn sval0pm_to_uval(val: i32) -> u32 {
    if val >= 0 { (val as u32) * 2 + 1 }
    else        { (-val as u32) * 2 }
}

fn write_unary210(bw: &mut BitWriter, val: u32) {
    bw.write_bit(val == 0);
    if val != 0 {
        bw.write_bit(val == 1);
    }
}

fn write_unary(bw: &mut BitWriter, val: u32, term: u32) {
    let term = term != 0;
    for _ in 0..val {
        bw.write_bit(!term);
    }
    bw.write_bit(term);
}

fn write_unary_lim(bw: &mut BitWriter, val: u32, maxval: u32, term: u32) {
    let term = term != 0;
    for _ in 0..val {
        bw.write_bit(!term);
    }
    if val < maxval {
        bw.write_bit(term);
    }
}

fn write_rice(bw: &mut BitWriter, val: u32, k: u8) {
    let mut exp = val >> k;
    while exp >= 16 {
        bw.write(0, 16);
        exp -= 16
    }
    if exp > 0 {
        bw.write(0, exp as u8);
    }
    bw.write1();
    if k > 0 {
        let mant = val & ((1 << k) - 1);
        bw.write(mant, k);
    }
}

fn write_golomb(bw: &mut BitWriter, val: u32, m: u8) {
    if m == 0 { return; }
    let nbits = (8 - m.leading_zeros()) as u8;
    if (m & (m - 1)) == 0 { return write_rice(bw, val, nbits); }
    let q = val / u32::from(m);
    let r = val % u32::from(m);
    let cutoff = u32::from((1 << nbits) - m);

    write_unary(bw, q, 0);
    if r < cutoff {
        bw.write(r, nbits - 1);
    } else {
        bw.write(r + cutoff, nbits);
    }
}

fn write_gamma(bw: &mut BitWriter, val: u32) {
    let val = val + 1;
    let bits = (32 - val.leading_zeros()) as u8;
    let mut mask = 1 << bits >> 2;
    while mask != 0 {
        bw.write0();
        bw.write_bit((val & mask) != 0);
        mask >>= 1;
    }
    bw.write1();
}

fn write_gammap(bw: &mut BitWriter, val: u32) {
    let bits = 31 - val.leading_zeros();
    write_unary(bw, bits, 1);
    bw.write(val - (1 << bits), bits as u8);
}

#[cfg(test)]
mod test {
    use super::*;
    use crate::io::bitreader::*;
    use crate::io::bitwriter::*;

    #[test]
    fn int_codes() {
        const GDATA: [u8; 6] = [0b000_001_01, 0b0_0110_011, 0b1_1000_100, 0b1_1010_101, 0b10_10111_1, 0b1000_0000];
        let src = &GDATA;
        let mut br = BitReader::new(src, BitReaderMode::BE);
        for i in 0..11 {
            assert_eq!(br.read_code(UintCodeType::Golomb(5)).unwrap(), i);
        }
    }
    #[test]
    fn rw_codes() {
        let mut bw = BitWriter::new(Vec::new(), BitWriterMode::BE);
        bw.write_code(UintCodeType::Golomb(5), 42);
        bw.write_code(UintCodeType::Gamma, 42);
        bw.write_code(UintCodeType::GammaP, 42);
        let data = bw.end();

        let mut br = BitReader::new(&data, BitReaderMode::BE);
        assert_eq!(br.read_code(UintCodeType::Golomb(5)).unwrap(), 42);
        assert_eq!(br.read_code(UintCodeType::Gamma).unwrap(), 42);
        assert_eq!(br.read_code(UintCodeType::GammaP).unwrap(), 42);
    }
}
Commit	Line	Data
5dd0d462	1	//! Some universal integer codes support for bitstream reader.
aca89041	2	use crate::io::bitreader::{BitReader, BitReaderError, BitReaderResult};
e450979b	3	use crate::io::bitwriter::BitWriter;
d7fcdd86	4
5dd0d462	5	/// Unsigned integer code types.
d7fcdd86 KS	6	#[derive(Debug)]
d7fcdd86 KS	7	pub enum UintCodeType {
5dd0d462	8	/// Code where number is represented as run of ones with terminating zero.
d7fcdd86	9	UnaryOnes,
5dd0d462	10	/// Code where number is represented as run of zeroes with terminating one.
d7fcdd86	11	UnaryZeroes,
5dd0d462	12	/// Code for 0, 1 and 2 coded as `0`, `10` and `11`
d7fcdd86	13	Unary012,
5dd0d462	14	/// Code for 0, 1 and 2 coded as `11`, `10` and `0`
d7fcdd86	15	Unary210,
5dd0d462	16	/// General limited unary code with defined run and terminating bit.
d7fcdd86	17	LimitedUnary(u32, u32),
5dd0d462 KS	18	/// Limited run of zeroes with terminating one (unless the code has maximum length).
	19	///
	20	/// [`Unary012`] is essentially an alias for `LimitedZeroes(2)`.
	21	///
	22	/// [`Unary012`]: #variant.Unary012
3b89a4d7	23	LimitedZeroes(u32),
5dd0d462	24	/// Limited run of one with terminating zero (unless the code has maximum length).
3b89a4d7	25	LimitedOnes(u32),
5dd0d462	26	/// Golomb code.
d7fcdd86	27	Golomb(u8),
5dd0d462	28	/// Rice code.
d7fcdd86	29	Rice(u8),
5dd0d462	30	/// Elias Gamma code (interleaved).
d7fcdd86	31	Gamma,
5dd0d462	32	/// Elias Gamma' code (sometimes incorrectly called exp-Golomb).
d7fcdd86 KS	33	GammaP,
	34	}
	35
5dd0d462	36	/// Signed integer code types.
d7fcdd86	37	pub enum IntCodeType {
5dd0d462	38	/// Golomb code. Last bit represents the sign.
d7fcdd86	39	Golomb(u8),
5dd0d462	40	/// Golomb code. Last bit represents the sign.
d7fcdd86	41	Rice(u8),
5dd0d462	42	/// Elias Gamma code. Unsigned values are remapped as 0, 1, -1, 2, -2, ...
d7fcdd86	43	Gamma,
5dd0d462	44	/// Elias Gamma' code. Unsigned values are remapped as 0, 1, -1, 2, -2, ...
d7fcdd86 KS	45	GammaP,
	46	}
	47
5dd0d462 KS	48	/// Universal integer code reader trait for bitstream reader.
	49	///
	50	/// # Examples
	51	///
	52	/// Read an unsigned Golomb code:
	53	/// ````
	54	/// use nihav_core::io::bitreader::*;
	55	/// use nihav_core::io::intcode::{IntCodeReader,UintCodeType};
	56	///
	57	/// # fn foo() -> BitReaderResult<()> {
	58	/// let mem: [u8; 4] = [ 0, 1, 2, 3];
	59	/// let mut br = BitReader::new(&mem, BitReaderMode::BE);
	60	/// let val = br.read_code(UintCodeType::Golomb(3))?;
	61	/// # Ok(())
	62	/// # }
	63	/// ````
	64	///
	65	/// Read signed Elias code:
	66	/// ````
	67	/// use nihav_core::io::bitreader::*;
	68	/// use nihav_core::io::intcode::{IntCodeReader,IntCodeType};
	69	///
	70	/// # fn foo() -> BitReaderResult<()> {
	71	/// let mem: [u8; 4] = [ 0, 1, 2, 3];
	72	/// let mut br = BitReader::new(&mem, BitReaderMode::BE);
	73	/// let val = br.read_code_signed(IntCodeType::Gamma)?;
	74	/// # Ok(())
	75	/// # }
	76	/// ````
d7fcdd86	77	pub trait IntCodeReader {
5dd0d462	78	/// Reads an unsigned integer code of requested type.
d7fcdd86	79	fn read_code(&mut self, t: UintCodeType) -> BitReaderResult<u32>;
5dd0d462	80	/// Reads signed integer code of requested type.
d7fcdd86 KS	81	fn read_code_signed(&mut self, t: IntCodeType) -> BitReaderResult<i32>;
	82	}
	83
	84	fn read_unary(br: &mut BitReader, terminator: u32) -> BitReaderResult<u32> {
	85	let mut res: u32 = 0;
	86	loop {
	87	if br.read(1)? == terminator { return Ok(res); }
e243ceb4	88	res += 1;
d7fcdd86 KS	89	}
	90	}
	91
8f879477	92	fn read_unary_lim(br: &mut BitReader, len: u32, terminator: u32) -> BitReaderResult<u32> {
d7fcdd86 KS	93	let mut res: u32 = 0;
	94	loop {
	95	if br.read(1)? == terminator { return Ok(res); }
e243ceb4	96	res += 1;
d7fcdd86 KS	97	if res == len { return Ok(res); }
	98	}
	99	}
	100
	101	fn read_unary210(br: &mut BitReader) -> BitReaderResult<u32> {
	102	let val = read_unary_lim(br, 2, 0)?;
	103	Ok(2 - val)
	104	}
	105
	106	fn read_golomb(br: &mut BitReader, m: u8) -> BitReaderResult<u32> {
	107	if m == 0 { return Err(BitReaderError::InvalidValue); }
	108	let nbits = (8 - m.leading_zeros()) as u8;
	109	if (m & (m - 1)) == 0 { return read_rice(br, nbits); }
e243ceb4	110	let cutoff = u32::from((1 << nbits) - m);
d7fcdd86 KS	111	let pfx = read_unary(br, 0)?;
	112	let tail = br.read(nbits - 1)?;
	113	if tail < cutoff {
e243ceb4	114	let res = pfx * u32::from(m) + tail;
d7fcdd86 KS	115	Ok (res)
	116	} else {
	117	let add = br.read(1)?;
e243ceb4	118	let res = pfx * u32::from(m) + (tail - cutoff) * 2 + add + cutoff;
d7fcdd86 KS	119	Ok (res)
	120	}
	121	}
	122
	123	fn read_rice(br: &mut BitReader, k: u8) -> BitReaderResult<u32> {
	124	let pfx = read_unary(br, 1)?;
	125	let ret = (pfx << k) + br.read(k)?;
	126	Ok(ret)
	127	}
	128
	129	fn read_gamma(br: &mut BitReader) -> BitReaderResult<u32> {
6036ce28	130	let mut ret = 1;
d7fcdd86 KS	131	while br.read(1)? != 1 {
	132	ret = (ret << 1) \| br.read(1)?;
	133	}
6036ce28	134	Ok(ret - 1)
d7fcdd86 KS	135	}
	136
	137	fn read_gammap(br: &mut BitReader) -> BitReaderResult<u32> {
	138	let pfx = read_unary(br, 1)?;
	139	if pfx > 32 { return Err(BitReaderError::InvalidValue); }
	140	let ret = (1 << pfx) + br.read(pfx as u8)?;
	141	Ok(ret)
	142	}
	143
	144	fn uval_to_sval0mp(uval: u32) -> i32 {
ffe6400e	145	if (uval & 1) != 0 { -(((uval + 1) >> 1) as i32) }
d7fcdd86 KS	146	else { (uval >> 1) as i32 }
	147	}
	148
	149	fn uval_to_sval0pm(uval: u32) -> i32 {
	150	if (uval & 1) != 0 { ((uval + 1) >> 1) as i32 }
	151	else { -((uval >> 1) as i32) }
	152	}
	153
	154	impl<'a> IntCodeReader for BitReader<'a> {
83b49341	155	#[inline(always)]
d7fcdd86 KS	156	fn read_code(&mut self, t: UintCodeType) -> BitReaderResult<u32> {
	157	match t {
	158	UintCodeType::UnaryOnes => read_unary(self, 0),
	159	UintCodeType::UnaryZeroes => read_unary(self, 1),
3b89a4d7 KS	160	UintCodeType::LimitedZeroes(len) => read_unary_lim(self, len, 1),
3b89a4d7 KS	161	UintCodeType::LimitedOnes(len) => read_unary_lim(self, len, 0),
8f879477	162	UintCodeType::LimitedUnary(len, term) => read_unary_lim(self, len, term),
d7fcdd86 KS	163	UintCodeType::Unary012 => read_unary_lim(self, 2, 0),
	164	UintCodeType::Unary210 => read_unary210(self),
	165	UintCodeType::Golomb(m) => read_golomb(self, m),
	166	UintCodeType::Rice(k) => read_rice(self, k),
	167	UintCodeType::Gamma => read_gamma(self),
	168	UintCodeType::GammaP => read_gammap(self),
	169	}
	170	}
d7fcdd86 KS	171	fn read_code_signed(&mut self, t: IntCodeType) -> BitReaderResult<i32> {
	172	let uval =
	173	match t {
	174	IntCodeType::Golomb(m) => read_golomb(self, m)?,
	175	IntCodeType::Rice(k) => read_rice(self, k)?,
	176	IntCodeType::Gamma => read_gamma(self)?,
	177	IntCodeType::GammaP => read_gammap(self)?,
	178	};
	179	match t {
1ff7036b KS	180	IntCodeType::Golomb(_) => Ok(uval_to_sval0mp(uval)),
1ff7036b KS	181	IntCodeType::Rice(_) => Ok(uval_to_sval0mp(uval)),
d7fcdd86 KS	182	IntCodeType::Gamma => Ok(uval_to_sval0pm(uval)),
	183	IntCodeType::GammaP => Ok(uval_to_sval0pm(uval)),
	184	}
	185	}
	186	}
	187
e450979b KS	188	/// Universal integer code writer trait for bitstream writer.
	189	///
	190	/// # Examples
	191	///
	192	/// Write an unsigned Golomb code:
	193	/// ````
	194	/// use nihav_core::io::bitwriter::*;
	195	/// use nihav_core::io::intcode::{IntCodeWriter,UintCodeType};
	196	///
	197	/// let mut bw = BitWriter::new(Vec::new(), BitWriterMode::BE);
	198	/// bw.write_code(UintCodeType::Golomb(3), 42);
	199	/// ````
	200	///
	201	/// Write signed Elias code:
	202	/// ````
	203	/// use nihav_core::io::bitwriter::*;
	204	/// use nihav_core::io::intcode::{IntCodeWriter,IntCodeType};
	205	///
	206	/// let mut bw = BitWriter::new(Vec::new(), BitWriterMode::BE);
712d6d9d	207	/// bw.write_code_signed(IntCodeType::Gamma, 42);
e450979b KS	208	/// ````
	209	pub trait IntCodeWriter {
	210	/// Writes an unsigned integer code of requested type.
	211	fn write_code(&mut self, t: UintCodeType, val: u32);
	212	/// Writes signed integer code of requested type.
	213	fn write_code_signed(&mut self, t: IntCodeType, val: i32);
	214	}
	215
	216	impl IntCodeWriter for BitWriter {
	217	#[inline(always)]
	218	fn write_code(&mut self, t: UintCodeType, val: u32) {
	219	match t {
	220	UintCodeType::UnaryOnes => write_unary(self, val, 0),
	221	UintCodeType::UnaryZeroes => write_unary(self, val, 1),
	222	UintCodeType::LimitedZeroes(len) => write_unary_lim(self, val, len, 1),
	223	UintCodeType::LimitedOnes(len) => write_unary_lim(self, val, len, 0),
	224	UintCodeType::LimitedUnary(len, term) => write_unary_lim(self, val, len, term),
	225	UintCodeType::Unary012 => write_unary_lim(self, val, 2, 0),
	226	UintCodeType::Unary210 => write_unary210(self, val),
	227	UintCodeType::Golomb(m) => write_golomb(self, val, m),
	228	UintCodeType::Rice(k) => write_rice(self, val, k),
	229	UintCodeType::Gamma => write_gamma(self, val),
	230	UintCodeType::GammaP => write_gammap(self, val),
	231	};
	232	}
	233	fn write_code_signed(&mut self, t: IntCodeType, val: i32) {
	234	match t {
	235	IntCodeType::Golomb(m) => write_golomb(self, sval0mp_to_uval(val), m),
	236	IntCodeType::Rice(k) => write_rice(self, sval0mp_to_uval(val), k),
	237	IntCodeType::Gamma => write_gamma(self, sval0pm_to_uval(val)),
	238	IntCodeType::GammaP => write_gammap(self, sval0pm_to_uval(val)),
	239	};
	240	}
	241	}
	242
	243	fn sval0mp_to_uval(val: i32) -> u32 {
	244	if val < 0 { (-val as u32) * 2 - 1 }
	245	else { (val as u32) * 2 }
	246	}
	247
	248	fn sval0pm_to_uval(val: i32) -> u32 {
	249	if val >= 0 { (val as u32) * 2 + 1 }
	250	else { (-val as u32) * 2 }
	251	}
	252
	253	fn write_unary210(bw: &mut BitWriter, val: u32) {
	254	bw.write_bit(val == 0);
	255	if val != 0 {
	256	bw.write_bit(val == 1);
	257	}
	258	}
	259
	260	fn write_unary(bw: &mut BitWriter, val: u32, term: u32) {
	261	let term = term != 0;
	262	for _ in 0..val {
	263	bw.write_bit(!term);
	264	}
	265	bw.write_bit(term);
	266	}
	267
	268	fn write_unary_lim(bw: &mut BitWriter, val: u32, maxval: u32, term: u32) {
	269	let term = term != 0;
	270	for _ in 0..val {
	271	bw.write_bit(!term);
272	}
273	if val < maxval {
274	bw.write_bit(term);
275	}
276	}
277
278	fn write_rice(bw: &mut BitWriter, val: u32, k: u8) {
279	let mut exp = val >> k;
280	while exp >= 16 {
281	bw.write(0, 16);
282	exp -= 16
283	}
284	if exp > 0 {
285	bw.write(0, exp as u8);
286	}
287	bw.write1();
288	if k > 0 {
289	let mant = val & ((1 << k) - 1);
290	bw.write(mant, k);
291	}
292	}
293
294	fn write_golomb(bw: &mut BitWriter, val: u32, m: u8) {
295	if m == 0 { return; }
296	let nbits = (8 - m.leading_zeros()) as u8;
297	if (m & (m - 1)) == 0 { return write_rice(bw, val, nbits); }
298	let q = val / u32::from(m);
299	let r = val % u32::from(m);
300	let cutoff = u32::from((1 << nbits) - m);
301
302	write_unary(bw, q, 0);
303	if r < cutoff {
304	bw.write(r, nbits - 1);
305	} else {
306	bw.write(r + cutoff, nbits);
307	}
308	}
309
310	fn write_gamma(bw: &mut BitWriter, val: u32) {
311	let val = val + 1;
312	let bits = (32 - val.leading_zeros()) as u8;
313	let mut mask = 1 << bits >> 2;
314	while mask != 0 {
315	bw.write0();
316	bw.write_bit((val & mask) != 0);
317	mask >>= 1;
318	}
319	bw.write1();
320	}
321
322	fn write_gammap(bw: &mut BitWriter, val: u32) {
323	let bits = 31 - val.leading_zeros();
324	write_unary(bw, bits, 1);
325	bw.write(val - (1 << bits), bits as u8);
326	}
327
d7fcdd86 KS	328	#[cfg(test)]
	329	mod test {
	330	use super::*;
aca89041	331	use crate::io::bitreader::*;
e450979b	332	use crate::io::bitwriter::*;
d7fcdd86 KS	333
	334	#[test]
	335	fn int_codes() {
	336	const GDATA: [u8; 6] = [0b000_001_01, 0b0_0110_011, 0b1_1000_100, 0b1_1010_101, 0b10_10111_1, 0b1000_0000];
	337	let src = &GDATA;
fa90ccfb	338	let mut br = BitReader::new(src, BitReaderMode::BE);
d7fcdd86 KS	339	for i in 0..11 {
	340	assert_eq!(br.read_code(UintCodeType::Golomb(5)).unwrap(), i);
	341	}
	342	}
e450979b KS	343	#[test]
	344	fn rw_codes() {
	345	let mut bw = BitWriter::new(Vec::new(), BitWriterMode::BE);
	346	bw.write_code(UintCodeType::Golomb(5), 42);
	347	bw.write_code(UintCodeType::Gamma, 42);
	348	bw.write_code(UintCodeType::GammaP, 42);
	349	let data = bw.end();
e450979b KS	350
	351	let mut br = BitReader::new(&data, BitReaderMode::BE);
	352	assert_eq!(br.read_code(UintCodeType::Golomb(5)).unwrap(), 42);
	353	assert_eq!(br.read_code(UintCodeType::Gamma).unwrap(), 42);
	354	assert_eq!(br.read_code(UintCodeType::GammaP).unwrap(), 42);
	355	}
d7fcdd86	356	}