feat(Header): Header::num_vars implementation

src/lib.rs

@@ -318,4 +318,29 @@ impl Header {
             _ => None,
         })
     }
+
+    /// Find the variable object at a specified path.
+    ///
+    /// ## Example
+    ///
+    /// ```rust
+    /// let mut parser = vcd::Parser::new(&b"
+    /// $scope module a $end
+    /// $var integer 32 y counter_b $end
+    /// $scope module b $end
+    /// $var integer 16 n0 counter $end
+    /// $upscope $end
+    /// $upscope $end
+    /// $enddefinitions $end
+    /// "[..]);
+    /// let header = parser.parse_header().unwrap();
+    /// let var_count = header.num_vars();
+    /// assert_eq!(var_count, 2);
+    /// ```
+    pub fn num_vars(&self) -> usize {
+        self.items
+            .iter()
+            .filter(|item| matches!(item, ScopeItem::Var(_) | ScopeItem::Scope(_)))
+            .fold(0, |acc, scope_item| acc + scope_item.num_vars())
+    }
 }

src/scope.rs

@@ -292,3 +292,14 @@ pub enum ScopeItem {
     /// `$comment` - Comment
     Comment(String),
 }
+
+impl ScopeItem {
+    /// Returns the number of variables in this scope item.
+    pub fn num_vars(&self) -> usize {
+        match self {
+            ScopeItem::Scope(s) => s.items.iter().map(ScopeItem::num_vars).sum(),
+            ScopeItem::Var(_) => 1,
+            ScopeItem::Comment(_) => 0,
+        }
+    }
+}