46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
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)
|
|
|