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

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

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

TITLE_SCREEN = True

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))

SUBTITLE_FONT_SIZE = 48
SUBTITLE_FONT = pygame.font.SysFont('Comic Sans MS', SUBTITLE_FONT_SIZE)
SUBTITLE_TEXT_SCREEN = SUBTITLE_FONT.render('You see, it\'s hard to clean up all this mess...', False, (255, 255, 255))
SUBTITLE_TEXT_RECT = SUBTITLE_TEXT_SCREEN.get_rect(center=(SCREEN_WIDTH/2, SCREEN_HEIGHT/2 + 100))
SUBTITLE_TEXT_SCREEN2 = SUBTITLE_FONT.render('Maybe use a bicycle instead of a car?', False, (255, 255, 255))
SUBTITLE_TEXT_RECT2 = SUBTITLE_TEXT_SCREEN2.get_rect(center=(SCREEN_WIDTH/2, SCREEN_HEIGHT/2 + 100 + SUBTITLE_FONT_SIZE))

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)
bicycle = Bicycle(0, SCREEN_HEIGHT - 40)


TITLE_IMAGE = pygame.image.load("sprites/title_screen.png")

while TITLE_SCREEN:
    SCREEN.fill(BACKGROUND)
    
    for event in pygame.event.get():
        if event.type == QUIT:
            pygame.quit()
            sys.exit()
        if event.type == pygame.KEYDOWN or event.type == pygame.MOUSEBUTTONDOWN:
            TITLE_SCREEN = False
    SCREEN.blit(TITLE_IMAGE, (0, 0))
    pygame.display.flip()
    CLOCK.tick(30)

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:
        SCREEN.blit(TEXT_SCREEN, TEXT_RECT)
        SCREEN.blit(SUBTITLE_TEXT_SCREEN, SUBTITLE_TEXT_RECT)
        SCREEN.blit(SUBTITLE_TEXT_SCREEN2, SUBTITLE_TEXT_RECT2)
        bicycle.update(SCREEN)
        bicycle.draw(SCREEN)
        


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