capture.py

Part of Clear Code Zelda.

#!/usr/bin/env python3
"""Capture headless screenshots at frames 30, 60, 120 for review.

Run from the repo root:
    uv run python examples/ports/clear_code_zelda/capture.py
"""

from __future__ import annotations

import sys
from pathlib import Path

_PORT_DIR = Path(__file__).resolve().parent
if str(_PORT_DIR) not in sys.path:
    sys.path.insert(0, str(_PORT_DIR))

from main import ZeldaRoot  # noqa: E402
from settings import HEIGHT, WIDTH  # noqa: E402

from simvx.graphics import App, save_png  # noqa: E402


def main():
    out = _PORT_DIR / "screenshots"
    out.mkdir(exist_ok=True)
    targets = [30, 60, 120]
    app = App(width=WIDTH, height=HEIGHT, title="Zelda capture", visible=False)
    frames = app.run_headless(ZeldaRoot(show_menu=False), frames=max(targets) + 1, capture_frames=targets)
    for idx, frame in zip(targets, frames, strict=False):
        path = out / f"frame_{idx:03d}.png"
        save_png(frame, str(path))
        print(f"saved {path}  shape={frame.shape}")


if __name__ == "__main__":
    main()