import pygame
import random
import sys
from pygame.locals import *

from Car  import Car
from Vacuum import Vacuum
from Lane import Lane

pygame.init()
pygame.font.init()

SCREEN_WIDTH = 800
SCREEN_HEIGHT = 600

FONT = pygame.font.SysFont('Comic Sans MS', 128)
TEXT_SCREEN = FONT.render('Game over!', False, (255, 255, 255))
TEXT_RECT = TEXT_SCREEN.get_rect(center=(SCREEN_WIDTH/2, SCREEN_HEIGHT/2))


BUBBLE_LIMIT = 100

SCREEN = pygame.display.set_mode((SCREEN_WIDTH,SCREEN_HEIGHT))
BACKGROUND = (70,70,70)
SCREEN.fill(BACKGROUND)
CLOCK = pygame.time.Clock()

game_objects = []
bubble_objects = []
for i in range(3):
    markings = True
    if i == 2:
        markings = False
    game_objects.append(Lane(i * 80 + 50, SCREEN_WIDTH, 80, add_markings=markings))
    game_objects.append(Car(random.uniform(0, SCREEN_WIDTH), i * 80 + 50, bubble_objects, directionX=random.uniform(1, 5)))

for i in range(3, 6):
    markings = True
    if i == 5:
        markings = False
    game_objects.append(Lane(i * 80 + 60, SCREEN_WIDTH, 80, add_markings=markings))
    game_objects.append(Car(random.uniform(0, SCREEN_WIDTH), i * 80 + 60, bubble_objects, directionX= - random.uniform(1, 5)))

vacuum = Vacuum(bubble_objects)

while True:
    SCREEN.fill(BACKGROUND)
    pygame.display.set_caption("Rush Hour CO2")

    if len(bubble_objects) < BUBBLE_LIMIT:
        for game_object in game_objects:
            game_object.update(SCREEN)
        for game_object in bubble_objects:
            game_object.update(SCREEN)
        vacuum.update(SCREEN)

        for game_object in game_objects:
            game_object.draw(SCREEN)
        for game_object in bubble_objects:
            game_object.draw(SCREEN)
        vacuum.draw(SCREEN)
    else:
        # TODO game over screen with message to reduce driving cars
        SCREEN.blit(TEXT_SCREEN, TEXT_RECT)
        


    pygame.display.flip()
    
    for event in pygame.event.get():
        if event.type == QUIT:
            pygame.quit()
            sys.exit()
    CLOCK.tick(60)