Use type_of() to categorize the moves
[stockfish] / 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 }