#[macro_use]
extern crate bencher;
extern crate paillier;

mod helpers;

#[cfg(feature="keygen")]
mod bench {

    use bencher::Bencher;
    use paillier::*;
    use helpers::*;

    pub fn bench_key_generation<KS: KeySize>(b: &mut Bencher) {
        b.iter(|| {
            Paillier::keypair_with_modulus_size(KS::size());
        });
    }

    benchmark_group!(group,
        self::bench_key_generation<KeySize512>,
        self::bench_key_generation<KeySize1024>,
        self::bench_key_generation<KeySize2048>,
        self::bench_key_generation<KeySize3072>,
        self::bench_key_generation<KeySize4096>
    );

}

#[cfg(feature="keygen")]
benchmark_main!(bench::group);

#[cfg(not(feature="keygen"))]
fn main() {}