import pygame, random
import math, sys

class Dot(object):
    """Dot object"""
    def __init__(self):
        self.surf = pygame.image.load("arrow.png").convert_alpha()
        self.rect = pygame.Rect(0,0,50,50)
        self.frame = 0
        self.dir = 0 # 0 = up, 1 = down, 2 = left, 3 = right
        self.start_time = pygame.time.get_ticks()
    def update(self, keymap):
        """updates the sprite"""
        if keymap[0] or keymap[1] or keymap[2] or keymap[3]:
            if pygame.time.get_ticks() - self.start_time > 250:
                self.frame += 1
                if self.frame > 3:
                    self.frame = 0
                self.start_time = pygame.time.get_ticks()
        else:
            self.frame = 0
            self.start_time = pygame.time.get_ticks()

class Game(object):
    def __init__(self):
        self.screen = pygame.display.set_mode((800, 600))
        self.screen.fill((0, 255, 255))
        self.game_over = False
        self.clock = pygame.time.Clock()
        self.dot_list = [Dot() for x in range(10)]
        self.keymap = [False, False, False, False] #up, down, left, right key pressed? Really should use a dictionary
        for dot in self.dot_list:
            dot.rect.topleft = (775*random.random(), 575*random.random())
        
        
    def process_input(self):
        for event in pygame.event.get():
            if event.type == pygame.KEYDOWN:
                if event.key == pygame.K_ESCAPE:
                    self.game_over = True
                if event.key == pygame.K_d:
                    self.keymap[3] = True
                if event.key == pygame.K_a:
                    self.keymap[2] = True
                if event.key == pygame.K_w:
                    self.keymap[0] = True
                if event.key == pygame.K_s:
                    self.keymap[1] = True
            if event.type == pygame.KEYUP:
                if event.key == pygame.K_d:
                    self.keymap[3] = False
                if event.key == pygame.K_a:
                    self.keymap[2] = False
                if event.key == pygame.K_w:
                    self.keymap[0] = False
                if event.key == pygame.K_s:
                    self.keymap[1] = False
        
    def update(self):
        if self.keymap[3]:
            for dot in self.dot_list:
                dot.dir = 3
                future_dot = dot.rect.move(10, 0)
                if self.screen.get_rect().contains(future_dot):
                    dot.rect = future_dot
                else:
                    dot.rect.right = 800
        if self.keymap[2]:
            for dot in self.dot_list:
                dot.dir = 2
                future_rect = dot.rect.move(-10,0)
                if self.screen.get_rect().contains(future_rect):
                    dot.rect = future_rect
                else:
                    dot.rect.left = 0
        if self.keymap[0]:
            for dot in self.dot_list:
                dot.dir = 0
                future_rect = dot.rect.move(0,-10)
                if self.screen.get_rect().contains(future_rect):
                    dot.rect = future_rect
                else:
                    dot.rect.top = 0
        if self.keymap[1]:
            for dot in self.dot_list:
                dot.dir = 1
                future_rect = dot.rect.move(0,10)
                if self.screen.get_rect().contains(future_rect):
                    dot.rect = future_rect
                else:
                    dot.rect.bottom = 600
        for dot in self.dot_list:
            dot.update(self.keymap)
        
    def draw(self):
        self.screen.fill((0, 255, 255))
        #self.screen.blit(self.dot, self.dot_rect)
        for dot in self.dot_list:
            self.screen.blit(dot.surf, dot.rect, pygame.Rect(dot.frame*50, dot.dir*50, 50, 50))
        
pygame.init()
g = Game()
while not g.game_over:
    g.clock.tick(30)
    g.process_input()
    g.update()
    g.draw()
    pygame.display.flip()
sys.exit()