From 8b746bf7d611f6910d656b312b4f6269fd63883c Mon Sep 17 00:00:00 2001
From: Kostya Shishkov <kostya.shishkov@gmail.com>
Date: Thu, 21 May 2020 10:57:22 +0200
Subject: [PATCH] core/formats: add NAPixelFormaton::get_total_depth()

---
 nihav-core/src/formats.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

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;
-- 
2.39.5