# scripts/make_toy_sensor.py
import numpy as np, pandas as pd

rng = np.random.default_rng(1)
rows = []
for subj in range(3):
    T = 400
    acc_x = rng.normal(0, 0.8, size=T)
    acc_y = rng.normal(0, 0.8, size=T)
    acc_z = rng.normal(0, 1.2, size=T)
    # 伪标签：0=rest,1=walk,2=run
    activity = np.zeros(T, dtype=object)
    activity[100:250] = "walk"
    activity[250:350] = "run"
    # 奖励表达式将惩罚加速度幅度（越平稳越好）
    for t in range(T):
        rows.append(dict(
            subject=f"S{subj}", frame=t,
            acc_x=acc_x[t], acc_y=acc_y[t], acc_z=acc_z[t],
            activity=activity[t]
        ))
df = pd.DataFrame(rows)
df.to_csv("./data/toy_sensor.csv", index=False)
print("[OK] wrote ./data/toy_sensor.csv with", len(df), "rows")