'''
Created on May 30, 2012

@author: David
'''
import os, pygame
from pygame.locals import *

#function to create our resources
def loadImage(name, colorkey=None):
    fullname = os.path.join(name)
    try:
        image = pygame.image.load(fullname).convert()
    except pygame.error, message:
        print 'Cannot load image:', name
        raise SystemExit, message
    if colorkey is not None:
        if colorkey is -1:
            colorkey = image.get_at((0,0))
        image.set_colorkey(colorkey, RLEACCEL)
    return image, image.get_rect()



# Class for paddle,
class Paddle (pygame.sprite.Sprite):
    def __init__(self):
        pygame.sprite.Sprite.__init__(self) #call Sprite initializer
        self.image, self.rect = loadImage ("paddle.jpg",-1)
        self.acceleration = 10 # X acceleration
        self.rect.midtop = [0,height]
       
    def update(self):
        "move the fist based on input"
        pos = 0
        # Go through the events pygame has
        for event in pygame.event.get():
            # If the event is a keydown and is value is == (x):
            if event.type == KEYDOWN and event.key == K_LEFT:
                pos = self.acceleration * -1
                return
            elif event.type == KEYDOWN and event.key == K_RIGHT:
                pos = self.acceleration
                return
            # Move the rectangle
            self.rect.move ([pos,0])

class Ball (pygame.sprite.Sprite):
    def __init__ (self):
        pygame.sprite.Sprite.__init__(self) # call Sprite initalizer
        self.image, self.rect = loadImage ("ball.jpg",-1)
        self.acceleration = [10,10] # X, Y accelerations
        self.velocity = [self.acceleration[0],self.acceleration[1]]
   
    def update (self):
        # Make sure it doesn't go off walls
        for i in range (2):
            if self.velocity [i] + self.rect [i] > screen.get_size() [i] or self.velocity [i] + self.rect [i] < i:
                self.velocity [i] *= -1
               
        # Check for collision with bricks or paddle
        didColide = 0
        for i in pygame.sprite.spritecollide(self, brickGroup, 1):
             # Could play brick break sound
            didColide = 1
            # Let bricks drop things
            i.die ()
           
        if didColide:
            self.colide()
        if pygame.sprite.spritecollide(self, paddleGroup, 0):
            # Bounce sound off paddle
            self.colide()
               
        self.rect.move (self.velocity)
       
    def colide (self):
        # Handles collision
        self.velocity [1] *= -1
       
class Brick (pygame.sprite.Sprite):
    def __init__ (self):
        pygame.sprite.Sprite.__init__(self) # Initalize
        self.image, self.rect = loadImage ("brick.jpg",-1)
   
    def die (self):pass # Drop loot if decided
   

# Global variable
screenSize = width, height = [500,600]
screen = pygame.display.set_mode(screenSize)
# Create groups
brickGroup = pygame.sprite.Group ()
ballGroup = pygame.sprite.Group ()
paddleGroup = pygame.sprite.Group ()   

def main ():
    # initalize the game
    pygame.init()
    clock = pygame.time.Clock()
    paddle = Paddle()
    ball = Ball ()
    ballGroup.add (ball)
    paddleGroup.add (paddle)
   
    # Make a grid of bricks
    brick = [[0,0,0,0,0],
             [0,0,0,0,0],
             [0,0,0,0,0],
             [0,0,0,0,0],
             [0,0,0,0,0] ]
    for y in range (4):
        for x in range (4):
            brick [y] [x]= Brick ()
            brickGroup.add (brick [y] [x])
           
    pygame.display.set_caption('Brick Breaker')
    pygame.mouse.set_visible(0)
           
   

    # Create background
    background = pygame.Surface(screen.get_size())
    background = background.convert()
    background.fill((255, 250, 250))

    #Display The Background
    screen.blit(background, (0, 0))
    pygame.display.flip()
   
    # main loop
    while 1:
        clock.tick(60) # keeps us from going faster than 60 fps
       
        # update objects
        paddleGroup.update()
        ballGroup.update()
        # draw objects
        brickGroup.draw (background)
        paddleGroup.draw(background)
        ballGroup.draw (background)
        screen.blit(background, (0,0))
        pygame.display.flip() # View.Update
        # handle exit
        for event in pygame.event.get():
            if event.type == QUIT:
                return
       
#this calls the 'main' function when this script is executed
if __name__ == '__main__': main()