scripts/reproduce_tables.py

from __future__ import annotations

import shutil
import subprocess
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
DATA = ROOT.parent / 'SimplexTasks-12-data'
RESULTS = ROOT / 'results' / 'tables'
OUTPUTS = ROOT / 'outputs' / 'tables'
PAPER = ROOT / 'paper' / 'rewrite_2026' / 'latex'


def stage_inputs() -> None:
    if not DATA.exists():
        raise SystemExit('SimplexTasks-12-data bundle not found next to the code bundle')
    RESULTS.mkdir(parents=True, exist_ok=True)
    OUTPUTS.mkdir(parents=True, exist_ok=True)
    PAPER.mkdir(parents=True, exist_ok=True)
    for subdir in ['figure_inputs', 'table_inputs']:
        src_dir = DATA / 'data' / 'summaries' / subdir
        if not src_dir.exists():
            continue
        for src in src_dir.iterdir():
            if src.is_file():
                shutil.copy2(src, RESULTS / src.name)


def main() -> None:
    stage_inputs()
    commands = [
        [sys.executable, str(ROOT / 'scripts' / 'make_tables.py'), '--results-dir', str(RESULTS), '--output-dir', str(OUTPUTS)],
        [
            sys.executable,
            str(ROOT / 'scripts' / 'summarize_real_strata_sensitivity.py'),
            '--input-dir', str(RESULTS),
            '--detail-csv', str(OUTPUTS / 'real_strata_sensitivity_detail.csv'),
            '--winner-json', str(OUTPUTS / 'real_strata_sensitivity_winners.json'),
            '--winner-md', str(OUTPUTS / 'real_strata_sensitivity_summary.md'),
            '--winner-tex', str(OUTPUTS / 'generated_real_strata_sensitivity.tex'),
        ],
        [sys.executable, str(ROOT / 'scripts' / 'make_appendix_tables.py')],
    ]
    for cmd in commands:
        subprocess.check_call(cmd, cwd=ROOT)
    appendix_tex = PAPER / 'generated_appendix_tables.tex'
    if appendix_tex.exists():
        shutil.copy2(appendix_tex, OUTPUTS / appendix_tex.name)


if __name__ == '__main__':
    main()