engine.py

import sys
import time
import signal
import argparse
import json

from concurrent.futures import ThreadPoolExecutor
from src.matrix import Matrix
from utils.utils import *
from save_report import save_report #TODO refactor


matrix = None

def update_from_env(config):
    for key, value in config.items():
        env_var = os.getenv(key)
        if env_var is not None:
            # Update the value from the environment variable
            config[key] = type(value)(env_var) if value is not None else env_var
    return config

def load_config():
    filename = "configs/defaults.json"
    with open(filename, 'r') as file:
        config = json.load(file)
    config = update_from_env(config)
    return config


def main():
    global matrix
    #Base.metadata.create_all(engine)
    # Parse Args
    parser = argparse.ArgumentParser(description='Matrix Simulation')
    parser.add_argument('--scenario', type=str, default='configs/def.json', help='Path to the scenario file')
    parser.add_argument('--environment', type=str, default='configs/largev2.tmj', help='Path to the env file')
    parser.add_argument('--id', type=str, default=None, help='Custom Simulation ID')
    args = parser.parse_args()

    #matrix_data = {"agents_file":args.agents, "world_file":args.world}
    #matrix = Matrix(matrix_data)
    config = load_config()
    config['scenario'] = args.scenario
    config['environment'] = args.environment
    if args.id:
        config['id'] = args.id
    matrix = Matrix(config)
    pd(f"model:#{MODEL}")


    # Run
    start_time = datetime.now()

    #matrix.run()
    matrix.boot()
    matrix.run_singlethread()
    end_time = datetime.now()
    matrix.run_interviews()

    # Log Runtime
    matrix.simulation_runtime = end_time - start_time

    # Save the environment state to a file for inspection
    if matrix.id is not None and matrix.id != '' and RUN_REPORTS != 0:
        save_report(matrix)
    #session.commit()

def signal_handler(signum, frame):
    global matrix,last_interrupt_time, ctrl_c_count
    current_time = time.time()
    #TODO on first control-c, run interview questions then quit, we need to pause the simulation
    if current_time - last_interrupt_time < 2:
        ctrl_c_count += 1
        if ctrl_c_count > 1 :
            pd("Exiting...")
            exit(0)
        else:
            pass

    else:
        ctrl_c_count = 1
        pd("stopping matrix, please wait for current step to finish")
        pd("*"*50)
        matrix.status = "stop"
    last_interrupt_time = current_time

ctrl_c_count = 0
last_interrupt_time = time.time()
signal.signal(signal.SIGINT, signal_handler)

if __name__ == "__main__":
    main()