scuffle_mp4/boxes/types/
trak.rs

1use std::io;
2
3use bytes::{Buf, Bytes};
4
5use super::edts::Edts;
6use super::mdia::Mdia;
7use super::tkhd::Tkhd;
8use crate::boxes::DynBox;
9use crate::boxes::header::BoxHeader;
10use crate::boxes::traits::BoxType;
11
12#[derive(Debug, Clone, PartialEq)]
13/// Track Box
14/// ISO/IEC 14496-12:2022(E) - 8.3.1
15pub struct Trak {
16    pub header: BoxHeader,
17    pub tkhd: Tkhd,
18    pub edts: Option<Edts>,
19    pub mdia: Mdia,
20    pub unknown: Vec<DynBox>,
21}
22
23impl Trak {
24    pub fn new(tkhd: Tkhd, edts: Option<Edts>, mdia: Mdia) -> Self {
25        Self {
26            header: BoxHeader::new(Self::NAME),
27            tkhd,
28            edts,
29            mdia,
30            unknown: Vec::new(),
31        }
32    }
33}
34
35impl BoxType for Trak {
36    const NAME: [u8; 4] = *b"trak";
37
38    fn demux(header: BoxHeader, data: Bytes) -> io::Result<Self> {
39        let mut reader = io::Cursor::new(data);
40        let mut tkhd = None;
41        let mut edts = None;
42        let mut mdia = None;
43        let mut unknown = Vec::new();
44
45        while reader.has_remaining() {
46            let dyn_box = DynBox::demux(&mut reader)?;
47
48            match dyn_box {
49                DynBox::Tkhd(b) => {
50                    tkhd = Some(*b);
51                }
52                DynBox::Edts(b) => {
53                    edts = Some(*b);
54                }
55                DynBox::Mdia(b) => {
56                    mdia = Some(*b);
57                }
58                _ => {
59                    unknown.push(dyn_box);
60                }
61            }
62        }
63
64        let tkhd = tkhd.ok_or_else(|| io::Error::new(io::ErrorKind::InvalidData, "trak box is missing tkhd box"))?;
65
66        let mdia = mdia.ok_or_else(|| io::Error::new(io::ErrorKind::InvalidData, "trak box is missing mdia box"))?;
67
68        Ok(Self {
69            header,
70            tkhd,
71            edts,
72            mdia,
73            unknown,
74        })
75    }
76
77    fn primitive_size(&self) -> u64 {
78        self.tkhd.size()
79            + self.edts.as_ref().map(|b| b.size()).unwrap_or(0)
80            + self.mdia.size()
81            + self.unknown.iter().map(|b| b.size()).sum::<u64>()
82    }
83
84    fn primitive_mux<T: io::Write>(&self, writer: &mut T) -> io::Result<()> {
85        self.tkhd.mux(writer)?;
86        if let Some(edts) = &self.edts {
87            edts.mux(writer)?;
88        }
89
90        self.mdia.mux(writer)?;
91
92        for box_ in &self.unknown {
93            box_.mux(writer)?;
94        }
95
96        Ok(())
97    }
98}
scuffle_mp4/boxes/types/trak.rs

scuffle_mp4/boxes/types/
trak.rs
