Use VecDeque in places
All unwrap()'s are after checks to make sure we won't be unwrap'ing a None
This commit is contained in:
parent
6732084928
commit
98e40aa44f
|
|
@ -13,7 +13,7 @@
|
||||||
#![allow(clippy::unit_arg)]
|
#![allow(clippy::unit_arg)]
|
||||||
#![allow(dead_code)]
|
#![allow(dead_code)]
|
||||||
|
|
||||||
use std::fmt;
|
use std::{collections::VecDeque, fmt};
|
||||||
|
|
||||||
use bitvec::prelude::*;
|
use bitvec::prelude::*;
|
||||||
use lazy_static::lazy_static;
|
use lazy_static::lazy_static;
|
||||||
|
|
@ -112,31 +112,31 @@ impl From<Vec<jubjub::Fq>> for NoteCommitmentTree {
|
||||||
|
|
||||||
let count = values.len() as u32;
|
let count = values.len() as u32;
|
||||||
let mut height = 0u8;
|
let mut height = 0u8;
|
||||||
let mut current_layer: Vec<[u8; 32]> =
|
let mut current_layer: VecDeque<[u8; 32]> =
|
||||||
values.into_iter().map(|cm_u| cm_u.to_bytes()).collect();
|
values.into_iter().map(|cm_u| cm_u.to_bytes()).collect();
|
||||||
|
|
||||||
while usize::from(height) < MERKLE_DEPTH {
|
while usize::from(height) < MERKLE_DEPTH {
|
||||||
let mut next_layer_up = vec![];
|
let mut next_layer_up = vec![];
|
||||||
|
|
||||||
while !current_layer.is_empty() {
|
while !current_layer.is_empty() {
|
||||||
let left = current_layer.remove(0);
|
let left = current_layer.pop_front().unwrap();
|
||||||
let right;
|
let right;
|
||||||
if current_layer.is_empty() {
|
if current_layer.is_empty() {
|
||||||
right = EMPTY_ROOTS[height as usize];
|
right = EMPTY_ROOTS[height as usize];
|
||||||
} else {
|
} else {
|
||||||
right = current_layer.remove(0);
|
right = current_layer.pop_front().unwrap();
|
||||||
}
|
}
|
||||||
next_layer_up.push(merkle_crh_sapling(height, left, right));
|
next_layer_up.push(merkle_crh_sapling(height, left, right));
|
||||||
}
|
}
|
||||||
|
|
||||||
height += 1;
|
height += 1;
|
||||||
current_layer = next_layer_up;
|
current_layer = next_layer_up.into();
|
||||||
}
|
}
|
||||||
|
|
||||||
assert!(current_layer.len() == 1);
|
assert!(current_layer.len() == 1);
|
||||||
|
|
||||||
NoteCommitmentTree {
|
NoteCommitmentTree {
|
||||||
root: Root(current_layer.remove(0)),
|
root: Root(current_layer.pop_front().unwrap()),
|
||||||
height,
|
height,
|
||||||
count,
|
count,
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue