From eacb65f120ca0fcbd920f14160404cb6c709b4ef Mon Sep 17 00:00:00 2001 From: Justin Worthe Date: Mon, 14 May 2018 23:31:07 +0200 Subject: Added running total of unoccupied cells --- src/strategy/sample.rs | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'src/strategy/sample.rs') diff --git a/src/strategy/sample.rs b/src/strategy/sample.rs index 2dad924..370df2f 100644 --- a/src/strategy/sample.rs +++ b/src/strategy/sample.rs @@ -9,7 +9,7 @@ pub fn choose_move(settings: &engine::settings::GameSettings, state: &engine::Ga 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); + let p_options = state.unoccupied_player_cells_in_row(y); if let Some(&p) = rng.choose(&p_options) { return Command::Build(p, BuildingType::Defence); } @@ -18,8 +18,7 @@ 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 = rng.choose(&options); + let option = rng.choose(&state.unoccupied_player_cells); let buildings = [BuildingType::Attack, BuildingType::Defence, BuildingType::Energy]; let building = rng.choose(&buildings); match (option, building) { -- cgit v1.2.3