mirror of
https://github.com/Ahp06/SUMO_Emissions.git
synced 2024-11-22 03:26:30 +00:00
30 lines
886 B
Python
30 lines
886 B
Python
'''
|
|
Created on 17 oct. 2018
|
|
|
|
@author: Axel Huynh-Phuc, Thibaud Gasser
|
|
'''
|
|
|
|
import traci
|
|
from shapely.geometry.linestring import LineString
|
|
|
|
|
|
class SUMOFactory(object):
|
|
|
|
def stop_vehicle(self, veh_id):
|
|
traci.vehicle.remove(veh_id, traci.constants.REMOVE_PARKING)
|
|
|
|
def lanes_in_area(self, area):
|
|
polygon_area = area.rectangle
|
|
for lane_id in traci.lane.getIDList():
|
|
polygon_lane = LineString(traci.lane.getShape(lane_id))
|
|
if polygon_area.intersects(polygon_lane):
|
|
yield lane_id
|
|
|
|
def lock_area(self, area):
|
|
for lane_id in self.lanes_in_area(area):
|
|
print(f'Setting max speed of {lane_id} to 9.')
|
|
traci.lane.setMaxSpeed(lane_id, 9)
|
|
|
|
for veh_id in traci.vehicle.getIDList():
|
|
traci.vehicle.rerouteTraveltime(veh_id, True)
|