Commit 2d481a89 authored by Jan Kiene's avatar Jan Kiene
Browse files

split markers for OMASA

parent fa524492

tests/conftest.py

+5 −1
Original line number Diff line number Diff line
@@ -1262,6 +1262,10 @@ def get_split_idx(input_file: str, sampling_rate_khz: int) -> Optional[np.ndarra 


    If no list of indices is available for the given input file, an empty array is returned.

    """

    input_file = input_file.lower()

    if "omasa" in input_file:

        format = "_".join(input_file.replace(".wav", "").split("_")[:-1])

    else:

        format = input_file.replace(".wav", "").split("_")[-1].lower()

    idx = SPLIT_IDX.get(format, np.empty(0))

tests/constants.py

+95 −0
Original line number Diff line number Diff line
@@ -288,6 +288,97 @@ SPLIT_IDX_LTV_MC_16KHZ = np.asarray( 
        2552400,

    ]

)

SPLIT_IDX_LTV_OMASA_1ISM_16KHZ = np.asarray(

    [

        182561,

        250497,

        321192,

        398875,

        462969,

        589104,

        634642,

        704772,

        761550,

        841565,

        905594,

        1077445,

        1147515,

        1204145,

        1284198,

        1348213,

        1474397,

        1519986,

        1590029,

        1686404,

        1726821,

        1790877,

        1872012,

    ]

)

SPLIT_IDX_LTV_OMASA_2ISM_16KHZ = np.asarray(

    [

        104025,

        188810,

        248000,

        600000,

        727048,

        832000,

        952951,

        970604,

        1037526,

        1104909,

        1233053,

        1295925,

        1361517,

        1414356,

        1476020,

        1549127,

        1605498,

        1727087,

        1876418,

    ]

)

SPLIT_IDX_LTV_OMASA_3ISM_16KHZ = np.asarray(

    [

        155662,

        238144,

        340776,

        415986,

        575167,

        707032,

        832054,

        975997,

        1104017,

        1197434,

        1247993,

        1344330,

        1482134,

        1549575,

        1688766,

        1745715,

    ]

)

SPLIT_IDX_LTV_OMASA_4ISM_16KHZ = np.asarray(

    [

        63985,

        192015,

        316768,

        391082,

        500584,

        597477,

        719995,

        838699,

        952531,

        1040041,

        1143121,

        1271008,

        1503986,

        1626407,

        1687966,

        1796807,

        1878304,

    ]

)



SPLIT_IDX = {

    "stereo": SPLIT_IDX_LTV_STEREO_16KHZ,
@@ -307,4 +398,8 @@ SPLIT_IDX = { 
    "mc514": SPLIT_IDX_LTV_MC_16KHZ,

    "mc71": SPLIT_IDX_LTV_MC_16KHZ,

    "mc714": SPLIT_IDX_LTV_MC_16KHZ,

    "omasa_1ism": SPLIT_IDX_LTV_OMASA_1ISM_16KHZ,

    "omasa_2ism": SPLIT_IDX_LTV_OMASA_2ISM_16KHZ,

    "omasa_3ism": SPLIT_IDX_LTV_OMASA_3ISM_16KHZ,

    "omasa_4ism": SPLIT_IDX_LTV_OMASA_4ISM_16KHZ,

}