mirror of
https://github.com/Ahp06/SUMO_Emissions.git
synced 2024-11-21 19:16:30 +00:00
Merge pull request #1 from Ahp06/dev
Change absolute to relative path using SUMO_HOME
This commit is contained in:
GitHub
commit
d8982b8c3b
@ -1,92 +1,94 @@
|
||||
'''
|
||||
Created on 11 oct. 2018
|
||||
|
||||
@author: Axel HUYNH-PHUC
|
||||
'''
|
||||
|
||||
import os, sys
|
||||
import traci
|
||||
from traci import polygon
|
||||
|
||||
'''Launch traci'''
|
||||
if 'SUMO_HOME' in os.environ:
|
||||
tools = os.path.join(os.environ['SUMO_HOME'], 'tools')
|
||||
sys.path.append(tools)
|
||||
else:
|
||||
sys.exit("please declare environment variable 'SUMO_HOME'")
|
||||
|
||||
sumoBinary = "C:\\Users\\Admin\\AppData\\Roaming\\Microsoft\\Installer\\{A63B306E-2B15-11E1-88C8-028037EC0200}\\sumogui.exe"
|
||||
sumoCmd = [sumoBinary, "-c", "mulhouse_simulation\\osm.sumocfg"]
|
||||
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
|
||||
|
||||
|
||||
'''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" )
|
||||
|
||||
|
||||
'''Simulation launch'''
|
||||
step = 0
|
||||
init_grid()
|
||||
while step < 100: # while traci.simulation.getMinExpectedNumber() > 0:
|
||||
traci.simulationStep()
|
||||
getAllEmissions()
|
||||
step += 1
|
||||
|
||||
'''Display emissions information and close simulation'''
|
||||
showEmissions()
|
||||
traci.close()
|
||||
sys.stdout.flush()
|
||||
|
||||
'''
|
||||
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()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user