core/formats: add NAPixelFormaton::get_total_depth()

1 files changed, 10 insertions, 0 deletions

diff --git a/nihav-core/src/formats.rs b/nihav-core/src/formats.rs
index 176f8d6..2c2e717 100644
--- a/nihav-core/src/formats.rs
+++ b/nihav-core/src/formats.rs
@@ -659,6 +659,16 @@ impl NAPixelFormaton {
         }
         mdepth
     }
+    /// Returns the total amount of bits needed for components.
+    pub fn get_total_depth(&self) -> u8 {
+        let mut depth = 0;
+        for chr in self.comp_info.iter() {
+            if let Some(ref chromaton) = chr {
+                depth += chromaton.depth;
+            }
+        }
+        depth
+    }
     /// Returns the maximum component subsampling.
     pub fn get_max_subsampling(&self) -> u8 {
         let mut ssamp = 0;