Corrected which player's attack towers to consider

1 files changed, 10 insertions, 0 deletions

diff --git a/src/engine/bitwise_engine.rs b/src/engine/bitwise_engine.rs
index 75acb7d..c82e268 100644
--- a/src/engine/bitwise_engine.rs
+++ b/src/engine/bitwise_engine.rs
@@ -450,4 +450,14 @@ impl Player {
         let mask = 255u64 << (y * SINGLE_MAP_WIDTH);
         (self.energy_towers & mask).count_ones()
     }
+
+    pub fn count_healthy_defence_in_row(&self, y: u8) -> u32 {
+        let mask = 255u64 << (y * SINGLE_MAP_WIDTH);
+        (self.buildings[1] & mask).count_ones()
+    }
+
+    pub fn count_towers_in_row(&self, y: u8) -> u32 {
+        let mask = 255u64 << (y * SINGLE_MAP_WIDTH);
+        (self.occupied & mask).count_ones()
+    }
 }