Changed config class

sumo_project/config.py

@@ -10,6 +10,12 @@ import json
 
 class config: 
     
+    # Total of emissions of all pollutants in mg for n steps of simulation without locking areas
+    # These constants are simulation dependant, you must change them according to your simulation 
+    total_emissions100 = 13615949.148296086
+    total_emissions200 = 43970763.15084738
+    total_emissions300 = 87382632.0821697
+    
     def __init__(self, config_file = None):
         if not (config_file is None):
             with open(config_file) as f:
@@ -20,6 +26,7 @@ class config:
             self.areas_number = data["areas_number"]
             self.emissions_threshold = data["emissions_threshold"]
             self.n_steps = data["n_steps"]
+            self.window_size = data["window_size"]
             self.without_actions_mode = data["without_actions_mode"]
             self.limit_speed_mode = data["limit_speed_mode"]
             self.speed_rf = data["speed_rf"]
@@ -27,98 +34,57 @@ class config:
             self.trafficLights_duration_rf = data["trafficLights_duration_rf"]
             self.weight_routing_mode = data["weight_routing_mode"]
             self.lock_area_mode = data["lock_area_mode"]
-    
-###############################################################################
-############################# SIMULATION FILE #################################
-###############################################################################
-
-    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'")
-    
-    _SUMOCMD = 'sumo' # use 'sumo-gui' cmd for UI 
-    _SUMOCFG = "mulhouse_simulation/osm.sumocfg"
-    sumo_binary = os.path.join(os.environ['SUMO_HOME'], 'bin', _SUMOCMD)
-    sumo_cmd = [sumo_binary, "-c", _SUMOCFG]
-
-###############################################################################
-################################## LOGS #######################################
-###############################################################################
-
-    now = datetime.datetime.now()
-    current_date = now.strftime("%Y_%m_%d_%H_%M_%S")
-    log_filename = f'logs/sumo_logs_{current_date}.log'
-    
-    # create logger
-    logger = logging.getLogger("sumo_logger")
-    logger.setLevel(logging.INFO)
-    # create handler and set level to info
-    handler = logging.FileHandler(log_filename)
-    # create formatter
-    formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
-    # add formatter to handler
-    handler.setFormatter(formatter)
-    # add handler to logger
-    logger.addHandler(handler)
-
-###############################################################################
-########################## SIMULATION CONFIGURATION ###########################
-###############################################################################
-
-    areas_number = 10 # Simulation boundary will be divided into areas_number x areas_number areas 
-    emissions_threshold = 500000
-    n_steps = 200 
-    window_size = 200
-
-###############################################################################
-########################## ACTIONS CONFIGURATION ##############################
-###############################################################################
-
-    #Set this mode to True if you want running a basic simulation without actions 
-    without_actions_mode = False 
-
-    #Limit the speed into areas when the threshold is exceeded
-    speed_rf = 0.1
-    limit_speed_mode = True
-    
-    #Decrease all traffic lights duration into the area when the threshold is exceeded
-    trafficLights_duration_rf = 0.2
-    adjust_traffic_light_mode = True
-    
-    #Vehicles are routed according to the less polluted route (HEAVY)
-    weight_routing_mode = False
-    
-    #Lock the area when the threshold is exceeded (NOT FIXED)
-    lock_area_mode = False 
-    
-    #Weight routing mode cannot be combinated with other actions 
-    if weight_routing_mode:
-        limit_speed_mode = False
-        adjust_traffic_light_mode = False
-        lock_area_mode = False
+            
+            self.check_config()
+                
+    def check_config(self):
+        #Weight routing mode cannot be combinated with other actions 
+        if self.weight_routing_mode:
+            self.limit_speed_mode = False
+            self.adjust_traffic_light_mode = False
+            self.lock_area_mode = False
+                
+        #If without_actions_mode is choosen 
+        if self.without_actions_mode:
+            self.limit_speed_mode = False
+            self.adjust_traffic_light_mode = False
+            self.weight_routing_mode = False
+            self.lock_area_mode = False
         
-    #If without_actions_mode is choosen 
-    if without_actions_mode:
-        limit_speed_mode = False
-        adjust_traffic_light_mode = False
-        weight_routing_mode = False
-        lock_area_mode = False
-
-###############################################################################
-########################## SIMULATION REFERENCES ##############################
-###############################################################################
-
-    # Total of emissions of all pollutants in mg for n steps of simulation without locking areas
-    # These constants are simulation dependant, you must change them according to your simulation 
-    total_emissions100 = 13615949.148296086
-    total_emissions200 = 43970763.15084738
-    total_emissions300 = 87382632.0821697
-
-###############################################################################
-########################## CONFIGURATION METHODS ##############################
-###############################################################################
+                
+    def __repr__(self) -> str:
+        return (str(f'Grid : {self.areas_number}x{self.areas_number}\n')
+        + str(f'step number = {self.n_steps}\n')
+        + str(f'window size = {self.window_size}\n')
+        + str(f'weight routing mode = {self.weight_routing_mode}\n')
+        + str(f'lock area mode = {self.lock_area_mode}\n')
+        + str(f'limit speed mode = {self.limit_speed_mode}, RF = {self.speed_rf*100}%\n')
+        + str(f'adjust traffic light mode = {self.adjust_traffic_light_mode} , RF = {self.trafficLights_duration_rf*100}%\n'))
+                
+    
+    def init_traci(self):
+        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'")
+        
+        sumo_binary = os.path.join(os.environ['SUMO_HOME'], 'bin', self._SUMOCMD)
+        self.sumo_cmd = [sumo_binary, "-c", self._SUMOCFG]
+    
+    def init_logger(self):
+        now = datetime.datetime.now()
+        current_date = now.strftime("%Y_%m_%d_%H_%M_%S")
+        log_filename = f'logs/sumo_logs_{current_date}.log'
+        
+        logger = logging.getLogger("sumo_logger")
+        logger.setLevel(logging.INFO)
+        handler = logging.FileHandler(log_filename)
+        formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+        handler.setFormatter(formatter)
+        logger.addHandler(handler)
+        
+        return logger
 
     def get_basics_emissions(self):
         if self.n_steps == 100:
@@ -128,12 +94,4 @@ class config:
         if self.n_steps == 300:
             return self.total_emissions300
     
-    def show_config(self):
-        return (str(f'Grid : {self.areas_number}x{self.areas_number}\n')
-        + str(f'step number = {self.n_steps}\n')
-        + str(f'window size = {self.window_size}\n')
-        + str(f'weight routing mode = {self.weight_routing_mode}\n')
-        + str(f'lock area mode = {self.lock_area_mode}\n')
-        + str(f'limit speed mode = {self.limit_speed_mode}, RF = {self.speed_rf*100}%\n')
-        + str(f'adjust traffic light mode = {self.adjust_traffic_light_mode} , RF = {self.trafficLights_duration_rf*100}%\n'))
     

sumo_project/emissions.py

@@ -12,8 +12,9 @@ import sys
 from model import Area, Vehicle, Lane , TrafficLight , Phase , Logic
 from traci import trafficlight
 
-config = config('C:\\Users\\Admin\\Desktop\\config.json')
-logger = config.logger
+config = config('config.json')
+config.init_traci()
+logger = config.init_logger()
 
 def init_grid(simulation_bounds, areas_number):
     grid = list()
@@ -155,7 +156,7 @@ def main():
                 diff_with_actions = (ref - total_emissions)/ref    
                 logger.info(f'Reduction percentage of emissions = {diff_with_actions*100} %')
         
-        logger.info('With the configuration : \n' + str(config.show_config()))
+        logger.info(f'With the configuration : \n{config}')
         logger.info('Logs END')