scripts/task_overview.py

"""Tile every task's opening frame into one high-resolution overview image.

Each cell keeps its full render resolution -- the point of this sheet is to be able to zoom in on
any single scene, so it is deliberately NOT downscaled to fit a screen.

    python scripts/task_overview.py                       # -> outputs/frames/overview.png
    python scripts/task_overview.py --cols 6 --scale 1.0
"""
import argparse
import glob
import os

from PIL import Image, ImageDraw, ImageFont

ap = argparse.ArgumentParser()
ap.add_argument("--frames", default="outputs/frames")
ap.add_argument("--out", default="outputs/frames/overview.png")
ap.add_argument("--cols", type=int, default=6)
ap.add_argument("--scale", type=float, default=1.0, help="per-cell scale; 1.0 keeps full res")
ap.add_argument("--pad", type=int, default=10)
ap.add_argument("--label_h", type=int, default=46)
a = ap.parse_args()

paths = sorted(p for p in glob.glob(os.path.join(a.frames, "*.png"))
               if os.path.basename(p) != os.path.basename(a.out))
if not paths:
    raise SystemExit(f"no frames in {a.frames} -- run yam_task.py --first-frame first")

first = Image.open(paths[0])
cw, ch = int(first.width*a.scale), int(first.height*a.scale)
cols = min(a.cols, len(paths))
rows = (len(paths)+cols-1)//cols
W = cols*cw+(cols+1)*a.pad
H = rows*(ch+a.label_h)+(rows+1)*a.pad

sheet = Image.new("RGB", (W, H), (17, 17, 20))
draw = ImageDraw.Draw(sheet)
try:
    font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf",
                              max(16, int(cw*0.045)))
except OSError:
    font = ImageFont.load_default()

for i, p in enumerate(paths):
    r, c = divmod(i, cols)
    x = a.pad+c*(cw+a.pad)
    y = a.pad+r*(ch+a.label_h+a.pad)
    im = Image.open(p).convert("RGB")
    if (im.width, im.height) != (cw, ch):
        im = im.resize((cw, ch), Image.LANCZOS)
    sheet.paste(im, (x, y))
    name = os.path.splitext(os.path.basename(p))[0]
    draw.text((x+6, y+ch+8), f"{i+1:2d}. {name}", fill=(245, 226, 120), font=font)

sheet.save(a.out)
print(f"[overview] {a.out}  {sheet.width} x {sheet.height} px, "
      f"{len(paths)} tasks in {cols} x {rows}")