logic/game.py

@@ -37,7 +37,6 @@ class Game:
     def game(self):
         logs = Logs(self.screen)
         level = Level(self.screen, logs)
-        stats = Stats(self.screen, level.list_knights_blue, level.list_knights_red)
 
         # setup clock for rounds
         NEXT_TURN = pygame.USEREVENT + 1
@@ -45,6 +44,7 @@ class Game:
 
         # create level
         level.create_map()
+        stats = Stats(self.screen, level.list_knights_blue, level.list_knights_red)
 
         print_numbers_flag = False
         running = True
@@ -60,8 +60,7 @@ class Game:
                         running = False
                     if event.key == pygame.K_n:
                         print_numbers_flag = not print_numbers_flag
-                    if event.key == pygame.K_r:
-                        stats.red_team_hp_bar.take_dmg(5)
+
                 if event.type == NEXT_TURN:  # is called every 'TURN_INTERVAL' milliseconds
                     level.handle_turn()
 

logic/health_bar.py

@@ -9,7 +9,6 @@ class HealthBar:
         self.rect = rect
         self.screen = screen
         self.current_hp = current_hp
-        self.target_hp = current_hp
         self.max_hp = max_hp
         self.x = self.rect.x
         self.y = self.rect.y
@@ -38,22 +37,22 @@ class HealthBar:
 
         self.health_ratio = self.max_hp / self.width
 
-    def take_dmg(self, dmg_taken):
-        if self.target_hp > 0:
-            self.target_hp -= dmg_taken
-        elif self.target_hp < 0:
-            self.target_hp = 0
+    def take_dmg(self, amount):
+        if self.current_hp - amount > 0:
+            self.current_hp -= amount
+        elif self.current_hp - amount <= 0:
+            self.current_hp = 0
 
     def heal(self, amount):
-        if self.target_hp < self.max_hp:
-            self.target_hp += amount
-        elif self.target_hp > self.max_hp:
-            self.target_hp = self.max_hp
+        if self.current_hp + amount < self.max_hp:
+            self.current_hp += amount
+        elif self.current_hp + amount > self.max_hp:
+            self.current_hp = self.max_hp
 
     def show(self):
         pygame.Surface.fill(self.screen, BLACK, (self.x-1, self.y-1, self.width+2, self.height+2))
         pygame.Surface.fill(self.screen, RED, (self.x, self.y, self.width, self.height))
-        pygame.Surface.fill(self.screen, GREEN, (self.x, self.y, int(self.target_hp / self.health_ratio), self.height))
+        pygame.Surface.fill(self.screen, GREEN, (self.x, self.y, int(self.current_hp / self.health_ratio), self.height))
 
 
 

models/knight.py

@@ -34,11 +34,10 @@ class Knight(pygame.sprite.Sprite):
 
         self.team = team
         self.max_hp = random.randint(7, 12)
-        self.current_hp = random.randint(1, self.max_hp)
         self.attack = random.randint(4, 7)
         self.defense = random.randint(1, 4)
         self.points = 1
-        self.health_bar = HealthBar(screen, self.rect, current_hp=self.current_hp, max_hp=self.max_hp, calculate_xy=True, calculate_size=True)
+        self.health_bar = HealthBar(screen, self.rect, current_hp=random.randint(1, self.max_hp), max_hp=self.max_hp, calculate_xy=True, calculate_size=True)
 
     def rotate_left(self):
         self.direction = self.direction.left()
@@ -48,6 +47,18 @@ class Knight(pygame.sprite.Sprite):
         self.direction = self.direction.right()
         self.image = self.states[self.direction.value]
 
+    def take_dmg(self, amount):
+        self.health_bar.take_dmg(amount)
+
+    def heal(self, amount):
+        self.health_bar.heal(amount)
+
+    def get_current_hp(self):
+        return self.health_bar.current_hp
+
+    def get_max_hp(self):
+        return self.health_bar.max_hp
+
     def step_forward(self):
         if self.direction.name == 'UP':
             self.position = (self.position[0], self.position[1] - 1)

ui/stats.py

@@ -15,8 +15,14 @@ class Stats:
         self.screen = screen
         self.x = (GRID_CELL_PADDING + GRID_CELL_SIZE) * COLUMNS + BORDER_WIDTH + 15
         self.y = 5
-        self.blue_team_hp_bar = HealthBar(self.screen, pygame.Rect(self.x + 30, self.y + 210, 100, 15), current_hp=50, max_hp=100)
-        self.red_team_hp_bar = HealthBar(self.screen, pygame.Rect(self.x + 210, self.y + 210, 100, 15), 100, 100)
+        self.blue_team_hp_bar = HealthBar(self.screen,
+                                          pygame.Rect(self.x + 30, self.y + 210, 100, 15),
+                                          current_hp=sum([knight.get_current_hp() for knight in self.list_knights_blue]),
+                                          max_hp=sum([knight.get_max_hp() for knight in self.list_knights_blue]))
+        self.red_team_hp_bar = HealthBar(self.screen,
+                                         pygame.Rect(self.x + 210, self.y + 210, 100, 15),
+                                         current_hp=sum([knight.get_current_hp() for knight in self.list_knights_red]),
+                                         max_hp=sum([knight.get_max_hp() for knight in self.list_knights_red]))
 
     def update(self):
 
@@ -35,6 +41,10 @@ class Stats:
         draw_text('VS', FONT_DARK, self.screen, self.x + 150, self.y + 120, 36)
 
         # HP bars
+        self.red_team_hp_bar.take_dmg(self.red_team_hp_bar.current_hp -
+                                      sum([knight.get_current_hp() for knight in self.list_knights_red]))
+        self.blue_team_hp_bar.take_dmg(self.blue_team_hp_bar.current_hp -
+                                       sum([knight.get_current_hp() for knight in self.list_knights_blue]))
         self.red_team_hp_bar.update()
         self.blue_team_hp_bar.update()