import numpy as np
import sys
import os
sys.path.insert(0, os.path.abspath('.'))

from hippocampaif.core.dynamics import ContinuousDynamics

cd = ContinuousDynamics(dt=0.01)

cd.add_level(
    hidden_dim=2, causal_dim=1, output_dim=2,
    g_fn=lambda x, v, theta: x,  # Identity observation
    f_fn=lambda x, v, theta: -0.1 * x,  # Stable decay
    obs_precision=100.0, state_precision=100.0
)

# Set initial state
cd.levels[0].x.position = np.array([1.0, 2.0])

# Generate 10 steps
obs = cd.forward_generate(n_steps=10, add_noise=False)

for i, o in enumerate(obs):
    print(f"Step {i}: {o}, norm={np.linalg.norm(o)}")