Track information about hidden leaves in taproot NodeInfo

This commit is contained in:
Dr Maxim Orlovsky 2022-04-01 08:05:41 +02:00
parent cb4d34fd40
commit 6add0dd9dc
No known key found for this signature in database
GPG Key ID: FFC0250947E5C6F7
1 changed files with 5 additions and 0 deletions

View File

@ -549,6 +549,8 @@ pub struct NodeInfo {
pub(crate) hash: sha256::Hash,
/// Information about leaves inside this node.
pub(crate) leaves: Vec<LeafInfo>,
/// Tracks information on hidden nodes below this node.
pub(crate) has_hidden_nodes: bool,
}
impl NodeInfo {
@ -557,6 +559,7 @@ impl NodeInfo {
Self {
hash: hash,
leaves: vec![],
has_hidden_nodes: true
}
}
@ -566,6 +569,7 @@ impl NodeInfo {
Self {
hash: leaf.hash(),
leaves: vec![leaf],
has_hidden_nodes: false,
}
}
@ -584,6 +588,7 @@ impl NodeInfo {
Ok(Self {
hash: sha256::Hash::from_inner(hash.into_inner()),
leaves: all_leaves,
has_hidden_nodes: a.has_hidden_nodes || b.has_hidden_nodes
})
}
}