import math
from decimal import Decimal
from datetime import timedelta
import numpy as np
from typing import Dict, List

from cadCAD.configuration import append_configs
from cadCAD.configuration.utils import bound_norm_random, ep_time_step, config_sim, access_block

from .genesis_states import genesis_states
from .partial_state_update_block import partial_state_update_block

params: Dict[str, List[int]] = {
    'drip_frequency': [30,60,90] # in days
}


sim_config = config_sim({
    'N': 5,
    'T': range(100), #day 
    'M': params,
})

seeds = {
    'p': np.random.RandomState(26042019),
}
env_processes = {}


append_configs(
    sim_configs=sim_config,
    initial_state=genesis_states,
    seeds=seeds,
    env_processes=env_processes,
    partial_state_update_blocks=partial_state_update_block
)