Merge UciGame and SerachParameters into UciParameters
This commit is contained in:
40
src/uci.rs
40
src/uci.rs
@@ -57,16 +57,18 @@ impl fmt::Display for Response {
|
||||
}
|
||||
}
|
||||
|
||||
struct SearchParameters {
|
||||
struct UciParameters {
|
||||
movetime: Option<usize>,
|
||||
depth: Option<u8>,
|
||||
game: Option<Game>,
|
||||
}
|
||||
|
||||
impl SearchParameters {
|
||||
impl UciParameters {
|
||||
const fn new() -> Self {
|
||||
Self {
|
||||
movetime: None,
|
||||
depth: None,
|
||||
game: None,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -77,9 +79,13 @@ impl SearchParameters {
|
||||
fn add_depth(&mut self, depth: u8) {
|
||||
self.depth = Some(depth);
|
||||
}
|
||||
|
||||
fn add_game(&mut self, game: Game) {
|
||||
self.game = Some(game);
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for SearchParameters {
|
||||
impl Default for UciParameters {
|
||||
fn default() -> Self {
|
||||
Self::new()
|
||||
}
|
||||
@@ -104,7 +110,7 @@ pub fn uci_position(mut position: SplitWhitespace) -> Result<Game, String> {
|
||||
const MAX_DEPTH: u8 = 5;
|
||||
|
||||
pub fn uci_go(mut go: SplitWhitespace, game: &mut Game) -> Result<Move, String> {
|
||||
let mut params = SearchParameters::new();
|
||||
let mut params = UciParameters::new();
|
||||
while let Some(subcommand) = go.next() {
|
||||
match subcommand {
|
||||
"depth" => params.add_depth(go.next().unwrap().parse::<u8>().ok().unwrap()),
|
||||
@@ -117,32 +123,12 @@ pub fn uci_go(mut go: SplitWhitespace, game: &mut Game) -> Result<Move, String>
|
||||
Ok(Move::new(0, 0))
|
||||
}
|
||||
|
||||
struct UciGame {
|
||||
game: Option<Game>,
|
||||
}
|
||||
|
||||
impl UciGame {
|
||||
const fn new() -> Self {
|
||||
Self { game: None }
|
||||
}
|
||||
|
||||
fn set_game_state(&mut self, game: Game) {
|
||||
self.game = Some(game);
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for UciGame {
|
||||
fn default() -> Self {
|
||||
Self::new()
|
||||
}
|
||||
}
|
||||
|
||||
pub fn uci_loop() -> Result<(), String> {
|
||||
let stdin = io::stdin();
|
||||
let stdout = io::stdout();
|
||||
let handle_in = stdin.lock();
|
||||
let mut handle_out = stdout.lock();
|
||||
let mut uci_game = UciGame::new();
|
||||
let mut params = UciParameters::new();
|
||||
|
||||
for line in handle_in.lines() {
|
||||
let line_str = line.unwrap_or_else(|_| "quit".to_string());
|
||||
@@ -152,11 +138,11 @@ pub fn uci_loop() -> Result<(), String> {
|
||||
Command::Uci => Response::UciOk,
|
||||
Command::IsReady => Response::ReadyOk,
|
||||
Command::Position => {
|
||||
uci_game.set_game_state(uci_position(parts)?);
|
||||
params.add_game(uci_position(parts)?);
|
||||
Response::Info("Initialized position.".to_string())
|
||||
}
|
||||
Command::Go => {
|
||||
if let Some(ref mut game) = uci_game.game {
|
||||
if let Some(ref mut game) = params.game {
|
||||
let best_move = uci_go(parts, game)?;
|
||||
Response::BestMove(best_move.parse_into_str())
|
||||
} else {
|
||||
|
||||
Reference in New Issue
Block a user