Quickly check for check in pseudo legality test

bdmendes · May 12, 2024 · 44bd300 · 44bd300
1 parent 7845abf
commit 44bd300
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/src/moves/mod.rs b/src/moves/mod.rs
@@ -1,6 +1,6 @@
 use self::{
     attacks::specials::pawn_attacks,
-    gen::{piece_attacks, MoveDirection},
+    gen::{piece_attacks, square_attackers, MoveDirection},
 };
 use crate::position::{
     bitboard::Bitboard, board::Piece, square::Square, CastlingRights, Color, Position,
@@ -130,6 +130,18 @@ impl Move {
                 return false;
             }
 
+            // Basic check test for king moves.
+            if piece == Piece::King
+                && square_attackers::<true>(
+                    &position.board,
+                    self.to(),
+                    position.side_to_move.opposite(),
+                )
+                .is_not_empty()
+            {
+                return false;
+            }
+
             match self.flag() {
                 MoveFlag::Quiet if piece == Piece::Pawn => {
                     to_color.is_none() && self.to().file() == self.from().file()

diff --git a/src/search/constraint.rs b/src/search/constraint.rs
@@ -124,11 +124,11 @@ mod tests {
             game_history: vec![],
         };
 
-        assert!(!constraint.should_stop_search());
+        assert!(!constraint.should_stop_search::<false>());
 
         stop_now.store(true, std::sync::atomic::Ordering::Release);
 
-        assert!(constraint.should_stop_search());
+        assert!(constraint.should_stop_search::<false>());
         assert!(constraint.remaining_time().unwrap() > Duration::from_millis(90));
     }
 }