Add tablebase probing in search

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## [Unreleased]
 
+* Add tablebase probing in search based on 'fathom' (18.97 +- 8.35 (5-man))
 * Use arrayvec for PrincipalVariation (15.52 +- 8.72)
 * Fix an accidental fail hard in quiescence - fail soft when eval >= beta (12.54 +- 7.15)
 * Add futility pruning (10.82 +- 6.41)

src/engine/search/iterative_deepening.rs

@@ -68,6 +68,7 @@ pub fn search(
                         state.nodes_visited,
                         time_control.elapsed(),
                     ),
+                    tbhits: state.tbhits,
                 },
             },
         );

src/engine/search/mod.rs

@@ -80,6 +80,7 @@ pub struct SearchState {
 
     nodes_visited: u64,
     max_depth_reached: u8,
+    tbhits: u64,
 }
 
 impl SearchState {
@@ -90,6 +91,7 @@ impl SearchState {
 
             max_depth_reached: 0,
             nodes_visited: 0,
+            tbhits: 0,
         }
     }
 }
@@ -134,6 +136,7 @@ pub struct SearchStats {
     pub time: Duration,
     pub nodes: u64,
     pub nodes_per_second: u64,
+    pub tbhits: u64,
 }
 
 pub trait Reporter {

src/engine/search/negamax.rs

@@ -9,6 +9,7 @@ use crate::engine::search::quiescence::quiescence;
 use crate::engine::search::tables::lmr_table::lmr_reduction;
 use crate::engine::search::time_control::TimeStrategy;
 use crate::engine::search::transposition::{NodeBound, SearchTranspositionTableData};
+use crate::engine::tablebases::Wdl;
 
 pub fn negamax(
     game: &mut Game,
@@ -80,6 +81,50 @@ pub fn negamax(
         previous_best_move = tt_entry.best_move;
     }
 
+    let tb_cardinality = persistent_state.tablebase.n_men() as u8;
+    if !is_root && tb_cardinality > 0 {
+        let piece_count = game.board.occupancy().count();
+
+        if piece_count < tb_cardinality || (piece_count <= tb_cardinality && depth >= 1) {
+            if let Some(wdl) = persistent_state.tablebase.wdl(game) {
+                state.tbhits += 1;
+
+                let score = match wdl {
+                    Wdl::Win => Eval::mate_in(plies),
+                    Wdl::Draw => Eval::DRAW,
+                    Wdl::Loss => Eval::mated_in(plies),
+                };
+
+                let tb_bound = match wdl {
+                    Wdl::Win => NodeBound::Lower,
+                    Wdl::Loss => NodeBound::Upper,
+                    Wdl::Draw => NodeBound::Exact,
+                };
+
+                if tb_bound == NodeBound::Exact
+                    || (tb_bound == NodeBound::Lower && score >= beta)
+                    || (tb_bound == NodeBound::Upper && score <= alpha)
+                {
+                    let tt_data = SearchTranspositionTableData {
+                        bound: tb_bound,
+                        eval: score,
+                        best_move: None,
+                        age: persistent_state.tt.generation,
+                        depth,
+                    };
+
+                    persistent_state.tt.insert(&game.zobrist, tt_data);
+
+                    return Ok(score);
+                }
+
+                if is_pv && tb_bound == NodeBound::Lower {
+                    alpha = alpha.max(score);
+                }
+            }
+        }
+    }
+
     let eval = eval::eval(game);
 
     if !is_root && !is_pv && !in_check {

src/engine/uci/mod.rs

@@ -64,6 +64,7 @@ impl UciReporter {
             time: Some(progress.stats.time),
             nodes: Some(progress.stats.nodes),
             nps: Some(progress.stats.nodes_per_second),
+            tbhits: Some(progress.stats.tbhits),
             hashfull: Some(progress.hashfull),
             ..Default::default()
         }));