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
  21 ////
  22 //// Includes
  23 ////
  24
  25 #include <cassert>
  26 #include <cctype>
  27
  28 #include "move.h"
  29 #include "piece.h"
  30 #include "position.h"
  31
  32
  33 ////
  34 //// Functions
  35 ////
  36
  37 /// move_from_uci() takes a position and a string as input, and attempts to
  38 /// convert the string to a move, using simple coordinate notation (g1f3,
  39 /// a7a8q, etc.). In order to correctly parse en passant captures and castling
  40 /// moves, we need the position. This function is not robust, and expects that
  41 /// the input move is legal and correctly formatted.
  42
  43 Move move_from_uci(const Position& pos, const std::string& str) {
  44
  45   Square from, to;
  46   Piece piece;
  47   Color us = pos.side_to_move();
  48
  49   if (str.length() < 4)
  50       return MOVE_NONE;
  51
  52   // Read the from and to squares
  53   from = make_square(file_from_char(str[0]), rank_from_char(str[1]));
  54   to   = make_square(file_from_char(str[2]), rank_from_char(str[3]));
  55
  56   // Find the moving piece
  57   piece = pos.piece_on(from);
  58
  59   // If the string has more than 4 characters, try to interpret the 5th
  60   // character as a promotion.
  61   if (str.length() > 4 && piece == piece_of_color_and_type(us, PAWN))
  62   {
  63       switch (tolower(str[4])) {
  64       case 'n':
  65           return make_promotion_move(from, to, KNIGHT);
  66       case 'b':
  67           return make_promotion_move(from, to, BISHOP);
  68       case 'r':
  69           return make_promotion_move(from, to, ROOK);
  70       case 'q':
  71           return make_promotion_move(from, to, QUEEN);
  72       }
  73   }
  74
  75   // En passant move? We assume that a pawn move is an en passant move
  76   // if the destination square is epSquare.
  77   if (to == pos.ep_square() && piece == piece_of_color_and_type(us, PAWN))
  78       return make_ep_move(from, to);
  79
  80   // Is this a castling move? A king move is assumed to be a castling move
  81   // if the destination square is occupied by a friendly rook, or if the
  82   // distance between the source and destination squares is more than 1.
  83   if (piece == piece_of_color_and_type(us, KING))
  84   {
  85       if (pos.piece_on(to) == piece_of_color_and_type(us, ROOK))
  86           return make_castle_move(from, to);
  87
  88       if (square_distance(from, to) > 1)
  89       {
  90           // This is a castling move, but we have to translate it to the
  91           // internal "king captures rook" representation.
  92           SquareDelta delta = (to > from ? DELTA_E : DELTA_W);
  93           Square s = from;
  94
  95           do s += delta;
  96           while (   pos.piece_on(s) != piece_of_color_and_type(us, ROOK)
  97                  && relative_rank(us, s) == RANK_1);
  98
  99           return relative_rank(us, s) == RANK_1 ? make_castle_move(from, s) : MOVE_NONE;
 100       }
 101   }
 102
 103   return make_move(from, to);
 104 }
 105
 106
 107 /// move_to_uci() converts a move to a string in coordinate notation
 108 /// (g1f3, a7a8q, etc.). The only special case is castling moves, where we
 109 /// print in the e1g1 notation in normal chess mode, and in e1h1 notation in
 110 /// Chess960 mode.
 111
 112 const std::string move_to_uci(Move move, bool chess960) {
 113
 114   std::string str;
 115   Square from = move_from(move);
 116   Square to = move_to(move);
 117
 118   if (move == MOVE_NONE)
 119       str = "(none)";
 120   else if (move == MOVE_NULL)
 121       str = "0000";
 122   else
 123   {
 124       if (move_is_short_castle(move) && !chess960)
 125           return (from == SQ_E1 ? "e1g1" : "e8g8");
 126
 127       if (move_is_long_castle(move) && !chess960)
 128           return (from == SQ_E1 ? "e1c1" : "e8c8");
 129
 130       str = square_to_string(from) + square_to_string(to);
 131       if (move_is_promotion(move))
 132           str += char(tolower(piece_type_to_char(move_promotion_piece(move))));
 133   }
 134   return str;
 135 }
 136
 137
 138 /// Overload the << operator, to make it easier to print moves
 139
 140 std::ostream& operator << (std::ostream& os, Move m) {
 141
 142   bool chess960 = (os.iword(0) != 0); // See set960()
 143   return os << move_to_uci(m, chess960);
 144 }
 145
 146
 147 /// move_is_ok(), for debugging
 148
 149 bool move_is_ok(Move m) {
 150
 151   return move_from(m) != move_to(m); // Catches also MOVE_NONE
 152 }