extend checks during quiescence

brunocodutra · Jul 17, 2023 · 6a0c391 · 6a0c391
1 parent 34aba42
commit 6a0c391
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/lib/search/pvs.rs b/lib/search/pvs.rs
@@ -175,13 +175,9 @@ impl Searcher {
             },
         };
 
-        let kind = if ply < depth {
-            MoveKind::ANY
-        } else if ply < Searcher::MAX_PLY {
-            MoveKind::CAPTURE | MoveKind::PROMOTION
-        } else {
+        if ply >= Searcher::MAX_PLY {
             return Ok(score);
-        };
+        }
 
         if !in_check {
             if let Some(d) = self.nmp(pos, score, beta, depth) {
@@ -195,6 +191,12 @@ impl Searcher {
             }
         }
 
+        let kind = if ply < depth || in_check {
+            MoveKind::ANY
+        } else {
+            MoveKind::CAPTURE | MoveKind::PROMOTION
+        };
+
         let mut moves = ArrayVec::<_, 256>::from_iter(pos.moves(kind).map(|m| {
             let mut next = pos.clone();
             next.play(*m).expect("expected legal move");