pellets.py

import pygame
from vector import Vector2
from constants import *

class Pellet(object):
    def __init__(self, x, y):
        self.name = "pellet"
        self.position = Vector2(x, y)
        self.color = WHITE
        self.radius = max(int(TILEWIDTH / 8), 1)
        self.points = 10
        self.visible = True
        
    def render(self, screen):
        if self.visible:
            p = self.position.asInt()
            p = (int(p[0]+TILEWIDTH/2), int(p[1]+TILEWIDTH/2))
            pygame.draw.circle(screen, self.color, p, self.radius)
        
        
class PowerPellet(Pellet):
    def __init__(self, x, y):
        Pellet.__init__(self, x, y)
        self.name = "powerpellet"
        self.radius = 8
        self.points = 50
        self.flashTime = 0.2
        self.timer= 0

    def update(self, dt):
        self.timer += dt
        if self.timer >= self.flashTime:
            self.visible = not self.visible
            self.timer = 0


class PelletGroup(object):
    def __init__(self, mazefile):
        self.pelletList = []
        self.powerpellets = []
        self.pelletSymbols = ["p", "n", "Y"]
        self.powerpelletSymbols = ["P", "N"]
        self.createPelletList(mazefile)
        self.dummy_list = [1 for pellet in self.pelletList]


    def update(self, dt):
        for powerpellet in self.powerpellets:
            powerpellet.update(dt)

        
    def createPelletList(self, mazefile):
        grid = self.readMazeFile(mazefile)
        rows = len(grid)
        cols = len(grid[0])
        for row in range(rows):
            for col in range(cols):
                if (grid[row][col] in self.pelletSymbols):
                    self.pelletList.append(Pellet(col*TILEWIDTH, row*TILEHEIGHT))
                if (grid[row][col] in self.powerpelletSymbols):
                    pp = PowerPellet(col*TILEWIDTH, row*TILEHEIGHT)
                    self.pelletList.append(pp)
                    self.powerpellets.append(pp)
                    
    def readMazeFile(self, textfile):
        f = open(textfile, "r")
        lines = [line.rstrip('\n') for line in f]
        return [line.split(' ') for line in lines]
    
    def isEmpty(self):
        if len(self.pelletList) == 0:
            return True
        return False
    
    def render(self, screen):
        for pellet in self.pelletList:
            pellet.render(screen)