1"""Export key numbers from results/tables as LaTeX macros (paper/tables/numbers.tex) so that prose never hard-codes values."""2from __future__ import annotations3import re, pandas as pd, numpy as np, json4from pathlib import Path5from ..util import ROOT, read_json67TAB = ROOT / "results/tables"; OUT = ROOT / "paper/tables/numbers.tex"89def _key(s: str) -> str:10 s = re.sub(r"[^A-Za-z0-9]", " ", s).title().replace(" ", "")11 return re.sub(r"\d", lambda m: "ZOTTFFSSEN"[int(m.group(0))], s) # LaTeX macro names cannot contain digits1213def main():14 lines = ["% auto-generated by wp30.analysis.numbers — do not edit"]15 def add(name, val, fmt="{:.2f}"):16 if val is None or (isinstance(val, float) and np.isnan(val)):17 lines.append(f"\\newcommand{{\\{name}}}{{--}}")18 else:19 lines.append(f"\\newcommand{{\\{name}}}{{{fmt.format(val) if not isinstance(val, str) else val}}}")20 prim = TAB / "E1_core_primary.csv"21 if prim.exists():22 t = pd.read_csv(prim)23 add("nCoreTrials", int(pd.read_csv(TAB / "E1_core_failures.csv").iloc[:, 1:].to_numpy().sum()) if (TAB / "E1_core_failures.csv").exists() else None, "{:,}")24 for _, r in t.iterrows():25 k = _key(f"{r['family']} {r['pair']} {r['condition']}")26 add("Q" + k, r["quality"] * 100, "{:.0f}"); add("Qlo" + k, r["q_lo"] * 100, "{:.0f}"); add("Qhi" + k, r["q_hi"] * 100, "{:.0f}")27 add("B" + k, r["payload_bytes_median"], "{:,.0f}"); add("Tcold" + k, r["cold_ms_p50"], "{:,.0f}"); add("Twarm" + k, r["warm_ms_p50"], "{:,.0f}")28 tests = pd.read_csv(TAB / "E1_core_primary_tests.csv") if (TAB / "E1_core_primary_tests.csv").exists() else None29 if tests is not None:30 for _, r in tests.iterrows():31 k = _key(r["name"].split(" ")[0])32 add("Diff" + k, r["diff"] * 100, "{:+.1f}"); add("Lo" + k, r["lo"] * 100, "{:+.1f}"); add("Hi" + k, r["hi"] * 100, "{:+.1f}"); add("P" + k, r["p_holm"], "{:.3f}"); add("N" + k, int(r["n"]), "{:d}")33 e6 = TAB / "E6_network.csv"34 if e6.exists():35 t6 = pd.read_csv(e6); t6 = t6[t6["label"] != "loopback"]36 for _, r in t6.iterrows():37 tag = _key(f"{r['method']} {'' if pd.isna(r['bits']) else int(r['bits'])} {r['codec'] if isinstance(r['codec'], str) else ''} {int(r['target_tokens'])}")38 add("Esix" + tag + "Rt", r["roundtrip_ms_p50"], "{:,.0f}"); add("Esix" + tag + "Bytes", r["bytes"] / 1e6 if not pd.isna(r["bytes"]) else None, "{:.0f}")39 add("Esix" + tag + "Send", r["send_ms_p50"], "{:,.0f}"); add("Esix" + tag + "Prefill", r["remote_prefill_p50"] if not pd.isna(r.get("remote_prefill_p50", float("nan"))) else r.get("local_prefill_p50"), "{:,.0f}")40 add("Esix" + tag + "Load", r["remote_load_ms_p50"], "{:,.0f}"); add("Esix" + tag + "Gbps", r["eff_gbps"], "{:.2f}")41 if (ROOT / "protocol/frozen.json").exists():42 add("protocolHash", read_json(ROOT / "protocol/frozen.json")["protocol_hash"][:16], "{}")43 OUT.parent.mkdir(parents=True, exist_ok=True); OUT.write_text("\n".join(lines) + "\n")44 print(f"{len(lines)-1} macros -> {OUT}")4546if __name__ == "__main__":47 main()48