From 727be96686a2c6c5747b26be15933e11c9cab9c6 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Sun, 28 Jan 2024 17:55:11 +0100
Subject: clean up some code + subrip support?

---
 remuxer/src/lib.rs | 27 +++++++++++++++++++++------
 1 file changed, 21 insertions(+), 6 deletions(-)

(limited to 'remuxer/src/lib.rs')

diff --git a/remuxer/src/lib.rs b/remuxer/src/lib.rs
index 96aeca1..b49dedc 100644
--- a/remuxer/src/lib.rs
+++ b/remuxer/src/lib.rs
@@ -61,13 +61,25 @@ pub fn ebml_track_entry(
         SourceTrackKind::Video {
             width,
             height,
-            fps: _,
+            display_height,
+            display_width,
+            fps,
         } => {
             els.push(MatroskaTag::TrackType(1));
-            els.push(MatroskaTag::Video(Master::Collected(vec![
+            let mut props = vec![
                 MatroskaTag::PixelWidth(width),
                 MatroskaTag::PixelHeight(height),
-            ])))
+            ];
+            if let Some(display_width) = display_width {
+                props.push(MatroskaTag::DisplayWidth(display_width))
+            }
+            if let Some(display_height) = display_height {
+                props.push(MatroskaTag::DisplayHeight(display_height))
+            }
+            if let Some(fps) = fps {
+                props.push(MatroskaTag::FrameRate(fps))
+            }
+            els.push(MatroskaTag::Video(Master::Collected(props)))
         }
         SourceTrackKind::Audio {
             channels,
@@ -75,11 +87,14 @@ pub fn ebml_track_entry(
             bit_depth,
         } => {
             els.push(MatroskaTag::TrackType(2));
-            els.push(MatroskaTag::Audio(Master::Collected(vec![
+            let mut props = vec![
                 MatroskaTag::SamplingFrequency(sample_rate),
                 MatroskaTag::Channels(channels.try_into().unwrap()),
-            ])));
-            els.push(MatroskaTag::BitDepth(bit_depth.try_into().unwrap()));
+            ];
+            if let Some(bit_depth) = bit_depth {
+                props.push(MatroskaTag::BitDepth(bit_depth.try_into().unwrap()));
+            }
+            els.push(MatroskaTag::Audio(Master::Collected(props)));
         }
         SourceTrackKind::Subtitles => {
             els.push(MatroskaTag::TrackType(17));
-- 
cgit v1.2.3-70-g09d2