from __future__ import absolute_import from __future__ import print_function import os from shutil import copyfile from testing_simulation import Simulation from generator import TrafficGenerator from model import TestModel from visualization import Visualization from utils import import_test_configuration, set_sumo, set_test_path if __name__ == "__main__": config = import_test_configuration(config_file='testing_settings.ini') sumo_cmd = set_sumo(config['gui'], config['sumocfg_file_name'], config['max_steps']) model_path, plot_path = set_test_path(config['models_path_name'], config['model_to_test']) Model = TestModel(input_dim=config['num_states'], model_path=model_path) TrafficGen = TrafficGenerator(config['max_steps'], config['n_v_generated']) Visualization = Visualization(plot_path, dpi=96) Simulation = Simulation(Model, TrafficGen, sumo_cmd, sumo_cmd, config['max_steps'], config['green_duration'], config['yellow_duration'], config['num_states'], config['num_actions'])
@property def CO2_episode(self): return self._CO2_episode @property def fuel_episode(self): return self._fuel_episode @property def waiting_times(self): return self._get_waiting_times() if __name__ == "__main__": config = import_test_configuration(config_file='settings/testing_one_settings.ini') sumo_cmd = set_sumo(config['gui'], config['simulation_folder'], config['sumocfg_file_name'], config['max_steps']) model_path, plot_path = set_test_path(config['models_path_name'], config['model_to_test']) Traffic_Generator = Traffic_Generator( config["flow_file"], config["route_file"], config["n_cars_generated"], config["simulation_time"] ) Visualization = Visualization( plot_path, dpi=96 )