X-Git-Url: http://git.scottworley.com/pluta-lesnura/blobdiff_plain/aa8493741589b32f3c4211919c0bda33eb02900d..2d3998b9d91b9016dd41fa871a2e2d9e2c15180a:/src/main.rs

diff --git a/src/main.rs b/src/main.rs
index 16bb4f7..b2924ee 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,5 +1,5 @@
-use clap::{Parser, Subcommand};
-use pluta_lesnura::{play, random_player, Game, Player};
+use clap::{Parser, Subcommand, ValueEnum};
+use pluta_lesnura::{momentum_player, play, random_player, Game, Player};
 
 #[derive(Parser)]
 #[command(author, version, about, long_about = None, arg_required_else_help = true)]
@@ -8,16 +8,28 @@ struct Cli {
     command: Option<Commands>,
 }
 
+#[derive(Clone, Debug, ValueEnum)]
+enum Strategy {
+    Random,
+    Momentum,
+}
+
 #[derive(Subcommand)]
 enum Commands {
     /// Runs simulations
     Sim {
+        /// For momentum play, draw how often?  0-1
+        #[arg(short = 'p', long, default_value_t = 0.5)]
+        draw_chance: f64,
         /// How many games?
         #[arg(short = 'g', long, default_value_t = 1)]
         num_games: usize,
         /// How many players?
         #[arg(short = 'p', long)]
         num_players: usize,
+        /// What strategy should players use?
+        #[arg(short = 's', long, value_enum)]
+        strategy: Strategy,
     },
 }
 
@@ -26,11 +38,17 @@ fn main() -> Result<(), &'static str> {
 
     match &cli.command {
         Some(Commands::Sim {
+            draw_chance,
             num_games,
             num_players,
+            strategy,
         }) => {
+            let player = || match strategy {
+                Strategy::Random => Player::new(random_player(*draw_chance)),
+                Strategy::Momentum => Player::new(momentum_player(*draw_chance)),
+            };
             for _ in 0..*num_games {
-                let players: Vec<_> = std::iter::from_fn(|| Some(Player::new(random_player)))
+                let players: Vec<_> = std::iter::from_fn(|| Some(player()))
                     .take(*num_players)
                     .collect();
                 let mut game = Game::default();