import sys
import pygame
import random

#Settings
width = 500
heigth = 500
#End settings

pygame.init()
pygame.display.set_caption("Snake")
display = pygame.display.set_mode((width,heigth))
bg = pygame.image.load("res/grass.jpg")

#Objects
class SnakeHead():
    def __init__(self,disp):
        self.moving = 0 #0 up,1 down,2 right,3 left
        self.display = disp
        self.image = pygame.image.load("res/head.png")
        self.movingImage = self.image
        self.rect = self.image.get_rect()
        self.move(200,200)
    def move(self,x,y):
        self.rect = self.rect.move(x,y)
    def render(self):
        self.display.blit(self.movingImage,self.rect)

class SnakePart():
    def __init__(self,disp,x,y,tail):
        self.display = disp
        if tail == 1:
            self.image = pygame.image.load("res/tail.png")
        else:
            self.image = pygame.image.load("res/body.png")
        self.movingImage = self.image
        self.rect = self.image.get_rect()
    def move(self,x,y):
        self.rect = self.rect.move(x,y)
    def render(self):
        self.display.blit(self.movingImage,self.rect)

class Apple():
    def __init__(self,disp,x,y):
        self.image = pygame.image.load("res/apple.png")
        self.rect = self.image.get_rect()
        self.rect = self.rect.move(x,y)
        self.display = disp
    def render(self):
        self.display.blit(self.image,self.rect)

#Functions
def rotateImage(image, angle):
    orig_rect = image.get_rect()
    rot_image = pygame.transform.rotate(image, angle)
    rot_rect = orig_rect.copy()
    rot_rect.center = rot_image.get_rect().center
    rot_image = rot_image.subsurface(rot_rect).copy()
    return rot_image

#Main Loop
while 1:

    #Create Objects
    player = SnakeHead(display)
    apple = Apple(display,10,10)
    snakeBody = [SnakePart(display,200,220,1),SnakePart(display,200,240,0)]

    #Game Loop
    while 1:
        #Get Events
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                sys.exit()
            if event.type == pygame.KEYDOWN:
                if event.key == pygame.K_LEFT:
                    player.moving = 3
                    player.movingImage = rotateImage(player.image,90)
                if event.key == pygame.K_RIGHT:
                    player.moving = 2
                    player.movingImage = rotateImage(player.image,270)
                if event.key == pygame.K_UP:
                    player.moving = 0
                    player.movingImage = player.image
                if event.key == pygame.K_DOWN:
                    player.moving = 1
                    player.movingImage = rotateImage(player.image,180)

        #Movement
        if player.moving == 0:
            player.move(0,-3)
        if player.moving == 1:
            player.move(0,3)
        if player.moving == 2:
            player.move(3,0)
        if player.moving == 3:
            player.move(-3,0)

        #Border
        if player.rect.x > width - 20:
            break
        if player.rect.x < 0:
            break
        if player.rect.y > heigth - 20:
            break
        if player.rect.y < 0:
            break

        #Calculate Snake
        i = len(snakeBody) - 1
        lastX = player.rect.x
        lastY = player.rect.y
        if player.moving == 0:
            lastY = player.rect.y - 20
        if player.moving == 1:
            lastY = player.rect.y + 20
        if player.moving == 2:
            lastX = player.rect.x + 20
        if player.moving == 3:
            lastX = player.rect.x - 20
        while i > -1:
            print "moved: x:%d y:%d"%(lastX,lastY)
            snakeBody[i].move(lastX,lastY)
            lastX = snakeBody[i].rect.x
            lastY = snakeBody[i].rect.y
            i = i - 1

        #Render
        display.fill((0,0,0))
        display.blit(bg,(0,0))
        for object in snakeBody[:]:
            object.render()
        player.render()
        apple.render()

        #Flip
        pygame.display.flip()
        #Delay
        pygame.time.delay(20)