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

1use crate::*;
2
3#[derive(Debug, Clone, PartialEq, Eq, Default)]
4#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
5pub struct Stss {
6    pub entries: Vec<u32>,
7}
8
9impl AtomExt for Stss {
10    type Ext = ();
11
12    const KIND_EXT: FourCC = FourCC::new(b"stss");
13
14    fn decode_body_ext<B: Buf>(buf: &mut B, _ext: ()) -> Result<Self> {
15        let entry_count = u32::decode(buf)?;
16
17        let mut entries = Vec::new();
18        for _ in 0..entry_count {
19            let sample_number = u32::decode(buf)?;
20            entries.push(sample_number);
21        }
22
23        Ok(Stss { 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 sample_number in self.entries.iter() {
29            sample_number.encode(buf)?;
30        }
31
32        Ok(())
33    }
34}
35
36#[cfg(test)]
37mod tests {
38    use super::*;
39
40    #[test]
41    fn test_stss() {
42        let expected = Stss {
43            entries: vec![1, 61, 121, 181, 241, 301, 361, 421, 481],
44        };
45        let mut buf = Vec::new();
46        expected.encode(&mut buf).unwrap();
47
48        let mut buf = buf.as_ref();
49        let decoded = Stss::decode(&mut buf).unwrap();
50        assert_eq!(decoded, expected);
51    }
52}