Add sumo factory to interact with the simulation

sumo_project/Area.py

@@ -0,0 +1,17 @@
+'''
+Created on 6 nov. 2018
+
+@author: Admin
+'''
+
+class Area:
+    
+    locked = False 
+    
+    def __init__(self, coords):
+        self.coords = coords
+    
+    
+    def getCoords(self):
+        return self.coords
+    

sumo_project/EmissionGetter.py

@@ -1,35 +1,20 @@
-'''
-Created on 11 oct. 2018
-
-@author: Axel HUYNH-PHUC
-'''
-
 import os, sys
-import traci
-from traci import polygon
+from sumo_project.SUMOFactory import SUMOFactory
 
-'''Launch traci'''
+# Traci launch
 if 'SUMO_HOME' in os.environ:
     tools = os.path.join(os.environ['SUMO_HOME'], 'tools')
-    sys.path.append(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'")
 
-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
+import traci
+from traci import polygon
 
 
-'''creating multiple zones of equal sizes'''
+# creating multiple zones of equal sizes
 def init_grid():
     default_color = (0, 255, 0)
     for i in range(cells_step):
@@ -40,53 +25,72 @@ def init_grid():
             polygon.add("area " + str(i) + "," + str(j), area, default_color, False, "rectangle")
 
 
-'''Emissions recovery by area'''
+# Emissions recovery by area
 def getEmissionsByArea(i, j):
     vehicles = []
-    '''Vehicle IDs retrieving into this area'''
+    
+    # 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'''
+    # 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'''
+    # Change area color if emisssions exceeds the threshold
     emissionsArea[i][j] += emissions
     if(emissionsArea[i][j] >= CO2_threshold):
         red = (255, 0, 0)
+        factory.lock_area(areas[i][j])
         polygon.setColor("area " + str(i) + "," + str(j), red)
         polygon.setFilled("area " + str(i) + "," + str(j), True)
 
-'''Recover emissions from all areas'''
+
+# 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'''
+
+# 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" )
+                  +" = " , str(emissionsArea[i][j]) + " mg")
 
 
-'''Simulation launch'''
-step = 0
-init_grid()      
-while step < 100:  # while traci.simulation.getMinExpectedNumber() > 0: 
-    traci.simulationStep()
-    getAllEmissions()
-    step += 1
+if __name__ == "__main__":
+    # Simulation launch
+    traci.start(sumoCmd)
 
-'''Display emissions information and close simulation'''
-showEmissions()
-traci.close()
+    # 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()
+    factory = SUMOFactory() 
+    
+    #factory.lock_area(areas[5][5])
+    
+    while traci.simulation.getMinExpectedNumber() > 0:
+        traci.simulationStep()
+        getAllEmissions()
+        step += 1
+
+    # Display emissions information and close simulation
+    showEmissions()
+    traci.close()
 sys.stdout.flush()
-

sumo_project/SUMOFactory.py

@@ -0,0 +1,40 @@
+'''
+Created on 17 oct. 2018
+
+@author: Admin
+'''
+import os, sys
+import traci 
+from shapely.geometry import Polygon
+from shapely.geometry import Point
+from shapely.geometry.linestring import LineString
+
+
+class SUMOFactory(object):
+    
+    def __init__(self):
+        '''Constructor'''
+    
+    def stopVehicle(self, veh_id):
+        traci.vehicle.remove(veh_id, traci.constants.REMOVE_PARKING)
+        
+    def getLanesIntoArea(self, area):
+        polygon_area = Polygon(area)
+        lanes = []
+        for lane_id in traci.lane.getIDList():
+            polygon_lane = LineString(traci.lane.getShape(lane_id))
+            if polygon_area.intersects(polygon_lane):
+                print("lane is in area : ", polygon_lane)
+                lanes.append(lane_id)
+        return lanes
+           
+    def lock_area(self, area):
+        lanes = self.getLanesIntoArea(area)
+        for lane_id in lanes:
+            '''print("Vehicles number into lane =  ", traci.lane.getLastStepVehicleNumber(lane_id))
+            if traci.lane.getLastStepVehicleNumber(lane_id) == 0:
+                traci.lane.setDisallowed(lane_id, "passenger")
+                print("lane blocked : ", lane_id)'''
+            traci.lane.setMaxSpeed(lane_id, 30) 
+            for veh_id in traci.vehicle.getIDList(): 
+                traci.vehicle.rerouteTraveltime(veh_id, True)