WMICraft/ui/stats.py

68 lines
3.3 KiB
Python
Raw Normal View History

2022-03-09 16:59:58 +01:00
import pygame
import time
2022-03-09 16:59:58 +01:00
from logic.health_bar import HealthBar
2022-03-11 19:42:17 +01:00
from common.colors import FONT_DARK, ORANGE, WHITE, RED
2022-04-10 20:28:50 +02:00
from common.constants import COLUMNS, GRID_CELL_PADDING, GRID_CELL_SIZE, BORDER_WIDTH, BORDER_RADIUS
2022-03-11 19:42:17 +01:00
from common.helpers import draw_text
2022-03-09 16:59:58 +01:00
class Stats:
2022-04-28 21:50:12 +02:00
def __init__(self, screen, list_knights_blue, list_knights_red):
2022-03-09 16:59:58 +01:00
self.grid = []
2022-04-28 21:50:12 +02:00
self.list_knights_blue = list_knights_blue
self.list_knights_red = list_knights_red
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=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]))
2022-06-09 23:56:26 +02:00
self.blue_team_points = 0
self.red_team_points = 0
2022-03-09 16:59:58 +01:00
2022-04-28 14:13:59 +02:00
def update(self):
2022-03-09 16:59:58 +01:00
# background
pygame.draw.rect(self.screen, WHITE, pygame.Rect(self.x, self.y, 340, 450), 0, BORDER_RADIUS)
2022-03-09 16:59:58 +01:00
# title
draw_text('STATS', FONT_DARK, self.screen, self.x + 120, self.y + 10, 36)
pygame.draw.rect(self.screen, ORANGE, pygame.Rect(self.x, self.y + 65, 340, 3))
2022-03-09 16:59:58 +01:00
# shields
2022-03-11 19:42:17 +01:00
shield_blue = pygame.image.load('./resources/textures/shield_blue.png')
shield_red = pygame.image.load('./resources/textures/shield_red.png')
self.screen.blit(shield_blue, (self.x + 20, self.y + 80))
self.screen.blit(shield_red, (self.x + 200, self.y + 80))
draw_text('VS', FONT_DARK, self.screen, self.x + 150, self.y + 120, 36)
2022-03-09 16:59:58 +01:00
# 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()
2022-03-09 16:59:58 +01:00
# texts
2022-04-28 21:50:12 +02:00
draw_text('Rycerze: ' + str(len(self.list_knights_blue)), FONT_DARK, self.screen, self.x + 35, self.y + 240, 18) # blue
2022-03-09 16:59:58 +01:00
draw_text('Rycerze: ' + str(len(self.list_knights_red)), FONT_DARK, self.screen, self.x + 215, self.y + 240, 18)
2022-03-09 16:59:58 +01:00
# points
pygame.draw.rect(self.screen, ORANGE, pygame.Rect(self.x, self.y + 390, 340, 3))
2022-06-09 23:56:26 +02:00
draw_text('PUNKTY: ' + str(self.blue_team_points), FONT_DARK, self.screen, self.x + 35, self.y + 408, 18, True)
draw_text('PUNKTY: ' + str(self.red_team_points), FONT_DARK, self.screen, self.x + 215, self.y + 408, 18, True)
def add_points(self, team, points):
if team == "blue":
self.blue_team_points += points
else:
self.red_team_points += points