#[macro_use]
extern crate criterion;
extern crate rand;
extern crate sha3;

extern crate ot_dalek;

use criterion::Criterion;
use ot_dalek::{Receiver, Sender};
use rand::{thread_rng, RngCore};
use sha3::Sha3_256;

fn bench_sender_new(c: &mut Criterion) {
    c.bench_function("sender setup", move |b| {
        let mut csprng = thread_rng();
        b.iter(|| {
            Sender::new(&mut csprng);
        })
    });
}

fn bench_sender_keys(c: &mut Criterion) {
    c.bench_function("sender keys", move |b| {
        let mut csprng = thread_rng();
        let (sender, s) = Sender::new(&mut csprng);
        let (_receiver, r) = Receiver::new(&mut csprng, 1, &s).unwrap();

        b.iter(|| {
            sender.keys::<Sha3_256>(&r, 2);
        })
    });
}

fn bench_receiver(c: &mut Criterion) {
    c.bench_function("receiver", move |b| {
        let mut csprng = thread_rng();
        let (_sender, s) = Sender::new(&mut csprng);

        b.iter(|| {
            let (receiver, _r) = Receiver::new(&mut csprng, 0, &s).unwrap();
            receiver.key::<Sha3_256>();
        })
    });
}

fn bench_transpose(c: &mut Criterion) {
    use ot_dalek::extension::transpose256;

    c.bench_function("transpose", move |b| {
        let mut rng = thread_rng();
        let mut src = [0u8; 128 * 32];
        rng.fill_bytes(&mut src);

        b.iter(|| {
            transpose256(&mut src);
        })
    });
}

criterion_group! {
    name = benches;
    config = Criterion::default();
    targets = bench_sender_new, bench_sender_keys, bench_receiver, bench_transpose
}
criterion_main!(benches);