"""Burst snapshot: capture N frames over T seconds, save each as PNG.

Useful for catching the ball in flight and getting a clean PCI frame for

tuning / template extraction. Saves to logs/burst_NN_tXXXXXXms.png.

"""

from __future__ import annotations

import argparse

import shutil

import sys

import time

from pathlib import Path

import cv2

from rich.console import Console

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

from capture.ingest import open_capture

from cv._common import load_config

console = Console()

OUT_DIR = Path(__file__).resolve().parents[1] / "logs" / "burst"

def main() -> int:

    ap = argparse.ArgumentParser(description="Burst frame capture.")

    ap.add_argument("--count", type=int, default=30, help="Total frames to save.")

    ap.add_argument("--duration", type=float, default=10.0, help="Spread frames over this many seconds.")

    ap.add_argument("--clear", action="store_true", help="Wipe logs/burst before starting.")

    args = ap.parse_args()

    if args.clear and OUT_DIR.exists():

        shutil.rmtree(OUT_DIR)

    OUT_DIR.mkdir(parents=True, exist_ok=True)

    cfg = load_config()

    cap = open_capture(cfg)

    if cap is None or not cap.isOpened():

        console.print("[red]Could not open capture card.[/red]")

        return 2

    for _ in range(5):

        cap.read()

    interval = args.duration / max(args.count - 1, 1)

    t0 = time.perf_counter()

    console.print(

        f"[bold cyan]Burst: {args.count} frames over {args.duration:.1f}s "

        f"(every {interval*1000:.0f}ms) -> {OUT_DIR}[/bold cyan]"

    )

    saved = 0

    for i in range(args.count):

        target = t0 + i * interval

        while time.perf_counter() < target - 0.002:

            cap.read()

        ok, frame = cap.read()

        if not ok or frame is None:

            continue

        elapsed_ms = int((time.perf_counter() - t0) * 1000)

        out = OUT_DIR / f"burst_{i:02d}_t{elapsed_ms:05d}ms.png"

        cv2.imwrite(str(out), frame)

        saved += 1

        if i % 5 == 0:

            console.print(f"  [{i+1}/{args.count}] t={elapsed_ms}ms  -> {out.name}")

    cap.release()

    console.print(f"[green]Saved {saved}/{args.count} frames to {OUT_DIR}[/green]")

    return 0

if __name__ == "__main__":

    sys.exit(main())