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

fixed ISM metadata preamble

parent ff88325e
Loading
Loading
Loading
Loading
Loading
+4 −8
Original line number Diff line number Diff line
@@ -40,7 +40,7 @@ from ivas_processing_scripts.audiotools import audio
from ivas_processing_scripts.audiotools.audio import fromtype
from ivas_processing_scripts.audiotools.audioarray import trim
from ivas_processing_scripts.audiotools.audiofile import read
from ivas_processing_scripts.audiotools.constants import IVAS_FRAME_LEN_MS, FORMAT_ISM_METADATA_CSV, NUMBER_COLUMNS_ISM_METADATA
from ivas_processing_scripts.audiotools.constants import IVAS_FRAME_LEN_MS, FORMAT_ISM_METADATA_CSV, NUMBER_COLUMNS_ISM_METADATA, DEFAULT_ISM_METADATA


class Metadata:
@@ -551,14 +551,10 @@ def add_remove_preamble(
    for obj_idx in range(len(metadata)):
        if metadata is not None and metadata[obj_idx] is not None:
            if add:
                metadata[obj_idx] = trim(
                    metadata[obj_idx],
                    limits=(-int(preamble_frames), 0),
                    samples=True,
                num_columns = metadata[obj_idx].shape[1]
                metadata[obj_idx] = np.vstack(
                    [np.repeat(np.array(DEFAULT_ISM_METADATA)[None, :num_columns], preamble_frames, 0), metadata[obj_idx]]
                )

                # add radius 1
                metadata[obj_idx][: int(preamble_frames), 2] = 1
            else:
                metadata[obj_idx] = trim(
                    metadata[obj_idx],