diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/lib.rs | 2 | ||||
| -rw-r--r-- | src/strategy/sample.rs | 9 |
2 files changed, 8 insertions, 3 deletions
@@ -4,6 +4,8 @@ extern crate serde_json; #[macro_use] extern crate serde_derive; +extern crate rand; + pub mod json; pub mod engine; pub mod strategy; diff --git a/src/strategy/sample.rs b/src/strategy/sample.rs index 3a311e0..2dad924 100644 --- a/src/strategy/sample.rs +++ b/src/strategy/sample.rs @@ -1,13 +1,16 @@ use engine; use engine::command::*; +use rand::{thread_rng, Rng}; pub fn choose_move(settings: &engine::settings::GameSettings, state: &engine::GameState) -> Command { + let mut rng = thread_rng(); + if state.player.can_afford_defence_buildings(settings) { for y in 0..settings.size.y { if is_under_attack(state, y) { let p_options = state.unoccupied_player_cells_in_row(settings, y); - if let Some(&p) = p_options.first() { + if let Some(&p) = rng.choose(&p_options) { return Command::Build(p, BuildingType::Defence); } } @@ -16,9 +19,9 @@ pub fn choose_move(settings: &engine::settings::GameSettings, state: &engine::Ga if state.player.can_afford_all_buildings(settings) { let options = state.unoccupied_player_cells(settings); - let option = options.first(); + let option = rng.choose(&options); let buildings = [BuildingType::Attack, BuildingType::Defence, BuildingType::Energy]; - let building = buildings.first(); + let building = rng.choose(&buildings); match (option, building) { (Some(&p), Some(&building)) => Command::Build(p, building), _ => Command::Nothing |