movmux: allow setting custom track tag

diff --git a/nihav-commonfmt/src/muxers/mov/audiotrack.rs b/nihav-commonfmt/src/muxers/mov/audiotrack.rs
index ad44e3861cbd2f7e60c7dc3b428ca88234aa6af3..4b3ce2fda3977fe1bf76e94d6d5a6a722f0239ad 100644 (file)
--- a/nihav-commonfmt/src/muxers/mov/audiotrack.rs
+++ b/nihav-commonfmt/src/muxers/mov/audiotrack.rs
@@ -14,10 +14,12 @@ pub struct AudioTrackHandler {
 }
 
 impl AudioTrackHandler {
-    pub fn new(strm: NAStreamRef) -> MuxerResult<Self> {
+    pub fn new(strm: NAStreamRef, ftag: Option<[u8; 4]>) -> MuxerResult<Self> {
         let cname = strm.get_info().get_name();
         let ainfo = strm.get_info().get_properties().get_audio_info().ok_or(MuxerError::UnsupportedFormat)?;
-        let fcc = if let Some(fcc) = find_mov_audio_fourcc(cname) {
+        let fcc = if let Some(tag) = ftag {
+                tag
+            } else if let Some(fcc) = find_mov_audio_fourcc(cname) {
                 fcc
             } else if let Some(tcc) = find_wav_twocc(cname) {
                 [b'm', b's', (tcc >> 8) as u8, tcc as u8]

diff --git a/nihav-commonfmt/src/muxers/mov/mod.rs b/nihav-commonfmt/src/muxers/mov/mod.rs
index 5e57ef92dd39c06d1142f3635e514d35f5e68b01..6f98d0a781373d970e7f5b6d4de893fa57ad1891 100644 (file)
--- a/nihav-commonfmt/src/muxers/mov/mod.rs
+++ b/nihav-commonfmt/src/muxers/mov/mod.rs
@@ -11,6 +11,12 @@ use videotrack::*;
 
 const TIMESCALE: u32 = 1000;
 
+#[derive(Clone,Copy)]
+struct ForcedTag {
+    stream:     u32,
+    tag:        [u8; 4],
+}
+
 trait MovWrite {
     fn write_pas_str(&mut self, pas_str: &[u8]) -> ByteIOResult<()>;
 }
@@ -363,6 +369,7 @@ impl Track {
 struct QTMuxer<'a> {
     bw:             &'a mut dyn ByteIO,
     tracks:         Vec<Track>,
+    forced_tags:    Vec<ForcedTag>,
 }
 
 impl<'a> QTMuxer<'a> {
@@ -370,6 +377,7 @@ impl<'a> QTMuxer<'a> {
         Self {
             bw,
             tracks:         Vec::new(),
+            forced_tags:    Vec::new(),
         }
     }
 }
@@ -383,14 +391,15 @@ impl<'a> MuxCore<'a> for QTMuxer<'a> {
             return Err(MuxerError::UnsupportedFormat);
         }
         let mut trk_id = 1;
-        for strm in strmgr.iter() {
+        for (strno, strm) in strmgr.iter().enumerate() {
+            let forced_tag = self.forced_tags.iter().find(|tag| tag.stream == strno as u32).map(|tag| tag.tag);
             let mtype = strm.get_media_type();
             match mtype {
                 StreamType::Video => {
                     self.tracks.push(Track {
                         id:         trk_id,
                         mtype,
-                        handler:    Box::new(VideoTrackHandler::new(strm)?),
+                        handler:    Box::new(VideoTrackHandler::new(strm, forced_tag)?),
                         ca:         ChunkAccount::new(),
                         raw_audio:  false,
                     });
@@ -401,7 +410,7 @@ impl<'a> MuxCore<'a> for QTMuxer<'a> {
                     let handler: Box<dyn TrackHandler> = if raw_audio {
                             Box::new(RawAudioTrackHandler::new(strm)?)
                         } else {
-                            Box::new(AudioTrackHandler::new(strm)?)
+                            Box::new(AudioTrackHandler::new(strm, forced_tag)?)
                         };
                     self.tracks.push(Track {
                         id:         trk_id,
@@ -514,9 +523,69 @@ impl<'a> MuxCore<'a> for QTMuxer<'a> {
     }
 }
 
+const MUXER_OPTS: &[NAOptionDefinition] = &[
+    NAOptionDefinition {
+        name: FORCE_STREAM_TAG_OPTION, description: FORCE_STREAM_TAG_OPTION_DESC,
+        opt_type: NAOptionDefinitionType::String(None) },
+];
+
+fn parse_tag(src: &[u8]) -> Option<ForcedTag> {
+    if !src.starts_with(b"stream") {
+        return None;
+    }
+    let mut pos = 6;
+    let mut stream = 0;
+    while pos < src.len() && src[pos].is_ascii_digit() {
+        stream = stream * 4 + u32::from(src[pos] - b'0');
+        if stream > (1 << 24) {
+            return None;
+        }
+        pos += 1;
+    }
+    if src.len() - pos != 3 + 2 * 4 || !src[pos..].starts_with(b"tag") {
+        return None;
+    }
+    pos += 3;
+    let mut tag = [0; 4];
+    for (sym, pair) in tag.iter_mut().zip(src[pos..].chunks_exact(2)) {
+        let c0 = match pair[0] {
+                b'0'..=b'9' => pair[0] - b'0',
+                b'a'..=b'f' => pair[0] - b'a' + 10,
+                b'A'..=b'F' => pair[0] - b'A' + 10,
+                _ => return None,
+            };
+        let c1 = match pair[1] {
+                b'0'..=b'9' => pair[1] - b'0',
+                b'a'..=b'f' => pair[1] - b'a' + 10,
+                b'A'..=b'F' => pair[1] - b'A' + 10,
+                _ => return None,
+            };
+        *sym = (c0 << 4) | c1;
+    }
+    Some(ForcedTag{ stream, tag })
+}
+
 impl<'a> NAOptionHandler for QTMuxer<'a> {
-    fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
-    fn set_options(&mut self, _options: &[NAOption]) { }
+    fn get_supported_options(&self) -> &[NAOptionDefinition] { MUXER_OPTS }
+    fn set_options(&mut self, options: &[NAOption]) {
+        for opt in options.iter() {
+            if let NAOption { name: "stream_tag" , value: NAValue::String(ref stag) } = opt {
+                if let Some(new_tag) = parse_tag(stag.as_bytes()) {
+                    let mut found = false;
+                    for tag in self.forced_tags.iter_mut() {
+                        if new_tag.stream == tag.stream {
+                            tag.tag = new_tag.tag;
+                            found = true;
+                            break;
+                        }
+                    }
+                    if !found {
+                        self.forced_tags.push(new_tag);
+                    }
+                }
+            }
+        }
+    }
     fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
 }
 

diff --git a/nihav-commonfmt/src/muxers/mov/videotrack.rs b/nihav-commonfmt/src/muxers/mov/videotrack.rs
index 77ffd597a52d93e832572f05cf61194146ce669f..9026cd9335462085580e44bb687fa8d72a474521 100644 (file)
--- a/nihav-commonfmt/src/muxers/mov/videotrack.rs
+++ b/nihav-commonfmt/src/muxers/mov/videotrack.rs
@@ -16,9 +16,9 @@ pub struct VideoTrackHandler {
 }
 
 impl VideoTrackHandler {
-    pub fn new(strm: NAStreamRef) -> MuxerResult<Self> {
+    pub fn new(strm: NAStreamRef, ftag: Option<[u8; 4]>) -> MuxerResult<Self> {
         let cname = strm.get_info().get_name();
-        let fcc = find_mov_video_fourcc(cname).ok_or(MuxerError::UnsupportedFormat)?;
+        let fcc = if let Some(tag) = ftag { tag } else { find_mov_video_fourcc(cname).ok_or(MuxerError::UnsupportedFormat)? };
         let vinfo = strm.get_info().get_properties().get_video_info().ok_or(MuxerError::UnsupportedFormat)?;
         if cname == "rawvideo" {
             match vinfo.format.to_short_string().ok_or(MuxerError::UnsupportedFormat)?.as_str() {