first commit

simulations/demo/simple_tracker.py

@@ -0,0 +1,92 @@
+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
+    )
+)