git.sesse.net Git - stockfish/blob - src/movepick.cpp

   1 /*
   2   Stockfish, a UCI chess playing engine derived from Glaurung 2.1
   3   Copyright (C) 2004-2023 The Stockfish developers (see AUTHORS file)
   4
   5   Stockfish is free software: you can redistribute it and/or modify
   6   it under the terms of the GNU General Public License as published by
   7   the Free Software Foundation, either version 3 of the License, or
   8   (at your option) any later version.
   9
  10   Stockfish is distributed in the hope that it will be useful,
  11   but WITHOUT ANY WARRANTY; without even the implied warranty of
  12   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13   GNU General Public License for more details.
  14
  15   You should have received a copy of the GNU General Public License
  16   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 #include "movepick.h"
  20
  21 #include <algorithm>
  22 #include <cassert>
  23 #include <iterator>
  24 #include <utility>
  25
  26 #include "bitboard.h"
  27 #include "position.h"
  28
  29 namespace Stockfish {
  30
  31 namespace {
  32
  33 enum Stages {
  34     // generate main search moves
  35     MAIN_TT,
  36     CAPTURE_INIT,
  37     GOOD_CAPTURE,
  38     REFUTATION,
  39     QUIET_INIT,
  40     QUIET,
  41     BAD_CAPTURE,
  42
  43     // generate evasion moves
  44     EVASION_TT,
  45     EVASION_INIT,
  46     EVASION,
  47
  48     // generate probcut moves
  49     PROBCUT_TT,
  50     PROBCUT_INIT,
  51     PROBCUT,
  52
  53     // generate qsearch moves
  54     QSEARCH_TT,
  55     QCAPTURE_INIT,
  56     QCAPTURE,
  57     QCHECK_INIT,
  58     QCHECK
  59 };
  60
  61 // Sort moves in descending order up to and including
  62 // a given limit. The order of moves smaller than the limit is left unspecified.
  63 void partial_insertion_sort(ExtMove* begin, ExtMove* end, int limit) {
  64
  65     for (ExtMove *sortedEnd = begin, *p = begin + 1; p < end; ++p)
  66         if (p->value >= limit)
  67         {
  68             ExtMove tmp = *p, *q;
  69             *p          = *++sortedEnd;
  70             for (q = sortedEnd; q != begin && *(q - 1) < tmp; --q)
  71                 *q = *(q - 1);
  72             *q = tmp;
  73         }
  74 }
  75
  76 }  // namespace
  77
  78
  79 // Constructors of the MovePicker class. As arguments, we pass information
  80 // to help it return the (presumably) good moves first, to decide which
  81 // moves to return (in the quiescence search, for instance, we only want to
  82 // search captures, promotions, and some checks) and how important a good
  83 // move ordering is at the current node.
  84
  85 // MovePicker constructor for the main search
  86 MovePicker::MovePicker(const Position&              p,
  87                        Move                         ttm,
  88                        Depth                        d,
  89                        const ButterflyHistory*      mh,
  90                        const CapturePieceToHistory* cph,
  91                        const PieceToHistory**       ch,
  92                        const PawnHistory*           ph,
  93                        Move                         cm,
  94                        const Move*                  killers) :
  95     pos(p),
  96     mainHistory(mh),
  97     captureHistory(cph),
  98     continuationHistory(ch),
  99     pawnHistory(ph),
 100     ttMove(ttm),
 101     refutations{{killers[0], 0}, {killers[1], 0}, {cm, 0}},
 102     depth(d) {
 103     assert(d > 0);
 104
 105     stage = (pos.checkers() ? EVASION_TT : MAIN_TT) + !(ttm && pos.pseudo_legal(ttm));
 106 }
 107
 108 // Constructor for quiescence search
 109 MovePicker::MovePicker(const Position&              p,
 110                        Move                         ttm,
 111                        Depth                        d,
 112                        const ButterflyHistory*      mh,
 113                        const CapturePieceToHistory* cph,
 114                        const PieceToHistory**       ch,
 115                        const PawnHistory*           ph,
 116                        Square                       rs) :
 117     pos(p),
 118     mainHistory(mh),
 119     captureHistory(cph),
 120     continuationHistory(ch),
 121     pawnHistory(ph),
 122     ttMove(ttm),
 123     recaptureSquare(rs),
 124     depth(d) {
 125     assert(d <= 0);
 126
 127     stage = (pos.checkers() ? EVASION_TT : QSEARCH_TT) + !(ttm && pos.pseudo_legal(ttm));
 128 }
 129
 130 // Constructor for ProbCut: we generate captures with SEE greater
 131 // than or equal to the given threshold.
 132 MovePicker::MovePicker(const Position& p, Move ttm, Value th, const CapturePieceToHistory* cph) :
 133     pos(p),
 134     captureHistory(cph),
 135     ttMove(ttm),
 136     threshold(th) {
 137     assert(!pos.checkers());
 138
 139     stage = PROBCUT_TT
 140           + !(ttm && pos.capture_stage(ttm) && pos.pseudo_legal(ttm) && pos.see_ge(ttm, threshold));
 141 }
 142
 143 // Assigns a numerical value to each move in a list, used
 144 // for sorting. Captures are ordered by Most Valuable Victim (MVV), preferring
 145 // captures with a good history. Quiets moves are ordered using the history tables.
 146 template<GenType Type>
 147 void MovePicker::score() {
 148
 149     static_assert(Type == CAPTURES || Type == QUIETS || Type == EVASIONS, "Wrong type");
 150
 151     [[maybe_unused]] Bitboard threatenedByPawn, threatenedByMinor, threatenedByRook,
 152       threatenedPieces;
 153     if constexpr (Type == QUIETS)
 154     {
 155         Color us = pos.side_to_move();
 156
 157         threatenedByPawn = pos.attacks_by<PAWN>(~us);
 158         threatenedByMinor =
 159           pos.attacks_by<KNIGHT>(~us) | pos.attacks_by<BISHOP>(~us) | threatenedByPawn;
 160         threatenedByRook = pos.attacks_by<ROOK>(~us) | threatenedByMinor;
 161
 162         // Pieces threatened by pieces of lesser material value
 163         threatenedPieces = (pos.pieces(us, QUEEN) & threatenedByRook)
 164                          | (pos.pieces(us, ROOK) & threatenedByMinor)
 165                          | (pos.pieces(us, KNIGHT, BISHOP) & threatenedByPawn);
 166     }
 167
 168     for (auto& m : *this)
 169         if constexpr (Type == CAPTURES)
 170             m.value =
 171               (7 * int(PieceValue[pos.piece_on(to_sq(m))])
 172                + (*captureHistory)[pos.moved_piece(m)][to_sq(m)][type_of(pos.piece_on(to_sq(m)))])
 173               / 16;
 174
 175         else if constexpr (Type == QUIETS)
 176         {
 177             Piece     pc   = pos.moved_piece(m);
 178             PieceType pt   = type_of(pos.moved_piece(m));
 179             Square    from = from_sq(m);
 180             Square    to   = to_sq(m);
 181
 182             // histories
 183             m.value = 2 * (*mainHistory)[pos.side_to_move()][from_to(m)];
 184             m.value += 2 * (*continuationHistory[0])[pc][to];
 185             m.value += (*continuationHistory[1])[pc][to];
 186             m.value += (*continuationHistory[2])[pc][to] / 4;
 187             m.value += (*continuationHistory[3])[pc][to];
 188             m.value += (*continuationHistory[5])[pc][to];
 189             m.value += (*pawnHistory)[pawn_structure(pos)][pc][to];
 190
 191             // bonus for checks
 192             m.value += bool(pos.check_squares(pt) & to) * 16384;
 193
 194             // bonus for escaping from capture
 195             m.value += threatenedPieces & from ? (pt == QUEEN && !(to & threatenedByRook)   ? 50000
 196                                                   : pt == ROOK && !(to & threatenedByMinor) ? 25000
 197                                                   : !(to & threatenedByPawn)                ? 15000
 198                                                                                             : 0)
 199                                                : 0;
 200
 201             // malus for putting piece en prise
 202             m.value -= !(threatenedPieces & from)
 203                        ? (pt == QUEEN ? bool(to & threatenedByRook) * 50000
 204                                           + bool(to & threatenedByMinor) * 10000
 205                                           + bool(to & threatenedByPawn) * 20000
 206                           : pt == ROOK ? bool(to & threatenedByMinor) * 25000
 207                                            + bool(to & threatenedByPawn) * 10000
 208                           : pt != PAWN ? bool(to & threatenedByPawn) * 15000
 209                                        : 0)
 210                        : 0;
 211         }
 212
 213         else  // Type == EVASIONS
 214         {
 215             if (pos.capture_stage(m))
 216                 m.value = PieceValue[pos.piece_on(to_sq(m))] - Value(type_of(pos.moved_piece(m)))
 217                         + (1 << 28);
 218             else
 219                 m.value = (*mainHistory)[pos.side_to_move()][from_to(m)]
 220                         + (*continuationHistory[0])[pos.moved_piece(m)][to_sq(m)]
 221                         + (*pawnHistory)[pawn_structure(pos)][pos.moved_piece(m)][to_sq(m)];
 222         }
 223 }
 224
 225 // Returns the next move satisfying a predicate function.
 226 // It never returns the TT move.
 227 template<MovePicker::PickType T, typename Pred>
 228 Move MovePicker::select(Pred filter) {
 229
 230     while (cur < endMoves)
 231     {
 232         if constexpr (T == Best)
 233             std::swap(*cur, *std::max_element(cur, endMoves));
 234
 235         if (*cur != ttMove && filter())
 236             return *cur++;
 237
 238         cur++;
 239     }
 240     return MOVE_NONE;
 241 }
 242
 243 // Most important method of the MovePicker class. It
 244 // returns a new pseudo-legal move every time it is called until there are no more
 245 // moves left, picking the move with the highest score from a list of generated moves.
 246 Move MovePicker::next_move(bool skipQuiets) {
 247
 248 top:
 249     switch (stage)
 250     {
 251
 252     case MAIN_TT :
 253     case EVASION_TT :
 254     case QSEARCH_TT :
 255     case PROBCUT_TT :
 256         ++stage;
 257         return ttMove;
 258
 259     case CAPTURE_INIT :
 260     case PROBCUT_INIT :
 261     case QCAPTURE_INIT :
 262         cur = endBadCaptures = moves;
 263         endMoves             = generate<CAPTURES>(pos, cur);
 264
 265         score<CAPTURES>();
 266         partial_insertion_sort(cur, endMoves, std::numeric_limits<int>::min());
 267         ++stage;
 268         goto top;
 269
 270     case GOOD_CAPTURE :
 271         if (select<Next>([&]() {
 272                 // Move losing capture to endBadCaptures to be tried later
 273                 return pos.see_ge(*cur, Value(-cur->value)) ? true
 274                                                             : (*endBadCaptures++ = *cur, false);
 275             }))
 276             return *(cur - 1);
 277
 278         // Prepare the pointers to loop over the refutations array
 279         cur      = std::begin(refutations);
 280         endMoves = std::end(refutations);
 281
 282         // If the countermove is the same as a killer, skip it
 283         if (refutations[0].move == refutations[2].move
 284             || refutations[1].move == refutations[2].move)
 285             --endMoves;
 286
 287         ++stage;
 288         [[fallthrough]];
 289
 290     case REFUTATION :
 291         if (select<Next>([&]() {
 292                 return *cur != MOVE_NONE && !pos.capture_stage(*cur) && pos.pseudo_legal(*cur);
 293             }))
 294             return *(cur - 1);
 295         ++stage;
 296         [[fallthrough]];
 297
 298     case QUIET_INIT :
 299         if (!skipQuiets)
 300         {
 301             cur      = endBadCaptures;
 302             endMoves = generate<QUIETS>(pos, cur);
 303
 304             score<QUIETS>();
 305             partial_insertion_sort(cur, endMoves, -3000 * depth);
 306         }
 307
 308         ++stage;
 309         [[fallthrough]];
 310
 311     case QUIET :
 312         if (!skipQuiets && select<Next>([&]() {
 313                 return *cur != refutations[0].move && *cur != refutations[1].move
 314                     && *cur != refutations[2].move;
 315             }))
 316             return *(cur - 1);
 317
 318         // Prepare the pointers to loop over the bad captures
 319         cur      = moves;
 320         endMoves = endBadCaptures;
 321
 322         ++stage;
 323         [[fallthrough]];
 324
 325     case BAD_CAPTURE :
 326         return select<Next>([]() { return true; });
 327
 328     case EVASION_INIT :
 329         cur      = moves;
 330         endMoves = generate<EVASIONS>(pos, cur);
 331
 332         score<EVASIONS>();
 333         ++stage;
 334         [[fallthrough]];
 335
 336     case EVASION :
 337         return select<Best>([]() { return true; });
 338
 339     case PROBCUT :
 340         return select<Next>([&]() { return pos.see_ge(*cur, threshold); });
 341
 342     case QCAPTURE :
 343         if (select<Next>(
 344               [&]() { return depth > DEPTH_QS_RECAPTURES || to_sq(*cur) == recaptureSquare; }))
 345             return *(cur - 1);
 346
 347         // If we did not find any move and we do not try checks, we have finished
 348         if (depth != DEPTH_QS_CHECKS)
 349             return MOVE_NONE;
 350
 351         ++stage;
 352         [[fallthrough]];
 353
 354     case QCHECK_INIT :
 355         cur      = moves;
 356         endMoves = generate<QUIET_CHECKS>(pos, cur);
 357
 358         ++stage;
 359         [[fallthrough]];
 360
 361     case QCHECK :
 362         return select<Next>([]() { return true; });
 363     }
 364
 365     assert(false);
 366     return MOVE_NONE;  // Silence warning
 367 }
 368
 369 }  // namespace Stockfish