git.sesse.net Git - stockfish/blob - src/direction.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-2010 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
  21 #if !defined(DIRECTION_H_INCLUDED)
  22 #define DIRECTION_H_INCLUDED
  23
  24 ////
  25 //// Includes
  26 ////
  27
  28 #include "square.h"
  29 #include "types.h"
  30
  31
  32 ////
  33 //// Types
  34 ////
  35
  36 enum Direction {
  37   DIR_E = 0, DIR_N = 1, DIR_NE = 2, DIR_NW = 3, DIR_NONE = 4
  38 };
  39
  40 enum SignedDirection {
  41   SIGNED_DIR_E  = 0, SIGNED_DIR_W  = 1,
  42   SIGNED_DIR_N  = 2, SIGNED_DIR_S  = 3,
  43   SIGNED_DIR_NE = 4, SIGNED_DIR_SW = 5,
  44   SIGNED_DIR_NW = 6, SIGNED_DIR_SE = 7,
  45   SIGNED_DIR_NONE = 8
  46 };
  47
  48
  49 ////
  50 //// Variables
  51 ////
  52
  53 extern uint8_t DirectionTable[64][64];
  54 extern uint8_t SignedDirectionTable[64][64];
  55
  56
  57 ////
  58 //// Inline functions
  59 ////
  60
  61 inline void operator++ (Direction& d, int) {
  62   d = Direction(int(d) + 1);
  63 }
  64
  65 inline void operator++ (SignedDirection& d, int) {
  66   d = SignedDirection(int(d) + 1);
  67 }
  68
  69 inline Direction direction_between_squares(Square s1, Square s2) {
  70   return Direction(DirectionTable[s1][s2]);
  71 }
  72
  73 inline SignedDirection signed_direction_between_squares(Square s1, Square s2) {
  74   return SignedDirection(SignedDirectionTable[s1][s2]);
  75 }
  76
  77 inline int direction_is_diagonal(Square s1, Square s2) {
  78   return DirectionTable[s1][s2] & 2;
  79 }
  80
  81 inline bool direction_is_straight(Square s1, Square s2) {
  82   return DirectionTable[s1][s2] < 2;
  83 }
  84
  85 ////
  86 //// Prototypes
  87 ////
  88
  89 extern void init_direction_table();
  90
  91
  92 #endif // !defined(DIRECTION_H_INCLUDED)