Commit 83e8b50b authored by Marek Szczerba's avatar Marek Szczerba
Browse files

PI data updates related to acoustic environments

parent b4324d63

tests/rtp/test_rtp.py

+22 −6
Original line number Diff line number Diff line
@@ -61,6 +61,7 @@ from tempfile import TemporaryDirectory 
from pathlib import Path

from ivasrtp import *

import numpy as np

import json

from pyaudio3dtools.audiofile import readfile



ROOT_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
@@ -233,15 +234,30 @@ def generatePiData(startTs: int, endTs: int) -> dict: 
    )

    someAcousticEnvLR = lambda: ACOUSTIC_ENVIRONMENT(

        aeid=random.randint(0, 127),

        rt60=[random.randint(0, len(rt60Value)) for _ in range(3)],

        dsr=[random.randint(0, len(dsrValue)) for _ in range(3)],

        rt60=(float(rt60Value[random.randint(0, len(rt60Value) - 1)]),

              float(rt60Value[random.randint(0, len(rt60Value) - 1)]),

              float(rt60Value[random.randint(0, len(rt60Value) - 1)])),

        dsr=(float(dsrValue[random.randint(0, len(dsrValue) - 1)]),

             float(dsrValue[random.randint(0, len(dsrValue) - 1)]),

             float(dsrValue[random.randint(0, len(dsrValue) - 1)])),

    )

    someAcousticEnvERLR = lambda: ACOUSTIC_ENVIRONMENT(

        aeid=random.randint(0, 127),

        rt60=[random.randint(0, len(rt60Value)) for _ in range(3)],

        dsr=[random.randint(0, len(dsrValue)) for _ in range(3)],

        dim=[random.randint(0, len(roomDimensionValue)) for _ in range(3)],

        abscoeff=[random.randint(0, len(absorptionCoeffValues)) for _ in range(6)],

        rt60=(float(rt60Value[random.randint(0, len(rt60Value) - 1)]),

              float(rt60Value[random.randint(0, len(rt60Value) - 1)]),

              float(rt60Value[random.randint(0, len(rt60Value) - 1)])),

        dsr=(float(dsrValue[random.randint(0, len(dsrValue) - 1)]),

             float(dsrValue[random.randint(0, len(dsrValue) - 1)]),

             float(dsrValue[random.randint(0, len(dsrValue) - 1)])),

        dim=(float(roomDimensionValue[random.randint(0, len(roomDimensionValue) - 1)]),

             float(roomDimensionValue[random.randint(0, len(roomDimensionValue) - 1)]),

             float(roomDimensionValue[random.randint(0, len(roomDimensionValue) - 1)])),

        abscoeff=(float(absorptionCoeffValues[random.randint(0, len(absorptionCoeffValues) - 1)]),

                  float(absorptionCoeffValues[random.randint(0, len(absorptionCoeffValues) - 1)]),

                  float(absorptionCoeffValues[random.randint(0, len(absorptionCoeffValues) - 1)]),

                  float(absorptionCoeffValues[random.randint(0, len(absorptionCoeffValues) - 1)]),

                  float(absorptionCoeffValues[random.randint(0, len(absorptionCoeffValues) - 1)]),

                  float(absorptionCoeffValues[random.randint(0, len(absorptionCoeffValues) - 1)])),

    )

    someAcousticEnvList = [someAcousticEnvAEID, someAcousticEnvLR, someAcousticEnvERLR]