From 1c8d9ddad9884f0de117b7d91d3e1c4e256d5586 Mon Sep 17 00:00:00 2001
From: Justin Worthe <justin@worthe-it.co.za>
Date: Sat, 30 Jun 2018 13:07:30 +0200
Subject: Added test of newly functioning tesla towers

---
 src/engine/mod.rs | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/engine/mod.rs b/src/engine/mod.rs
index 7a39214..8078ca9 100644
--- a/src/engine/mod.rs
+++ b/src/engine/mod.rs
@@ -127,7 +127,7 @@ impl GameState {
         GameState::update_construction(&mut self.player_unconstructed_buildings, &mut self.player_buildings, &mut self.player);
         GameState::update_construction(&mut self.opponent_unconstructed_buildings, &mut self.opponent_buildings, &mut self.opponent);
 
-        GameState::fire_teslas(&mut self.player, &mut self.player_buildings, &mut self.opponent, &mut self.opponent_buildings, &settings);
+        GameState::fire_teslas(&mut self.player, &mut self.player_buildings, &mut self.unoccupied_player_cells, &mut self.opponent, &mut self.opponent_buildings, &mut self.unoccupied_opponent_cells, &settings);
 
         GameState::add_missiles(&mut self.player_buildings, &mut self.player_missiles);
         GameState::add_missiles(&mut self.opponent_buildings, &mut self.opponent_missiles);
@@ -202,7 +202,7 @@ impl GameState {
         unconstructed_buildings.truncate(buildings_len);
     }
 
-    fn fire_teslas(player: &mut Player, player_buildings: &mut Vec<Building>, opponent: &mut Player, opponent_buildings: &mut Vec<Building>, settings: &GameSettings) {
+    fn fire_teslas(player: &mut Player, player_buildings: &mut Vec<Building>, player_unoccupied_cells: &mut Vec<Point>, opponent: &mut Player, opponent_buildings: &mut Vec<Building>, opponent_unoccupied_cells: &mut Vec<Point>,settings: &GameSettings) {
         for tesla in player_buildings.iter_mut().filter(|b| b.weapon_damage == 20) {
             if tesla.weapon_cooldown_time_left > 0 {
                 tesla.weapon_cooldown_time_left -= 1;
@@ -250,7 +250,17 @@ impl GameState {
                 }
             }
         }
+        
+        for building in player_buildings.iter().filter(|b| b.health == 0) {
+            player_unoccupied_cells.push(building.pos);
+            player.energy_generated -= building.energy_generated_per_turn;
+        }
         player_buildings.retain(|b| b.health > 0);
+
+        for building in opponent_buildings.iter().filter(|b| b.health == 0) {
+            opponent_unoccupied_cells.push(building.pos);
+            opponent.energy_generated -= building.energy_generated_per_turn;
+        }
         opponent_buildings.retain(|b| b.health > 0);
     }
         
-- 
cgit v1.2.3