use std::iter;

use bromberg_sl2::*;
use criterion::{black_box, criterion_group, criterion_main, BenchmarkId, Criterion, Throughput};
use digest::Update;
use sha3::{Digest, Sha3_512};

// This isjust stolen straight from the Criterion documentation
fn from_elem(c: &mut Criterion) {
    static KB: usize = 1024;
    let sizes = [KB, 2 * KB, 4 * KB, 8 * KB, 16 * KB, 4096 * KB];

    let mut group = c.benchmark_group("byte_hashing");
    for size in sizes.iter() {
        group.throughput(Throughput::Bytes(*size as u64));
        group.bench_with_input(BenchmarkId::new("bromberg", size), size, |b, &size| {
            b.iter(|| {
                black_box(
                    iter::repeat(5u8)
                        .take(size)
                        .collect::<Vec<u8>>()
                        .bromberg_hash(),
                )
            });
        });
    }
    for size in sizes.iter() {
        group.throughput(Throughput::Bytes(*size as u64));
        group.bench_with_input(
            BenchmarkId::new("bromberg_strict", size),
            size,
            |b, &size| {
                b.iter(
                    || black_box(hash_strict(&iter::repeat(5u8).take(size).collect::<Vec<u8>>()))
                );
            },
        );
    }
    #[cfg(feature = "std")]
    {
        for size in sizes.iter() {
            group.throughput(Throughput::Bytes(*size as u64));
            group.bench_with_input(BenchmarkId::new("bromberg_par", size), size, |b, &size| {
                b.iter(|| black_box(hash_par(&iter::repeat(5u8).take(size).collect::<Vec<u8>>())));
            });
        }
    }
    for size in sizes.iter() {
        group.throughput(Throughput::Bytes(*size as u64));
        group.bench_with_input(BenchmarkId::new("sha", size), size, |b, &size| {
            b.iter(|| {
                let mut hasher = Sha3_512::new();
                black_box(Update::update(
                    &mut hasher,
                    iter::repeat(5u8).take(size).collect::<Vec<u8>>().as_slice(),
                ));
                black_box(hasher.finalize());
            });
        });
    }
    group.finish();
}

criterion_group!(benches, from_elem);
criterion_main!(benches);