mirror of
https://gitlab.com/pulsechaincom/lighthouse-pulse.git
synced 2024-12-29 07:17:17 +00:00
Remove FastBeaconState
This commit is contained in:
parent
b211e39331
commit
cdc03f1749
File diff suppressed because it is too large
Load Diff
@ -1,128 +0,0 @@
|
||||
use crate::{
|
||||
validator_registry::get_active_validator_indices, BeaconState, ChainSpec, Epoch, Hash256,
|
||||
};
|
||||
use honey_badger_split::SplitExt;
|
||||
use serde_derive::Serialize;
|
||||
use swap_or_not_shuffle::get_permutated_index;
|
||||
|
||||
pub const CACHED_EPOCHS: usize = 3;
|
||||
|
||||
#[derive(Debug, PartialEq, Clone, Default, Serialize)]
|
||||
pub struct CommitteesCache {
|
||||
cache_index_offset: usize,
|
||||
active_validator_indices_cache: Vec<Option<Vec<usize>>>,
|
||||
shuffling_cache: Vec<Option<Option<Vec<Vec<usize>>>>>,
|
||||
previous_epoch: Epoch,
|
||||
current_epoch: Epoch,
|
||||
next_epoch: Epoch,
|
||||
}
|
||||
|
||||
impl CommitteesCache {
|
||||
pub fn new(current_epoch: Epoch, spec: &ChainSpec) -> Self {
|
||||
let previous_epoch = if current_epoch == spec.genesis_epoch {
|
||||
current_epoch
|
||||
} else {
|
||||
current_epoch - 1
|
||||
};
|
||||
let next_epoch = current_epoch + 1;
|
||||
|
||||
Self {
|
||||
cache_index_offset: 0,
|
||||
active_validator_indices_cache: vec![None; CACHED_EPOCHS],
|
||||
shuffling_cache: vec![None; CACHED_EPOCHS],
|
||||
previous_epoch,
|
||||
current_epoch,
|
||||
next_epoch,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn advance_epoch(&mut self) {
|
||||
let previous_cache_index = self.cache_index(self.previous_epoch);
|
||||
|
||||
self.active_validator_indices_cache[previous_cache_index] = None;
|
||||
self.shuffling_cache[previous_cache_index] = None;
|
||||
|
||||
self.cache_index_offset += 1;
|
||||
self.cache_index_offset %= CACHED_EPOCHS;
|
||||
}
|
||||
|
||||
pub fn cache_index(&self, epoch: Epoch) -> usize {
|
||||
let base_index = match epoch {
|
||||
e if e == self.previous_epoch => 0,
|
||||
e if e == self.current_epoch => 1,
|
||||
e if e == self.next_epoch => 2,
|
||||
_ => panic!("Bad cache index."),
|
||||
};
|
||||
|
||||
(base_index + self.cache_index_offset) % CACHED_EPOCHS
|
||||
}
|
||||
|
||||
pub fn get_active_validator_indices(
|
||||
&mut self,
|
||||
state: &BeaconState,
|
||||
epoch: Epoch,
|
||||
) -> &Vec<usize> {
|
||||
let i = self.cache_index(epoch);
|
||||
|
||||
if self.active_validator_indices_cache[i] == None {
|
||||
self.active_validator_indices_cache[i] = Some(get_active_validator_indices(
|
||||
&state.validator_registry,
|
||||
epoch,
|
||||
));
|
||||
}
|
||||
|
||||
self.active_validator_indices_cache[i]
|
||||
.as_ref()
|
||||
.expect("Cache cannot be None")
|
||||
}
|
||||
|
||||
pub fn get_shuffling(
|
||||
&mut self,
|
||||
state: &BeaconState,
|
||||
seed: Hash256,
|
||||
epoch: Epoch,
|
||||
spec: &ChainSpec,
|
||||
) -> Option<&Vec<Vec<usize>>> {
|
||||
let cache_index = self.cache_index(epoch);
|
||||
|
||||
if self.shuffling_cache[cache_index] == None {
|
||||
let active_validator_indices = self.get_active_validator_indices(&state, epoch);
|
||||
|
||||
if active_validator_indices.is_empty() {
|
||||
return None;
|
||||
}
|
||||
|
||||
let committees_per_epoch =
|
||||
state.get_epoch_committee_count(active_validator_indices.len(), spec);
|
||||
|
||||
let mut shuffled_active_validator_indices = vec![0; active_validator_indices.len()];
|
||||
for &i in active_validator_indices {
|
||||
let shuffled_i = get_permutated_index(
|
||||
i,
|
||||
active_validator_indices.len(),
|
||||
&seed[..],
|
||||
spec.shuffle_round_count,
|
||||
)?;
|
||||
shuffled_active_validator_indices[i] = active_validator_indices[shuffled_i]
|
||||
}
|
||||
|
||||
let committees: Vec<Vec<usize>> = shuffled_active_validator_indices
|
||||
.honey_badger_split(committees_per_epoch as usize)
|
||||
.map(|slice: &[usize]| slice.to_vec())
|
||||
.collect();
|
||||
|
||||
self.shuffling_cache[cache_index] = Some(Some(committees));
|
||||
}
|
||||
|
||||
match self.shuffling_cache[cache_index] {
|
||||
Some(_) => Some(
|
||||
self.shuffling_cache[cache_index]
|
||||
.as_ref()
|
||||
.expect("Cache cannot be None")
|
||||
.as_ref()
|
||||
.expect("Shuffling cannot be None."),
|
||||
),
|
||||
None => None,
|
||||
}
|
||||
}
|
||||
}
|
@ -16,7 +16,6 @@ pub mod deposit_input;
|
||||
pub mod eth1_data;
|
||||
pub mod eth1_data_vote;
|
||||
pub mod exit;
|
||||
pub mod fast_beacon_state;
|
||||
pub mod fork;
|
||||
pub mod free_attestation;
|
||||
pub mod pending_attestation;
|
||||
@ -53,7 +52,6 @@ pub use crate::deposit_input::DepositInput;
|
||||
pub use crate::eth1_data::Eth1Data;
|
||||
pub use crate::eth1_data_vote::Eth1DataVote;
|
||||
pub use crate::exit::Exit;
|
||||
pub use crate::fast_beacon_state::FastBeaconState;
|
||||
pub use crate::fork::Fork;
|
||||
pub use crate::free_attestation::FreeAttestation;
|
||||
pub use crate::pending_attestation::PendingAttestation;
|
||||
|
Loading…
Reference in New Issue
Block a user