From dafb9cf03cc5e8ad0a2b2d1857bb635d237f47b0 Mon Sep 17 00:00:00 2001 From: Justin Worthe Date: Sat, 1 Sep 2018 22:01:28 +0200 Subject: Corrected which player's attack towers to consider --- src/engine/bitwise_engine.rs | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'src/engine/bitwise_engine.rs') 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() + } } -- cgit v1.2.3