use std::cmp::min;
use std::default::Default;
use bitcoin_hashes::{sha256d, Hash};
use consensus::encode::Encodable;
pub trait MerkleRoot {
fn merkle_root(&self) -> sha256d::Hash;
}
pub fn bitcoin_merkle_root(data: Vec<sha256d::Hash>) -> sha256d::Hash {
if data.len() < 1 {
return Default::default();
}
if data.len() < 2 {
return data[0];
}
let mut next = vec![];
for idx in 0..((data.len() + 1) / 2) {
let idx1 = 2 * idx;
let idx2 = min(idx1 + 1, data.len() - 1);
let mut encoder = sha256d::Hash::engine();
data[idx1].consensus_encode(&mut encoder).unwrap();
data[idx2].consensus_encode(&mut encoder).unwrap();
next.push(sha256d::Hash::from_engine(encoder));
}
bitcoin_merkle_root(next)
}
impl<'a, T: BitcoinHash> MerkleRoot for &'a [T] {
fn merkle_root(&self) -> sha256d::Hash {
bitcoin_merkle_root(self.iter().map(|obj| obj.bitcoin_hash()).collect())
}
}
impl <T: BitcoinHash> MerkleRoot for Vec<T> {
fn merkle_root(&self) -> sha256d::Hash {
(&self[..]).merkle_root()
}
}
pub trait BitcoinHash {
fn bitcoin_hash(&self) -> sha256d::Hash;
}