]> git.sesse.net Git - stockfish/blob - src/piece.h
Templetize functions to get pieces by type
[stockfish] / src / piece.h
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 #if !defined(PIECE_H_INCLUDED)
22 #define PIECE_H_INCLUDED
23
24 ////
25 //// Includes
26 ////
27
28 #include "color.h"
29 #include "square.h"
30
31
32 ////
33 //// Types
34 ////
35
36 enum PieceType {
37   NO_PIECE_TYPE = 0,
38   PAWN = 1, KNIGHT = 2, BISHOP = 3, ROOK = 4, QUEEN = 5, KING = 6,
39   BISHOP_AND_QUEEN = 8, ROOK_AND_QUEEN = 9
40 };
41
42 enum Piece {
43   NO_PIECE = 0, WP = 1, WN = 2, WB = 3, WR = 4, WQ = 5, WK = 6,
44   BP = 9, BN = 10, BB = 11, BR = 12, BQ = 13, BK = 14,
45   EMPTY = 16, OUTSIDE = 17
46 };
47
48
49 ////
50 //// Constants
51 ////
52
53 const int SlidingArray[18] = {
54   0, 0, 0, 1, 2, 3, 0, 0, 0, 0, 0, 1, 2, 3, 0, 0, 0, 0
55 };
56
57
58 ////
59 //// Inline functions
60 ////
61
62 inline Piece operator+ (Piece p, int i) { return Piece(int(p) + i); }
63 inline void operator++ (Piece &p, int) { p = Piece(int(p) + 1); }
64 inline Piece operator- (Piece p, int i) { return Piece(int(p) - i); }
65 inline void operator-- (Piece &p, int) { p = Piece(int(p) - 1); }
66 inline PieceType operator+ (PieceType p, int i) {return PieceType(int(p) + i);}
67 inline void operator++ (PieceType &p, int) { p = PieceType(int(p) + 1); }
68 inline PieceType operator- (PieceType p, int i) {return PieceType(int(p) - i);}
69 inline void operator-- (PieceType &p, int) { p = PieceType(int(p) - 1); }
70
71 inline PieceType type_of_piece(Piece p)  {
72   return PieceType(int(p) & 7);
73 }
74
75 inline Color color_of_piece(Piece p) {
76   return Color(int(p) >> 3);
77 }
78
79 inline Piece piece_of_color_and_type(Color c, PieceType pt) {
80   return Piece((int(c) << 3) | int(pt));
81 }
82
83 inline int piece_is_slider(Piece p) {
84   return SlidingArray[int(p)];
85 }
86
87 inline int piece_is_slider(PieceType pt) {
88   return SlidingArray[int(pt)];
89 }
90
91 inline SquareDelta pawn_push(Color c) {
92     return (c == WHITE ? DELTA_N : DELTA_S);
93 }
94
95 inline bool piece_type_is_ok(PieceType pc) {
96   return pc >= PAWN && pc <= KING;
97 }
98
99 inline bool piece_is_ok(Piece pc) {
100   return piece_type_is_ok(type_of_piece(pc)) && color_is_ok(color_of_piece(pc));
101 }
102
103
104 ////
105 //// Prototypes
106 ////
107
108 extern char piece_type_to_char(PieceType pt, bool upcase = false);
109 extern PieceType piece_type_from_char(char c);
110
111
112 #endif // !defined(PIECE_H_INCLUDED)