X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=src%2Fsquare.h;h=0e314b04ac809f017d42d162a8d381caad9b8d99;hb=4ead60e2a77892dbeedb36f4557da5eba2f0558b;hp=5ebfcc39b96365e09aef8f25b2a11e421bdfb652;hpb=72c7595f8ac72c7831ee319b8b0bc46404c5fc27;p=stockfish
diff --git a/src/square.h b/src/square.h
index 5ebfcc39..0e314b04 100644
--- a/src/square.h
+++ b/src/square.h
@@ -1,7 +1,7 @@
/*
Stockfish, a UCI chess playing engine derived from Glaurung 2.1
Copyright (C) 2004-2008 Tord Romstad (Glaurung author)
- Copyright (C) 2008 Marco Costalba
+ Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad
Stockfish is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
@@ -17,25 +17,15 @@
along with this program. If not, see .
*/
-
#if !defined(SQUARE_H_INCLUDED)
#define SQUARE_H_INCLUDED
-////
-//// Includes
-////
-
#include // for abs()
#include
#include "color.h"
#include "misc.h"
-
-////
-//// Types
-////
-
enum Square {
SQ_A1, SQ_B1, SQ_C1, SQ_D1, SQ_E1, SQ_F1, SQ_G1, SQ_H1,
SQ_A2, SQ_B2, SQ_C2, SQ_D2, SQ_E2, SQ_F2, SQ_G2, SQ_H2,
@@ -49,63 +39,37 @@ enum Square {
};
enum File {
- FILE_A, FILE_B, FILE_C, FILE_D, FILE_E, FILE_F, FILE_G, FILE_H, FILE_NONE
+ FILE_A, FILE_B, FILE_C, FILE_D, FILE_E, FILE_F, FILE_G, FILE_H
};
enum Rank {
- RANK_1, RANK_2, RANK_3, RANK_4, RANK_5, RANK_6, RANK_7, RANK_8, RANK_NONE
+ RANK_1, RANK_2, RANK_3, RANK_4, RANK_5, RANK_6, RANK_7, RANK_8
};
enum SquareDelta {
- DELTA_SSW = -021, DELTA_SS = -020, DELTA_SSE = -017, DELTA_SWW = -012,
- DELTA_SW = -011, DELTA_S = -010, DELTA_SE = -07, DELTA_SEE = -06,
- DELTA_W = -01, DELTA_ZERO = 0, DELTA_E = 01, DELTA_NWW = 06, DELTA_NW = 07,
- DELTA_N = 010, DELTA_NE = 011, DELTA_NEE = 012, DELTA_NNW = 017,
- DELTA_NN = 020, DELTA_NNE = 021
+
+ DELTA_N = 8, DELTA_E = 1, DELTA_S = -8, DELTA_W = -1, DELTA_NONE = 0,
+
+ DELTA_NN = DELTA_N + DELTA_N,
+ DELTA_NE = DELTA_N + DELTA_E,
+ DELTA_SE = DELTA_S + DELTA_E,
+ DELTA_SS = DELTA_S + DELTA_S,
+ DELTA_SW = DELTA_S + DELTA_W,
+ DELTA_NW = DELTA_N + DELTA_W
};
+ENABLE_OPERATORS_ON(Square)
+ENABLE_OPERATORS_ON(File)
+ENABLE_OPERATORS_ON(Rank)
+ENABLE_OPERATORS_ON(SquareDelta)
-////
-//// Constants
-////
-
-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));
-}
+const int FlipMask = 56;
+const int FlopMask = 7;
+
+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));
@@ -131,12 +95,21 @@ inline Square relative_square(Color c, Square s) {
return Square(int(s) ^ (int(c) * FlipMask));
}
+inline Rank relative_rank(Color c, Rank r) {
+ return Rank(int(r) ^ (int(c) * 7));
+}
+
inline Rank relative_rank(Color c, Square s) {
- return square_rank(relative_square(c, s));
+ return relative_rank(c, square_rank(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 opposite_color_squares(Square s1, Square s2) {
+ int s = int(s1) ^ int(s2);
+ return ((s >> 3) ^ s) & 1;
}
inline int file_distance(File f1, File f2) {
@@ -164,7 +137,7 @@ inline File file_from_char(char c) {
}
inline char file_to_char(File f) {
- return char(f - FILE_A + 'a');
+ return char(f - FILE_A + int('a'));
}
inline Rank rank_from_char(char c) {
@@ -172,18 +145,12 @@ inline Rank rank_from_char(char c) {
}
inline char rank_to_char(Rank r) {
- return char(r - RANK_1 + '1');
-}
-
-inline Square square_from_string(const std::string& str) {
- return make_square(file_from_char(str[0]), rank_from_char(str[1]));
+ return char(r - RANK_1 + int('1'));
}
inline const std::string square_to_string(Square s) {
- std::string str;
- str += file_to_char(square_file(s));
- str += rank_to_char(square_rank(s));
- return str;
+ 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) {
@@ -195,7 +162,7 @@ inline bool rank_is_ok(Rank r) {
}
inline bool square_is_ok(Square s) {
- return file_is_ok(square_file(s)) && rank_is_ok(square_rank(s));
+ return s >= SQ_A1 && s <= SQ_H8;
}
#endif // !defined(SQUARE_H_INCLUDED)