import argparse
import os
from pathlib import Path
import subprocess
import sys

ROOT = Path(__file__).resolve().parent
ap = argparse.ArgumentParser(description="Plot the saved paper results")
ap.add_argument(
    "--only", choices=["all", "mb", "aib9_robustness", "p53"], default="all"
)
args = ap.parse_args()
commands = {
    "mb": [
        ["make_mb_panels.py", "--from-cache"],
        ["make_mb_unconditional.py", "--from-cache"],
    ],
    "aib9_robustness": [["replot_aib9_robustness.py"]],
    "p53": [
        ["paper_figures/plot/p53_discovery_plot.py"],
        ["paper_figures/plot/p53_preference_plot.py"],
    ],
}
env = os.environ.copy()
for name in ["OMP_NUM_THREADS", "OPENBLAS_NUM_THREADS", "MKL_NUM_THREADS"]:
    env[name] = "1"
env["MPLBACKEND"] = "Agg"
env["MPLCONFIGDIR"] = str(ROOT / ".mplconfig")
for group, jobs in commands.items():
    if args.only not in ["all", group]:
        continue
    for command in jobs:
        print("Running:", " ".join(command), flush=True)
        subprocess.run([sys.executable, *command], cwd=ROOT, env=env, check=True)