class Sword:
"""
Main sword weapon.
"""
def __init__(self,
parent_rect: pygame.Rect,
depth: int = 20,
slash_range: int = 40,
extent: int = 10):
# TODO: These Default parameters should be set in the configuration file
self.parent_rect = parent_rect
self.depth = depth
self.slash_range = slash_range
self.extent = extent
self.horizontal_rect = pygame.Rect(0, 0, self.slash_range, self.depth)
self.vertical_rect = pygame.Rect(0, 0, self.depth, self.slash_range)
self.current_rect = self.horizontal_rect
self.image = None
self.action = Action(PlayerSprite.ATTACK_FRAME_DURATION *
PlayerSprite.FRAME_NUMBER)
self._update_current_rect_and_direction(Direction.DOWN)
def attack(self, direction: Direction):
if self.action.in_progress():
return
self._update_current_rect_and_direction(direction)
self.action.restart()
def has_hit(self, enemy: Enemy):
if self.action.in_progress():
return self.current_rect.colliderect(enemy.hitbox.rect)
else:
return False
def update(self):
self.action.update()
def _update_current_rect_and_direction(self, direction: Direction):
if direction in (Direction.DOWN, Direction.UP):
self.current_rect = self.horizontal_rect
else:
self.current_rect = self.vertical_rect
relative_position = self.extent * direction.value
self.current_rect.center = self.parent_rect.center + relative_position
self.set_image(BLUE)
def set_image(self, color):
self.image = pygame.Surface(self.current_rect.size)
self.image.fill(color)
self.image.set_alpha(100)
class Bow:
"""
Bow weapon and arrows
"""
SHOOT_COOLDOWN = 20
def __init__(self, parent_rect: pygame.Rect, parent_group: pygame.sprite.Group):
self.parent_rect = parent_rect
self.action = Action(self.SHOOT_COOLDOWN)
self.quiver = pygame.sprite.Group()
self.parent_group = parent_group
self.proyectile_image = pygame.image.load('assets/sprites/arrow.png').convert_alpha()
def attack(self, direction: Direction):
if self.action.in_progress():
return
arrow = Arrow(self.parent_rect.center, direction, self.proyectile_image)
arrow.add(self.parent_group, self.quiver)
self.action.restart()
def update(self, delta: float, physical_obstacles: List[pygame.Rect]):
self.action.update()
self.quiver.update(delta, physical_obstacles)