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)