use criterion::{criterion_group, criterion_main, Criterion};
use para_graph::algorithms::radix_sort::{radix_sort_par_cpu, radix_sort_serial};
use rand::seq::SliceRandom;
use rand::thread_rng;

const MAX: usize = 10_000_000;
const ORDER: usize = MAX;

fn test_vector() -> Vec<usize> {
    let mut vec: Vec<usize> = (0..MAX).map(|x| x * ORDER).collect();
    vec.shuffle(&mut thread_rng());
    vec
}

fn bench_radix_sort(c: &mut Criterion) {
    let mut group = c.benchmark_group("Radix Sort");
    group.sampling_mode(criterion::SamplingMode::Flat);
    group.sample_size(10);
    group.measurement_time(std::time::Duration::from_secs(60));

    let input = test_vector();

    group.bench_with_input("Serial", &input, |b, vec| {
        let mut vec = vec.clone();
        b.iter(|| radix_sort_serial(&mut vec))
    });
    group.bench_with_input("CPU", &input, |b, vec| {
        let mut vec = vec.clone();
        b.iter(|| radix_sort_par_cpu(&mut vec))
    });
    group.finish();
}

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