134 lines
2.6 KiB
Rust
134 lines
2.6 KiB
Rust
use std::cmp::Ordering;
|
|
|
|
use anyhow::{bail, Result};
|
|
use termion::color::{self};
|
|
|
|
mod agents;
|
|
mod board;
|
|
mod util;
|
|
use board::Board;
|
|
use util::{Player, Symb};
|
|
|
|
fn play(
|
|
maxi: &mut dyn agents::MaximizerAgent,
|
|
maxi_player: Player,
|
|
mini: &mut dyn agents::MinimizerAgent,
|
|
mini_player: Player,
|
|
) -> Result<Board> {
|
|
let mut board = Board::new();
|
|
let mut is_first_turn = true;
|
|
let mut is_maxi_turn = true;
|
|
|
|
while !board.is_done() {
|
|
// Print board
|
|
println!(
|
|
"\r{}{}{}{}",
|
|
" ".repeat(6),
|
|
if is_first_turn { '╓' } else { '║' },
|
|
board.prettyprint()?,
|
|
if is_first_turn { '╖' } else { '║' },
|
|
);
|
|
is_first_turn = false;
|
|
|
|
// Take action
|
|
let action = if is_maxi_turn {
|
|
maxi.step_max(&board)?
|
|
} else {
|
|
mini.step_min(&board)?
|
|
};
|
|
|
|
if !board.play(
|
|
action,
|
|
if is_maxi_turn {
|
|
maxi_player
|
|
} else {
|
|
mini_player
|
|
},
|
|
) {
|
|
bail!("agent made invalid move")
|
|
}
|
|
|
|
is_maxi_turn = !is_maxi_turn;
|
|
}
|
|
|
|
println!("\r{}║{}║", " ".repeat(6), board.prettyprint()?);
|
|
println!("\r{}╙{}╜", " ".repeat(6), " ".repeat(board.size()));
|
|
Ok(board)
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
rayon::ThreadPoolBuilder::new()
|
|
.num_threads(4)
|
|
.build_global()
|
|
.unwrap();
|
|
|
|
let mut maxi = agents::PlayerAgent::new(Player::Human);
|
|
let mut mini = agents::Diffuse {};
|
|
|
|
let a = play(&mut maxi, Player::Human, &mut mini, Player::Computer)?;
|
|
if a.is_done() {
|
|
println!(
|
|
"\r\n{}Your score:{} {:.2}\n\n",
|
|
color::Fg(Player::Human.color()),
|
|
color::Fg(color::Reset),
|
|
a.evaluate().unwrap()
|
|
);
|
|
} else {
|
|
println!(
|
|
"\r\n{}Quitting{}\r\n",
|
|
color::Fg(color::Red),
|
|
color::Fg(color::Reset),
|
|
);
|
|
return Ok(());
|
|
}
|
|
|
|
let mut mini = agents::PlayerAgent::new(Player::Human);
|
|
let mut maxi = agents::Diffuse {};
|
|
|
|
let b = play(&mut maxi, Player::Computer, &mut mini, Player::Human)?;
|
|
if b.is_done() {
|
|
println!(
|
|
"\r\n{}Computer score:{} {:.2}\n\n",
|
|
color::Fg(Player::Human.color()),
|
|
color::Fg(color::Reset),
|
|
b.evaluate().unwrap()
|
|
);
|
|
} else {
|
|
println!(
|
|
"\r\n{}Quitting{}\r\n",
|
|
color::Fg(color::Red),
|
|
color::Fg(color::Reset),
|
|
);
|
|
return Ok(());
|
|
}
|
|
|
|
match a.evaluate().partial_cmp(&b.evaluate()) {
|
|
Some(Ordering::Equal) => {
|
|
println!("\r\nTie");
|
|
}
|
|
Some(Ordering::Greater) => {
|
|
println!(
|
|
"\r\n{}Human wins{}",
|
|
color::Fg(Player::Human.color()),
|
|
color::Fg(color::Reset),
|
|
);
|
|
}
|
|
Some(Ordering::Less) => {
|
|
println!(
|
|
"\r\n{}Computer wins{}",
|
|
color::Fg(Player::Computer.color()),
|
|
color::Fg(color::Reset),
|
|
);
|
|
}
|
|
None => {
|
|
println!(
|
|
"\r\n{}Error{}",
|
|
color::Fg(color::Red),
|
|
color::Fg(color::Reset),
|
|
);
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|