#!/usr/bin/env python3
"""L80 Surface World A/A and no-op evaluator-version control timings."""
from __future__ import annotations

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

ROOT = Path(__file__).resolve().parents[4]
OUT = ROOT / "Saved/Linux/l137/timing/aa"
DATA = ROOT / "Standalone/data/owner_world.json"
BINARIES = {
    "aa_a": ROOT / "Saved/Linux/l137/bin/VoxelForgeCoreHeadless-l137-aa-a",
    "aa_b": ROOT / "Saved/Linux/l137/bin/VoxelForgeCoreHeadless-l137-aa-b",
    "v228": ROOT / "Saved/Linux/l137/bin/VoxelForgeCoreHeadless-l137-v228",
}
CAMERA = "8.000000000000,16.000000000000,-372.262695312500,0,0"
PIPELINE = re.compile(r"^\[StreamPipeline\] (.*)$", re.M)
FIELDS = re.compile(r"([a-zA-Z0-9_]+)=([^ ]+)")
NEAR = re.compile(
    r"^\[WorldAssemblerNearLODReady\] lods=0-2 tiles=(\d+)/(\d+) cold_ready_s=([0-9.]+)$",
    re.M)
STAMP = re.compile(r"^\[VoxelForgeSourceStamp\] sha256=([0-9a-f]+)$", re.M)


def run(workers: int, version: str, phase: str, replicate: int) -> dict:
    OUT.mkdir(parents=True, exist_ok=True)
    label = f"sw-w{workers}-{version}-{phase}{replicate}"
    log = OUT / f"{label}.log"
    command = [
        "taskset", "-c", "0-11", "prlimit", "--as=6442450944", "--",
        "timeout", "1740s", str(BINARIES[version]), str(DATA),
        "--world-assembler-prototype", "--stream-benchmark", "--stream-pipeline",
        f"--workers={workers}", "--clip-radius=3", "--max-clip-level=8",
        "--max-level=8", f"--camera={CAMERA}", "--profile-lod-work",
        "--stream-scheduler-policy=3",
    ]
    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)
    content = log.read_text(encoding="utf-8", errors="replace")
    pipeline, near, stamp = PIPELINE.search(content), NEAR.search(content), STAMP.search(content)
    if result.returncode or pipeline is None or near is None or stamp is None:
        raise SystemExit(f"L80 run failed or trace missing: {log}; exit={result.returncode}")
    fields = dict(FIELDS.findall(pipeline.group(1)))
    row = {
        "strate": 0, "workers": workers, "version": version,
        "phase": phase, "replicate": replicate, "source_stamp": stamp.group(1),
        "lod02_ready_s": float(near.group(3)),
        "whole_view_ready_s": float(fields["whole_view_ready"].removesuffix("s")),
        "whole_tiles": int(fields["whole_tiles"].split("/")[0]),
        "wall_s": round(time.monotonic() - started, 3),
        "log": str(log.relative_to(ROOT)),
    }
    print("[L137AARun] " + json.dumps(row, sort_keys=True), flush=True)
    return row


def quiet_machine() -> tuple[float, float]:
    load_1m = os.getloadavg()[0]

    def totals() -> tuple[int, int]:
        fields = Path("/proc/stat").read_text().splitlines()[0].split()
        values = [int(value) for value in fields[1:]]
        return sum(values), values[3] + values[4]

    before = totals()
    time.sleep(2.0)
    after = totals()
    idle = 100.0 * (after[1] - before[1]) / max(1, after[0] - before[0])
    return load_1m, idle


def main() -> int:
    missing = [str(path) for path in [DATA, *BINARIES.values()] if not path.is_file()]
    if missing:
        raise SystemExit("missing L137 A/A input: " + ", ".join(missing))
    OUT.mkdir(parents=True, exist_ok=True)
    load_1m, idle = quiet_machine()
    (OUT / "machine-load-before.txt").write_text(
        f"1-minute load: {load_1m:.2f}\n2-second CPU idle: {idle:.1f}%\n")
    print(f"[L137AAGate] load_1m={load_1m:.2f} cpu_idle_2s={idle:.1f}%", flush=True)
    if load_1m >= 0.9 or idle < 97.0:
        raise SystemExit("timing refused: machine is above the L80 quiet-load gate")

    rows = []
    versions = tuple(BINARIES)
    for workers in (8, 12):
        for version in versions:
            run(workers, version, "warmup", 0)
        orders = (versions, ("v228", "aa_b", "aa_a"),
                  ("aa_b", "aa_a", "v228"), ("aa_a", "v228", "aa_b"),
                  ("v228", "aa_a", "aa_b"))
        for replicate, order in enumerate(orders, 1):
            for version in order:
                rows.append(run(workers, version, "measured", replicate))
                (OUT / "runs-partial.json").write_text(
                    json.dumps(rows, indent=2) + "\n", encoding="utf-8")

    summary = []
    for workers in (8, 12):
        group = [row for row in rows if row["workers"] == workers]
        medians = {
            version: {
                key: statistics.median(row[key] for row in group
                                       if row["version"] == version)
                for key in ("lod02_ready_s", "whole_view_ready_s")
            }
            for version in versions
        }
        item = {"strate": 0, "workers": workers, "replicates_per_version": 5,
                "medians": medians}
        for metric in ("lod02_ready_s", "whole_view_ready_s"):
            for lhs, rhs in (("aa_b", "aa_a"), ("v228", "aa_a")):
                item[f"{lhs}_vs_{rhs}_{metric}_delta_pct"] = round(
                    100.0 * (medians[lhs][metric] / medians[rhs][metric] - 1.0), 3)
            paired = []
            for replicate in range(1, 6):
                values = {row["version"]: row[metric] for row in group
                          if row["phase"] == "measured" and row["replicate"] == replicate}
                paired.append({"replicate": replicate,
                               "aa_b_vs_aa_a_pct": round(
                                   100.0 * (values["aa_b"] / values["aa_a"] - 1.0), 3),
                               "v228_vs_aa_a_pct": round(
                                   100.0 * (values["v228"] / values["aa_a"] - 1.0), 3)})
            item[f"{metric}_paired_deltas_pct"] = paired
        summary.append(item)
    (OUT / "summary.json").write_text(
        json.dumps({"runs": rows, "summary": summary}, indent=2) + "\n",
        encoding="utf-8")
    return 0


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