from decimal import Decimal
import numpy as np
from datetime import timedelta

from SimCAD import configs
from SimCAD.configuration import Configuration
from SimCAD.configuration.utils import exo_update_per_ts, proc_trigger, bound_norm_random, \
    ep_time_step

seed = {
}

# Behaviors
def tracker(step, sL, s):
    currentSignal = s['signal']
    currentFollow = s['follow']
    dif = currentSignal - currentFollow
    return {'value': dif}

# Mechanisms
def add(step, sL, s, _input):
    y = 'follow'
    x = s['follow'] + _input['value']
    return (y, x)

# Exogenous States
# amplitude = 0.7
period = 50
def sinusoid(step, sL, s, _input):
    y = 'signal'
    x = np.sin(s['elapsed_time'] * 2 * np.pi / period)
    return (y, x)

def delta_time(step, sL, s, _input):
    y = 'elapsed_time'
    x = s['elapsed_time'] + t_delta.seconds
    return (y, x)

ts_format = '%Y-%m-%d %H:%M:%S'
t_delta = timedelta(days=0, minutes=0, seconds=1)
def time_model(step, sL, s, _input):
    y = 'timestamp'
    x = ep_time_step(s, dt_str=s['timestamp'], fromat_str=ts_format, _timedelta=t_delta)
    return (y, x)

# Genesis States
genesis_states = {
    'signal': float(0),
    'follow': float(0),
    'elapsed_time': float(0),
    'timestamp': '2018-01-01 00:00:00'
}

# remove `exo_update_per_ts` to update every ts
exogenous_states = exo_update_per_ts(
    {
    "signal": sinusoid,
    'elapsed_time': delta_time,
    "timestamp": time_model
    }
)

#	make env proc trigger field agnostic
env_processes = {
}

mechanisms = {
    "add": {
        "behaviors": {
            "tracker": tracker
        },
        "states": { 
            "follow": add
        }
    }
}

sim_config = {
    "N": 1,
    "T": range(50)
}

configs.append(
    Configuration(
        sim_config=sim_config,
        state_dict=genesis_states,
        seed=seed,
        exogenous_states=exogenous_states,
        env_processes=env_processes,
        mechanisms=mechanisms
    )
)