agent.h

Go to the documentation of this file.

/*  */
#include <vector>

#include "chessboard.h"

class Node {
   public:
    Node(Chessboard state, std::pair<int, int> move) : board_state(state), move(move), score(0) {}
    void reset_tree_recursive();

    Chessboard board_state;
    std::pair<int, int> move;
    int score;
    std::vector<Node *> children;

   private:
    Node(Node &&other) = delete;
    Node &operator=(Node &&other) = delete;
};

class Agent {
   public:
    Agent(Chessboard initial_board);
    Node *root;
    std::pair<int, int> find_best_move(int depth);

    void reset_tree(Chessboard state);

   private:
    void initialize_piece_structure_bonus();
    std::vector<std::pair<int, int>> generate_possible_moves(Node *node, bool b_team);

    void generate_tree(Node *node, int depth, bool b_team);
    int minimax(Node *node, int depth, int alpha, int beta, bool maximizingPlayer);
    int min(int a, int b);
    int max(int a, int b);

    int evaluate(Chessboard state);

    std::vector<int> get_piece_structure(Piece piece);
    int get_piece_value(Type type);

    std::vector<int> piece_values;

    std::vector<std::vector<int>> piece_structures;
    std::vector<int> pawn_structure_white;
    std::vector<int> pawn_structure_black;
    std::vector<int> knight_structure_white;
    std::vector<int> knight_structure_black;
    std::vector<int> bishop_structure_white;
    std::vector<int> bishop_structure_black;
    std::vector<int> rook_structure_white;
    std::vector<int> rook_structure_black;
    std::vector<int> king_structure_white;
    std::vector<int> king_structure_black;
    std::vector<int> queen_structure_white;
    std::vector<int> queen_structure_black;

    std::vector<int> end_king_structure_white;
    std::vector<int> end_king_structure_black;

    Agent(const Agent &other) = delete;
    Agent &operator=(const Agent &other) = delete;
    Agent(Agent &&other) = delete;
    Agent &operator=(Agent &&other) = delete;
};