X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fsquare.h;h=4d427e8e1072d5400bf5c89e3aaea15cde0802ef;hp=c158c89ed74768e328d011aeecd0735d885ad03c;hb=a5ae7fe26030bdd73a95fde3ebe841abfe84ec5e;hpb=f56af8e84db25c0d26fe762fbe171ec5518177bb diff --git a/src/square.h b/src/square.h index c158c89e..4d427e8e 100644 --- a/src/square.h +++ b/src/square.h @@ -1,17 +1,18 @@ /* - Glaurung, a UCI chess playing engine. - Copyright (C) 2004-2008 Tord Romstad + Stockfish, a UCI chess playing engine derived from Glaurung 2.1 + Copyright (C) 2004-2008 Tord Romstad (Glaurung author) + Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad - Glaurung is free software: you can redistribute it and/or modify + Stockfish is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. - - Glaurung is distributed in the hope that it will be useful, + + Stockfish is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. - + You should have received a copy of the GNU General Public License along with this program. If not, see . */ @@ -24,6 +25,7 @@ //// Includes //// +#include // for abs() #include #include "color.h" @@ -44,7 +46,7 @@ enum Square { SQ_A7, SQ_B7, SQ_C7, SQ_D7, SQ_E7, SQ_F7, SQ_G7, SQ_H7, SQ_A8, SQ_B8, SQ_C8, SQ_D8, SQ_E8, SQ_F8, SQ_G8, SQ_H8, SQ_NONE -}; +}; enum File { FILE_A, FILE_B, FILE_C, FILE_D, FILE_E, FILE_F, FILE_G, FILE_H, FILE_NONE @@ -62,6 +64,11 @@ enum SquareDelta { DELTA_NN = 020, DELTA_NNE = 021 }; +ENABLE_OPERATORS_ON(Square); +ENABLE_OPERATORS_ON(File); +ENABLE_OPERATORS_ON(Rank); +ENABLE_OPERATORS_ON(SquareDelta); + //// //// Constants @@ -71,39 +78,14 @@ const int FlipMask = 070; const int FlopMask = 07; -//// +//// //// Inline functions //// -inline File operator+ (File x, int i) { return File(int(x) + i); } -inline File operator+ (File x, File y) { return x + int(y); } -inline void operator++ (File &x, int) { x = File(int(x) + 1); } -inline void operator+= (File &x, int i) { x = File(int(x) + i); } -inline File operator- (File x, int i) { return File(int(x) - i); } -inline void operator-- (File &x, int) { x = File(int(x) - 1); } -inline void operator-= (File &x, int i) { x = File(int(x) - i); } - -inline Rank operator+ (Rank x, int i) { return Rank(int(x) + i); } -inline Rank operator+ (Rank x, Rank y) { return x + int(y); } -inline void operator++ (Rank &x, int) { x = Rank(int(x) + 1); } -inline void operator+= (Rank &x, int i) { x = Rank(int(x) + i); } -inline Rank operator- (Rank x, int i) { return Rank(int(x) - i); } -inline void operator-- (Rank &x, int) { x = Rank(int(x) - 1); } -inline void operator-= (Rank &x, int i) { x = Rank(int(x) - i); } - -inline Square operator+ (Square x, int i) { return Square(int(x) + i); } -inline void operator++ (Square &x, int) { x = Square(int(x) + 1); } -inline void operator+= (Square &x, int i) { x = Square(int(x) + i); } -inline Square operator- (Square x, int i) { return Square(int(x) - i); } -inline void operator-- (Square &x, int) { x = Square(int(x) - 1); } -inline void operator-= (Square &x, int i) { x = Square(int(x) - i); } -inline Square operator+ (Square x, SquareDelta i) { return Square(int(x) + i); } -inline void operator+= (Square &x, SquareDelta i) { x = Square(int(x) + i); } -inline Square operator- (Square x, SquareDelta i) { return Square(int(x) - i); } -inline void operator-= (Square &x, SquareDelta i) { x = Square(int(x) - i); } -inline SquareDelta operator- (Square x, Square y) { - return SquareDelta(int(x) - int(y)); -} +inline Square operator+ (Square x, SquareDelta i) { return x + Square(i); } +inline void operator+= (Square& x, SquareDelta i) { x = x + Square(i); } +inline Square operator- (Square x, SquareDelta i) { return x - Square(i); } +inline void operator-= (Square& x, SquareDelta i) { x = x - Square(i); } inline Square make_square(File f, Rank r) { return Square(int(f) | (int(r) << 3)); @@ -133,8 +115,13 @@ inline Rank relative_rank(Color c, Square s) { return square_rank(relative_square(c, s)); } -inline Color square_color(Square s) { - return Color((int(square_file(s)) + int(square_rank(s))) & 1); +inline SquareColor square_color(Square s) { + return SquareColor((int(square_file(s)) + int(square_rank(s))) & 1); +} + +inline bool same_color_squares(Square s1, Square s2) { + int s = int(s1) ^ int(s2); + return (((s >> 3) ^ s) & 1) == 0; } inline int file_distance(File f1, File f2) { @@ -157,21 +144,41 @@ inline int square_distance(Square s1, Square s2) { return Max(file_distance(s1, s2), rank_distance(s1, s2)); } +inline File file_from_char(char c) { + return File(c - 'a') + FILE_A; +} -//// -//// Prototypes -//// +inline char file_to_char(File f) { + return char(f - FILE_A + int('a')); +} -extern File file_from_char(char c); -extern char file_to_char(File f); -extern Rank rank_from_char(char c); -extern char rank_to_char(Rank r); -extern Square square_from_string(const std::string &str); -extern const std::string square_to_string(Square s); +inline Rank rank_from_char(char c) { + return Rank(c - '1') + RANK_1; +} + +inline char rank_to_char(Rank r) { + return char(r - RANK_1 + int('1')); +} -extern bool file_is_ok(File f); -extern bool rank_is_ok(Rank r); -extern bool square_is_ok(Square s); +inline Square square_from_string(const std::string& str) { + return make_square(file_from_char(str[0]), rank_from_char(str[1])); +} + +inline const std::string square_to_string(Square s) { + return std::string(1, file_to_char(square_file(s))) + + std::string(1, rank_to_char(square_rank(s))); +} +inline bool file_is_ok(File f) { + return f >= FILE_A && f <= FILE_H; +} + +inline bool rank_is_ok(Rank r) { + return r >= RANK_1 && r <= RANK_8; +} + +inline bool square_is_ok(Square s) { + return file_is_ok(square_file(s)) && rank_is_ok(square_rank(s)); +} #endif // !defined(SQUARE_H_INCLUDED)