#![allow(clippy::all)]
#[macro_use]
extern crate criterion;
use hex;
use vdf_classgroup::{gmp_classgroup::GmpClassGroup, ClassGroup};
use criterion::Criterion;
use std::{cell::RefCell, env, rc::Rc};
use vdf_rs::create_discriminant;
fn bench_square(c: &mut Criterion) {
let bench_params = |c: &mut Criterion, len: u16, seed: &[u8]| {
let i = Rc::new(RefCell::new(GmpClassGroup::generator_for_discriminant(
create_discriminant(seed, len),
)));
{
let i = i.clone();
c.bench_function(
&format!("square with seed {}: {}", hex::encode(seed), len),
move |b| b.iter(|| i.borrow_mut().square()),
);
}
{
let multiplier = i.borrow().clone();
c.bench_function(
&format!("multiply with seed {}: {}", hex::encode(seed), len),
move |b| b.iter(|| *i.borrow_mut() *= &multiplier),
);
}
};
let seed = env::var("VDF_BENCHMARK_SEED")
.ok()
.and_then(|x| hex::decode(x).ok())
.expect("bug in calling script");
for &i in &[512, 1024, 2048] {
bench_params(c, i, &seed)
}
}
criterion_group!(benches, bench_square);
criterion_main!(benches);