Commit 75063657 authored by Fabian Müller's avatar Fabian Müller
Browse files

Copy missing reference files for P.800 preliminaries

parent bc8e3c2b
Loading
Loading
Loading
Loading
Loading
+16 −9
Original line number Diff line number Diff line
@@ -55,19 +55,26 @@ def copy_preliminaries(root_dir, config_yaml):

        for item in exp_cfg["preliminaries"]:
            cat = item["category"]
            cond = item["label"]
            # need to copy the reference files for every condition as well
            conds = ["c01", item["label"]]

            # Find the condition folder in proc_output
            for proc_out in proc_out_list:
                src = proc_out / f"{cat}*" / f"out_{LOUDNESS_MAP[cat]}LKFS" / cond
                for cond in conds:
                    src = proc_out / f"{cat}*" / f"out_{LOUDNESS_MAP[cat]}LKFS" / cond / "*s07.c*.wav"

                for file in glob.glob(str(src / "*s07.c*.wav")):
                    for file in glob.glob(str(src), recursive=True):
                        print(f"  Copying {file} -> {out_dir}")

                    if (out_dir / Path(file).name).is_file():
                        filename = Path(file).name

                        if (out_dir / filename).is_file():
                            if cond != "c01":
                                raise FileExistsError(
                                    f"File {file} already exists in {out_dir}"
                                )
                            else:
                                print(f"    c01 condition already copied, skipping: {filename}")

                        shutil.copy(file, out_dir)