adding json based perft tests

1 files changed, 51 insertions, 3 deletions

diff --git a/src/main.cpp b/src/main.cpp
index 4469e40..a2af33f 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -1,13 +1,61 @@
+#include <cassert>
+#include <chrono>
+#include <cstdint>
+#include <cstdlib>
+#include <iostream>
+#include <sstream>
+#include <string>
+
+#include "board/board.hpp"
 #include "misc.hpp"
 #include "uci.hpp"
 #include "zobrist.hpp"
-#include <iostream>
 
-int main() {
+static void perft(int argc, char *argv[]) {
+  if (argc != 4 || std::string(argv[1]) != "perft") {
+    std::cout << "Usage: ./mono\n";
+    std::cout << "       ./mono perft <non-negative depth> <fen>\n";
+    exit(1);
+  }
+
+  int depth = -1;
+  try {
+    depth = std::stoi(argv[2]);
+  } catch (const std::exception &) {
+    std::cout << "info string usage: ./mono perft <non-negative depth>\n";
+    exit(1);
+  }
+
+  if (depth < 0) {
+    std::cout << "info string usage: ./mono perft <non-negative depth>\n";
+    exit(1);
+  }
+
+  std::string fen = argv[3];
+
+  auto start = std::chrono::steady_clock::now();
+
+  Game game = initBoard(fen, nullptr);
+  uint64_t count = MoveGenTest(depth, &game);
+
+  auto end = std::chrono::steady_clock::now();
+  auto ms = std::chrono::duration_cast<std::chrono::milliseconds>(end - start)
+                .count();
+  std::ostringstream oss;
+  oss << "{\"ms\": " << ms << ", \"result\": " << count << "}";
+
+  std::cout << oss.str() << std::endl;
+}
+
+int main(int argc, char *argv[]) {
   InitZobrist();
 
-  std::cout << engine_info();
+  if (argc != 1) {
+    perft(argc, argv);
+    return 0;
+  }
 
+  std::cout << engine_info();
   Uci();
   return 0;
 }