"""Run ball and PCI detectors on a single saved frame and save annotated PNG."""

from __future__ import annotations

import sys

from pathlib import Path

import cv2

from rich.console import Console

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

from cv._common import load_config

from cv.ball_tracker import detect_ball

from cv.pci_tracker import detect_by_hsv as pci_detect

console = Console()

def main() -> int:

    if len(sys.argv) < 2:

        console.print("usage: detect_on_frame.py <frame.png>")

        return 2

    fp = Path(sys.argv[1])

    frame = cv2.imread(str(fp), cv2.IMREAD_COLOR)

    if frame is None:

        console.print(f"[red]Can't read {fp}[/red]")

        return 2

    cfg = load_config()

    overlay = frame.copy()

    h, w = frame.shape[:2]

    plate_y = int(h * float(cfg["cv"].get("plate_y_frac", 0.72)))

    cv2.line(overlay, (0, plate_y), (w, plate_y), (255, 255, 255), 1)

    ball = detect_ball(frame, cfg)

    if ball is not None:

        cv2.circle(overlay, (int(ball.x), int(ball.y)), max(int(ball.r), 4), (0, 255, 255), 2)

        cv2.putText(overlay, f"ball r={ball.r:.0f} s={ball.score:.2f}",

                    (int(ball.x) + 10, int(ball.y) - 10),

                    cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 255), 2, cv2.LINE_AA)

        console.print(f"ball: x={ball.x:.0f} y={ball.y:.0f} r={ball.r:.1f} score={ball.score:.2f}")

    else:

        console.print("ball: none")

    pci = pci_detect(frame, cfg["cv"]["pci"])

    if pci is not None:

        cv2.circle(overlay, (int(pci["x"]), int(pci["y"])), int(pci["r"]), (0, 255, 0), 2)

        cv2.putText(overlay, f"pci r={pci['r']:.0f} s={pci['score']:.2f}",

                    (int(pci["x"]) + 10, int(pci["y"]) + 20),

                    cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2, cv2.LINE_AA)

        console.print(f"pci:  x={pci['x']:.0f} y={pci['y']:.0f} r={pci['r']:.1f} score={pci['score']:.2f}")

    else:

        console.print("pci:  none")

    out = Path(__file__).resolve().parents[1] / "logs" / f"detect_{fp.stem}.png"

    cv2.imwrite(str(out), overlay)

    console.print(f"saved -> {out}")

    return 0

if __name__ == "__main__":

    sys.exit(main())