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

   1 /*
   2   Stockfish, a UCI chess playing engine derived from Glaurung 2.1
   3   Copyright (C) 2004-2008 Tord Romstad (Glaurung author)
   4   Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad
   5
   6   Stockfish is free software: you can redistribute it and/or modify
   7   it under the terms of the GNU General Public License as published by
   8   the Free Software Foundation, either version 3 of the License, or
   9   (at your option) any later version.
  10
  11   Stockfish is distributed in the hope that it will be useful,
  12   but WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14   GNU General Public License for more details.
  15
  16   You should have received a copy of the GNU General Public License
  17   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include <cassert>
  21 #include <cstring>
  22 #include <iomanip>
  23 #include <string>
  24 #include <sstream>
  25
  26 #include "move.h"
  27 #include "movegen.h"
  28 #include "search.h"
  29
  30 using std::string;
  31
  32 namespace {
  33   const string time_string(int milliseconds);
  34   const string score_string(Value v);
  35 }
  36
  37
  38 /// move_to_uci() converts a move to a string in coordinate notation
  39 /// (g1f3, a7a8q, etc.). The only special case is castling moves, where we
  40 /// print in the e1g1 notation in normal chess mode, and in e1h1 notation in
  41 /// Chess960 mode.
  42
  43 const string move_to_uci(Move m, bool chess960) {
  44
  45   Square from = move_from(m);
  46   Square to = move_to(m);
  47   string promotion;
  48
  49   if (m == MOVE_NONE)
  50       return "(none)";
  51
  52   if (m == MOVE_NULL)
  53       return "0000";
  54
  55   if (move_is_short_castle(m) && !chess960)
  56       return from == SQ_E1 ? "e1g1" : "e8g8";
  57
  58   if (move_is_long_castle(m) && !chess960)
  59       return from == SQ_E1 ? "e1c1" : "e8c8";
  60
  61   if (move_is_promotion(m))
  62       promotion = char(tolower(piece_type_to_char(move_promotion_piece(m))));
  63
  64   return square_to_string(from) + square_to_string(to) + promotion;
  65 }
  66
  67
  68 /// move_from_uci() takes a position and a string representing a move in
  69 /// simple coordinate notation and returns an equivalent Move if any.
  70 /// Moves are guaranteed to be legal.
  71
  72 Move move_from_uci(const Position& pos, const string& str) {
  73
  74   MoveStack mlist[MAX_MOVES];
  75   MoveStack* last = generate<MV_LEGAL>(pos, mlist);
  76
  77   for (MoveStack* cur = mlist; cur != last; cur++)
  78       if (str == move_to_uci(cur->move, pos.is_chess960()))
  79           return cur->move;
  80
  81   return MOVE_NONE;
  82 }
  83
  84
  85 /// move_to_san() takes a position and a move as input, where it is assumed
  86 /// that the move is a legal move from the position. The return value is
  87 /// a string containing the move in short algebraic notation.
  88
  89 const string move_to_san(Position& pos, Move m) {
  90
  91   assert(pos.is_ok());
  92   assert(move_is_ok(m));
  93
  94   MoveStack mlist[MAX_MOVES];
  95   Square from = move_from(m);
  96   Square to = move_to(m);
  97   PieceType pt = pos.type_of_piece_on(from);
  98   string san;
  99
 100   if (m == MOVE_NONE)
 101       return "(none)";
 102
 103   if (m == MOVE_NULL)
 104       return "(null)";
 105
 106   if (move_is_long_castle(m))
 107       san = "O-O-O";
 108   else if (move_is_short_castle(m))
 109       san = "O-O";
 110   else
 111   {
 112       if (pt != PAWN)
 113       {
 114           san = piece_type_to_char(pt);
 115
 116           // Collect all legal moves of piece type 'pt' with destination 'to'
 117           MoveStack* last = generate<MV_LEGAL>(pos, mlist);
 118           int f = 0, r = 0;
 119
 120           for (MoveStack* cur = mlist; cur != last; cur++)
 121               if (   move_to(cur->move) == to
 122                   && pos.type_of_piece_on(move_from(cur->move)) == pt)
 123               {
 124                   if (square_file(move_from(cur->move)) == square_file(from))
 125                       f++;
 126
 127                   if (square_rank(move_from(cur->move)) == square_rank(from))
 128                       r++;
 129               }
 130
 131           assert(f > 0 && r > 0);
 132
 133           // Disambiguation if we have more then one piece with destination 'to'
 134           if (f == 1 && r > 1)
 135               san += file_to_char(square_file(from));
 136           else if (f > 1 && r == 1)
 137               san += rank_to_char(square_rank(from));
 138           else if (f > 1 && r > 1)
 139               san += square_to_string(from);
 140       }
 141
 142       if (pos.move_is_capture(m))
 143       {
 144           if (pt == PAWN)
 145               san += file_to_char(square_file(from));
 146
 147           san += 'x';
 148       }
 149
 150       san += square_to_string(to);
 151
 152       if (move_is_promotion(m))
 153       {
 154           san += '=';
 155           san += piece_type_to_char(move_promotion_piece(m));
 156       }
 157   }
 158
 159   // The move gives check? We don't use pos.move_gives_check() here
 160   // because we need to test for a mate after the move is done.
 161   StateInfo st;
 162   pos.do_move(m, st);
 163   if (pos.in_check())
 164       san += pos.is_mate() ? "#" : "+";
 165   pos.undo_move(m);
 166
 167   return san;
 168 }
 169
 170
 171 /// pretty_pv() creates a human-readable string from a position and a PV.
 172 /// It is used to write search information to the log file (which is created
 173 /// when the UCI parameter "Use Search Log" is "true").
 174
 175 const string pretty_pv(Position& pos, int depth, Value score, int time, Move pv[]) {
 176
 177   const int64_t K = 1000;
 178   const int64_t M = 1000000;
 179   const int startColumn = 28;
 180   const size_t maxLength = 80 - startColumn;
 181   const string lf = string("\n") + string(startColumn, ' ');
 182
 183   StateInfo state[PLY_MAX_PLUS_2], *st = state;
 184   Move* m = pv;
 185   string san;
 186   std::stringstream s;
 187   size_t length = 0;
 188
 189   // First print depth, score, time and searched nodes...
 190   s << std::setw(2) << depth
 191     << std::setw(8) << score_string(score)
 192     << std::setw(8) << time_string(time);
 193
 194   if (pos.nodes_searched() < M)
 195       s << std::setw(8) << pos.nodes_searched() / 1 << "  ";
 196   else if (pos.nodes_searched() < K * M)
 197       s << std::setw(7) << pos.nodes_searched() / K << "K  ";
 198   else
 199       s << std::setw(7) << pos.nodes_searched() / M << "M  ";
 200
 201   // ...then print the full PV line in short algebraic notation
 202   while (*m != MOVE_NONE)
 203   {
 204       san = move_to_san(pos, *m);
 205       length += san.length() + 1;
 206
 207       if (length > maxLength)
 208       {
 209           length = san.length() + 1;
 210           s << lf;
 211       }
 212       s << san << ' ';
 213
 214       pos.do_move(*m++, *st++);
 215   }
 216
 217   // Restore original position before to leave
 218   while (m != pv) pos.undo_move(*--m);
 219
 220   return s.str();
 221 }
 222
 223
 224 namespace {
 225
 226   const string time_string(int millisecs) {
 227
 228     const int MSecMinute = 1000 * 60;
 229     const int MSecHour   = 1000 * 60 * 60;
 230
 231     int hours = millisecs / MSecHour;
 232     int minutes =  (millisecs % MSecHour) / MSecMinute;
 233     int seconds = ((millisecs % MSecHour) % MSecMinute) / 1000;
 234
 235     std::stringstream s;
 236
 237     if (hours)
 238         s << hours << ':';
 239
 240     s << std::setfill('0') << std::setw(2) << minutes << ':' << std::setw(2) << seconds;
 241     return s.str();
 242   }
 243
 244
 245   const string score_string(Value v) {
 246
 247     std::stringstream s;
 248
 249     if (v >= VALUE_MATE - 200)
 250         s << "#" << (VALUE_MATE - v + 1) / 2;
 251     else if (v <= -VALUE_MATE + 200)
 252         s << "-#" << (VALUE_MATE + v) / 2;
 253     else
 254         s << std::setprecision(2) << std::fixed << std::showpos << float(v) / PawnValueMidgame;
 255
 256     return s.str();
 257   }
 258 }