boxcars 0.11.0

Rocket league replay parser
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

use bitter::{BitReader, LittleEndianReader};

#[inline]
pub(crate) const fn bit_width(input: u64) -> u32 {
    (core::mem::size_of::<u64>() as u32) * 8 - input.leading_zeros()
}

pub(crate) trait RlBits {
    fn peek_and_consume(&mut self, bits: u32) -> u64;
    fn peek_bits_max_computed(&mut self, bits: u32, max: u64) -> u64;
    fn read_bits_max_computed(&mut self, bits: u32, max: u64) -> Option<u64>;
    fn if_get<T, F>(&mut self, f: F) -> Option<Option<T>>
    where
        F: FnMut(&mut Self) -> Option<T>;
}

impl RlBits for LittleEndianReader<'_> {
    #[inline]
    fn peek_and_consume(&mut self, bits: u32) -> u64 {
        let res = self.peek(bits);
        self.consume(bits);
        res
    }

    #[inline]
    fn if_get<T, F>(&mut self, mut f: F) -> Option<Option<T>>
    where
        F: FnMut(&mut Self) -> Option<T>,
    {
        self.read_bit()
            .and_then(|bit| if bit { f(self).map(Some) } else { Some(None) })
    }

    #[inline]
    fn peek_bits_max_computed(&mut self, bits: u32, max: u64) -> u64 {
        debug_assert!(core::cmp::max(bit_width(max), 1) == bits + 1);

        let data = self.peek_and_consume(bits);
        let up = data + (1 << bits);
        if up >= max {
            data
        } else if self.peek_and_consume(1) != 0 {
            up
        } else {
            data
        }
    }

    #[inline]
    fn read_bits_max_computed(&mut self, bits: u32, max: u64) -> Option<u64> {
        debug_assert!(core::cmp::max(bit_width(max), 1) == bits + 1);
        self.read_bits(bits).and_then(|data| {
            let up = data + (1 << bits);
            if up >= max {
                Some(data)
            } else {
                // Check the next bit
                self.read_bit().map(|x| if x { up } else { data })
            }
        })
    }
}