ref: 85c371df3ea73f75612b876907140d1fd3d845f7
dir: /tests/multi_channel_test.c/
/* ** Copyright (C) 2002-2004 Erik de Castro Lopo <[email protected]> ** ** This program is free software; you can redistribute it and/or modify ** it under the terms of the GNU General Public License as published by ** the Free Software Foundation; either version 2 of the License, or ** (at your option) any later version. ** ** This program is distributed in the hope that it will be useful, ** but WITHOUT ANY WARRANTY; without even the implied warranty of ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ** GNU General Public License for more details. ** ** You should have received a copy of the GNU General Public License ** along with this program; if not, write to the Free Software ** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA. */ #include "config.h" #include <stdio.h> #include <stdlib.h> #include <string.h> #include <math.h> #include <samplerate.h> #include "util.h" #define BUFFER_LEN (1<<15) #define BLOCK_LEN (12) static void simple_test (int converter, int channel_count, double target_snr) ; static void process_test (int converter, int channel_count, double target_snr) ; static void callback_test (int converter, int channel_count, double target_snr) ; int main (void) { puts ("\n Zero Order Hold interpolator :") ; simple_test (SRC_ZERO_ORDER_HOLD, 1, 45.0) ; process_test (SRC_ZERO_ORDER_HOLD, 1, 45.0) ; callback_test (SRC_ZERO_ORDER_HOLD, 1, 45.0) ; simple_test (SRC_ZERO_ORDER_HOLD, 2, 44.0) ; process_test (SRC_ZERO_ORDER_HOLD, 2, 44.0) ; callback_test (SRC_ZERO_ORDER_HOLD, 2, 44.0) ; simple_test (SRC_ZERO_ORDER_HOLD, 3, 44.0) ; process_test (SRC_ZERO_ORDER_HOLD, 3, 44.0) ; callback_test (SRC_ZERO_ORDER_HOLD, 3, 44.0) ; puts ("\n Linear interpolator :") ; simple_test (SRC_LINEAR, 1, 92.0) ; process_test (SRC_LINEAR, 1, 92.0) ; callback_test (SRC_LINEAR, 1, 92.0) ; simple_test (SRC_LINEAR, 2, 90.0) ; process_test (SRC_LINEAR, 2, 90.0) ; callback_test (SRC_LINEAR, 2, 90.0) ; simple_test (SRC_LINEAR, 3, 88.0) ; process_test (SRC_LINEAR, 3, 88.0) ; callback_test (SRC_LINEAR, 3, 88.0) ; puts ("\n Sinc interpolator :") ; simple_test (SRC_SINC_FASTEST, 1, 100.0) ; process_test (SRC_SINC_FASTEST, 1, 100.0) ; callback_test (SRC_SINC_FASTEST, 1, 100.0) ; simple_test (SRC_SINC_FASTEST, 2, 100.0) ; process_test (SRC_SINC_FASTEST, 2, 100.0) ; callback_test (SRC_SINC_FASTEST, 2, 100.0) ; simple_test (SRC_SINC_FASTEST, 3, 100.0) ; process_test (SRC_SINC_FASTEST, 3, 100.0) ; callback_test (SRC_SINC_FASTEST, 3, 100.0) ; puts ("") ; return 0 ; } /* main */ /*============================================================================== */ static float input_serial [BUFFER_LEN] ; static float input_interleaved [BUFFER_LEN] ; static float output_interleaved [BUFFER_LEN] ; static float output_serial [BUFFER_LEN] ; static void simple_test (int converter, int channel_count, double target_snr) { SRC_DATA src_data ; double freq, snr ; int ch, error, frames ; printf ("\t%-22s (%d channel%c) ............. ", "simple_test", channel_count, channel_count > 1 ? 's' : ' ') ; fflush (stdout) ; memset (input_serial, 0, sizeof (input_serial)) ; memset (input_interleaved, 0, sizeof (input_interleaved)) ; memset (output_interleaved, 0, sizeof (output_interleaved)) ; memset (output_serial, 0, sizeof (output_serial)) ; frames = MIN (ARRAY_LEN (input_serial) / channel_count, 1 << 16) ; /* Calculate channel_count separate windowed sine waves. */ for (ch = 0 ; ch < channel_count ; ch++) { freq = (200.0 + 33.333333333 * ch) / 44100.0 ; gen_windowed_sines (1, &freq, 1.0, input_serial + ch * frames, frames) ; } ; /* Interleave the data in preparation for SRC. */ interleave_data (input_serial, input_interleaved, frames, channel_count) ; /* Choose a converstion ratio <= 1.0. */ src_data.src_ratio = 0.95 ; src_data.data_in = input_interleaved ; src_data.input_frames = frames ; src_data.data_out = output_interleaved ; src_data.output_frames = frames ; if ((error = src_simple (&src_data, converter, channel_count))) { printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ; exit (1) ; } ; if (fabs (src_data.output_frames_gen - src_data.src_ratio * src_data.input_frames) > 2) { printf ("\n\nLine %d : bad output data length %ld should be %d.\n", __LINE__, src_data.output_frames_gen, (int) floor (src_data.src_ratio * src_data.input_frames)) ; printf ("\tsrc_ratio : %.4f\n", src_data.src_ratio) ; printf ("\tinput_len : %ld\n", src_data.input_frames) ; printf ("\toutput_len : %ld\n\n", src_data.output_frames_gen) ; exit (1) ; } ; /* De-interleave data so SNR can be calculated for each channel. */ deinterleave_data (output_interleaved, output_serial, frames, channel_count) ; for (ch = 0 ; ch < channel_count ; ch++) { snr = calculate_snr (output_serial + ch * frames, frames, 1) ; if (snr < target_snr) { printf ("\n\nLine %d: channel %d snr %f should be %f\n", __LINE__, ch, snr, target_snr) ; save_oct_float ("output.dat", input_serial, channel_count * frames, output_serial, channel_count * frames) ; exit (1) ; } ; } ; puts ("ok") ; return ; } /* simple_test */ /*============================================================================== */ static void process_test (int converter, int channel_count, double target_snr) { SRC_STATE *src_state ; SRC_DATA src_data ; double freq, snr ; int ch, error, frames, current_in, current_out ; printf ("\t%-22s (%d channel%c) ............. ", "process_test", channel_count, channel_count > 1 ? 's' : ' ') ; fflush (stdout) ; memset (input_serial, 0, sizeof (input_serial)) ; memset (input_interleaved, 0, sizeof (input_interleaved)) ; memset (output_interleaved, 0, sizeof (output_interleaved)) ; memset (output_serial, 0, sizeof (output_serial)) ; frames = MIN (ARRAY_LEN (input_serial) / channel_count, 1 << 16) ; /* Calculate channel_count separate windowed sine waves. */ for (ch = 0 ; ch < channel_count ; ch++) { freq = (200.0 + 33.333333333 * ch) / 44100.0 ; gen_windowed_sines (1, &freq, 1.0, input_serial + ch * frames, frames) ; } ; /* Interleave the data in preparation for SRC. */ interleave_data (input_serial, input_interleaved, frames, channel_count) ; /* Perform sample rate conversion. */ if ((src_state = src_new (converter, channel_count, &error)) == NULL) { printf ("\n\nLine %d : src_new() failed : %s\n\n", __LINE__, src_strerror (error)) ; exit (1) ; } ; src_data.end_of_input = 0 ; /* Set this later. */ /* Choose a converstion ratio < 1.0. */ src_data.src_ratio = 0.95 ; src_data.data_in = input_interleaved ; src_data.data_out = output_interleaved ; current_in = current_out = 0 ; while (1) { src_data.input_frames = MAX (MIN (BLOCK_LEN, frames - current_in), 0) ; src_data.output_frames = MAX (MIN (BLOCK_LEN, frames - current_out), 0) ; if ((error = src_process (src_state, &src_data))) { printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ; exit (1) ; } ; if (src_data.end_of_input && src_data.output_frames_gen == 0) break ; current_in += src_data.input_frames_used ; current_out += src_data.output_frames_gen ; src_data.data_in += src_data.input_frames_used * channel_count ; src_data.data_out += src_data.output_frames_gen * channel_count ; src_data.end_of_input = (current_in >= frames) ? 1 : 0 ; } ; src_state = src_delete (src_state) ; if (fabs (current_out - src_data.src_ratio * current_in) > 2) { printf ("\n\nLine %d : bad output data length %d should be %d.\n", __LINE__, current_out, (int) floor (src_data.src_ratio * current_in)) ; printf ("\tsrc_ratio : %.4f\n", src_data.src_ratio) ; printf ("\tinput_len : %d\n", frames) ; printf ("\toutput_len : %d\n\n", current_out) ; exit (1) ; } ; /* De-interleave data so SNR can be calculated for each channel. */ deinterleave_data (output_interleaved, output_serial, frames, channel_count) ; for (ch = 0 ; ch < channel_count ; ch++) { snr = calculate_snr (output_serial + ch * frames, frames, 1) ; if (snr < target_snr) { printf ("\n\nLine %d: channel %d snr %f should be %f\n", __LINE__, ch, snr, target_snr) ; save_oct_float ("output.dat", input_serial, channel_count * frames, output_serial, channel_count * frames) ; exit (1) ; } ; } ; puts ("ok") ; return ; } /* process_test */ /*============================================================================== */ typedef struct { int channels ; long total_frames ; long current_frame ; float *data ; } TEST_CB_DATA ; static long test_callback_func (void *cb_data, float **data) { TEST_CB_DATA *pcb_data ; long frames ; if ((pcb_data = cb_data) == NULL) return 0 ; if (data == NULL) return 0 ; *data = pcb_data->data + (pcb_data->current_frame * pcb_data->channels) ; if (pcb_data->total_frames - pcb_data->current_frame < BLOCK_LEN) frames = pcb_data->total_frames - pcb_data->current_frame ; else frames = BLOCK_LEN ; pcb_data->current_frame += frames ; return frames ; } /* test_callback_func */ static void callback_test (int converter, int channel_count, double target_snr) { TEST_CB_DATA test_callback_data ; SRC_STATE *src_state = NULL ; double freq, snr, src_ratio ; int ch, error, frames, read_total, read_count ; printf ("\t%-22s (%d channel%c) ............. ", "callback_test", channel_count, channel_count > 1 ? 's' : ' ') ; fflush (stdout) ; memset (input_serial, 0, sizeof (input_serial)) ; memset (input_interleaved, 0, sizeof (input_interleaved)) ; memset (output_interleaved, 0, sizeof (output_interleaved)) ; memset (output_serial, 0, sizeof (output_serial)) ; memset (&test_callback_data, 0, sizeof (test_callback_data)) ; frames = MIN (ARRAY_LEN (input_serial) / channel_count, 1 << 16) ; /* Calculate channel_count separate windowed sine waves. */ for (ch = 0 ; ch < channel_count ; ch++) { freq = (200.0 + 33.333333333 * ch) / 44100.0 ; gen_windowed_sines (1, &freq, 1.0, input_serial + ch * frames, frames) ; } ; /* Interleave the data in preparation for SRC. */ interleave_data (input_serial, input_interleaved, frames, channel_count) ; /* Perform sample rate conversion. */ src_ratio = 0.95 ; test_callback_data.channels = channel_count ; test_callback_data.total_frames = frames ; test_callback_data.current_frame = 0 ; test_callback_data.data = input_interleaved ; if ((src_state = src_callback_new (test_callback_func, converter, channel_count, &error, &test_callback_data)) == NULL) { printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ; exit (1) ; } ; read_total = 0 ; while (read_total < frames) { read_count = src_callback_read (src_state, src_ratio, frames - read_total, output_interleaved + read_total * channel_count) ; if (read_count <= 0) break ; read_total += read_count ; } ; if ((error = src_error (src_state)) != 0) { printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ; exit (1) ; } ; src_state = src_delete (src_state) ; if (fabs (read_total - src_ratio * frames) > 2) { printf ("\n\nLine %d : bad output data length %d should be %d.\n", __LINE__, read_total, (int) floor (src_ratio * frames)) ; printf ("\tsrc_ratio : %.4f\n", src_ratio) ; printf ("\tinput_len : %d\n", frames) ; printf ("\toutput_len : %d\n\n", read_total) ; exit (1) ; } ; /* De-interleave data so SNR can be calculated for each channel. */ deinterleave_data (output_interleaved, output_serial, frames, channel_count) ; for (ch = 0 ; ch < channel_count ; ch++) { snr = calculate_snr (output_serial + ch * frames, frames, 1) ; if (snr < target_snr) { printf ("\n\nLine %d: channel %d snr %f should be %f\n", __LINE__, ch, snr, target_snr) ; save_oct_float ("output.dat", input_serial, channel_count * frames, output_serial, channel_count * frames) ; exit (1) ; } ; } ; puts ("ok") ; return ; } /* callback_test */ /* ** Do not edit or modify anything in this comment block. ** The arch-tag line is a file identity tag for the GNU Arch ** revision control system. ** ** arch-tag: 48c58d36-baed-4f24-89f1-027a939b240a */