improving check generation

1 files changed, 13 insertions, 0 deletions

diff --git a/src/board.cpp b/src/board.cpp
index ce116a1..4a9531a 100644
--- a/src/board.cpp
+++ b/src/board.cpp
@@ -1,4 +1,5 @@
 #include "board.hpp"
+#include "moves.hpp"
 #include "zobrist.hpp"
 #include <cassert>
 
@@ -11,6 +12,18 @@ Piece createPiece(PieceType type, bool color) {
 
 int PositionToIndex(Position i) { return i.rank * 8 + i.file; }
 
+Position FindKing(Game *b, bool white) {
+  for (int i = 0; i < 64; i++) {
+    Piece piece = b->pieces[i];
+
+    if (piece.type == KING && piece.color == white) {
+      return IndexToPosition(i);
+    }
+  }
+
+  assert(false && "King not found");
+  return {0, 0};
+}
 UndoMove MakeMove(Move move, Game *g) {
   UndoMove undo = {};