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