forked from s464965/WMICraft
84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
import random
|
|
|
|
import pygame.image
|
|
|
|
from common.constants import GRID_CELL_SIZE, Direction
|
|
from common.helpers import parse_cord
|
|
from logic.health_bar import HealthBar
|
|
|
|
|
|
def load_knight_textures(team):
|
|
if team == "blue":
|
|
random_index = 3
|
|
else:
|
|
random_index = 4
|
|
states = [
|
|
pygame.image.load(f'resources/textures/knight_{random_index}_up.png').convert_alpha(), # up = 0
|
|
pygame.image.load(f'resources/textures/knight_{random_index}_right.png').convert_alpha(), # right = 1
|
|
pygame.image.load(f'resources/textures/knight_{random_index}_down.png').convert_alpha(), # down = 2
|
|
pygame.image.load(f'resources/textures/knight_{random_index}_left.png').convert_alpha(), # left = 3
|
|
]
|
|
return states
|
|
|
|
|
|
class Knight(pygame.sprite.Sprite):
|
|
|
|
def __init__(self, screen, position, group, team):
|
|
super().__init__(group)
|
|
|
|
self.direction = Direction.DOWN
|
|
self.states = load_knight_textures(team)
|
|
|
|
self.image = self.states[self.direction.value]
|
|
self.position = position
|
|
self._layer = 1
|
|
position_in_px = (parse_cord(position[0]), parse_cord(position[1]))
|
|
self.rect = self.image.get_rect(topleft=position_in_px)
|
|
|
|
self.team = team
|
|
self.max_hp = random.randint(9, 13)
|
|
self.attack = random.randint(2, 4)
|
|
self.defense = random.randint(1, 4)
|
|
self.points = 1
|
|
self.health_bar = HealthBar(screen, self.rect, current_hp=self.max_hp, max_hp=self.max_hp, calculate_xy=True, calculate_size=True)
|
|
|
|
def rotate_left(self):
|
|
self.direction = self.direction.left()
|
|
self.image = self.states[self.direction.value]
|
|
|
|
def update(self):
|
|
self.health_bar.update()
|
|
|
|
def rotate_right(self):
|
|
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)
|
|
self.rect.y = self.rect.y - GRID_CELL_SIZE - 5
|
|
elif self.direction.name == 'RIGHT':
|
|
self.position = (self.position[0] + 1, self.position[1])
|
|
self.rect.x = self.rect.x + GRID_CELL_SIZE + 5
|
|
elif self.direction.name == 'DOWN':
|
|
self.position = (self.position[0], self.position[1] + 1)
|
|
self.rect.y = self.rect.y + GRID_CELL_SIZE + 5
|
|
elif self.direction.name == 'LEFT':
|
|
self.position = (self.position[0] - 1, self.position[1])
|
|
self.rect.x = self.rect.x - GRID_CELL_SIZE - 5
|
|
|
|
def team_alias(self) -> str:
|
|
return "k_b" if self.team == "blue" else "k_r"
|