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

added logging in preprocessing2

parent 6d37df12

ivas_processing_scripts/audiotools/wrappers/bs1770.py

+1 −1
Original line number Diff line number Diff line
@@ -252,7 +252,7 @@ def loudness_norm( 
    # log loudness before and after adjustment

    if logger:

        if file_name_logging:

            logger.debug(f"File {file_name_logging} loudness; before: {loudness_before}, after: {loudness_after}")

            logger.info(f"File {file_name_logging} loudness; before: {loudness_before}, after: {loudness_after}")

        else:

            logger.debug(f"Loudness; before: {loudness_before}, after: {loudness_after}")

ivas_processing_scripts/processing/preprocessing_2.py

+13 −3
Original line number Diff line number Diff line
@@ -66,6 +66,7 @@ class Preprocessing2(Processing): 


        # add preamble

        if self.preamble:

            logger.debug(f"Add preamble of length {self.preamble}ms")

            # also apply preamble to ISM metadata

            if self.in_fmt.startswith("ISM"):

                # read out old
@@ -91,14 +92,21 @@ class Preprocessing2(Processing): 


        # add background noise

        if self.background_noise:

            audio_object.audio = self.add_background_noise(audio_object, in_meta)

            logger.debug(f"Add background noise from file {self.background_noise.get('background_noise_path', 'file missing')} and SNR {self.background_noise.get('snr', 'snr missing')}")

            audio_object.audio = self.add_background_noise(audio_object, in_meta, logger)



        # save file

        write(out_file, audio_object.audio, fs=audio_object.fs)



        return



    def add_background_noise(self, audio_object: audio.Audio, in_meta) -> np.ndarray:

    def add_background_noise(self, audio_object: audio.Audio, in_meta, logger) -> np.ndarray:

        # check if SNR and background noise are given

        if not self.background_noise["background_noise_path"] or not Path(self.background_noise["background_noise_path"]).exists():

            raise ValueError("Background noise does not exist")

        if not self.background_noise["snr"]:

            raise ValueError("SNR value needed for using background noise")



        # range for random delay

        range_delay = (1, 2400000)
@@ -133,6 +141,7 @@ class Preprocessing2(Processing): 
            out_format = self.out_fmt



        loudness_signal, _ = get_loudness(audio_object, loudness_format=out_format)

        logger.debug(f"Loudness of audio signal: {loudness_signal}LKFS")



        # compute desired loudness of background noise

        loudness_noise = loudness_signal - self.background_noise["snr"]
@@ -149,7 +158,8 @@ class Preprocessing2(Processing): 
        )[: len(audio_object.audio)]



        # scale background noise to desired loudness based on output format

        noise_object.audio = loudness_norm(noise_object, loudness_noise, out_format)

        logger.debug(f"Scaling of background noise to {self.background_noise['snr']}dB SNR")

        noise_object.audio = loudness_norm(noise_object, loudness_noise, out_format, logger)



        # add array to signal

        audio_object.audio = noise_object.audio + audio_object.audio

ivas_processing_scripts/processing/processing.py

+1 −0
Original line number Diff line number Diff line
@@ -324,6 +324,7 @@ def preprocess_2(cfg, logger): 
def reverse_process_2(cfg, logger):

    # remove preamble

    if cfg.pre2.preamble:

        logger.info("Remove preamble")

        remove_preamble(cfg)



    # reverse concatenation