mp4-edit 0.1.1

mp4 read/write library designed with audiobooks in mind
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

use crate::{
    atom::{
        atom_ref::{unwrap_atom_data, AtomRef, AtomRefMut},
        hdlr::HDLR,
        mdhd::MDHD,
        HandlerReferenceAtom, MediaHeaderAtom, MinfAtomRef, MinfAtomRefMut, MINF,
    },
    Atom, AtomData, FourCC,
};

pub const MDIA: FourCC = FourCC::new(b"mdia");

#[derive(Debug, Clone, Copy)]
pub struct MdiaAtomRef<'a>(pub(crate) AtomRef<'a>);

impl<'a> MdiaAtomRef<'a> {
    pub fn children(&self) -> impl Iterator<Item = &'a Atom> {
        self.0.children()
    }

    /// Finds the MDHD atom
    pub fn header(&self) -> Option<&'a MediaHeaderAtom> {
        let atom = self.0.find_child(MDHD)?;
        match atom.data.as_ref()? {
            AtomData::MediaHeader(data) => Some(data),
            _ => None,
        }
    }

    /// Finds the HDLR atom
    pub fn handler_reference(&self) -> Option<&'a HandlerReferenceAtom> {
        let atom = self.0.find_child(HDLR)?;
        match atom.data.as_ref()? {
            AtomData::HandlerReference(data) => Some(data),
            _ => None,
        }
    }

    /// Finds the MINF atom
    pub fn media_information(&self) -> MinfAtomRef<'a> {
        let atom = self.0.find_child(MINF);
        MinfAtomRef(AtomRef(atom))
    }
}

#[derive(Debug)]
pub struct MdiaAtomRefMut<'a>(pub(crate) AtomRefMut<'a>);

impl<'a> MdiaAtomRefMut<'a> {
    pub fn as_ref(&self) -> MdiaAtomRef<'_> {
        MdiaAtomRef(self.0.as_ref())
    }

    pub fn into_ref(self) -> MdiaAtomRef<'a> {
        MdiaAtomRef(self.0.into_ref())
    }

    pub fn children(&mut self) -> impl Iterator<Item = &'_ mut Atom> {
        self.0.children()
    }

    pub fn into_children(self) -> impl Iterator<Item = &'a mut Atom> {
        self.0.into_children()
    }

    /// Finds or inserts the MDHD atom
    pub fn header(&mut self) -> &mut MediaHeaderAtom {
        unwrap_atom_data!(
            self.0
                .find_or_insert_child(MDHD)
                .insert_data(AtomData::MediaHeader(MediaHeaderAtom::default()))
                .call(),
            AtomData::MediaHeader,
        )
    }

    /// Finds or inserts the HDLR atom
    pub fn handler_reference(&mut self) -> &mut HandlerReferenceAtom {
        unwrap_atom_data!(
            self.0
                .find_or_insert_child(HDLR)
                .insert_data(AtomData::HandlerReference(HandlerReferenceAtom::default()))
                .call(),
            AtomData::HandlerReference,
        )
    }

    /// Finds or inserts the MINF atom
    pub fn media_information(&mut self) -> MinfAtomRefMut<'_> {
        MinfAtomRefMut(
            self.0
                .find_or_insert_child(MINF)
                .insert_after(vec![HDLR, MDHD])
                .call(),
        )
    }

    /// Finds the MINF atom
    pub fn into_media_information(self) -> Option<MinfAtomRefMut<'a>> {
        let atom = self.0.into_child(MINF)?;
        Some(MinfAtomRefMut(AtomRefMut(atom)))
    }
}