compactly 0.1.7

Compactly encode data types using adaptive arithmetic coding
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use crate::Sorted;

use super::Encode;
use super::{bit_context::BitContext, EncodingStrategy};

impl Encode for bool {
    type Context = BitContext;
    #[inline]
    fn encode<E: super::EntropyCoder>(&self, writer: &mut E, ctx: &mut Self::Context) {
        // println!("Encoding {self:?}");
        writer.encode_bit(ctx.probability(), *self);
        *ctx = ctx.adapt(*self);
    }
    #[inline]
    fn decode<D: super::EntropyDecoder>(
        reader: &mut D,
        ctx: &mut Self::Context,
    ) -> Result<Self, std::io::Error> {
        let b = reader.decode_bit(ctx)?;
        // println!("Decoding {b:?}");
        Ok(b)
    }
}

impl EncodingStrategy<bool> for Sorted {
    type Context = BitContext;
    fn decode<D: super::EntropyDecoder>(
        reader: &mut D,
        ctx: &mut Self::Context,
    ) -> Result<bool, std::io::Error> {
        bool::decode(reader, ctx)
    }
    fn encode<E: super::EntropyCoder>(value: &bool, writer: &mut E, ctx: &mut Self::Context) {
        value.encode(writer, ctx)
    }
}

#[test]
fn size() {
    use super::assert_bits;
    assert_bits!(true, 1);
    assert_bits!(false, 1);
    assert_bits!([false; 128], 7);
    assert_bits!([true; 2], 1);
    assert_bits!([true; 3], 1);
    assert_bits!([true; 16], 3);
    assert_bits!([true; 64], 5);
    assert_bits!([false, true], 3);
}

#[test]
fn millibits_required() {
    use super::Millibits;
    let mut bc = BitContext::default();
    assert_eq!(bc.probability().as_f64(), 0.5);

    assert_eq!(false.millibits(), Millibits::bits(1));
    assert_eq!(true.millibits(), Millibits::bits(1));

    macro_rules! assert_millibits {
        ($bit:literal, $ctx:expr, $expected:expr) => {{
            let mut mb = Millibits::new(0);
            $bit.encode(&mut mb, $ctx);
            assert_eq!(mb, $expected);
        }};
    }

    assert_millibits!(true, &mut bc, Millibits::bits(1));

    assert_eq!(bc, BitContext::True1False0);
    assert!(bc.probability().as_f64() < 0.5);

    assert_millibits!(true, &mut BitContext::True1False0, Millibits::new(582));
    assert_millibits!(false, &mut BitContext::True1False0, Millibits::new(1590));

    assert_millibits!(true, &mut bc, Millibits::new(582));
    assert_millibits!(true, &mut bc, Millibits::new(415));
    assert_millibits!(false, &mut bc, Millibits::new(2327));
    assert_millibits!(false, &mut bc, Millibits::new(1590));
    assert_millibits!(false, &mut bc, Millibits::new(1218));
    assert_millibits!(false, &mut bc, Millibits::new(1000));
}