use codec::{Encode, WrapperTypeEncode};
use polkadot_primitives::{
CheckedMultiDisputeStatementSet, MultiDisputeStatementSet, UncheckedSignedAvailabilityBitfield,
UncheckedSignedAvailabilityBitfields,
};
use super::{BackedCandidate, Config, DisputeStatementSet, Weight};
pub trait WeightInfo {
fn enter_variable_disputes(v: u32) -> Weight;
fn enter_bitfields() -> Weight;
fn enter_backed_candidates_variable(v: u32) -> Weight;
fn enter_backed_candidate_code_upgrade() -> Weight;
}
pub struct TestWeightInfo;
#[cfg(not(feature = "runtime-benchmarks"))]
impl WeightInfo for TestWeightInfo {
fn enter_variable_disputes(v: u32) -> Weight {
Weight::from_parts(80_000 * v as u64 + 80_000, 0)
}
fn enter_bitfields() -> Weight {
Weight::from_parts(40_000u64, 0)
}
fn enter_backed_candidates_variable(v: u32) -> Weight {
Weight::from_parts(40_000 * v as u64 + 40_000, 0)
}
fn enter_backed_candidate_code_upgrade() -> Weight {
Weight::zero()
}
}
#[cfg(feature = "runtime-benchmarks")]
impl WeightInfo for TestWeightInfo {
fn enter_variable_disputes(_v: u32) -> Weight {
Weight::zero()
}
fn enter_bitfields() -> Weight {
Weight::zero()
}
fn enter_backed_candidates_variable(_v: u32) -> Weight {
Weight::zero()
}
fn enter_backed_candidate_code_upgrade() -> Weight {
Weight::zero()
}
}
pub fn paras_inherent_total_weight<T: Config>(
backed_candidates: &[BackedCandidate<<T as frame_system::Config>::Hash>],
bitfields: &UncheckedSignedAvailabilityBitfields,
disputes: &MultiDisputeStatementSet,
) -> Weight {
backed_candidates_weight::<T>(backed_candidates)
.saturating_add(signed_bitfields_weight::<T>(bitfields))
.saturating_add(multi_dispute_statement_sets_weight::<T>(disputes))
}
pub fn multi_dispute_statement_sets_weight<T: Config>(
disputes: &MultiDisputeStatementSet,
) -> Weight {
set_proof_size_to_tx_size(
disputes
.iter()
.map(|d| dispute_statement_set_weight::<T, _>(d))
.fold(Weight::zero(), |acc_weight, weight| acc_weight.saturating_add(weight)),
disputes,
)
}
pub fn checked_multi_dispute_statement_sets_weight<T: Config>(
disputes: &CheckedMultiDisputeStatementSet,
) -> Weight {
set_proof_size_to_tx_size(
disputes
.iter()
.map(|d| dispute_statement_set_weight::<T, _>(d))
.fold(Weight::zero(), |acc_weight, weight| acc_weight.saturating_add(weight)),
disputes,
)
}
pub fn dispute_statement_set_weight<T, D>(statement_set: D) -> Weight
where
T: Config,
D: AsRef<DisputeStatementSet> + WrapperTypeEncode + Sized + Encode,
{
set_proof_size_to_tx_size(
<<T as Config>::WeightInfo as WeightInfo>::enter_variable_disputes(
statement_set.as_ref().statements.len() as u32,
),
statement_set,
)
}
pub fn signed_bitfields_weight<T: Config>(
bitfields: &UncheckedSignedAvailabilityBitfields,
) -> Weight {
set_proof_size_to_tx_size(
<<T as Config>::WeightInfo as WeightInfo>::enter_bitfields()
.saturating_mul(bitfields.len() as u64),
bitfields,
)
}
pub fn signed_bitfield_weight<T: Config>(bitfield: &UncheckedSignedAvailabilityBitfield) -> Weight {
set_proof_size_to_tx_size(
<<T as Config>::WeightInfo as WeightInfo>::enter_bitfields(),
bitfield,
)
}
pub fn backed_candidate_weight<T: frame_system::Config + Config>(
candidate: &BackedCandidate<T::Hash>,
) -> Weight {
set_proof_size_to_tx_size(
if candidate.candidate().commitments.new_validation_code.is_some() {
<<T as Config>::WeightInfo as WeightInfo>::enter_backed_candidate_code_upgrade()
} else {
<<T as Config>::WeightInfo as WeightInfo>::enter_backed_candidates_variable(
candidate.validity_votes().len() as u32,
)
},
candidate,
)
}
pub fn backed_candidates_weight<T: frame_system::Config + Config>(
candidates: &[BackedCandidate<T::Hash>],
) -> Weight {
candidates
.iter()
.map(|c| backed_candidate_weight::<T>(c))
.fold(Weight::zero(), |acc, x| acc.saturating_add(x))
}
fn set_proof_size_to_tx_size<Arg: Encode>(weight: Weight, arg: Arg) -> Weight {
weight.set_proof_size(arg.encoded_size() as u64)
}