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 Marco Costalba
   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
  27 #include "move.h"
  28 #include "piece.h"
  29 #include "position.h"
  30 #include "ucioption.h"
  31
  32
  33 ////
  34 //// Functions
  35 ////
  36
  37 /// move_from_string() 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_string(const Position &pos, const std::string &str) {
  44   Square from, to;
  45   Piece piece;
  46   Color us = pos.side_to_move();
  47
  48   if(str.length() < 4) return MOVE_NONE;
  49
  50   // Read the from and to squares:
  51   from = square_from_string(str.substr(0, 2));
  52   to = square_from_string(str.substr(2, 4));
  53
  54   // Find the moving piece:
  55   piece = pos.piece_on(from);
  56
  57   // If the string has more than 4 characters, try to interpret the 5th
  58   // character as a promotion:
  59   if(type_of_piece(piece) == PAWN && str.length() >= 5) {
  60     switch(str[4]) {
  61     case 'n': case 'N':
  62       return make_promotion_move(from, to, KNIGHT);
  63     case 'b': case 'B':
  64       return make_promotion_move(from, to, BISHOP);
  65     case 'r': case 'R':
  66       return make_promotion_move(from, to, ROOK);
  67     case 'q': case 'Q':
  68       return make_promotion_move(from, to, QUEEN);
  69     }
  70   }
  71
  72   if(piece == king_of_color(us)) {
  73     // Is this a castling move?  A king move is assumed to be a castling
  74     // move if the destination square is occupied by a friendly rook, or
  75     // if the distance between the source and destination squares is more
  76     // than 1.
  77     if(pos.piece_on(to) == rook_of_color(us))
  78       return make_castle_move(from, to);
  79     else if(square_distance(from, to) > 1) {
  80       // This is a castling move, but we have to translate it to the
  81       // internal "king captures rook" representation.
  82       SquareDelta delta = (to > from)? DELTA_E : DELTA_W;
  83       Square s;
  84       for(s = from + delta;
  85           relative_rank(us, s) == RANK_1 && pos.piece_on(s) != rook_of_color(us);
  86           s += delta);
  87       if(relative_rank(us, s) == RANK_1 && pos.piece_on(s) == rook_of_color(us))
  88         return make_castle_move(from, s);
  89     }
  90   }
  91   else if(piece == pawn_of_color(us)) {
  92     // En passant move?  We assume that a pawn move is an en passant move
  93     // without further testing if the destination square is epSquare.
  94     if(to == pos.ep_square())
  95       return make_ep_move(from, to);
  96   }
  97
  98   return make_move(from, to);
  99 }
 100
 101
 102 /// move_to_string() converts a move to a string in coordinate notation
 103 /// (g1f3, a7a8q, etc.).  The only special case is castling moves, where we
 104 /// print in the e1g1 notation in normal chess mode, and in e1h1 notation in
 105 /// Chess960 mode.
 106
 107 const std::string move_to_string(Move move) {
 108   std::string str;
 109
 110   if(move == MOVE_NONE)
 111     str = "(none)";
 112   else if(move == MOVE_NULL)
 113     str = "0000";
 114   else {
 115     if(!Chess960) {
 116       if(move_from(move) == SQ_E1 && move_is_short_castle(move)) {
 117         str = "e1g1"; return str;
 118       }
 119       else if(move_from(move) == SQ_E1 && move_is_long_castle(move)) {
 120         str = "e1c1"; return str;
 121       }
 122       if(move_from(move) == SQ_E8 && move_is_short_castle(move)) {
 123         str = "e8g8"; return str;
 124       }
 125       else if(move_from(move) == SQ_E8 && move_is_long_castle(move)) {
 126         str = "e8c8"; return str;
 127       }
 128     }
 129     str = square_to_string(move_from(move)) + square_to_string(move_to(move));
 130     if(move_promotion(move))
 131       str += piece_type_to_char(move_promotion(move), false);
 132   }
 133   return str;
 134 }
 135
 136
 137 /// Overload the << operator, to make it easier to print moves.
 138
 139 std::ostream &operator << (std::ostream &os, Move m) {
 140   return os << move_to_string(m);
 141 }
 142
 143
 144 /// move_is_ok(), for debugging.
 145
 146 bool move_is_ok(Move m) {
 147   return square_is_ok(move_from(m)) && square_is_ok(move_to(m));
 148 }