scuffle_mp4/boxes/types/
mfhd.rs

1use std::io;
2
3use byteorder::{BigEndian, ReadBytesExt, WriteBytesExt};
4use bytes::Bytes;
5
6use crate::boxes::header::{BoxHeader, FullBoxHeader};
7use crate::boxes::traits::BoxType;
8
9#[derive(Debug, Clone, PartialEq)]
10/// Movie Fragment Header Box
11/// ISO/IEC 14496-12:2022(E) - 8.8.5
12pub struct Mfhd {
13    pub header: FullBoxHeader,
14    pub sequence_number: u32,
15}
16
17impl Mfhd {
18    pub fn new(sequence_number: u32) -> Self {
19        Self {
20            header: FullBoxHeader::new(Self::NAME, 0, 0),
21            sequence_number,
22        }
23    }
24}
25
26impl BoxType for Mfhd {
27    const NAME: [u8; 4] = *b"mfhd";
28
29    fn demux(header: BoxHeader, data: Bytes) -> io::Result<Self> {
30        let mut reader = io::Cursor::new(data);
31
32        let header = FullBoxHeader::demux(header, &mut reader)?;
33
34        let sequence_number = reader.read_u32::<BigEndian>()?;
35
36        Ok(Self { header, sequence_number })
37    }
38
39    fn primitive_size(&self) -> u64 {
40        self.header.size() + 4 // sequence_number
41    }
42
43    fn primitive_mux<T: io::Write>(&self, writer: &mut T) -> io::Result<()> {
44        self.header.mux(writer)?;
45        writer.write_u32::<BigEndian>(self.sequence_number)?;
46
47        Ok(())
48    }
49
50    fn validate(&self) -> io::Result<()> {
51        if self.header.version != 0 {
52            return Err(io::Error::new(io::ErrorKind::InvalidData, "mfhd box version must be 0"));
53        }
54
55        if self.header.flags != 0 {
56            return Err(io::Error::new(io::ErrorKind::InvalidData, "mfhd box flags must be 0"));
57        }
58
59        Ok(())
60    }
61}