mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
automated test for FreeDV resampler
This commit is contained in:
parent
653713918d
commit
b95562740a
|
@ -22,6 +22,12 @@ set(FRAMESPERBURST 3)
|
||||||
set(BURSTS 1)
|
set(BURSTS 1)
|
||||||
set(TESTFRAMES 3)
|
set(TESTFRAMES 3)
|
||||||
|
|
||||||
|
add_test(NAME 000_resampler
|
||||||
|
COMMAND sh -c "export LD_LIBRARY_PATH=${CODEC2_BUILD_DIR}/src;
|
||||||
|
cd ${CMAKE_CURRENT_SOURCE_DIR}/test/000_resampler;
|
||||||
|
python3 t48_8_short.py")
|
||||||
|
set_tests_properties(000_resampler PROPERTIES PASS_REGULAR_EXPRESSION "PASS")
|
||||||
|
|
||||||
add_test(NAME 001_highsnr_stdio_P_C_SM
|
add_test(NAME 001_highsnr_stdio_P_C_SM
|
||||||
COMMAND sh -c "export LD_LIBRARY_PATH=${CODEC2_BUILD_DIR}/src;
|
COMMAND sh -c "export LD_LIBRARY_PATH=${CODEC2_BUILD_DIR}/src;
|
||||||
PATH=$PATH:${CODEC2_BUILD_DIR}/src;
|
PATH=$PATH:${CODEC2_BUILD_DIR}/src;
|
||||||
|
|
|
@ -38,25 +38,72 @@ AMP = 16000 # sine wave amplitude
|
||||||
FTEST8 = 800 # input test frequency at FS=8kHz
|
FTEST8 = 800 # input test frequency at FS=8kHz
|
||||||
FINTER48 = 10000 # interferer frequency at FS=48kHz
|
FINTER48 = 10000 # interferer frequency at FS=48kHz
|
||||||
|
|
||||||
|
# Due to the design of these resamplers, the processing buffer (at 8kHz)
|
||||||
|
# must be an integer multiple of oversampling ratio
|
||||||
|
assert N8 % FDMDV_OS_48 == 0
|
||||||
|
|
||||||
in8k = np.zeros(MEM8 + N8, dtype=np.int16)
|
in8k = np.zeros(MEM8 + N8, dtype=np.int16)
|
||||||
out48k = np.zeros(N48, dtype=np.int16)
|
out48k = np.zeros(N48, dtype=np.int16)
|
||||||
in48k = np.zeros(MEM48 + N48, dtype=np.int16)
|
in48k = np.zeros(MEM48 + N48, dtype=np.int16)
|
||||||
out8k = np.zeros(N8, dtype=np.int16)
|
out8k = np.zeros(N8, dtype=np.int16)
|
||||||
|
|
||||||
|
# time indexes, we advance every frame
|
||||||
t = 0
|
t = 0
|
||||||
|
t1 = 0
|
||||||
|
|
||||||
fin8 = open("in8.raw", mode='wb')
|
fin8 = open("in8.raw", mode='wb')
|
||||||
f48 = open("out48.raw", mode='wb')
|
f48 = open("out48.raw", mode='wb')
|
||||||
|
fout8 = open("out8.raw", mode='wb')
|
||||||
|
|
||||||
for f in range(FRAMES):
|
for f in range(FRAMES):
|
||||||
|
|
||||||
|
# input sine wave
|
||||||
in8k[MEM8:] = AMP*np.cos(2*np.pi*np.arange(t,t+N8)*FTEST8/FS8)
|
in8k[MEM8:] = AMP*np.cos(2*np.pi*np.arange(t,t+N8)*FTEST8/FS8)
|
||||||
t += N8
|
t += N8
|
||||||
in8k[MEM8:].tofile(fin8)
|
in8k[MEM8:].tofile(fin8)
|
||||||
|
|
||||||
|
# upsample
|
||||||
pin8k,flag = in8k.__array_interface__['data']
|
pin8k,flag = in8k.__array_interface__['data']
|
||||||
pin8k += 2*MEM8
|
pin8k += 2*MEM8
|
||||||
codec2.api.fdmdv_8_to_48_short(out48k.ctypes, pin8k, N8);
|
codec2.api.fdmdv_8_to_48_short(out48k.ctypes, pin8k, N8);
|
||||||
out48k.tofile(f48)
|
out48k.tofile(f48)
|
||||||
|
|
||||||
|
# add interfering sine wave (down sampling filter should remove)
|
||||||
|
in48k[MEM48:] = out48k + (AMP/2)*np.cos(2*np.pi*np.arange(t1,t1+N48)*FINTER48/FS48)
|
||||||
|
t1 += N48
|
||||||
|
|
||||||
|
# downsample
|
||||||
|
pin48k,flag = in48k.__array_interface__['data']
|
||||||
|
pin48k += 2*MEM48
|
||||||
|
codec2.api.fdmdv_48_to_8_short(out8k.ctypes, pin48k, N8);
|
||||||
|
out8k.tofile(fout8)
|
||||||
|
|
||||||
fin8.close()
|
fin8.close()
|
||||||
f48.close()
|
f48.close()
|
||||||
|
fout8.close()
|
||||||
|
|
||||||
|
# Automated test evaluation --------------------------------------------
|
||||||
|
|
||||||
|
# The input and output signals will not be time aligned due to the filter
|
||||||
|
# delays, so compare the magnitude spectrum
|
||||||
|
|
||||||
|
in8k = np.fromfile("in8.raw", dtype=np.int16)
|
||||||
|
out8k = np.fromfile("out8.raw", dtype=np.int16)
|
||||||
|
assert len(in8k) == len(out8k)
|
||||||
|
|
||||||
|
n = len(in8k)
|
||||||
|
|
||||||
|
h = np.hanning(len(in8k))
|
||||||
|
S1 = np.abs(np.fft.fft(in8k * h))
|
||||||
|
S2 = np.abs(np.fft.fft(out8k * h))
|
||||||
|
|
||||||
|
error = S1-S2
|
||||||
|
error_energy = np.dot(error,error)
|
||||||
|
ratio = error_energy/np.dot(S1,S1)
|
||||||
|
ratio_dB = 10*np.log10(ratio);
|
||||||
|
print("ratio_dB: %4.2f" % (ratio_dB));
|
||||||
|
threshdB = -40
|
||||||
|
if ratio_dB < threshdB:
|
||||||
|
print("PASS")
|
||||||
|
else:
|
||||||
|
print("FAIL")
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
|
||||||
|
@ -7,8 +7,6 @@ from ctypes import *
|
||||||
import pathlib
|
import pathlib
|
||||||
import pyaudio
|
import pyaudio
|
||||||
import time
|
import time
|
||||||
import threading
|
|
||||||
import audioop
|
|
||||||
import argparse
|
import argparse
|
||||||
import sys
|
import sys
|
||||||
sys.path.insert(0,'..')
|
sys.path.insert(0,'..')
|
||||||
|
|
Loading…
Reference in a new issue