#include <catch2/catch_test_macros.hpp>
#include <fmt/core.h>
#include <string>
#include "algos/matrix.hpp"
#include "algos/rand.hpp"
#include "constants.hpp"
#include "algos/maze.hpp"

#define DUMP 0

using std::string;
using matrix::Matrix;

TEST_CASE("hunt-and-kill", "[mazes]") {
  Map map(21, 21);
  maze::Builder maze(map);

  maze.hunt_and_kill();
  if(DUMP) maze.dump("BASIC MAZE");

  maze.randomize_rooms(ROOM_SIZE);
  maze.hunt_and_kill();
  maze.make_doors();
  if(DUMP) maze.dump("ROOM MAZE");

  REQUIRE(map.$dead_ends.size() > 0);
  REQUIRE(map.$rooms.size() > 0);
}

TEST_CASE("hunt-and-kill box", "[mazes]") {
  for(int i = 25; i < 65; i += 2) {
    Map map(i, i);
    maze::Builder maze(map);

    maze.hunt_and_kill();
    maze.clear();
    maze.inner_box(6, 4);
    maze.randomize_rooms(ROOM_SIZE+2);

    maze.hunt_and_kill();
    maze.open_box(6);
    maze.make_doors();

    if(i == 41 && DUMP) {
      maze.dump("INNER BOX");
    }
  }
}

TEST_CASE("hunt-and-kill ring", "[mazes]") {
  Map map(21, 21);
  maze::Builder maze(map);

  maze.inner_donut(5.5, 3.5);
  maze.hunt_and_kill();

  if(DUMP) maze.dump("INNER RING");

  REQUIRE(maze.$rooms.size() == 0);
}

TEST_CASE("hunt-and-kill fissure", "[mazes]") {
  Map map(21, 21);
  maze::Builder maze(map);

  maze.divide({3,3}, {19,18});
  maze.hunt_and_kill();

  if(DUMP) maze.dump("FISSURE MAZE");

  REQUIRE(maze.$rooms.size() == 0);
}

TEST_CASE("hunt-and-kill no-dead-ends", "[mazes]") {
  Map map(21, 21);
  maze::Builder maze(map);

  maze.hunt_and_kill();
  maze.remove_dead_ends();
  maze.enclose();

  if(DUMP) maze.dump("NO DEAD ENDS");
}

TEST_CASE("hunt-and-kill too much", "[mazes]") {
  for(int i = 25; i < 65; i += 2) {
    Map map(i, i);
    maze::Builder maze(map);

    maze.hunt_and_kill();
    maze.randomize_rooms(ROOM_SIZE);
    maze.clear();
    maze.inner_donut(9, 4);
    maze.divide({3,3}, {15,16});
    maze.hunt_and_kill();
    maze.make_doors();

    if(i == 41 && DUMP) {
      maze.dump("COMBINED");
    }
  }
}

TEST_CASE("hunt-and-kill validator", "[mazes]") {
  Map map(33, 33);
  maze::Builder maze(map);

  maze.hunt_and_kill();
  maze.clear();
  maze.inner_box(6, 4);
  maze.randomize_rooms(ROOM_SIZE);
  maze.hunt_and_kill();
  maze.open_box(6);
  maze.make_doors();
  maze.enclose();

  bool valid = maze.validate();

  maze.dump("VALIDATED");

  REQUIRE(valid == true);
}