similarity-core 0.5.0

Core library for code similarity detection using AST-based comparison
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

use std::rc::Rc;

#[derive(Debug, Clone)]
pub struct TreeNode {
    pub label: String,
    pub value: String,
    pub children: Vec<Rc<TreeNode>>,
    pub id: usize,
    pub subtree_size: Option<usize>,
}

impl TreeNode {
    #[must_use]
    pub fn new(label: String, value: String, id: usize) -> Self {
        TreeNode { label, value, children: Vec::new(), id, subtree_size: None }
    }

    pub fn add_child(&mut self, child: Rc<TreeNode>) {
        self.children.push(child);
    }

    #[must_use]
    pub fn get_subtree_size(&self) -> usize {
        // Since we can't mutate through Rc, we'll calculate it each time
        // In a real implementation, you might want to use RefCell for interior mutability
        let mut size = 1;
        for child in &self.children {
            size += child.get_subtree_size();
        }
        size
    }
}