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 ENABLE_OPERATORS_ON(SignedDirection);
  49
  50
  51 ////
  52 //// Variables
  53 ////
  54
  55 extern uint8_t DirectionTable[64][64];
  56 extern uint8_t SignedDirectionTable[64][64];
  57
  58
  59 ////
  60 //// Inline functions
  61 ////
  62
  63 inline Direction direction_between_squares(Square s1, Square s2) {
  64   return Direction(DirectionTable[s1][s2]);
  65 }
  66
  67 inline SignedDirection signed_direction_between_squares(Square s1, Square s2) {
  68   return SignedDirection(SignedDirectionTable[s1][s2]);
  69 }
  70
  71 inline int direction_is_diagonal(Square s1, Square s2) {
  72   return DirectionTable[s1][s2] & 2;
  73 }
  74
  75 inline bool direction_is_straight(Square s1, Square s2) {
  76   return DirectionTable[s1][s2] < 2;
  77 }
  78
  79 ////
  80 //// Prototypes
  81 ////
  82
  83 extern void init_direction_table();
  84
  85
  86 #endif // !defined(DIRECTION_H_INCLUDED)