Added running total of unoccupied cells

1 files changed, 2 insertions, 3 deletions

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) {