mirror of
https://github.com/Ahp06/SUMO_Emissions.git
synced 2024-12-26 11:26:30 +00:00
83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
"""
|
|
This module defines the global configuration for the simulation
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
|
|
from data import Data
|
|
from model import Emission
|
|
|
|
|
|
class Config:
|
|
"""
|
|
The Config class defines all simulation properties that can be changed
|
|
"""
|
|
|
|
def __init__(self,config_file, data : Data):
|
|
"""
|
|
Default constructor
|
|
"""
|
|
self.import_config_file(config_file)
|
|
self.init_traci(data.dir)
|
|
self.check_config()
|
|
|
|
def import_config_file(self, config_file):
|
|
"""
|
|
Import your configuration file in JSON format
|
|
:param config_file: The path to your configuration file
|
|
:return:
|
|
"""
|
|
with open(f'{config_file}', 'r') as f:
|
|
data = json.load(f)
|
|
|
|
for option in data:
|
|
self.__setattr__(option, data[option])
|
|
self.config_filename = os.path.basename(f.name)
|
|
self.check_config()
|
|
|
|
def check_config(self):
|
|
"""
|
|
Check the relevance of user configuration choices
|
|
:return:
|
|
"""
|
|
# Weight routing mode cannot be combined 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 chosen
|
|
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
|
|
|
|
def __repr__(self) -> str:
|
|
"""
|
|
:return: All properties chosen by the user
|
|
"""
|
|
return (
|
|
f'step number = {self.n_steps}\n'
|
|
f'window size = {self.window_size}\n'
|
|
f'weight routing mode = {self.weight_routing_mode}\n'
|
|
f'lock area mode = {self.lock_area_mode}\n'
|
|
f'limit speed mode = {self.limit_speed_mode}, RF = {self.speed_rf * 100}%\n'
|
|
f'adjust traffic light mode = {self.adjust_traffic_light_mode},'
|
|
f'RF = {self.trafficLights_duration_rf * 100}%\n'
|
|
)
|
|
|
|
def init_traci(self, simulation_dir):
|
|
"""
|
|
Init the Traci API
|
|
:param simulation_dir: The path to the simulation directory
|
|
:return:
|
|
"""
|
|
simdir = os.path.join(os.path.dirname(__file__), f'{simulation_dir}')
|
|
|
|
for f in os.listdir(simdir):
|
|
if f.endswith('.sumocfg'):
|
|
self._SUMOCFG = os.path.join(simdir, f)
|
|
sumo_binary = os.path.join(os.environ['SUMO_HOME'], 'bin', self._SUMOCMD)
|
|
self.sumo_cmd = [sumo_binary, "-c", self._SUMOCFG] |