100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
|
# Copyright 2024-2025 XMOS LIMITED.
|
||
|
|
# This Software is subject to the terms of the XMOS Public Licence: Version 1.
|
||
|
|
|
||
|
|
from audio_dsp.dsp.generic import Q_SIG
|
||
|
|
import audio_dsp.dsp.reverb as reverb
|
||
|
|
import audio_dsp.dsp.signal_gen as gen
|
||
|
|
import numpy as np
|
||
|
|
from pathlib import Path
|
||
|
|
import pytest
|
||
|
|
import shutil
|
||
|
|
import soundfile as sf
|
||
|
|
import subprocess
|
||
|
|
from test.test_utils import xdist_safe_bin_write, float_to_qxx, qxx_to_float, q_convert_flt
|
||
|
|
|
||
|
|
BIN_DIR = Path(__file__).parent / "bin"
|
||
|
|
GEN_DIR = Path(__file__).parent / "autogen"
|
||
|
|
FS = 48000
|
||
|
|
|
||
|
|
|
||
|
|
def get_sig(len=0.05):
|
||
|
|
sig_fl = gen.log_chirp(FS, len, 0.5)
|
||
|
|
sig_fl = q_convert_flt(sig_fl, 23, Q_SIG)
|
||
|
|
|
||
|
|
sig_int = float_to_qxx(sig_fl)
|
||
|
|
|
||
|
|
name = "rv_sig_48k"
|
||
|
|
sig_path = BIN_DIR / str(name + ".bin")
|
||
|
|
|
||
|
|
xdist_safe_bin_write(sig_int, sig_path)
|
||
|
|
|
||
|
|
# wav file does not need to be locked as it is only used for debugging outside pytest
|
||
|
|
wav_path = GEN_DIR / str(name + ".wav")
|
||
|
|
sf.write(wav_path, sig_fl, int(FS), "PCM_32")
|
||
|
|
return sig_fl
|
||
|
|
|
||
|
|
|
||
|
|
def get_c_wav(dir_name, app_name, verbose=False, sim=True):
|
||
|
|
app = "xsim" if sim else "xrun --io"
|
||
|
|
run_cmd = app + " " + str(BIN_DIR / app_name)
|
||
|
|
stdout = subprocess.check_output(run_cmd, cwd=dir_name, shell=True)
|
||
|
|
if verbose: print("run msg:\n", stdout.decode())
|
||
|
|
|
||
|
|
sig_bin = dir_name / "rv_sig_out.bin"
|
||
|
|
assert sig_bin.is_file(), f"Could not find output bin {sig_bin}"
|
||
|
|
sig_int = np.fromfile(sig_bin, dtype=np.int32)
|
||
|
|
|
||
|
|
sig_fl = qxx_to_float(sig_int)
|
||
|
|
sf.write(GEN_DIR / "sig_c.wav", sig_fl, FS, "PCM_32")
|
||
|
|
return sig_fl
|
||
|
|
|
||
|
|
|
||
|
|
def run_py(uut: reverb.reverb_room, sig_fl, use_float_sig=True):
|
||
|
|
out_int = np.zeros(sig_fl.size)
|
||
|
|
sig_int = float_to_qxx(sig_fl)
|
||
|
|
|
||
|
|
if use_float_sig:
|
||
|
|
for n in range(sig_fl.size):
|
||
|
|
out_int[n] = uut.process_xcore(sig_fl[n])
|
||
|
|
else:
|
||
|
|
for n in range(sig_fl.size):
|
||
|
|
out_int[n] = uut.process_xcore(sig_int[n])
|
||
|
|
out_int = qxx_to_float(out_int)
|
||
|
|
|
||
|
|
# sf.write(GEN_DIR / "sig_py_int.wav", out_int, FS, "PCM_32")
|
||
|
|
|
||
|
|
return out_int
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope="module")
|
||
|
|
def in_signal():
|
||
|
|
BIN_DIR.mkdir(exist_ok=True, parents=True)
|
||
|
|
GEN_DIR.mkdir(exist_ok=True, parents=True)
|
||
|
|
return get_sig()
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("decay, damping", [[1.0, 1.0],
|
||
|
|
[0.1, 0.5]])
|
||
|
|
@pytest.mark.parametrize("wet, dry, pregain", [[-1.0, -1.0, 0.015]])
|
||
|
|
def test_reverb_room(in_signal, decay, damping, wet, dry, pregain):
|
||
|
|
n_chans = 1
|
||
|
|
fs = FS
|
||
|
|
max_room_size = 1.0
|
||
|
|
room_size = 1.0
|
||
|
|
predelay = 10
|
||
|
|
|
||
|
|
uut = reverb.reverb_room(fs, n_chans, max_room_size, room_size, decay, damping, wet, dry, pregain, predelay)
|
||
|
|
test_name = f"reverb_room_{decay}_{damping}_{wet}_{dry}_{pregain}"
|
||
|
|
|
||
|
|
test_dir = BIN_DIR / test_name
|
||
|
|
test_dir.mkdir(exist_ok = True, parents = True)
|
||
|
|
|
||
|
|
rv_info = [uut.pregain_int, uut.wet_int, uut.dry_int, uut.combs[0].feedback_int, uut.combs[0].damp1_int]
|
||
|
|
rv_info = np.array(rv_info, dtype=np.int32)
|
||
|
|
rv_info.tofile(test_dir / "rv_info.bin")
|
||
|
|
|
||
|
|
out_py_int = run_py(uut, in_signal)
|
||
|
|
out_c = get_c_wav(test_dir, "reverb_test.xe")
|
||
|
|
shutil.rmtree(test_dir)
|
||
|
|
|
||
|
|
np.testing.assert_allclose(out_c, out_py_int, rtol=0, atol=0)
|