#[macro_use]
extern crate criterion;

use rand::{thread_rng, Rng};
use rand_distr::Uniform;

use argmm::ArgMinMax;
use criterion::{black_box, Criterion};

fn get_array_u8() -> Vec<u8> {
    let rng = thread_rng();
    let uni = Uniform::from(std::u8::MIN..std::u8::MAX);
    rng.sample_iter(uni).take(512).collect()
}

fn max_u8(c: &mut Criterion) {
    let data = get_array_u8();
    c.bench_function("simple_argmax_u8", |b| {
        b.iter(|| argmm::generic::simple_argmax(black_box(data.as_slice())))
    });
    let data = get_array_u8();
    c.bench_function("argmax_simd_u8", |b| {
        b.iter(|| black_box(data.as_slice().argmax()))
    });
}

fn min_u8(c: &mut Criterion) {
    let data = get_array_u8();
    c.bench_function("simple_argmin_u8", |b| {
        b.iter(|| argmm::generic::simple_argmin(black_box(data.as_slice())))
    });
    let data = get_array_u8();
    c.bench_function("argmin_simd_u8", |b| {
        b.iter(|| black_box(data.as_slice().argmin()))
    });
}

criterion_group!(benches, max_u8, min_u8);
criterion_main!(benches);