Initial commitHEADtrunk

1 files changed, 76 insertions, 0 deletions

diff --git a/src/boxes/avcc.rs b/src/boxes/avcc.rs
new file mode 100644
index 0000000..e932543
--- /dev/null
+++ b/src/boxes/avcc.rs
@@ -0,0 +1,76 @@
+use byteorder::{BigEndian, WriteBytesExt};
+
+use four_cc::FourCC;
+
+use bytes::{BufMut, BytesMut};
+
+use crate::Mp4Box;
+use crate::Mp4BoxError;
+
+use std::mem::size_of;
+
+pub struct AvcConfigurationBox {
+    pub config: AvcDecoderConfigurationRecord,
+}
+
+impl Mp4Box for AvcConfigurationBox {
+    const NAME: FourCC = FourCC(*b"avcC");
+
+    fn content_size(&self) -> u64 {
+        self.config.size()
+    }
+
+    fn write_box_contents(&self, writer: &mut BytesMut) -> Result<(), Mp4BoxError> {
+        self.config.write(writer)?;
+
+        Ok(())
+    }
+}
+
+pub struct AvcDecoderConfigurationRecord {
+    pub profile_indication: u8,
+    pub profile_compatibility: u8,
+    pub level_indication: u8,
+    pub sequence_parameter_set: Vec<u8>,
+    pub picture_parameter_set: Vec<u8>,
+}
+
+impl AvcDecoderConfigurationRecord {
+    fn size(&self) -> u64 {
+        size_of::<u8>() as u64 // configurationVersion
+            + size_of::<u8>() as u64 // AVCProfileIndication
+            + size_of::<u8>() as u64 // profile_compatibility
+            + size_of::<u8>() as u64 // AVCLevelIndication
+            + size_of::<u8>() as u64 // lengthSizeMinusOne
+            + size_of::<u8>() as u64 // numOfSequenceParameterSets
+            + size_of::<u16>() as u64 // sequenceParameterSetLength
+            + self.sequence_parameter_set.len() as u64
+            + size_of::<u8>() as u64 // numOfPictureParameterSets
+            + size_of::<u16>() as u64 // pictureParameterSetLength
+            + self.picture_parameter_set.len() as u64
+    }
+
+    fn write(&self, writer: &mut BytesMut) -> Result<(), Mp4BoxError> {
+        let mut v = Vec::new();
+
+        v.push(1);
+        v.push(self.profile_indication);
+        v.push(self.profile_compatibility);
+        v.push(self.level_indication);
+        v.push(0b1111_1100 | 3);
+
+        v.push(0b1110_0000 | 1);
+        v.write_u16::<BigEndian>(self.sequence_parameter_set.len() as u16)?;
+        v.extend(&self.sequence_parameter_set);
+
+        v.push(1);
+        v.write_u16::<BigEndian>(self.picture_parameter_set.len() as u16)?;
+        v.extend(&self.picture_parameter_set);
+
+        assert_eq!(self.size(), v.len() as u64);
+
+        writer.put_slice(&v);
+
+        Ok(())
+    }
+}