SPB Git

spb/wp30_uqo

Public
2.3 KB · 40 lines python
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    if (ROOT / "protocol/frozen.json").exists():34        add("protocolHash", read_json(ROOT / "protocol/frozen.json")["protocol_hash"][:16], "{}")35    OUT.parent.mkdir(parents=True, exist_ok=True); OUT.write_text("\n".join(lines) + "\n")36    print(f"{len(lines)-1} macros -> {OUT}")3738if __name__ == "__main__":39    main()40
Files · e6d9a3b2e6aad04d133df41f962eaa0b3171272e
Menu

Theme