import os, sys
if 'SUMO_HOME' in os.environ:
    tools = os.path.join(os.environ['SUMO_HOME'], 'tools')
    sys.path.append(tools)
else:   
    print("please declare environment variable 'SUMO_HOME'")

import pygame as pg
from pygame.locals import *

from Map import Map
from CarController import CarController

pg.init()

screen = pg.display.set_mode((1280, 480), pg.SCALED)
pg.display.set_caption("Traffic")

clk = pg.time.Clock()

netSurf = pg.Surface(screen.get_size(),pg.SRCALPHA)
netSurf = netSurf.convert()
m = Map("test.net.xml",netSurf)

carSurf = pg.Surface(screen.get_size())
carSurf.set_colorkey((0, 0, 0))
carSurf = carSurf.convert()
cc = CarController("test3.rou.xml",m,carSurf)

running = True
while running:
    for event in pg.event.get():
        if event.type == pg.QUIT:
            running = False
        elif event.type == pg.KEYDOWN and event.key == pg.K_ESCAPE:
            running = False
        elif event.type == pg.KEYDOWN and event.key == pg.K_q:
            running = False

    m.draw(screen)
    cc.draw(screen)
    pg.display.flip()

    clk.tick(60)