"""Scan every burst frame for PCI-like green pixels; report per-frame counts."""

from __future__ import annotations

import sys

from pathlib import Path

import cv2

import numpy as np

from rich.console import Console

console = Console()

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

def main() -> int:

    lo = np.array([45, 140, 140], np.uint8)

    hi = np.array([75, 255, 255], np.uint8)

    frames = sorted(BURST_DIR.glob("burst_*.png"))

    console.print(f"scanning {len(frames)} frames with HSV {list(lo)}..{list(hi)}")

    hits = []

    for f in frames:

        img = cv2.imread(str(f))

        if img is None:

            continue

        hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

        h, w = img.shape[:2]

        mask = cv2.inRange(hsv, lo, hi)

        window = np.zeros_like(mask)

        window[int(0.3 * h):int(0.75 * h), int(0.35 * w):int(0.65 * w)] = 255

        mask = cv2.bitwise_and(mask, window)

        n = int((mask > 0).sum())

        if n >= 100:

            hits.append((f.name, n))

    hits.sort(key=lambda x: -x[1])

    console.print(f"[bold]Found {len(hits)} frames with >=100 central green px:[/bold]")

    for name, n in hits[:15]:

        console.print(f"  {name}  {n} px")

    return 0

if __name__ == "__main__":

    sys.exit(main())