das-research/study.py

#! /bin/python3

import time, sys, random, copy
import importlib
from DAS import *


def study():
    if len(sys.argv) < 2:
        print("You need to pass a configuration file in parameter")
        exit(1)

    try:
        config = importlib.import_module(sys.argv[1])
    except ModuleNotFoundError as e:
        try:
            config = importlib.import_module(str(sys.argv[1]).replace(".py", ""))
        except ModuleNotFoundError as e:
            print(e)
            print("You need to pass a configuration file in parameter")
            exit(1)

    shape = Shape(0, 0, 0, 0, 0, 0)
    sim = Simulator(shape, config)
    sim.initLogger()
    results = []
    simCnt = 0

    now = datetime.now()
    if config.deterministic:
        random.seed("DASsimulator")
    else:
        random.seed(str(now).split(".")[1])
    execID = now.strftime("%Y-%m-%d_%H-%M-%S_")+str(random.randint(100,999))

    sim.logger.info("Starting simulations:", extra=sim.format)
    start = time.time()

    for shape in config.nextShape():

        sim.resetShape(shape)
        sim.initValidators()
        sim.initNetwork()
        result = sim.run()
        sim.logger.info("Shape: %s ... Block Available: %d in %d steps" % (str(sim.shape.__dict__), result.blockAvailable, len(result.missingVector)), extra=sim.format)
        results.append(copy.deepcopy(result))
        simCnt += 1

    end = time.time()
    sim.logger.info("A total of %d simulations ran in %d seconds" % (simCnt, end-start), extra=sim.format)

    if config.dumpXML:
        for res in results:
            res.dump(execID)
        sim.logger.info("Results dumped into results/%s/" % (execID), extra=sim.format)

    visualization = 1
    if visualization:
        vis = Visualizer(execID)
        vis.plotHeatmaps()


study()