updated the code of main

moves

cpp/src/ais/ai.hpp

@@ -0,0 +1,23 @@
+#pragma once
+
+#include "../board.hpp"
+
+#include <string>
+
+#define DECLARE_AI(x)                                                          \
+    struct x : public AI {                                                     \
+        std::string search(std::string, int) override;                         \
+        int minimax(const Board&, int) override;                               \
+        int eval(const Board&) override;                                       \
+    };
+
+namespace ai {
+    struct AI {
+        virtual std::string search(std::string, int) = 0;
+        virtual int minimax(const Board&, int) = 0;
+        virtual int eval(const Board&) = 0;
+    };
+
+    DECLARE_AI(v0_random)
+    DECLARE_AI(v1_simple)
+} // namespace ai

cpp/src/ais/v0_random.cpp

@@ -0,0 +1,8 @@
+#include "ai.hpp"
+
+std::string ai::v1_simple::search(std::string pos, int depth) {
+    Board b = Board::setup_fen_position(pos);
+    std::vector<Move> moves = b.all_legal_moves();
+
+    return moves[rand() % moves.size()].to_string();
+}

cpp/src/ais/v1_simple.cpp

@@ -1,14 +1,13 @@
-#include "stickfosh.hpp"
-
-#include "pieces/piece.hpp"
-#include "threadpool.hpp"
+#include "../pieces/piece.hpp"
+#include "../threadpool.hpp"
+#include "ai.hpp"
 
 #include <future>
 #include <map>
 
 static int INFINITY = std::numeric_limits<int>::max();
 
-std::string search(std::string pos, int depth) {
+std::string ai::v1_simple::search(std::string pos, int depth) {
     Board b = Board::setup_fen_position(pos);
 
     ThreadPool pool(std::thread::hardware_concurrency());
@@ -18,7 +17,9 @@ std::string search(std::string pos, int depth) {
     for (const Move& move : moves) {
         Board tmp_board = b.make_move(move);
         futures.insert(
-            {move.to_string(), pool.enqueue(minimax, tmp_board, depth - 1)}
+            {move.to_string(), pool.enqueue([this, tmp_board, depth]() {
+                 return minimax(tmp_board, depth - 1);
+             })}
         );
     }
 
@@ -35,7 +36,7 @@ std::string search(std::string pos, int depth) {
     return best_move;
 }
 
-int minimax(const Board& b, int depth) {
+int ai::v1_simple::minimax(const Board& b, int depth) {
     if (b.is_checkmate_for(b.white_to_play ? White : Black))
         return -INFINITY;
 
@@ -84,7 +85,7 @@ int count_material(const Board& b, int8_t colour) {
     return ret;
 }
 
-int eval(const Board& b) {
+int ai::v1_simple::eval(const Board& b) {
     int white_eval = count_material(b, Colour::White);
     int black_eval = count_material(b, Colour::Black);
 

cpp/src/main.cpp

@@ -1,11 +1,14 @@
-#include "stickfosh.hpp"
+#include "ais/ai.hpp"
 
 #include <iostream>
 
 int main(int argc, char* argv[]) {
     std::string pos =
         "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1";
-    std::string move = search(pos, 4);
+
+    ai::v1_simple ai;
+
+    std::string move = ai.search(pos, 4);
     std::cout << move << std::endl;
     return 0;
 }

cpp/src/stickfosh.hpp

@@ -1,9 +0,0 @@
-#pragma once
-
-#include "board.hpp"
-
-#include <string>
-
-std::string search(std::string, int);
-int minimax(const Board&, int);
-int eval(const Board&);