from robot import Robot

from math import degrees
import sys, pygame
from pygame.locals import *
pygame.init()

size = width, height = 1000, 300
black = 0, 0, 0

screen = pygame.display.set_mode(size)

r = Robot()

while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT: sys.exit()
        if event.type == pygame.KEYDOWN:
            if event.key == K_q:
                sys.exit()
    
    screen.fill(black)
    r.update(1/60)
    robotShape = pygame.Rect(0,0, 10,20);
    rsurf = pygame.Surface((10,20)).convert_alpha();
    pygame.draw.rect(rsurf, (255,255,255), robotShape);
    rsurf_r = pygame.transform.rotate(rsurf, degrees(r.rot));
    center = rsurf_r.get_rect().center
    screen.blit(rsurf_r, (r.pos[0]*100-center[0],r.pos[1]*100-center[1]))
    pygame.display.flip()