Implement node depth calculation (#29)

crates/clarirs_core/src/ast/node.rs

@@ -68,14 +68,15 @@ impl<'c, O: Op<'c> + Serialize> AstNode<'c, O> {
             .child_iter()
             .flat_map(|child| child.variables().clone().into_iter())
             .collect::<HashSet<String>>();
+        let depth = op.depth();
 
         Self {
             op,
             ctx,
             hash,
             symbolic,
             variables,
-            depth: 0, // TODO: Implement depth calculation
+            depth,
         }
     }
 
@@ -94,17 +95,17 @@ impl<'c, O: Op<'c> + Serialize> AstNode<'c, O> {
     pub fn variables(&self) -> &HashSet<String> {
         &self.variables
     }
-
-    pub fn depth(&self) -> u32 {
-        self.depth
-    }
 }
 
 impl<'c, O: Op<'c>> Op<'c> for AstNode<'c, O> {
     fn child_iter(&self) -> IntoIter<VarAst<'c>> {
         self.op.child_iter()
     }
 
+    fn depth(&self) -> u32 {
+        self.depth
+    }
+
     fn is_true(&self) -> bool {
         self.op.is_true()
     }
@@ -142,6 +143,15 @@ impl<'c> Op<'c> for VarAst<'c> {
         }
     }
 
+    fn depth(&self) -> u32 {
+        match self {
+            VarAst::Boolean(ast) => ast.depth(),
+            VarAst::BitVec(ast) => ast.depth(),
+            VarAst::Float(ast) => ast.depth(),
+            VarAst::String(ast) => ast.depth(),
+        }
+    }
+
     fn is_true(&self) -> bool {
         match self {
             VarAst::Boolean(ast) => ast.is_true(),

crates/clarirs_core/src/ast/op.rs

@@ -14,6 +14,10 @@ pub trait Op<'c>: Debug + Hash + Serialize {
         self.child_iter().collect()
     }
 
+    fn depth(&self) -> u32 {
+        1 + self.children().iter().map(|c| c.depth()).max().unwrap_or(0)
+    }
+
     fn is_true(&self) -> bool {
         false
     }