44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
import pygame
|
|
from enums import PillType
|
|
|
|
class Pill(pygame.sprite.Sprite):
|
|
def __init__(self, x, y, pill_type: PillType):
|
|
super().__init__()
|
|
self.x = x
|
|
self.y = y
|
|
self.pill_type = pill_type
|
|
self.eaten = False
|
|
self.radius = 2 if pill_type == PillType.NORMAL else 6
|
|
self.color = (255, 255, 255)
|
|
self.image = pygame.Surface((self.radius * 2, self.radius * 2), pygame.SRCALPHA)
|
|
pygame.draw.circle(self.image, self.color, (self.radius, self.radius), self.radius)
|
|
self.rect = self.image.get_rect(center=(x, y))
|
|
|
|
def check_eaten(self, pacman_rect, game_state):
|
|
if not self.eaten and self.rect.colliderect(pacman_rect):
|
|
self.eaten = True
|
|
self.kill()
|
|
return self.on_eaten(game_state)
|
|
return 0
|
|
|
|
def on_eaten(self, game_state):
|
|
"""Override this in subclasses to apply effects when eaten"""
|
|
return 0
|
|
|
|
class NormalPill(Pill):
|
|
def __init__(self, x, y):
|
|
super().__init__(x, y, PillType.NORMAL)
|
|
|
|
def on_eaten(self, game_state):
|
|
game_state["score"] += 100
|
|
return 100
|
|
|
|
class PowerPill(Pill):
|
|
def __init__(self, x, y):
|
|
super().__init__(x, y, PillType.POWER)
|
|
|
|
def on_eaten(self, game_state):
|
|
game_state["score"] += 500
|
|
game_state["ghost_mode"] = "frightened"
|
|
game_state["frightened_timer"] = pygame.time.get_ticks() # Optional
|
|
return 500 |