Merge pull request #96 from nuttycom/frontier_size

incrementalmerkletree: Add `Frontier::tree_size`

incrementalmerkletree/CHANGELOG.md

@@ -7,6 +7,9 @@ and this project adheres to Rust's notion of
 
 ## Unreleased
 
+### Added
+- `incrementalmerkletree::Frontier::tree_size`
+
 ## [0.5.0] - 2023-09-08
 
 ### Added

incrementalmerkletree/src/frontier.rs

@@ -225,6 +225,13 @@ impl<H, const DEPTH: u8> Frontier<H, DEPTH> {
             size_of::<usize>() + (f.ommers.capacity() + 1) * size_of::<H>()
         })
     }
+
+    /// Returns the size of the Merkle tree that this frontier corresponds to.
+    pub fn tree_size(&self) -> u64 {
+        self.frontier
+            .as_ref()
+            .map_or(0, |f| u64::from(f.position()) + 1)
+    }
 }
 
 impl<H: Hashable + Clone, const DEPTH: u8> Frontier<H, DEPTH> {