Commit 37eaa4bf authored by Archit Tamarapu's avatar Archit Tamarapu
Browse files

[fix] metadata/scene desc. and MASA tests

parent cbccc131

.gitignore

+1 −0
Original line number Diff line number Diff line
@@ -33,3 +33,4 @@ tests/temp_output_* 
tests/cut

tests/ref

tests/concatenation_folder

tests/data/testv

ivas_processing_scripts/audiotools/metadata.py

+4 −5
Original line number Diff line number Diff line
@@ -119,13 +119,12 @@ class Metadata: 
        start = int(f.readline().strip()) - 1

        masa_tc = int(f.readline().strip())



        masa = fromtype(f"MASA{masa_tc}")

        masa = fromtype(f"MASA{masa_tc}DIR1")  # assume DIR1

        masa.audio = self.audio_array[:, start : start + masa_tc]

        masa.fs = self.fs

        masa.metadata_files = [

            self.meta_file.parent.joinpath(f.readline().strip()).absolute()

        ]

        masa.init_metadata()

        masa.metadata_file = self.meta_file.parent.joinpath(

            f.readline().strip()

        ).absolute()



        self.audio.append(masa)

tests/constants.py

+17 −14
Original line number Diff line number Diff line
@@ -40,7 +40,7 @@ CODEC_TEST_VECTOR_DIR = TEST_VECTOR_DIR.joinpath("testv") 
EXPERIMENTS_DIR = "../experiments/selection"



ISM_METADATA_DIR = TEST_VECTOR_DIR.joinpath("ism_metadata")

MASA_METADATA_DIR = TEST_VECTOR_DIR.joinpath("masa")

MASA_METADATA_DIR = TEST_VECTOR_DIR.joinpath("masa_metadata")

SCENE_DESC_DIR = TEST_VECTOR_DIR.joinpath("scene_description")



OUTPUT_PATH_REF = TESTS_DIR.joinpath("ref")
@@ -113,8 +113,10 @@ FORMAT_TO_FILE = { 
    "ISM2": NCHAN_TO_FILE[2],

    "ISM3": NCHAN_TO_FILE[3],

    "ISM4": NCHAN_TO_FILE[4],

    "MASA1": NCHAN_TO_FILE[1],

    "MASA2": NCHAN_TO_FILE[2],

    "MASA1DIR1": NCHAN_TO_FILE[1],

    "MASA1DIR2": NCHAN_TO_FILE[1],

    "MASA2DIR1": NCHAN_TO_FILE[2],

    "MASA2DIR2": NCHAN_TO_FILE[2],

    # "MASA1": TEST_VECTOR_DIR.joinpath("stv_IVASMASA_1dir1TC.pcm"),

    # "MASA2": TEST_VECTOR_DIR.joinpath("stv_IVASMASA_2dir2TC.pcm"),

    "META": TEST_VECTOR_DIR.joinpath("mixed_scene.txt"),
@@ -146,15 +148,17 @@ FORMAT_TO_METADATA_FILES = { 
        str(ISM_METADATA_DIR.joinpath("stvISM3.csv")),

        str(ISM_METADATA_DIR.joinpath("stvISM4.csv")),

    ],

    "MASA1": [str(MASA_METADATA_DIR.joinpath("stv_IVASMASA_1dir1TC.met"))],

    "MASA2": [str(MASA_METADATA_DIR.joinpath("stv_IVASMASA_2dir2TC.met"))],

    "MASA1DIR1": [str(MASA_METADATA_DIR.joinpath("stv_IVASMASA_1dir1TC.met"))],

    "MASA1DIR2": [str(MASA_METADATA_DIR.joinpath("stv_IVASMASA_2dir1TC.met"))],

    "MASA2DIR1": [str(MASA_METADATA_DIR.joinpath("stv_IVASMASA_1dir2TC.met"))],

    "MASA2DIR2": [str(MASA_METADATA_DIR.joinpath("stv_IVASMASA_2dir2TC.met"))],

}



""" Input formats """

INPUT_FORMATS_AMBI = ["FOA", "HOA2", "HOA3"]

INPUT_FORMATS_MC = ["MONO", "STEREO", "5_1", "5_1_2", "5_1_4", "7_1", "7_1_4"]

INPUT_FORMATS_ISM = ["ISM1", "ISM2", "ISM3", "ISM4"]

INPUT_FORMATS_MASA = ["MASA1", "MASA2"]

INPUT_FORMATS_MASA = ["MASA1DIR1", "MASA1DIR2", "MASA2DIR1", "MASA2DIR2"]



""" Non binaural / parametric output formats """

OUTPUT_FORMATS = [
@@ -184,10 +188,8 @@ CUSTOM_LS_TO_TEST = [ 
]



""" Mixed scene ( metadata ) rendering """

METADATA_SCENES_TO_TEST = [

    SCENE_DESC_DIR.joinpath(s) for s in ["mixed_scene", "mixed_scene_simple"]

]

METADATA_SCENES_TO_TEST_NO_BE = [SCENE_DESC_DIR.joinpath(s) for s in ["masa_scene"]]

METADATA_SCENES_TO_TEST = ["mixed_scene", "mixed_scene_simple"]

METADATA_SCENES_TO_TEST_NO_BE = ["masa_scene"]



""" Binaural rendering """

INPUT_FORMATS_BINAURAL = OUTPUT_FORMATS[2:]
@@ -210,11 +212,12 @@ HR_TRAJECTORIES_TO_TEST = [ 
]



""" Generate Test Items Configs """

INPUT_CONFIG_DIR = TEST_VECTOR_DIR

INPUT_CONFIG_FILES = [

    str(TEST_VECTOR_DIR.joinpath("test_ISM.yml")),

    str(TEST_VECTOR_DIR.joinpath("test_MASA.yml")),

    str(TEST_VECTOR_DIR.joinpath("test_MC.yml")),

    str(TEST_VECTOR_DIR.joinpath("test_SBA.yml")),

    "test_ISM.yml",

    "test_MASA.yml",

    "test_MC.yml",

    "test_SBA.yml",

]



INPUT_EXPERIMENT_NAMES = [

tests/data/masa_metadata/stv_IVASMASA_1dir1TC.met

0 → 100644LFS
+131 B

File added.

No diff preview for this file type.

View file

tests/data/masa_metadata/stv_IVASMASA_1dir2TC.met

0 → 100644LFS
+131 B

File added.

No diff preview for this file type.

View file