import pygame
from pygame.locals import *
from Battlebots import *
import sys

BLACK=(0,0,0)

xpos = [10, 20, 30, 40, 50]

class Game(object):
    def __init__(self, width, height, fps):
        pygame.init()
        self.w = width
        self.h = height
        self.screen = pygame.display.set_mode((self.w,self.h))
        self.clock = pygame.time.Clock()

    def gameLoop(self):
        self.loop = True
        while self.loop:
            background = (pygame.Surface(self.screen.get_size())).convert()
            #background.fill(BLACK)
            self.screen.fill(BLACK)
            label = pygame.font.SysFont("Arial", 14)
            caption = "Battlebots"
            pygame.display.set_caption(caption)

            self.pausespeed = 5



            for r in robot:
                pygame.draw.circle(self.screen, (255,255,255), r.getPos()[:2], 10)
                if r.x < 0: r.x = 0
                if r.x > self.w: r.x = self.w
                if r.y < 0: r.y = 0
                if r.y > self.h: r.y = self.h
                position = "(%d, %d)" % (r.getPos()[0], r.getPos()[1])
                name = label.render(r.name, True, (255,255,255))
                self.pos = label.render(position, True, (255,255,255))
                self.screen.blit(name,(r.x,r.y))
                self.screen.blit(self.pos,(r.x, r.y+12))
                #pygame.draw.line(self.screen,(255,255,255),(r.x+50,r.y+10), (r.x,r.y))

            for r in range(len(robot)):
                robot[r].update(robot[r].getPos()[0], robot[r].getPos()[1])
            pygame.display.flip()


            self.eventHandler()


    def eventHandler(self):
        for event in pygame.event.get():
            if event.type == QUIT:
                pygame.quit()
                sys.exit()
            elif event.type == KEYUP:
                if event.key == K_ESCAPE:
                    pygame.quit()
                    sys.exit()



if __name__=='__main__':
    game = Game(800,600,60)
    game.gameLoop()