use std::convert::TryInto;

use criterion::{black_box, criterion_group, criterion_main, Criterion};
use magpie::othello::{Bitboard, Board, Position, Stone};

fn bench_clone(c: &mut Criterion) {
    let board = Board::standard();
    c.bench_function("clone", |b| b.iter(|| board.clone()));
}

fn bench_legal_moves(c: &mut Criterion) {
    let board = board_for_legal_moves();
    c.bench_function("legal_moves", |b| {
        b.iter(|| board.moves_for(black_box(Stone::Black)));
    });
}

fn bench_play(c: &mut Criterion) {
    let board = board_for_play();
    let pos: Position = (0x00_00_00_00_08_00_00_00).try_into().unwrap();
    c.bench_function("play", |b| {
        b.iter(|| {
            board.clone().play(black_box(Stone::Black), black_box(pos));
        });
    });
}

fn bench_legal_move_check(c: &mut Criterion) {
    let board = board_for_play();
    let pos: Position = (0x00_00_00_00_08_00_00_00).try_into().unwrap();
    c.bench_function("legal_move_check", |b| {
        b.iter(|| board.is_legal_move(black_box(Stone::Black), black_box(pos)));
    });
}

fn bench_bits_extraction(c: &mut Criterion) {
    let board = board_for_legal_moves();
    let moves = board.moves_for(Stone::Black);
    c.bench_function("bits_extraction", |b| {
        b.iter(|| moves.hot_bits().collect::<Vec<Position>>());
    });
}

fn bench_hot_bits_extraction(c: &mut Criterion) {
    let board = board_for_legal_moves();
    let moves = board.moves_for(Stone::Black);
    c.bench_function("hot_bits_extraction", |b| {
        b.iter(|| moves.bits().collect::<Vec<Bitboard>>());
    });
}

criterion_group!(
    benches,
    bench_clone,
    bench_legal_moves,
    bench_play,
    bench_legal_move_check,
    bench_bits_extraction,
    bench_hot_bits_extraction,
);
criterion_main!(benches);

fn board_for_play() -> Board {
    let black_pos = 0x88_01_00_00_81_00_00_49;
    let white_pos = 0x00_48_2a_1c_76_1c_2a_00;

    Board::try_from((black_pos, white_pos)).unwrap()
}

fn board_for_legal_moves() -> Board {
    let black_pos = 0x00_11_66_0c_3c_2c_00_00;
    let white_pos = 0x00_66_00_52_40_52_56_00;

    Board::try_from((black_pos, white_pos)).unwrap()
}