detect video bit depth from codec param

1 files changed, 18 insertions, 2 deletions

diff --git a/remuxer/src/codec_param/mod.rs b/remuxer/src/codec_param/mod.rs
index 4fdb181..4613c88 100644
--- a/remuxer/src/codec_param/mod.rs
+++ b/remuxer/src/codec_param/mod.rs
@@ -11,6 +11,12 @@ mod av1;
 mod hevc;
 mod vp9;
 
+#[derive(Debug, PartialEq, Eq)]
+struct CodecParam {
+    bit_depth: Option<u8>,
+    string: String,
+}
+
 pub fn codec_param(te: &TrackEntry) -> String {
     let empty_cp = vec![];
     let cp = te.codec_private.as_ref().unwrap_or(&empty_cp);
@@ -20,9 +26,9 @@ pub fn codec_param(te: &TrackEntry) -> String {
         "A_OPUS" => "opus".to_string(),
         "A_VORBIS" => "vorbis".to_string(),
 
-        "V_AV1" => av1_codec_param(cp),
+        "V_AV1" => av1_codec_param(cp).string,
         "V_MPEG4/ISO/AVC" => format!("avc1.{:02x}{:02x}{:02x}", cp[1], cp[2], cp[3]),
-        "V_MPEGH/ISO/HEVC" => hevc_codec_param(cp),
+        "V_MPEGH/ISO/HEVC" => hevc_codec_param(cp).string,
         "V_VP9" => vp9_codec_param(te),
 
         "D_WEBVTT/SUBTITLES" => "webvtt".to_string(),
@@ -31,3 +37,13 @@ pub fn codec_param(te: &TrackEntry) -> String {
         x => todo!("{x:?}"),
     }
 }
+
+pub fn codec_param_bit_depth(te: &TrackEntry) -> Option<u8> {
+    let empty_cp = vec![];
+    let cp = te.codec_private.as_ref().unwrap_or(&empty_cp);
+    match te.codec_id.as_str() {
+        "V_AV1" => av1_codec_param(cp).bit_depth,
+        "V_MPEGH/ISO/HEVC" => hevc_codec_param(cp).bit_depth,
+        _ => None,
+    }
+}