use std::{fs, path::Path};

use criterion::{black_box, criterion_group, criterion_main, Criterion};
use nalgebra::Vector2;
use soukoban::{deadlock::*, path_finding::box_move_waypoints, solver::Solver, Level, Map};

pub fn load_level_from_file<P: AsRef<Path>>(path: P, id: usize) -> Level {
    debug_assert!(id >= 1);
    Level::load_nth(&fs::read_to_string(path).unwrap(), id).unwrap()
}

// Title: World Cup 2014 (MF8 61st Sokoban Competition, Extra)
// Author: laizhufu
const WORLDCUP2014: &str = r#"
    -------#########-------
    -----##---------##-----
    ---##---#####--#--##---
    --#---##------#--#--#--
    --####---##--#--#--##--
    -#-----##---#--#--#--#-
    -######----#--#--#--##-
    #-------##---#--#--#--#
    ########----#--#--#--##
    #-------.*#---#--#--#-#
    #-#-#-#-*-*-$*--*--**-#
    #-#-#-#---*-*-*-*-*-*-#
    #--#-#-#-*--*-*-*-*@*-#
    ##-#-#-#-*-**-*-*-$***#
    -#--#-#-#-*-*-*-*---*#-
    -##-#-#-#----*--.-#-##-
    --#--#-#-#-#-#-#-#--#--
    --##-#-#-#-#-#-#-#-##--
    ---##-#-#-#-#-#-#-##---
    ------#-#-#-#-#-#------
    -----#-#-#-#-#-#-#-----
    -----#-#-#-#-#-#-#-----
    ------#-#-#-#-#-#------
    ------#-#-#-#-#-#------
    ------#--#-#-#-#-#-----
    ------##-#-#-#-#-#-----
    ------##--#-#-#-#------
    ------###-#-#-#-#------
    -----####--#-#-#-#-----
    -----#####-#-#-#-#-----
    -----#####--#-#-#-#----
    ----#######-#-#-#-#----
    ----#######--#-#-#-#---
    -----#######---#-#-#---
    ---#--########----#-#--
    -#--#--##########----#-
    --#--#--#############--
    ---#--#-###########----
    -------#--######-------
"#;

// Title: 一箭十万 (102547步)
// Author: 20603
pub const PATH: &str = r#"
    ##################################################
    #   #@    #   ##  #   ##  #   ##  #   ##  #   #  #
    # # #     #       #       #       #       #      #
    # #  # ## ## ###  ## ###  ## ###  ## ###  ## ##  #
    #  # #$#      ## ###  ## ###  ## ###  ## ###  #  #
    ## #   #   #  #       #       #       #       #  #
    #  #########  #   ##  #   ##  #   ##  #   ##  #  #
    # #  #   #  ######  ######  ######  ######  ### ##
    # #      #       #       #       #       #       #
    # #  ## ##  ###  #  ###  #  ###  #  ###  #  ## # #
    # ## #  ### ##  ### ##  ### ##  ### ##  ### ##   #
    #  # #       #       #       #       #        ####
    ## # #  ##   #  ##   #  ##   #  ##   #  ###   #  #
    #  # ###  ######  ######  ######  ######  #####  #
    # #       #       #       #       #       #      #
    # #   ##  #  ###  #  ###  #  ###  #  ###  #  ## ##
    #  ##### ###  ## ###  ## ###  ## ###  ## ###  # ##
    ## #  #       #       #       #       #       # ##
    #  #  #   ##  #   ##  #   ##  #   ##  #   ##  # ##
    # ##  ######################################### ##
    #   #                                            #
    ### # ########################################   #
    #   # #   ##  #   ##  #   ##  #   ##  #   ##  ####
    # ### #       #       #       #       #       #  #
    #  ## ## ###  ## ###  ## ###  ## ###  ## ###  #  #
    ## #      ## ###  ## ###  ## ###  ## ###  ## ##  #
    #  #   #  #       #       #       #       #     ##
    # ######  #   ##  #   ##  #   ##  #   ##  #   # ##
    #     ######################################### ##
    ##### #  ##   #  ##   #  ##   #  ##   #  ##   # ##
    #     #       #       #       #       #       # ##
    # #####  ### ##  ### ##  ### ##  ### ##  ### ## ##
    #  #  ## ##  ### ##  ### ##  ### ##  ### ##  ## ##
    ## #      #       #       #       #       #      #
    #  #  #   #  ##   #  ##   #  ##   #  ##   #  #   #
    # ### ############################################
    #  ## ##  #   ##  #   ##  #   ##  #   ##  #   #  #
    ## #      #       #       #       #       #      #
    #  #   #  ## ###  ## ###  ## ###  ## ###  ## ##  #
    #  ##### ###  ## ###  ## ###  ## ###  ## ###  #  #
    # #   #       #       #       #       #       #  #
    #   # #   ##  #   ##  #   ##  #   ##  #   ##  #  #
    ##### #########################################  #
    #   # #  ##   #  ##   #  ##   #  ##   #  ##   #  #
    #     #       #       #       #       #       #  #
    ## # .#  ### ##  ### ##  ### ##  ### ##  ### ##  #
    #  ##### ##  ### ##  ### ##  ### ##  ### ##  ## ##
    #         #       #       #       #       #      #
    #  ####   #  ##   #  ##   #  ##   #  ##   #  #   #
    ##################################################
"#;

fn tunnel_benchmark(c: &mut Criterion) {
    let map = Map::from_str(PATH).unwrap();
    let solver = Solver::new(map.clone());
    solver.lower_bounds();
    c.bench_function("calculate tunnels", |b| {
        let solver = solver.clone();
        b.iter(|| {
            black_box(solver.tunnels());
        })
    });
}

fn solver_benchmark(c: &mut Criterion) {
    let map = Map::from_str(PATH).unwrap();
    c.bench_function("solve level 'PATH'", |b| {
        b.iter(|| {
            let solver = black_box(Solver::new(map.clone()));
            black_box(solver.a_star_search(&map).unwrap());
        })
    });
    let map = load_level_from_file("assets/BoxWorld_100.xsb", 3)
        .map()
        .clone();
    c.bench_function("solve level 'BoxWorld #3'", |b| {
        b.iter(|| {
            let solver = black_box(Solver::new(map.clone()));
            black_box(solver.a_star_search(&map).unwrap());
        })
    });
}

fn deadlock_benchmark(c: &mut Criterion) {
    let map = Map::from_str(WORLDCUP2014).unwrap();
    c.bench_function("calculate unused floors", |b| {
        b.iter(|| black_box(calculate_unused_floors(black_box(map.clone()))))
    });
    c.bench_function("calculate dead positions", |b| {
        b.iter(|| black_box(calculate_dead_positions(black_box(&map))))
    });
}

fn path_finding_benchmark(c: &mut Criterion) {
    let map = Map::from_str(PATH).unwrap();
    c.bench_function("box move waypoints", |b| {
        b.iter(|| box_move_waypoints(black_box(&map), Vector2::new(6, 4)))
    });
}

fn map_benchmark(c: &mut Criterion) {
    let map = Map::from_str(WORLDCUP2014).unwrap();
    c.bench_function("normalize map", |b| {
        b.iter(|| black_box(black_box(map.clone()).normalize()))
    });
}

fn level_benchmark(c: &mut Criterion) {
    let mut string = String::new();
    for entry in fs::read_dir("assets/").unwrap() {
        let path = entry.unwrap().path();
        string += &(fs::read_to_string(path).unwrap() + "\n\n");
    }

    c.bench_function("load levels", |b| {
        b.iter(|| black_box(Level::load(black_box(&string)).collect::<Vec<_>>()))
    });
    c.bench_function("load the nth level", |b| {
        b.iter(|| black_box(Level::load_nth(black_box(&string), black_box(3371)).unwrap()))
    });
}

criterion_group!(
    benches,
    tunnel_benchmark,
    solver_benchmark,
    deadlock_benchmark,
    path_finding_benchmark,
    map_benchmark,
    level_benchmark
);
criterion_main!(benches);