109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
from random import sample
|
||
import pygame
|
||
from coords import Coords
|
||
from common import DrawableGameObject, FONT_NAME
|
||
from maze import get_maze_sz
|
||
|
||
|
||
class Coin(DrawableGameObject):
|
||
"""объект монетки"""
|
||
|
||
def __init__(
|
||
self,
|
||
coords: Coords,
|
||
parent: DrawableGameObject,
|
||
assets: dict | None = None,
|
||
):
|
||
super().__init__(coords, parent, assets)
|
||
self._surface = pygame.image.load(self.assets["coin.png"]).convert_alpha()
|
||
self.rect = self.surface.get_rect()
|
||
self.rect.topleft = coords
|
||
# уменьшаем размер монетки
|
||
sf = Coords(0.7, 0.7)
|
||
self._surface, self.rect = self.scene.scale_box(self.surface, self.rect, sf)
|
||
|
||
def draw(self):
|
||
self.parent.surface.blit(self.surface, self.rect)
|
||
|
||
@property
|
||
def bounding_rect(self) -> pygame.Rect:
|
||
new_rect = self.surface.get_bounding_rect()
|
||
new_rect.center = self.rect.center
|
||
return new_rect
|
||
|
||
|
||
class Coins(DrawableGameObject):
|
||
"""объект коллекции монеток"""
|
||
|
||
def __init__(
|
||
self,
|
||
parent: DrawableGameObject,
|
||
maze: list[list[int]],
|
||
box_sz: Coords,
|
||
count: int,
|
||
assets: dict | None = None,
|
||
):
|
||
super().__init__(Coords.zero(), parent, assets)
|
||
self.box_sz = box_sz
|
||
self._capacity = count
|
||
|
||
free_points = []
|
||
excluded = Coords(0, 1), get_maze_sz(maze) - Coords(1, 2)
|
||
for i, row in enumerate(maze):
|
||
for j, item in enumerate(row):
|
||
p = Coords(j, i)
|
||
if item < 1 and p not in excluded:
|
||
free_points.append(p)
|
||
continue
|
||
|
||
coin_points = sample(free_points, min(count, len(free_points)))
|
||
self.coins = [
|
||
Coin(point.transform(box_sz), self, self.assets) for point in coin_points
|
||
]
|
||
self.collected_coins = []
|
||
|
||
# Надпись, если все монетки собраны
|
||
font = pygame.font.SysFont(FONT_NAME, 30)
|
||
text = "Все монетки собраны!"
|
||
self.done_txt = font.render(text, 1, "#050366e3")
|
||
self.done_txt_rect = self.done_txt.get_rect()
|
||
self.done_txt_rect.topleft = Coords(10, 10)
|
||
|
||
@property
|
||
def capacity(self) -> int:
|
||
return self._capacity
|
||
|
||
@property
|
||
def coins_left(self) -> int:
|
||
return len(self.coins)
|
||
|
||
@property
|
||
def coins_collected(self) -> int:
|
||
return self.capacity - self.coins_left
|
||
|
||
@property
|
||
def all_collected(self) -> int:
|
||
return self.coins_left == 0
|
||
|
||
def draw(self):
|
||
for coin in self.collected_coins:
|
||
coin.draw()
|
||
for coin in self.coins:
|
||
coin.draw()
|
||
if self.all_collected:
|
||
self.parent.surface.blit(self.done_txt, self.done_txt_rect)
|
||
|
||
def add_to_collected(self, coin: Coin):
|
||
last_pos = Coords(10, 10)
|
||
if self.collected_coins:
|
||
last_pos = Coords(*self.collected_coins[-1].rect.topright)
|
||
last_pos -= Coords(coin.rect.width // 2, 0)
|
||
coin.coords = last_pos
|
||
self.collected_coins.append(coin)
|
||
|
||
def collect(self, actor: DrawableGameObject):
|
||
mined = [*filter(lambda coin: coin.overlap(actor.rect, actor.mask), self.coins)]
|
||
for coin in mined:
|
||
self.coins.remove(coin)
|
||
self.add_to_collected(coin)
|