RustPhysicsMQ/configurations/gen.py

32 lines
934 B
Python

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)