Commit a4848584 authored by Anika Treffehn's avatar Anika Treffehn
Browse files

changed renaming of files at the end

parent 76c223cd

ivas_processing_scripts/__init__.py

+0 −5
Original line number Diff line number Diff line
@@ -48,7 +48,6 @@ from ivas_processing_scripts.processing.processing import ( 
    preprocess_2,

    preprocess_background_noise,

    process_item,

    rename_generated_conditions,

    reorder_items_list,

)

from ivas_processing_scripts.utils import DirManager, apply_func_parallel
@@ -182,10 +181,6 @@ def main(args): 
                "mp" if cfg.multiprocessing else None,

            )



        # rename output with condition name

        if cfg.condition_in_output_filename:

            rename_generated_conditions(cfg.output_path)



    # copy configuration to output directory

    with open(cfg.output_path.joinpath(f"{cfg.name}.yml"), "w") as f:

        yaml.safe_dump(cfg._yaml_dump, f)

ivas_processing_scripts/processing/chains.py

+1 −0
Original line number Diff line number Diff line
@@ -452,6 +452,7 @@ def get_processing_chain( 
                "loudness": post_cfg.get("loudness", None),

                "loudness_fmt": post_cfg.get("loudness_fmt", None),

                "tx_condition": tx_condition,

                "condition_in_output_filename": cfg.condition_in_output_filename,

            }

        )

    )

ivas_processing_scripts/processing/processing.py

+0 −18
Original line number Diff line number Diff line
@@ -533,21 +533,3 @@ def multiple_of_frame_size( 
                )



def rename_generated_conditions(output_path: Path):

    """

    Rename the output files. Only renames the files in directories that contain "cXX" in thier names.

    The "XX" in "cXX" stands for the condition number, for example "c01"



    Parameters

    ----------

    output_path: Path

        Path to output directory

    """

    directory = output_path

    pattern = re.compile(r"^c\d{2}")

    for subdirectory in directory.iterdir():

        if subdirectory.is_dir() and pattern.match(subdirectory.name):

            for file_path in subdirectory.iterdir():

                new_filename = f"{file_path.stem}.{subdirectory.name}{file_path.suffix}"

                file_path.rename(subdirectory / new_filename)

ivas_processing_scripts/processing/processing_splitting_scaling.py

+4 −1
Original line number Diff line number Diff line
@@ -80,6 +80,9 @@ class Processing_splitting_scaling(Processing): 
            regex_filter = r"[/\\][t][m][p][_]\w+[/\\]"

            tmp_name = re.search(regex_filter, str(f)).group().strip()[1:-1]

            out_name = tmp_name.replace("tmp_", "")

            if self.condition_in_output_filename:

                f_out = Path(str(f).replace(tmp_name, out_name).replace(".processing_splitting_scaling.", f".{out_name}."))

            else:

                f_out = Path(str(f).replace(tmp_name, out_name).replace(".processing_splitting_scaling.", "."))

            out_out_files.append(f_out)

            if meta_splits: