#!/usr/bin/env python3
"""Capture sampled legacy FI/TN field-term costs for the four owner views."""
from __future__ import annotations

import json
import os
import re
import subprocess
import time
from pathlib import Path


ROOT = Path(__file__).resolve().parents[4]
OUT = ROOT / os.environ.get("VOXELFORGE_L76_PROFILE_OUT",
                            "Saved/Linux/l76/term-profile-before")
DATA = ROOT / "Standalone/data/owner_world.json"
BIN = ROOT / "Standalone/bin/linux/VoxelForgeCoreHeadless-l76-termprofile-clang"
CAMERAS = {
    0: (8.0, 16.0, -372.2626953125),
    1: (25.8369140625, 3.19921875, -737.0),
    2: (19.166015625, -14.2001953125, -753.3740234375),
    3: (-20.0625, -5.068359375, -900.353515625),
}
KV = re.compile(r"([a-zA-Z0-9_]+)=([^ ]+)")


def run(strate: int) -> dict:
    OUT.mkdir(parents=True, exist_ok=True)
    log = OUT / f"strate{strate}-w8.log"
    camera = ",".join(f"{value:.12f}" for value in CAMERAS[strate]) + ",0,0"
    command = [
        "taskset", "-c", "0-11", "prlimit", "--as=6442450944", "--", "timeout",
        "1740s", str(BIN), str(DATA), "--world-assembler-prototype", "--stream-benchmark",
        "--stream-pipeline", "--force-generate", "--workers=8", "--clip-radius=3",
        "--max-clip-level=8", "--max-level=8", f"--camera={camera}",
        "--profile-lod-work", "--instrument-tile-drain", "--profile-passage-composer",
    ]
    started = time.monotonic()
    with log.open("w", encoding="utf-8") as output:
        result = subprocess.run(command, cwd=ROOT, stdout=output,
                                stderr=subprocess.STDOUT, check=False, timeout=1740)
    rows: dict[int, dict[str, dict]] = {}
    pipeline = {}
    with log.open(encoding="utf-8", errors="replace") as stream:
        for line in stream:
            if line.startswith("[WorldAssemblerLODLegacyField]"):
                row = dict(KV.findall(line))
                rows.setdefault(int(row["lod"]), {})["field"] = row
            elif line.startswith("[WorldAssemblerLODLegacyTerm]"):
                row = dict(KV.findall(line))
                rows.setdefault(int(row["lod"]), {}).setdefault("terms", {})[row["term"]] = row
            elif line.startswith("[WorldAssemblerLODClassification]"):
                row = dict(KV.findall(line))
                rows.setdefault(int(row["lod"]), {})["classification"] = row
            elif line.startswith("[WorldAssemblerLODWorker]"):
                row = dict(KV.findall(line))
                rows.setdefault(int(row["lod"]), {})["worker"] = row
            elif line.startswith("[StreamPipeline]"):
                pipeline = dict(KV.findall(line))
    if result.returncode != 0 or not pipeline or not rows:
        raise SystemExit(f"profile run failed or summary missing; see {log}")
    row = {
        "strate": strate,
        "exit_code": result.returncode,
        "process_wall_s": round(time.monotonic() - started, 3),
        "whole_view_ready_s": float(pipeline["whole_view_ready"].removesuffix("s")),
        "whole_tiles": int(pipeline["whole_tiles"].split("/")[0]),
        "lod": rows,
        "log": str(log.relative_to(ROOT)),
    }
    (OUT / f"strate{strate}-w8.json").write_text(json.dumps(row, indent=2) + "\n",
                                                   encoding="utf-8")
    print("[L76TermProfileRun] " + json.dumps({k: v for k, v in row.items() if k != "lod"},
                                               sort_keys=True), flush=True)
    return row


def main() -> int:
    missing = [str(path) for path in (DATA, BIN) if not path.is_file()]
    if missing:
        raise SystemExit("missing input: " + ", ".join(missing))
    rows = [run(strate) for strate in range(4)]
    (OUT / "summary.json").write_text(json.dumps(rows, indent=2) + "\n",
                                        encoding="utf-8")
    return 0


if __name__ == "__main__":
    raise SystemExit(main())