#!/usr/bin/env python3
"""Pre-generate + cache synthetic scenes (renders GT with the rasterizer).

    python scripts/gen_synthetic.py --config configs/synthetic_smoke.yaml --split train --n-scenes 64
"""

import argparse

from mapgs.config import load_config
from mapgs.data import SyntheticDataset


def main():
    ap = argparse.ArgumentParser()
    ap.add_argument("--config", type=str, default="configs/synthetic_smoke.yaml")
    ap.add_argument("--override", nargs="*", default=[])
    ap.add_argument("--split", type=str, default="train")
    ap.add_argument("--n-scenes", type=int, default=None)
    ap.add_argument("--regenerate", action="store_true")
    args = ap.parse_args()

    cfg = load_config(args.config, args.override)
    ds = SyntheticDataset(cfg, split=args.split, n_scenes=args.n_scenes,
                          device=cfg.device, regenerate=args.regenerate)
    print(f"prepared {len(ds)} {args.split} scenes under {ds.root}")


if __name__ == "__main__":
    main()