from copy import deepcopy
from fn import op, _

def getColResults(step, sL, s, funcs):
    return list(map(lambda f: f(step, sL, s), funcs))


def getBehaviorInput(step, sL, s, funcs):
    return op.foldr(_ + _)(getColResults(step, sL, s, funcs))


def apply_env_proc(env_processes, state_dict, step):
    for state in state_dict.keys():
        if state in list(env_processes.keys()):
            state_dict[state] = env_processes[state](step)(state_dict[state])


def mech_step(m_step, sL, state_funcs, behavior_funcs, env_processes, t_step):
    in_copy, mutatable_copy, out_copy = deepcopy(sL), deepcopy(sL), deepcopy(sL)
    last_in_obj, last_mut_obj = in_copy[-1], mutatable_copy[-1]

    _input = getBehaviorInput(m_step, sL, last_in_obj, behavior_funcs)

    # OLD: no bueno! Mutation Bad
    # for f in state_funcs:
    #     f(m_step, sL, last_mut_obj, _input)

    # New
    last_mut_obj = dict([ f(m_step, sL, last_mut_obj, _input) for f in state_funcs ])

    apply_env_proc(env_processes, last_mut_obj, last_mut_obj['timestamp'])

    last_mut_obj["mech_step"], last_mut_obj["time_step"] = m_step, t_step
    out_copy.append(last_mut_obj)

    del last_in_obj, last_mut_obj, in_copy, mutatable_copy,

    return out_copy


def block_gen(states_list, configs, env_processes, t_step):
    m_step = 0
    states_list_copy = deepcopy(states_list)
    genesis_states = states_list_copy[-1]
    genesis_states['mech_step'], genesis_states['time_step'] = m_step, t_step
    states_list = [genesis_states]

    m_step += 1
    for config in configs:
        s_conf, b_conf = config[0], config[1]
        states_list = mech_step(m_step, states_list, s_conf, b_conf, env_processes, t_step)
        m_step += 1

    t_step += 1

    return states_list


def pipeline(states_list, configs, env_processes, time_seq):
    time_seq = [x + 1 for x in time_seq]
    simulation_list = [states_list]
    for time_step in time_seq:
        pipeline_run = block_gen(simulation_list[-1], configs, env_processes, time_step)
        head, *pipeline_run = pipeline_run
        simulation_list.append(pipeline_run)

    return simulation_list


def simulation(states_list, configs, env_processes, time_seq, runs):
    pipeline_run = []
    for run in range(runs):
        if run == 0:
            head, *tail = pipeline(states_list, configs, env_processes, time_seq)
            head[-1]['mech_step'], head[-1]['time_step'] = 0, 0
            simulation_list = [head] + tail
            pipeline_run += simulation_list
        else:
            transient_states_list = [pipeline_run[-1][-1]]
            head, *tail = pipeline(transient_states_list, configs, env_processes, time_seq)
            pipeline_run += tail

    return pipeline_run