src/ais/daniel/aidaniel.cpp

   1 #include "aidaniel.h"
   2
   3 AIDaniel::AIDaniel(Position *p) : position(p), isInterrupted(false), cutoffDepth(DEFAULT_CUTOFF_DEPTH) {
   4 }
   5
   6 inline int AIDaniel::alphaBeta(Position *pos, int depth, int alpha, int beta, bool max) {
   7     if (isInterrupted)
   8     {
   9         return 0;
  10     }
  11     if ((depth >= cutoffDepth && pos->getNoisiness() < 50)
  12         || depth >= cutoffDepth + QUIESCENCE_SEARCH_MAX_ADDITIONAL_DEPTH
  13         || pos->isGameOver()) {
  14             return pos->evaluate();
  15     }
  16     if (max) {
  17         for (unsigned int i = 0; i < pos->getLegalMoves()->size(); ++i) {
  18             Position child(pos, i);
  19             beta = std::min(beta, alphaBeta(&child, depth + 1, alpha, beta, !max));
  20             if (beta <= alpha) {
  21                 return alpha;
  22             }
  23         }
  24         return beta;
  25     }
  26     else {
  27         for (unsigned int i = 0; i < pos->getLegalMoves()->size(); ++i) {
  28             Position child(pos, i);
  29             alpha = std::max(alpha, alphaBeta(&child, depth + 1, alpha, beta, !max));
  30             if (beta <= alpha) {
  31                 return beta;
  32             }
  33         }
  34         return alpha;
  35     }
  36 }
  37
  38 int AIDaniel::getNextMove() {
  39     isInterrupted = false;
  40     std::vector<int> values;
  41     int best = 0;
  42     if (position->isWhiteToMove()) {
  43         for (unsigned int i = 0; i < position->getLegalMoves()->size(); ++i) {
  44             Position child(position, i);
  45             values.push_back(alphaBeta(&child, 1, std::numeric_limits<int>::min(),std::numeric_limits<int>::max(),true));
  46             //std::cout << "white: " << i << " " << values.at(i) << " " << position->getLegalMoves()->at(i) << std::endl;
  47         }
  48         int max = std::numeric_limits<int>::min();
  49         for (unsigned int i = 0; i < values.size(); ++i) {
  50             if (max < values.at(i)) {
  51                 max = values.at(i);
  52                 best = i;
  53             }
  54         }
  55     }
  56     else {
  57         for (unsigned int i = 0; i < position->getLegalMoves()->size(); ++i) {
  58             Position child(position, i);
  59             values.push_back(alphaBeta(&child, 1, std::numeric_limits<int>::min(),std::numeric_limits<int>::max(),false));
  60             //std::cout << "black: " << i << " " << values.at(i) << " " << position->getLegalMoves()->at(i) << std::endl;
  61         }
  62         int min = std::numeric_limits<int>::max();
  63         for (unsigned int i = 0; i < values.size(); ++i) {
  64             if (min > values.at(i)) {
  65                 min = values.at(i);
  66                 best = i;
  67             }
  68         }
  69     }
  70     bestMove = position->getLegalMoves()->at(best);
  71     position->move(bestMove, 0);
  72     return bestMove;
  73 }
  74
  75 void AIDaniel::setPosition(Position *p) {
  76     delete position;
  77     position = p;
  78 }
  79
  80 void AIDaniel::interrupt()
  81 {
  82     isInterrupted = true;
  83 }
  84
  85 void AIDaniel::setCutoffDepth(int cutoffDepth)
  86 {
  87     this->cutoffDepth = cutoffDepth;
  88 }
  89
  90 AIDaniel::~AIDaniel() {
  91     delete position;
  92 }