git.sesse.net Git - stockfish/blob - src/notation.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-2012 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 <string>
  22
  23 #include "movegen.h"
  24 #include "position.h"
  25
  26 using std::string;
  27
  28 /// move_to_uci() converts a move to a string in coordinate notation
  29 /// (g1f3, a7a8q, etc.). The only special case is castling moves, where we print
  30 /// in the e1g1 notation in normal chess mode, and in e1h1 notation in chess960
  31 /// mode. Internally castle moves are always coded as "king captures rook".
  32
  33 const string move_to_uci(Move m, bool chess960) {
  34
  35   Square from = from_sq(m);
  36   Square to = to_sq(m);
  37
  38   if (m == MOVE_NONE)
  39       return "(none)";
  40
  41   if (m == MOVE_NULL)
  42       return "0000";
  43
  44   if (type_of(m) == CASTLE && !chess960)
  45       to = (to > from ? FILE_G : FILE_C) | rank_of(from);
  46
  47   string move = square_to_string(from) + square_to_string(to);
  48
  49   if (type_of(m) == PROMOTION)
  50       move += char(tolower(piece_type_to_char(promotion_type(m))));
  51
  52   return move;
  53 }
  54
  55
  56 /// move_from_uci() takes a position and a string representing a move in
  57 /// simple coordinate notation and returns an equivalent legal Move if any.
  58
  59 Move move_from_uci(const Position& pos, string& str) {
  60
  61   if (str.length() == 5) // Junior could send promotion piece in uppercase
  62       str[4] = char(tolower(str[4]));
  63
  64   for (MoveList<LEGAL> ml(pos); !ml.end(); ++ml)
  65       if (str == move_to_uci(ml.move(), pos.is_chess960()))
  66           return ml.move();
  67
  68   return MOVE_NONE;
  69 }
  70
  71
  72 /// move_to_san() takes a position and a legal Move as input and returns its
  73 /// short algebraic notation representation.
  74
  75 const string move_to_san(Position& pos, Move m) {
  76
  77   if (m == MOVE_NONE)
  78       return "(none)";
  79
  80   if (m == MOVE_NULL)
  81       return "(null)";
  82
  83   assert(pos.move_is_legal(m));
  84
  85   Bitboard attackers;
  86   bool ambiguousMove, ambiguousFile, ambiguousRank;
  87   string san;
  88   Square from = from_sq(m);
  89   Square to = to_sq(m);
  90   PieceType pt = type_of(pos.piece_on(from));
  91
  92   if (type_of(m) == CASTLE)
  93       san = to > from ? "O-O" : "O-O-O";
  94   else
  95   {
  96       if (pt != PAWN)
  97       {
  98           san = piece_type_to_char(pt);
  99
 100           // Disambiguation if we have more then one piece with destination 'to'
 101           // note that for pawns is not needed because starting file is explicit.
 102           attackers = pos.attackers_to(to) & pos.pieces(pos.side_to_move(), pt);
 103           attackers ^= from;
 104           ambiguousMove = ambiguousFile = ambiguousRank = false;
 105
 106           while (attackers)
 107           {
 108               Square sq = pop_1st_bit(&attackers);
 109
 110               // Pinned pieces are not included in the possible sub-set
 111               if (!pos.pl_move_is_legal(make_move(sq, to), pos.pinned_pieces()))
 112                   continue;
 113
 114               ambiguousFile |= file_of(sq) == file_of(from);
 115               ambiguousRank |= rank_of(sq) == rank_of(from);
 116               ambiguousMove = true;
 117           }
 118
 119           if (ambiguousMove)
 120           {
 121               if (!ambiguousFile)
 122                   san += file_to_char(file_of(from));
 123
 124               else if (!ambiguousRank)
 125                   san += rank_to_char(rank_of(from));
 126
 127               else
 128                   san += square_to_string(from);
 129           }
 130       }
 131
 132       if (pos.is_capture(m))
 133       {
 134           if (pt == PAWN)
 135               san += file_to_char(file_of(from));
 136
 137           san += 'x';
 138       }
 139
 140       san += square_to_string(to);
 141
 142       if (type_of(m) == PROMOTION)
 143           san += string("=") + piece_type_to_char(promotion_type(m));
 144   }
 145
 146   if (pos.move_gives_check(m, CheckInfo(pos)))
 147   {
 148       StateInfo st;
 149       pos.do_move(m, st);
 150       san += MoveList<LEGAL>(pos).size() ? "+" : "#";
 151       pos.undo_move(m);
 152   }
 153
 154   return san;
 155 }