av1-obu-parser 0.1.0-alpha.1

A pure Rust parser for AV1 OBU bitstreams and IVF containers.
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

use super::Buffer;

#[derive(Debug, Clone, Copy)]
pub struct TileListEntry {
    pub anchor_frame_idx: u8,
    pub anchor_tile_row: u8,
    pub anchor_tile_col: u8,
    pub tile_data_size: u16,
    pub coded_tile_data: u32,
}

impl TileListEntry {
    pub fn decode(buf: &mut Buffer) -> Self {
        // anchor_frame_idx	f(8)
        let anchor_frame_idx = buf.get_bits(8) as u8;

        // anchor_tile_row	f(8)
        let anchor_tile_row = buf.get_bits(8) as u8;

        // anchor_tile_col	f(8)
        let anchor_tile_col = buf.get_bits(8) as u8;

        // tile_data_size_minus_1	f(16)
        let tile_data_size = buf.get_bits(16) as u16 + 1;

        // coded_tile_data	f(N)
        let coded_tile_data = buf.get_bits(8 * tile_data_size as usize);
        Self {
            anchor_frame_idx,
            anchor_tile_col,
            anchor_tile_row,
            tile_data_size,
            coded_tile_data,
        }
    }
}

#[derive(Debug, Clone)]
pub struct TileList {
    pub output_frame_width_in_tiles: u8,
    pub output_frame_height_in_tiles: u8,
    pub tile_list_entrys: Vec<TileListEntry>,
}

impl TileList {
    pub fn decode(buf: &mut Buffer) -> Self {
        // output_frame_width_in_tiles_minus_1	f(8)
        let output_frame_width_in_tiles = buf.get_bits(8) as u8;

        // output_frame_height_in_tiles_minus_1	f(8)
        let output_frame_height_in_tiles = buf.get_bits(8) as u8;

        // tile_count_minus_1	f(16)
        let tile_count = buf.get_bits(16) as usize;
        let mut tile_list_entrys = Vec::with_capacity(tile_count);
        for _ in 0..tile_count {
            tile_list_entrys.push(TileListEntry::decode(buf));
        }

        Self {
            output_frame_height_in_tiles,
            output_frame_width_in_tiles,
            tile_list_entrys,
        }
    }
}