"""Crop a region of a frame at native pixel resolution and save it."""

from __future__ import annotations

import sys

from pathlib import Path

import cv2

from rich.console import Console

console = Console()

def main() -> int:

    if len(sys.argv) < 6:

        console.print("usage: crop.py <in.png> <x> <y> <w> <h> [out.png]")

        return 2

    inp = Path(sys.argv[1])

    x, y, w, h = int(sys.argv[2]), int(sys.argv[3]), int(sys.argv[4]), int(sys.argv[5])

    out = Path(sys.argv[6]) if len(sys.argv) >= 7 else (

        Path(__file__).resolve().parents[1] / "logs" / f"crop_{inp.stem}_{x}_{y}_{w}x{h}.png"

    )

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

    if frame is None:

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

        return 2

    H, W = frame.shape[:2]

    x0, y0 = max(0, x), max(0, y)

    x1, y1 = min(W, x + w), min(H, y + h)

    if x1 <= x0 or y1 <= y0:

        console.print(f"[red]Invalid crop: image is {W}x{H}[/red]")

        return 2

    crop = frame[y0:y1, x0:x1]

    cv2.imwrite(str(out), crop)

    console.print(f"saved {crop.shape[1]}x{crop.shape[0]} -> {out}")

    return 0

if __name__ == "__main__":

    sys.exit(main())