139 lines
4.9 KiB
Python
139 lines
4.9 KiB
Python
import pygame
|
||
from coords import Coords, Direction
|
||
from common import DrawableGameObject, EventHandler
|
||
|
||
|
||
class Hero(DrawableGameObject, EventHandler):
|
||
"""объект главного героя"""
|
||
|
||
def __init__(
|
||
self,
|
||
coords: Coords,
|
||
parent: DrawableGameObject,
|
||
assets: dict | None = None,
|
||
):
|
||
super().__init__(coords, parent, assets)
|
||
self._surface = pygame.image.load(self.assets["ghost.png"])
|
||
self.rect = self.surface.get_rect()
|
||
sf = Coords(0.8, 0.8)
|
||
self._surface, self.rect = self.scene.scale_box(self.surface, self.rect, sf)
|
||
self.rect.topleft = coords
|
||
self.active = True
|
||
self.looking_right = False
|
||
self._speed = 1
|
||
self.direction = Direction.RIGHT
|
||
self.mouse_active = False
|
||
# картинка изначально влево, а надо бы начинать со взгляда вправо
|
||
self.flip()
|
||
|
||
def draw(self):
|
||
self.parent.surface.blit(self.surface, self.rect)
|
||
|
||
def _check_collision(self, coords):
|
||
"""Проверка пересечения со стенами"""
|
||
new_rect = self.rect.copy()
|
||
new_rect.topleft = coords
|
||
return self.scene.walls.check_collision(new_rect, self.mask)
|
||
|
||
@property
|
||
def speed(self):
|
||
return max(self._speed, 1)
|
||
|
||
@speed.setter
|
||
def speed(self, value):
|
||
self._speed = min(value, 15)
|
||
|
||
def _reduce_step(self, coords):
|
||
"""Уменьшение шага движения, с целью подойти вплотную к стене"""
|
||
delta = coords - self.coords
|
||
dx, dy = 0, 0
|
||
if abs(delta.x) > 1:
|
||
dx = 1 * (delta.x < 0 or -1)
|
||
if abs(delta.y) > 1:
|
||
dy = 1 * (delta.y < 0 or -1)
|
||
return coords + Coords(dx, dy)
|
||
|
||
def set_coords(self, coords: Coords):
|
||
# проверка колизии
|
||
has_collision = self._check_collision(coords)
|
||
if not has_collision:
|
||
super().set_coords(coords)
|
||
self.scene.coins.collect(self)
|
||
return
|
||
|
||
# уменьшение шага
|
||
while has_collision and coords != self.coords:
|
||
coords_new = self._reduce_step(coords)
|
||
if coords_new == coords:
|
||
return # не могу уменьшить шаг
|
||
coords = coords_new
|
||
has_collision = self._check_collision(coords)
|
||
super().set_coords(coords)
|
||
self.scene.coins.collect(self)
|
||
|
||
def flip(self):
|
||
self.looking_right = not self.looking_right
|
||
self._surface = pygame.transform.flip(self.surface, flip_x=True, flip_y=False)
|
||
|
||
def update_direction(self, direction: Direction):
|
||
if direction in (Direction.LEFT, Direction.RIGHT):
|
||
going_right = direction == Direction.RIGHT
|
||
if self.looking_right != going_right:
|
||
self.flip()
|
||
|
||
if direction != self.direction:
|
||
self.speed = 0
|
||
self.direction = direction
|
||
else:
|
||
self.speed += 1
|
||
|
||
def move(self, direction: Direction, step: int = 1):
|
||
self.update_direction(direction)
|
||
self.coords += direction.as_coords() * step * self.speed // 3
|
||
|
||
def handle_mouse_event(self, event: pygame.event.Event):
|
||
if event.type not in (
|
||
pygame.MOUSEBUTTONDOWN,
|
||
pygame.MOUSEBUTTONUP,
|
||
pygame.MOUSEMOTION,
|
||
):
|
||
return
|
||
match event.type:
|
||
case pygame.MOUSEBUTTONDOWN:
|
||
self.mouse_active = self.rect.collidepoint(event.pos)
|
||
case pygame.MOUSEBUTTONUP:
|
||
self.mouse_active = False
|
||
case pygame.MOUSEMOTION if self.mouse_active:
|
||
rel = Coords(*event.rel)
|
||
direction = Direction.from_coords(rel)
|
||
if direction:
|
||
self.update_direction(direction)
|
||
self.coords += rel
|
||
|
||
def handle_event(self, event: pygame.event.Event):
|
||
if not self.active:
|
||
return
|
||
self.handle_mouse_event(event)
|
||
|
||
def handle_keys(self, keys_pressed):
|
||
if not self.active:
|
||
return
|
||
|
||
wide, short = 3, 1
|
||
if keys_pressed[pygame.K_UP]:
|
||
self.move(Direction.UP, wide)
|
||
if keys_pressed[pygame.K_DOWN]:
|
||
self.move(Direction.DOWN, wide)
|
||
if keys_pressed[pygame.K_LEFT]:
|
||
self.move(Direction.LEFT, wide)
|
||
if keys_pressed[pygame.K_RIGHT]:
|
||
self.move(Direction.RIGHT, wide)
|
||
if keys_pressed[pygame.K_w]:
|
||
self.move(Direction.UP, short)
|
||
if keys_pressed[pygame.K_s]:
|
||
self.move(Direction.DOWN, short)
|
||
if keys_pressed[pygame.K_a]:
|
||
self.move(Direction.LEFT, short)
|
||
if keys_pressed[pygame.K_d]:
|
||
self.move(Direction.RIGHT, short)
|