diff options
Diffstat (limited to 'cli/src')
| -rw-r--r-- | cli/src/main.rs | 37 | ||||
| -rw-r--r-- | cli/src/perft.rs | 26 |
2 files changed, 63 insertions, 0 deletions
diff --git a/cli/src/main.rs b/cli/src/main.rs new file mode 100644 index 0000000..f7fcddf --- /dev/null +++ b/cli/src/main.rs @@ -0,0 +1,37 @@ +use ai::CheckersBitBoard; +use clap::{App, Arg, SubCommand}; + +mod perft; + +fn main() { + let matches = App::new("Ampere") + .version("0.1") + .author("Botahamec <botahamec@outlook.com>") + .about("An American Checkers AI") + .subcommand( + SubCommand::with_name("perft") + .about("Calculate the number of possible moves") + .arg( + Arg::with_name("depth") + .required(true) + .short("d") + .takes_value(true) + .help("The depth to go to"), + ), + ) + .get_matches(); + + if let Some(matches) = matches.subcommand_matches("perft") { + println!( + "{}", + perft::positions( + CheckersBitBoard::starting_position(), + matches + .value_of("depth") + .unwrap() + .parse::<usize>() + .expect("Error: not a valid number") + ) + ); + } +} diff --git a/cli/src/perft.rs b/cli/src/perft.rs new file mode 100644 index 0000000..eba640e --- /dev/null +++ b/cli/src/perft.rs @@ -0,0 +1,26 @@ +use ai::{CheckersBitBoard, Move, PossibleMoves};
+use rayon::prelude::*;
+use std::fmt::{Display, Formatter};
+
+#[derive(Clone)]
+struct PerftResult {
+ result: Vec<(Move, usize)>,
+}
+
+pub fn positions(board: CheckersBitBoard, depth: usize) -> usize {
+ let moves = PossibleMoves::moves(board);
+
+ if depth == 0 {
+ 1
+ } else {
+ let mut total = 0;
+
+ for current_move in moves {
+ // safety: we got this move out of the list of possible moves, so it's definitely valid
+ let board = unsafe { current_move.apply_to(board) };
+ total += positions(board, depth - 1);
+ }
+
+ total
+ }
+}
|
