wav1c 0.2.0

Wondrous AV1 encoder written in safe Rust.
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
83
84
85
86
87

#[repr(u8)]
#[derive(Clone, Copy)]
pub enum ObuType {
    SequenceHeader = 1,
    TemporalDelimiter = 2,
    Frame = 6,
}

pub fn leb128_encode(mut value: u64) -> Vec<u8> {
    let mut result = Vec::new();
    loop {
        let mut byte = (value & 0x7F) as u8;
        value >>= 7;
        if value != 0 {
            byte |= 0x80;
        }
        result.push(byte);
        if value == 0 {
            break;
        }
    }
    result
}

pub fn obu_wrap(obu_type: ObuType, payload: &[u8]) -> Vec<u8> {
    let header_byte = (obu_type as u8) << 3 | (1 << 1);
    let size_bytes = leb128_encode(payload.len() as u64);
    let mut result = Vec::with_capacity(1 + size_bytes.len() + payload.len());
    result.push(header_byte);
    result.extend_from_slice(&size_bytes);
    result.extend_from_slice(payload);
    result
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn leb128_zero() {
        assert_eq!(leb128_encode(0), vec![0x00]);
    }

    #[test]
    fn leb128_small() {
        assert_eq!(leb128_encode(6), vec![0x06]);
    }

    #[test]
    fn leb128_127() {
        assert_eq!(leb128_encode(127), vec![0x7F]);
    }

    #[test]
    fn leb128_128() {
        assert_eq!(leb128_encode(128), vec![0x80, 0x01]);
    }

    #[test]
    fn leb128_300() {
        assert_eq!(leb128_encode(300), vec![0xAC, 0x02]);
    }

    #[test]
    fn obu_temporal_delimiter() {
        let result = obu_wrap(ObuType::TemporalDelimiter, &[]);
        assert_eq!(result, vec![0x12, 0x00]);
    }

    #[test]
    fn obu_sequence_header_6bytes() {
        let payload = vec![0x18, 0x15, 0x7f, 0xfc, 0x00, 0x08];
        let result = obu_wrap(ObuType::SequenceHeader, &payload);
        assert_eq!(result[0], 0x0A);
        assert_eq!(result[1], 0x06);
        assert_eq!(&result[2..], &payload[..]);
    }

    #[test]
    fn obu_frame_16bytes() {
        let payload = vec![0u8; 16];
        let result = obu_wrap(ObuType::Frame, &payload);
        assert_eq!(result[0], 0x32);
        assert_eq!(result[1], 0x10);
        assert_eq!(result.len(), 2 + 16);
    }
}