Add basic iterative deepening implementation
This commit is contained in:
53
src/search/iterative_deepening.rs
Normal file
53
src/search/iterative_deepening.rs
Normal file
@@ -0,0 +1,53 @@
|
||||
use crate::{
|
||||
board::{game::Game, transposition_table::TranspositionTable},
|
||||
evaluation::{MAX_SCORE, MIN_SCORE},
|
||||
movegen::r#move::Move,
|
||||
};
|
||||
|
||||
use super::negamax;
|
||||
|
||||
pub fn iterative_deepening(
|
||||
game: &mut Game,
|
||||
max_depth: u8,
|
||||
move_time: u128,
|
||||
mut tt: TranspositionTable,
|
||||
) -> Result<Move, String> {
|
||||
let mut nodes = 0;
|
||||
let (mut mv, _): (Option<Move>, i32) = (None, 0);
|
||||
let time_now = std::time::Instant::now();
|
||||
|
||||
for depth in 0..max_depth {
|
||||
if move_time < time_now.elapsed().as_millis() {
|
||||
return Ok(mv.expect("No move selected"));
|
||||
}
|
||||
|
||||
(mv, _) = negamax::negamax(game, MIN_SCORE, MAX_SCORE, depth, 0, &mut tt, &mut nodes);
|
||||
dbg!(depth, nodes);
|
||||
}
|
||||
|
||||
Ok(mv.expect("No move selected"))
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use crate::{
|
||||
board::{fen::from_fen, transposition_table::TranspositionTable},
|
||||
movegen::attack_generator::init_attacks,
|
||||
search,
|
||||
};
|
||||
|
||||
const FEN: &str = "1r2k2r/2P1pq1p/2npb3/1p3ppP/p3P3/P2B1Q2/1P1PNPP1/R3K2R w KQk g6 0 1";
|
||||
|
||||
#[test]
|
||||
fn test_iterative_deepening() -> Result<(), String> {
|
||||
init_attacks();
|
||||
let mut game = from_fen(FEN)?;
|
||||
let tt = TranspositionTable::new(1000000);
|
||||
let time_now = std::time::Instant::now();
|
||||
|
||||
let _ = search::iterative_deepening::iterative_deepening(&mut game, 6, 100000, tt);
|
||||
dbg!(time_now.elapsed());
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
@@ -1,3 +1,4 @@
|
||||
pub mod iterative_deepening;
|
||||
pub mod move_ordering;
|
||||
pub mod negamax;
|
||||
pub mod perft;
|
||||
|
||||
Reference in New Issue
Block a user