38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
import sumolib
|
|
from Car import Car
|
|
|
|
class CarController:
|
|
def __init__(self,parentMap):
|
|
self.map=parentMap
|
|
self.cars=[]
|
|
self.path = None
|
|
|
|
def setPath(self,path):
|
|
self.path = path
|
|
|
|
def fromPath(self,path=None):
|
|
if(path == None):
|
|
if(self.path == None):
|
|
return
|
|
else:
|
|
path = self.path
|
|
for vehicle in sumolib.xml.parse(path,"vehicle"):
|
|
route=vehicle.route[0].edges.split()
|
|
self.cars.append(Car(vehicle.id,route,self.map,self))
|
|
|
|
def getCarsOnEdge(self,edgeID): # on devrai probablement utiliser une map ici (dict en python?) mais flemme
|
|
res=[]
|
|
for car in self.cars:
|
|
cid=car.route[car.index].getID()
|
|
if(cid==edgeID):
|
|
res.append(car)
|
|
return res
|
|
|
|
def update(self):
|
|
for car in self.cars:
|
|
car.update(1.0/60)
|
|
|
|
def draw(self,painter):
|
|
for car in self.cars:
|
|
car.draw(painter)
|