Commit f71a5ba5 authored by Jan Kiene's avatar Jan Kiene
Browse files

add silence_pre/post keys to config

this allows to add padding between files during concatenation
parent a0c18bdd
Loading
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -166,8 +166,8 @@ def write(
def concat(
    in_filenames: list,
    out_file: str,
    silence_pre: Optional[int] = 0,
    silence_post: Optional[int] = 0,
    silence_pre: int = 0,
    silence_post: int = 0,
    in_fs: Optional[int] = 48000,
    num_channels: Optional[int] = None,
    pad_noise: Optional[bool] = False,
+2 −0
Original line number Diff line number Diff line
@@ -201,6 +201,8 @@ def get_preprocessing_2(cfg: TestConfig) -> dict:
                "in_mask": pre2_cfg.get("mask", None),
                "multiprocessing": cfg.multiprocessing,
                "repeat_signal": pre2_cfg.get("repeat_signal", False),
                "silence_pre": pre2_cfg.get("silence_pre", 0),
                "silence_post": pre2_cfg.get("silence_post", 0),
            }
        )
    )
+9 −3
Original line number Diff line number Diff line
@@ -163,11 +163,17 @@ def concat_setup(cfg: TestConfig, chain, logger: logging.Logger):
    tmp_audio = audio.fromtype(cfg_pre2.in_fmt)
    tmp_num_chans = tmp_audio.num_channels

    # convert from milisecs to samples
    silence_pre = int(cfg_pre2.silence_pre / 1000 * cfg_pre2.in_fs)
    silence_post = int(cfg_pre2.silence_post / 1000 * cfg_pre2.in_fs)

    cfg.splits, fs = concat(
        cfg.items_list,
        cfg.concat_file,
        in_fs=cfg_pre2.in_fs,
        num_channels=tmp_num_chans,
        silence_pre=silence_pre,
        silence_post=silence_post,
    )

    # save item naming for splits naming in the end
@@ -633,8 +639,8 @@ def preprocess_background_noise(cfg):

    # save result in cfg
    # cfg.preprocessing_2["background_noise"].update({"background_object": output_audio})
    cfg.proc_chains[0]["processes"][0].background_noise[
        "background_object"
    ] = output_audio
    cfg.proc_chains[0]["processes"][0].background_noise["background_object"] = (
        output_audio
    )

    return