Fix message count in folded info

This commit is contained in:
Joscha 2022-08-09 15:12:49 +02:00
parent 87a14eedf2
commit c41ab742d3
2 changed files with 21 additions and 9 deletions

View file

@ -94,6 +94,15 @@ impl<M: Msg> Tree<M> {
self.children.get(id).map(|c| c as &[M::Id])
}
pub fn subtree_size(&self, id: &M::Id) -> usize {
let children = self.children(id).unwrap_or_default();
let mut result = children.len();
for child in children {
result += self.subtree_size(child);
}
result
}
pub fn siblings(&self, id: &M::Id) -> Option<&[M::Id]> {
if let Some(parent) = self.parent(id) {
self.children(&parent)