mirror of
https://github.com/alfg/mp4-rust.git
synced 2024-05-20 01:08:06 +00:00
148 lines
3.7 KiB
Rust
148 lines
3.7 KiB
Rust
use byteorder::{BigEndian, WriteBytesExt};
|
|
use serde::Serialize;
|
|
use std::io::Write;
|
|
|
|
use crate::mp4box::vp09::Vp09Box;
|
|
use crate::mp4box::*;
|
|
use crate::mp4box::{avc1::Avc1Box, hev1::Hev1Box, mp4a::Mp4aBox, tx3g::Tx3gBox};
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq, Default, Serialize)]
|
|
pub struct StsdBox {
|
|
pub version: u8,
|
|
pub flags: u32,
|
|
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub avc1: Option<Avc1Box>,
|
|
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub hev1: Option<Hev1Box>,
|
|
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub vp09: Option<Vp09Box>,
|
|
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub mp4a: Option<Mp4aBox>,
|
|
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub tx3g: Option<Tx3gBox>,
|
|
}
|
|
|
|
impl StsdBox {
|
|
pub fn get_type(&self) -> BoxType {
|
|
BoxType::StsdBox
|
|
}
|
|
|
|
pub fn get_size(&self) -> u64 {
|
|
let mut size = HEADER_SIZE + HEADER_EXT_SIZE + 4;
|
|
if let Some(ref avc1) = self.avc1 {
|
|
size += avc1.box_size();
|
|
} else if let Some(ref hev1) = self.hev1 {
|
|
size += hev1.box_size();
|
|
} else if let Some(ref vp09) = self.vp09 {
|
|
size += vp09.box_size();
|
|
} else if let Some(ref mp4a) = self.mp4a {
|
|
size += mp4a.box_size();
|
|
} else if let Some(ref tx3g) = self.tx3g {
|
|
size += tx3g.box_size();
|
|
}
|
|
|
|
size
|
|
}
|
|
}
|
|
|
|
impl Mp4Box for StsdBox {
|
|
const TYPE: BoxType = BoxType::StsdBox;
|
|
|
|
fn box_size(&self) -> u64 {
|
|
self.get_size()
|
|
}
|
|
|
|
fn to_json(&self) -> Result<String> {
|
|
Ok(serde_json::to_string(&self).unwrap())
|
|
}
|
|
|
|
fn summary(&self) -> Result<String> {
|
|
let s = String::new();
|
|
Ok(s)
|
|
}
|
|
}
|
|
|
|
impl BlockReader for StsdBox {
|
|
fn read_block<'a>(reader: &mut impl Reader<'a>) -> Result<Self> {
|
|
let (version, flags) = read_box_header_ext(reader);
|
|
|
|
reader.get_u32(); // XXX entry_count
|
|
|
|
let mut avc1 = None;
|
|
let mut hev1 = None;
|
|
let mut vp09 = None;
|
|
let mut mp4a = None;
|
|
let mut tx3g = None;
|
|
|
|
while let Some(mut bx) = reader.get_box()? {
|
|
match bx.kind {
|
|
BoxType::Avc1Box => {
|
|
avc1 = Some(bx.read()?);
|
|
}
|
|
|
|
BoxType::Hev1Box => {
|
|
hev1 = Some(bx.read()?);
|
|
}
|
|
|
|
BoxType::Vp09Box => {
|
|
vp09 = Some(bx.read()?);
|
|
}
|
|
|
|
BoxType::Mp4aBox => {
|
|
mp4a = Some(bx.read()?);
|
|
}
|
|
|
|
BoxType::Tx3gBox => {
|
|
tx3g = Some(bx.read()?);
|
|
}
|
|
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
Ok(StsdBox {
|
|
version,
|
|
flags,
|
|
avc1,
|
|
hev1,
|
|
vp09,
|
|
mp4a,
|
|
tx3g,
|
|
})
|
|
}
|
|
|
|
fn size_hint() -> usize {
|
|
8
|
|
}
|
|
}
|
|
|
|
impl<W: Write> WriteBox<&mut W> for StsdBox {
|
|
fn write_box(&self, writer: &mut W) -> Result<u64> {
|
|
let size = self.box_size();
|
|
BoxHeader::new(Self::TYPE, size).write(writer)?;
|
|
|
|
write_box_header_ext(writer, self.version, self.flags)?;
|
|
|
|
writer.write_u32::<BigEndian>(1)?; // entry_count
|
|
|
|
if let Some(ref avc1) = self.avc1 {
|
|
avc1.write_box(writer)?;
|
|
} else if let Some(ref hev1) = self.hev1 {
|
|
hev1.write_box(writer)?;
|
|
} else if let Some(ref vp09) = self.vp09 {
|
|
vp09.write_box(writer)?;
|
|
} else if let Some(ref mp4a) = self.mp4a {
|
|
mp4a.write_box(writer)?;
|
|
} else if let Some(ref tx3g) = self.tx3g {
|
|
tx3g.write_box(writer)?;
|
|
}
|
|
|
|
Ok(size)
|
|
}
|
|
}
|