use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
use othello::{
    logic::FutureMoves,
    repr::{Board, Piece},
};
use std::time::Duration;

fn extend_layers_test(depth: usize, expire: usize) {
    let mut fut = FutureMoves::new(Piece::Black, depth, expire);
    fut.set_root_from_board(Board::new().starting_pos());
    fut.extend_layers();
}

fn criterion_benchmark(c: &mut Criterion) {
    const EXPIRE: usize = 4;
    let mut group = c.benchmark_group(format!("extend_layers (expire {})", EXPIRE));
    group.measurement_time(Duration::from_secs(10));
    group.sample_size(1000);

    for (depth, expire) in (4..8).zip([EXPIRE].into_iter().cycle()) {
        // TODO! maybe somehow get throughput from `extend_layers`?
        // group.throughput(Throughput::Elements(depth as u64));
        group.bench_with_input(BenchmarkId::from_parameter(depth), &depth, |b, depth| {
            b.iter(|| extend_layers_test(*depth, expire));
        });
    }
    group.finish();
}

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