feat: add leaf count to SimpleSmt (#302)

0xPolygonMiden · Apr 2, 2024 · b4dc373 · b4dc373
1 parent 4885f88
commit b4dc373
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/merkle/smt/simple/mod.rs b/src/merkle/smt/simple/mod.rs
@@ -152,6 +152,11 @@ impl<const DEPTH: u8> SimpleSmt<DEPTH> {
         <Self as SparseMerkleTree<DEPTH>>::open(self, key)
     }
 
+    /// Returns a count of non-empty leaves.
+    pub fn leaf_count(&self) -> usize {
+        self.leaves.len()
+    }
+
     // ITERATORS
     // --------------------------------------------------------------------------------------------