mp4_atom/moov/trak/mdia/minf/stbl/
stco.rs

1use crate::*;
2
3#[derive(Debug, Clone, PartialEq, Eq, Default)]
4#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
5pub struct Stco {
6    pub entries: Vec<u32>,
7}
8
9impl AtomExt for Stco {
10    type Ext = ();
11
12    const KIND_EXT: FourCC = FourCC::new(b"stco");
13
14    fn decode_body_ext<B: Buf>(buf: &mut B, _ext: ()) -> Result<Self> {
15        let count = u32::decode(buf)?;
16        let mut entries = Vec::new();
17
18        for _ in 0..count {
19            let chunk_offset = u32::decode(buf)?;
20            entries.push(chunk_offset);
21        }
22
23        Ok(Stco { entries })
24    }
25
26    fn encode_body_ext<B: BufMut>(&self, buf: &mut B) -> Result<()> {
27        (self.entries.len() as u32).encode(buf)?;
28        for chunk_offset in self.entries.iter() {
29            (chunk_offset).encode(buf)?;
30        }
31
32        Ok(())
33    }
34}
35
36#[cfg(test)]
37mod tests {
38    use super::*;
39
40    #[test]
41    fn test_stco() {
42        let expected = Stco {
43            entries: vec![267, 1970, 2535, 2803, 11843, 22223, 33584],
44        };
45        let mut buf = Vec::new();
46        expected.encode(&mut buf).unwrap();
47
48        let mut buf = buf.as_ref();
49        let decoded = Stco::decode(&mut buf).unwrap();
50        assert_eq!(decoded, expected);
51    }
52}