media-codec-h264 0.1.1

An H.264 decoder implementation for media-codec
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

//! H.264/AVC Parameter Sets Container

use std::array;

use media_core::Result;

use crate::{
    constants::{MAX_PPS_COUNT, MAX_SPS_COUNT},
    pps::Pps,
    sps::Sps,
};

/// Container for all SPS and PPS parameter sets
pub struct ParameterSets {
    /// SPS list (indexed by seq_parameter_set_id)
    pub sps_list: [Option<Box<Sps>>; MAX_SPS_COUNT],
    /// PPS list (indexed by pic_parameter_set_id)
    pub pps_list: [Option<Box<Pps>>; MAX_PPS_COUNT],
}

impl ParameterSets {
    pub fn new() -> Self {
        Self {
            sps_list: array::from_fn(|_| None),
            pps_list: array::from_fn(|_| None),
        }
    }

    pub fn get_sps(&self, id: u32) -> Option<&Sps> {
        self.sps_list.get(id as usize)?.as_deref()
    }

    pub fn get_pps(&self, id: u32) -> Option<&Pps> {
        self.pps_list.get(id as usize)?.as_deref()
    }

    pub fn add_sps(&mut self, sps: Sps) {
        let id = sps.seq_parameter_set_id as usize;
        if id < MAX_SPS_COUNT {
            self.sps_list[id] = Some(Box::new(sps));
        }
    }

    pub fn add_pps(&mut self, pps: Pps) -> Result<()> {
        let pps_id = pps.pic_parameter_set_id as usize;
        self.pps_list[pps_id] = Some(Box::new(pps));
        Ok(())
    }
}

impl Default for ParameterSets {
    fn default() -> Self {
        Self::new()
    }
}