From 96da47a8bb0a58bc2732c35eef86702f5cd2b5b2 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Wed, 16 Apr 2025 21:08:26 +0200
Subject: send actual remux bitrate; close #18

---
 remuxer/src/seek_index.rs | 7 +++++++
 1 file changed, 7 insertions(+)

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

diff --git a/remuxer/src/seek_index.rs b/remuxer/src/seek_index.rs
index 7296d93..82f62fb 100644
--- a/remuxer/src/seek_index.rs
+++ b/remuxer/src/seek_index.rs
@@ -53,6 +53,13 @@ pub fn get_seek_index(path: &Path) -> anyhow::Result<Arc<BTreeMap<u64, Arc<SeekI
     })
 }
 
+pub fn get_track_sizes(path: &Path) -> Result<BTreeMap<u64, usize>> {
+    Ok(get_seek_index(path)?
+        .iter()
+        .map(|(k, v)| (*k, v.blocks.iter().map(|b| b.size).sum::<usize>()))
+        .collect())
+}
+
 pub fn import_seek_index(input: &mut EbmlReader) -> Result<BTreeMap<u64, SeekIndex>> {
     let mut seek_index = BTreeMap::new();
     while let Some(item) = input.next() {
-- 
cgit v1.2.3-70-g09d2