import argparse
import sys
import time
from pypdevs.minimal import CoupledDEVS
from pypdevs.minimal import Simulator

from devstone import LI, HI, HO, HOmod
from generator import Generator

sys.setrecursionlimit(10000)


class DEVStoneEnvironment(CoupledDEVS):

    def __init__(self, name, devstone_model, num_gen_outputs=1, gen_period=None):
        super(DEVStoneEnvironment, self).__init__(name=name)
        generator = Generator("generator", num_gen_outputs, gen_period)
        self.addSubModel(generator)
        self.addSubModel(devstone_model)

        self.connectPorts(generator.o_out, devstone_model.i_in)

        if isinstance(devstone_model, HO) or isinstance(devstone_model, HOmod):
            self.connectPorts(generator.o_out, devstone_model.i_in2)


MODEL_TYPES = ("LI", "HI", "HO", "HOmod")


def parse_args():
    parser = argparse.ArgumentParser(description='Script to compare DEVStone implementations with different engines')

    parser.add_argument('-m', '--model-type', required=True, help='DEVStone model type (LI, HI, HO, HOmod)')
    parser.add_argument('-d', '--depth', type=int, required=True, help='Number of recursive levels of the model.')
    parser.add_argument('-w', '--width', type=int, required=True, help='Width of each coupled model.')
    parser.add_argument('-i', '--int-cycles', type=int, default=0, help='Dhrystone cycles executed in internal transitions')
    parser.add_argument('-e', '--ext-cycles', type=int, default=0, help='Dhrystone cycles executed in external transitions')
    parser.add_argument('-f', '--flatten', action="store_true", help='Activate flattening on model')
    parser.add_argument('-c', '--chained', action="store_true", help='Use chained coordinator')

    args = parser.parse_args()

    if args.model_type not in MODEL_TYPES:
        raise RuntimeError("Unrecognized model type.")

    return args


if __name__ == '__main__':

    args = parse_args()

    devstone_model = None

    start_time = time.time()
    if args.model_type == "LI":
        devstone_model = LI("LI_root", args.depth, args.width, args.int_cycles, args.ext_cycles)
    elif args.model_type == "HI":
        devstone_model = HI("HI_root", args.depth, args.width, args.int_cycles, args.ext_cycles)
    elif args.model_type == "HO":
        devstone_model = HO("HO_root", args.depth, args.width, args.int_cycles, args.ext_cycles)
    elif args.model_type == "HOmod":
        devstone_model = HOmod("HOmod_root", args.depth, args.width, args.int_cycles, args.ext_cycles)

    env = DEVStoneEnvironment("DEVStoneEnvironment", devstone_model)
    model_created_time = time.time()

    sim = Simulator(env)
    sim.setVerbose(None)
    # sim.setTerminationTime(10.0)
    # sim.setStateSaving("custom")
    engine_setup_time = time.time()

    sim.simulate()
    sim_time = time.time()

    print("Model creation time: {}".format(model_created_time - start_time))
    print("Engine setup time: {}".format(engine_setup_time - model_created_time))
    print("Simulation time: {}".format(sim_time - engine_setup_time))