use criterion::{black_box, criterion_group, criterion_main, Criterion};

use pui_arena::base::sparse::Arena;

#[allow(clippy::many_single_char_names)]
pub fn sparse(c: &mut Criterion) {
    c.bench_function("sparse insertion", |b| {
        b.iter(|| {
            let mut arena = Arena::new();
            let arena = black_box(&mut arena);
            let _: usize = arena.insert(());
        })
    });
    c.bench_function("sparse re-insertion", |b| {
        let mut arena = Arena::new();
        b.iter(|| {
            let arena = black_box(&mut arena);
            let a: usize = arena.insert(());
            let b: usize = arena.insert(());
            let c: usize = arena.insert(());
            let d: usize = arena.insert(());
            let e: usize = arena.insert(());

            let (a, b, c, d, e) = black_box((a, b, c, d, e));

            arena.remove(b);
            arena.remove(d);
            arena.remove(a);
            arena.remove(c);
            arena.remove(e);

            let _ = black_box((a, b, c, d, e));

            let a: usize = arena.insert(());
            let b: usize = arena.insert(());
            let c: usize = arena.insert(());
            let d: usize = arena.insert(());
            let e: usize = arena.insert(());

            let (a, b, c, d, e) = black_box((a, b, c, d, e));

            arena.remove(b);
            arena.remove(d);
            arena.remove(a);
            arena.remove(c);
            arena.remove(e);

            let _ = black_box((a, b, c, d, e));
        });
    });
    c.bench_function("sparse iteration non-contigious", |b| {
        let mut arena = Arena::new();

        for i in 0..1000 {
            let _: usize = arena.insert(i);
        }

        for i in (0..1000).filter(|&i| i % 2 == 0 || i % 11 == 0) {
            arena.remove(i);
        }

        b.iter(|| black_box(&arena).iter().sum::<i32>())
    });
    c.bench_function("sparse iteration contigious", |b| {
        let mut arena = Arena::new();

        for i in 0..1000 {
            let _: usize = arena.insert(i);
        }

        for i in (1..=1000).filter(|&i| i % 11 < 6) {
            arena.remove(i);
        }

        b.iter(|| black_box(&arena).iter().sum::<i32>())
    });
    c.bench_function("sparse iteration packed", |b| {
        let mut arena = Arena::new();

        for i in 0..545 {
            let _: usize = arena.insert(i);
        }

        b.iter(|| black_box(&arena).iter().sum::<i32>())
    });
}

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