mirror of
https://github.com/Ahp06/SUMO_Emissions.git
synced 2024-11-24 20:46:29 +00:00
95 lines
3.0 KiB
Python
95 lines
3.0 KiB
Python
'''
|
|
Created on 11 oct. 2018
|
|
|
|
@author: Axel HUYNH-PHUC
|
|
'''
|
|
|
|
import os, sys
|
|
|
|
'''Launch traci'''
|
|
if 'SUMO_HOME' in os.environ:
|
|
tools = os.path.join(os.environ['SUMO_HOME'], 'tools')
|
|
sys.path.append(os.path.join(os.environ['SUMO_HOME'], tools))
|
|
sumoBinary = os.path.join(os.environ['SUMO_HOME'], 'bin', 'sumo-gui')
|
|
sumoCmd = [sumoBinary, "-c", "mulhouse_simulation/osm.sumocfg"]
|
|
else:
|
|
sys.exit("please declare environment variable 'SUMO_HOME'")
|
|
|
|
import traci
|
|
from traci import polygon
|
|
|
|
|
|
'''creating multiple zones of equal sizes'''
|
|
def init_grid():
|
|
default_color = (0, 255, 0)
|
|
for i in range(cells_step):
|
|
for j in range(cells_step):
|
|
area = ((i * width, j * height), (i * width, (j + 1) * height),
|
|
((i + 1) * width, (j + 1) * height), ((i + 1) * width, j * height))
|
|
areas[i][j] = area
|
|
polygon.add("area " + str(i) + "," + str(j), area, default_color, False, "rectangle")
|
|
|
|
|
|
'''Emissions recovery by area'''
|
|
def getEmissionsByArea(i, j):
|
|
vehicles = []
|
|
'''Vehicle IDs retrieving into this area'''
|
|
for veh_id in traci.vehicle.getIDList():
|
|
pos = traci.vehicle.getPosition(veh_id)
|
|
if((i * width < pos[0] and (i + 1) * width > pos[0])
|
|
and (j * height < pos[1] and (j + 1) * height > pos[1])):
|
|
vehicles.append(veh_id)
|
|
|
|
'''Sum all emissions'''
|
|
emissions = 0.0
|
|
for veh_id in vehicles:
|
|
emission = traci.vehicle.getCO2Emission(veh_id)
|
|
emissions += emission
|
|
|
|
'''Change area color if emisssions exceeds the threshold'''
|
|
emissionsArea[i][j] += emissions
|
|
if(emissionsArea[i][j] >= CO2_threshold):
|
|
red = (255, 0, 0)
|
|
polygon.setColor("area " + str(i) + "," + str(j), red)
|
|
polygon.setFilled("area " + str(i) + "," + str(j), True)
|
|
|
|
'''Recover emissions from all areas'''
|
|
def getAllEmissions():
|
|
for i in range(cells_step):
|
|
for j in range(cells_step):
|
|
getEmissionsByArea(i, j)
|
|
|
|
'''Display emissions information'''
|
|
def showEmissions():
|
|
for i in range(cells_step):
|
|
for j in range(cells_step):
|
|
print("Total CO2 emissions into Area " + str(i) + "," + str(j)
|
|
+ " = " , str(emissionsArea[i][j]) + " mg" )
|
|
|
|
|
|
if __name__ == "__main__":
|
|
'''Simulation launch'''
|
|
traci.start(sumoCmd)
|
|
|
|
'''Variables and constants declaration'''
|
|
cells_step = 10
|
|
boundary = traci.simulation.getNetBoundary()
|
|
areas = [[0] * cells_step for _ in range(cells_step)]
|
|
emissionsArea = [[0] * cells_step for _ in range(cells_step)]
|
|
width = boundary[1][0] / cells_step # width/step
|
|
height = boundary[1][1] / cells_step # height/step
|
|
CO2_threshold = 500000
|
|
|
|
step = 0
|
|
init_grid()
|
|
while traci.simulation.getMinExpectedNumber() > 0:
|
|
print("Vehicle count :", traci.vehicle.getIDCount())
|
|
traci.simulationStep()
|
|
getAllEmissions()
|
|
step += 1
|
|
|
|
'''Display emissions information and close simulation'''
|
|
showEmissions()
|
|
traci.close()
|
|
sys.stdout.flush()
|