scripts/make_example.py

"""Bake a synthetic multi-camera fly video into APP_TMP_DIR for example + smoke."""
from __future__ import annotations

import sys

import tifffile

from core.io import APP_TMP_DIR, register_protected
from core.synth import fly_multiview


def main() -> int:
    out = APP_TMP_DIR / "example.tif"
    frames, kp3d, coxa, cams, meta = fly_multiview(T=120, fps=20.0, seed=0)
    tifffile.imwrite(out, frames)
    register_protected(out)
    print(f"wrote {out}  shape={frames.shape}  cams={meta['n_cams']}  keypoints={len(meta['names'])}")
    return 0


if __name__ == "__main__":
    sys.exit(main())