cannam@126: /*
cannam@126: ** Copyright (c) 2002-2016, Erik de Castro Lopo <erikd@mega-nerd.com>
cannam@126: ** All rights reserved.
cannam@126: **
cannam@126: ** This code is released under 2-clause BSD license. Please see the
cannam@126: ** file at : https://github.com/erikd/libsamplerate/blob/master/COPYING
cannam@126: */
cannam@126: 
cannam@126: #include "config.h"
cannam@126: 
cannam@126: #include <stdio.h>
cannam@126: #include <stdlib.h>
cannam@126: #include <string.h>
cannam@126: #include <math.h>
cannam@126: #include <assert.h>
cannam@126: 
cannam@126: #if (HAVE_FFTW3)
cannam@126: #include <fftw3.h>
cannam@126: #else
cannam@126: static inline void
cannam@126: fftw_cleanup (void)
cannam@126: {	return ;
cannam@126: }
cannam@126: #endif
cannam@126: 
cannam@126: #include <samplerate.h>
cannam@126: 
cannam@126: #include "util.h"
cannam@126: #define	BUFFER_LEN		50000
cannam@126: #define	BLOCK_LEN		(12)
cannam@126: 
cannam@126: #define	MAX_CHANNELS	10
cannam@126: 
cannam@126: static void simple_test (int converter, int channel_count, double target_snr) ;
cannam@126: static void process_test (int converter, int channel_count, double target_snr) ;
cannam@126: static void callback_test (int converter, int channel_count, double target_snr) ;
cannam@126: 
cannam@126: int
cannam@126: main (void)
cannam@126: {	double target ;
cannam@126: 	int k ;
cannam@126: 
cannam@126: 	puts ("\n    Zero Order Hold interpolator :") ;
cannam@126: 	target = 38.0 ;
cannam@126: 	for (k = 1 ; k <= 3 ; k++)
cannam@126: 	{	simple_test		(SRC_ZERO_ORDER_HOLD, k, target) ;
cannam@126: 		process_test	(SRC_ZERO_ORDER_HOLD, k, target) ;
cannam@126: 		callback_test	(SRC_ZERO_ORDER_HOLD, k, target) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	puts ("\n    Linear interpolator :") ;
cannam@126: 	target = 79.0 ;
cannam@126: 	for (k = 1 ; k <= 3 ; k++)
cannam@126: 	{	simple_test		(SRC_LINEAR, k, target) ;
cannam@126: 		process_test	(SRC_LINEAR, k, target) ;
cannam@126: 		callback_test	(SRC_LINEAR, k, target) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	puts ("\n    Sinc interpolator :") ;
cannam@126: 	target = 100.0 ;
cannam@126: 	for (k = 1 ; k <= MAX_CHANNELS ; k++)
cannam@126: 	{	simple_test		(SRC_SINC_FASTEST, k, target) ;
cannam@126: 		process_test	(SRC_SINC_FASTEST, k, target) ;
cannam@126: 		callback_test	(SRC_SINC_FASTEST, k, target) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	fftw_cleanup () ;
cannam@126: 	puts ("") ;
cannam@126: 
cannam@126: 	return 0 ;
cannam@126: } /* main */
cannam@126: 
cannam@126: /*==============================================================================
cannam@126: */
cannam@126: 
cannam@126: static float input_serial		[BUFFER_LEN * MAX_CHANNELS] ;
cannam@126: static float input_interleaved	[BUFFER_LEN * MAX_CHANNELS] ;
cannam@126: static float output_interleaved	[BUFFER_LEN * MAX_CHANNELS] ;
cannam@126: static float output_serial		[BUFFER_LEN * MAX_CHANNELS] ;
cannam@126: 
cannam@126: static void
cannam@126: simple_test (int converter, int channel_count, double target_snr)
cannam@126: {	SRC_DATA	src_data ;
cannam@126: 
cannam@126: 	double	freq, snr ;
cannam@126: 	int		ch, error, frames ;
cannam@126: 
cannam@126: 	printf ("\t%-22s (%2d channel%c) ............ ", "simple_test", channel_count, channel_count > 1 ? 's' : ' ') ;
cannam@126: 	fflush (stdout) ;
cannam@126: 
cannam@126: 	assert (channel_count <= MAX_CHANNELS) ;
cannam@126: 
cannam@126: 	memset (input_serial, 0, sizeof (input_serial)) ;
cannam@126: 	memset (input_interleaved, 0, sizeof (input_interleaved)) ;
cannam@126: 	memset (output_interleaved, 0, sizeof (output_interleaved)) ;
cannam@126: 	memset (output_serial, 0, sizeof (output_serial)) ;
cannam@126: 
cannam@126: 	frames = BUFFER_LEN ;
cannam@126: 
cannam@126: 	/* Calculate channel_count separate windowed sine waves. */
cannam@126: 	for (ch = 0 ; ch < channel_count ; ch++)
cannam@126: 	{	freq = (200.0 + 33.333333333 * ch) / 44100.0 ;
cannam@126: 		gen_windowed_sines (1, &freq, 1.0, input_serial + ch * frames, frames) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	/* Interleave the data in preparation for SRC. */
cannam@126: 	interleave_data (input_serial, input_interleaved, frames, channel_count) ;
cannam@126: 
cannam@126: 	/* Choose a converstion ratio <= 1.0. */
cannam@126: 	src_data.src_ratio = 0.95 ;
cannam@126: 
cannam@126: 	src_data.data_in = input_interleaved ;
cannam@126: 	src_data.input_frames = frames ;
cannam@126: 
cannam@126: 	src_data.data_out = output_interleaved ;
cannam@126: 	src_data.output_frames = frames ;
cannam@126: 
cannam@126: 	if ((error = src_simple (&src_data, converter, channel_count)))
cannam@126: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	if (fabs (src_data.output_frames_gen - src_data.src_ratio * src_data.input_frames) > 2)
cannam@126: 	{	printf ("\n\nLine %d : bad output data length %ld should be %d.\n", __LINE__,
cannam@126: 					src_data.output_frames_gen, (int) floor (src_data.src_ratio * src_data.input_frames)) ;
cannam@126: 		printf ("\tsrc_ratio  : %.4f\n", src_data.src_ratio) ;
cannam@126: 		printf ("\tinput_len  : %ld\n", src_data.input_frames) ;
cannam@126: 		printf ("\toutput_len : %ld\n\n", src_data.output_frames_gen) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	/* De-interleave data so SNR can be calculated for each channel. */
cannam@126: 	deinterleave_data (output_interleaved, output_serial, frames, channel_count) ;
cannam@126: 
cannam@126: 	for (ch = 0 ; ch < channel_count ; ch++)
cannam@126: 	{	snr = calculate_snr (output_serial + ch * frames, frames, 1) ;
cannam@126: 		if (snr < target_snr)
cannam@126: 		{	printf ("\n\nLine %d: channel %d snr %f should be %f\n", __LINE__, ch, snr, target_snr) ;
cannam@126: 			save_oct_float ("output.dat", input_serial, channel_count * frames, output_serial, channel_count * frames) ;
cannam@126: 			exit (1) ;
cannam@126: 			} ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	puts ("ok") ;
cannam@126: 
cannam@126: 	return ;
cannam@126: } /* simple_test */
cannam@126: 
cannam@126: /*==============================================================================
cannam@126: */
cannam@126: 
cannam@126: static void
cannam@126: process_test (int converter, int channel_count, double target_snr)
cannam@126: {	SRC_STATE	*src_state ;
cannam@126: 	SRC_DATA	src_data ;
cannam@126: 
cannam@126: 	double	freq, snr ;
cannam@126: 	int		ch, error, frames, current_in, current_out ;
cannam@126: 
cannam@126: 	printf ("\t%-22s (%2d channel%c) ............ ", "process_test", channel_count, channel_count > 1 ? 's' : ' ') ;
cannam@126: 	fflush (stdout) ;
cannam@126: 
cannam@126: 	assert (channel_count <= MAX_CHANNELS) ;
cannam@126: 
cannam@126: 	memset (input_serial, 0, sizeof (input_serial)) ;
cannam@126: 	memset (input_interleaved, 0, sizeof (input_interleaved)) ;
cannam@126: 	memset (output_interleaved, 0, sizeof (output_interleaved)) ;
cannam@126: 	memset (output_serial, 0, sizeof (output_serial)) ;
cannam@126: 
cannam@126: 	frames = BUFFER_LEN ;
cannam@126: 
cannam@126: 	/* Calculate channel_count separate windowed sine waves. */
cannam@126: 	for (ch = 0 ; ch < channel_count ; ch++)
cannam@126: 	{	freq = (400.0 + 11.333333333 * ch) / 44100.0 ;
cannam@126: 		gen_windowed_sines (1, &freq, 1.0, input_serial + ch * frames, frames) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	/* Interleave the data in preparation for SRC. */
cannam@126: 	interleave_data (input_serial, input_interleaved, frames, channel_count) ;
cannam@126: 
cannam@126: 	/* Perform sample rate conversion. */
cannam@126: 	if ((src_state = src_new (converter, channel_count, &error)) == NULL)
cannam@126: 	{	printf ("\n\nLine %d : src_new() failed : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	src_data.end_of_input = 0 ; /* Set this later. */
cannam@126: 
cannam@126: 	/* Choose a converstion ratio < 1.0. */
cannam@126: 	src_data.src_ratio = 0.95 ;
cannam@126: 
cannam@126: 	src_data.data_in = input_interleaved ;
cannam@126: 	src_data.data_out = output_interleaved ;
cannam@126: 
cannam@126: 	current_in = current_out = 0 ;
cannam@126: 
cannam@126: 	while (1)
cannam@126: 	{	src_data.input_frames	= MAX (MIN (BLOCK_LEN, frames - current_in), 0) ;
cannam@126: 		src_data.output_frames	= MAX (MIN (BLOCK_LEN, frames - current_out), 0) ;
cannam@126: 
cannam@126: 		if ((error = src_process (src_state, &src_data)))
cannam@126: 		{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@126: 			exit (1) ;
cannam@126: 			} ;
cannam@126: 
cannam@126: 		if (src_data.end_of_input && src_data.output_frames_gen == 0)
cannam@126: 			break ;
cannam@126: 
cannam@126: 		current_in	+= src_data.input_frames_used ;
cannam@126: 		current_out += src_data.output_frames_gen ;
cannam@126: 
cannam@126: 		src_data.data_in	+= src_data.input_frames_used * channel_count ;
cannam@126: 		src_data.data_out	+= src_data.output_frames_gen * channel_count ;
cannam@126: 
cannam@126: 		src_data.end_of_input = (current_in >= frames) ? 1 : 0 ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	src_state = src_delete (src_state) ;
cannam@126: 
cannam@126: 	if (fabs (current_out - src_data.src_ratio * current_in) > 2)
cannam@126: 	{	printf ("\n\nLine %d : bad output data length %d should be %d.\n", __LINE__,
cannam@126: 					current_out, (int) floor (src_data.src_ratio * current_in)) ;
cannam@126: 		printf ("\tsrc_ratio  : %.4f\n", src_data.src_ratio) ;
cannam@126: 		printf ("\tinput_len  : %d\n", frames) ;
cannam@126: 		printf ("\toutput_len : %d\n\n", current_out) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	/* De-interleave data so SNR can be calculated for each channel. */
cannam@126: 	deinterleave_data (output_interleaved, output_serial, frames, channel_count) ;
cannam@126: 
cannam@126: 	for (ch = 0 ; ch < channel_count ; ch++)
cannam@126: 	{	snr = calculate_snr (output_serial + ch * frames, frames, 1) ;
cannam@126: 		if (snr < target_snr)
cannam@126: 		{	printf ("\n\nLine %d: channel %d snr %f should be %f\n", __LINE__, ch, snr, target_snr) ;
cannam@126: 			save_oct_float ("output.dat", input_serial, channel_count * frames, output_serial, channel_count * frames) ;
cannam@126: 			exit (1) ;
cannam@126: 			} ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	puts ("ok") ;
cannam@126: 
cannam@126: 	return ;
cannam@126: } /* process_test */
cannam@126: 
cannam@126: /*==============================================================================
cannam@126: */
cannam@126: 
cannam@126: typedef struct
cannam@126: {	int channels ;
cannam@126: 	long total_frames ;
cannam@126: 	long current_frame ;
cannam@126: 	float *data ;
cannam@126: } TEST_CB_DATA ;
cannam@126: 
cannam@126: static long
cannam@126: test_callback_func (void *cb_data, float **data)
cannam@126: {	TEST_CB_DATA *pcb_data ;
cannam@126: 
cannam@126: 	long frames ;
cannam@126: 
cannam@126: 	if ((pcb_data = cb_data) == NULL)
cannam@126: 		return 0 ;
cannam@126: 
cannam@126: 	if (data == NULL)
cannam@126: 		return 0 ;
cannam@126: 
cannam@126: 	*data = pcb_data->data + (pcb_data->current_frame * pcb_data->channels) ;
cannam@126: 
cannam@126: 	if (pcb_data->total_frames - pcb_data->current_frame < BLOCK_LEN)
cannam@126: 		frames = pcb_data->total_frames - pcb_data->current_frame ;
cannam@126: 	else
cannam@126: 		frames = BLOCK_LEN ;
cannam@126: 
cannam@126: 	pcb_data->current_frame += frames ;
cannam@126: 
cannam@126: 	return frames ;
cannam@126: } /* test_callback_func */
cannam@126: 
cannam@126: static void
cannam@126: callback_test (int converter, int channel_count, double target_snr)
cannam@126: {	TEST_CB_DATA test_callback_data ;
cannam@126: 	SRC_STATE	*src_state = NULL ;
cannam@126: 
cannam@126: 	double	freq, snr, src_ratio ;
cannam@126: 	int		ch, error, frames, read_total, read_count ;
cannam@126: 
cannam@126: 	printf ("\t%-22s (%2d channel%c) ............ ", "callback_test", channel_count, channel_count > 1 ? 's' : ' ') ;
cannam@126: 	fflush (stdout) ;
cannam@126: 
cannam@126: 	assert (channel_count <= MAX_CHANNELS) ;
cannam@126: 
cannam@126: 	memset (input_serial, 0, sizeof (input_serial)) ;
cannam@126: 	memset (input_interleaved, 0, sizeof (input_interleaved)) ;
cannam@126: 	memset (output_interleaved, 0, sizeof (output_interleaved)) ;
cannam@126: 	memset (output_serial, 0, sizeof (output_serial)) ;
cannam@126: 	memset (&test_callback_data, 0, sizeof (test_callback_data)) ;
cannam@126: 
cannam@126: 	frames = BUFFER_LEN ;
cannam@126: 
cannam@126: 	/* Calculate channel_count separate windowed sine waves. */
cannam@126: 	for (ch = 0 ; ch < channel_count ; ch++)
cannam@126: 	{	freq = (200.0 + 33.333333333 * ch) / 44100.0 ;
cannam@126: 		gen_windowed_sines (1, &freq, 1.0, input_serial + ch * frames, frames) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	/* Interleave the data in preparation for SRC. */
cannam@126: 	interleave_data (input_serial, input_interleaved, frames, channel_count) ;
cannam@126: 
cannam@126: 	/* Perform sample rate conversion. */
cannam@126: 	src_ratio = 0.95 ;
cannam@126: 	test_callback_data.channels = channel_count ;
cannam@126: 	test_callback_data.total_frames = frames ;
cannam@126: 	test_callback_data.current_frame = 0 ;
cannam@126: 	test_callback_data.data = input_interleaved ;
cannam@126: 
cannam@126: 	if ((src_state = src_callback_new (test_callback_func, converter, channel_count, &error, &test_callback_data)) == NULL)
cannam@126: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	read_total = 0 ;
cannam@126: 	while (read_total < frames)
cannam@126: 	{	read_count = src_callback_read (src_state, src_ratio, frames - read_total, output_interleaved + read_total * channel_count) ;
cannam@126: 
cannam@126: 		if (read_count <= 0)
cannam@126: 			break ;
cannam@126: 
cannam@126: 		read_total += read_count ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	if ((error = src_error (src_state)) != 0)
cannam@126: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	src_state = src_delete (src_state) ;
cannam@126: 
cannam@126: 	if (fabs (read_total - src_ratio * frames) > 2)
cannam@126: 	{	printf ("\n\nLine %d : bad output data length %d should be %d.\n", __LINE__,
cannam@126: 					read_total, (int) floor (src_ratio * frames)) ;
cannam@126: 		printf ("\tsrc_ratio  : %.4f\n", src_ratio) ;
cannam@126: 		printf ("\tinput_len  : %d\n", frames) ;
cannam@126: 		printf ("\toutput_len : %d\n\n", read_total) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	/* De-interleave data so SNR can be calculated for each channel. */
cannam@126: 	deinterleave_data (output_interleaved, output_serial, frames, channel_count) ;
cannam@126: 
cannam@126: 	for (ch = 0 ; ch < channel_count ; ch++)
cannam@126: 	{	snr = calculate_snr (output_serial + ch * frames, frames, 1) ;
cannam@126: 		if (snr < target_snr)
cannam@126: 		{	printf ("\n\nLine %d: channel %d snr %f should be %f\n", __LINE__, ch, snr, target_snr) ;
cannam@126: 			save_oct_float ("output.dat", input_serial, channel_count * frames, output_serial, channel_count * frames) ;
cannam@126: 			exit (1) ;
cannam@126: 			} ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	puts ("ok") ;
cannam@126: 
cannam@126: 	return ;
cannam@126: } /* callback_test */
cannam@126: