From b4dc373925e69511cadc8f2e1dbb71ac2a3dcf83 Mon Sep 17 00:00:00 2001
From: polydez <155382956+polydez@users.noreply.github.com>
Date: Wed, 3 Apr 2024 00:07:00 +0500
Subject: [PATCH] feat: add leaf count to `SimpleSmt` (#302)

---
 src/merkle/smt/simple/mod.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/merkle/smt/simple/mod.rs b/src/merkle/smt/simple/mod.rs
index 350ceaa..4d14e9c 100644
--- 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
     // --------------------------------------------------------------------------------------------