RustPhysicsMQ/configurations/gen.py

import os
import subprocess

FOLDER_PREFIX = "configurations/"
MOT_FOLDER = FOLDER_PREFIX + "mot/"
SIM_FOLDER = FOLDER_PREFIX + "sim/"
PID_FOLDER = FOLDER_PREFIX + "pid_cont/"
FINAL_FOLDER = FOLDER_PREFIX + "final/"

subprocess.run(f"rm -rf {FINAL_FOLDER} && mkdir {FINAL_FOLDER}", shell=True)


mot_files = os.listdir(MOT_FOLDER)
sim_files = os.listdir(SIM_FOLDER)
pid_files = os.listdir(PID_FOLDER)

prod: list[tuple[str, str, str]] = [
    (y, x, z) for x in mot_files for y in sim_files for z in pid_files
]

for case in prod:
    settings = ""
    with open(SIM_FOLDER + case[0], "r") as f1:
        with open(MOT_FOLDER + case[1], "r") as f2:
            with open(PID_FOLDER + case[2], "r") as f3:
                settings = f1.read() + "\n" + f2.read() + "\n" + f3.read()
    with open(
        FINAL_FOLDER + "".join(case).replace(".toml", "_").removesuffix("_") + ".toml",
        "w",
    ) as f:
        f.write(settings)