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 <time.h>
cannam@126: 
cannam@126: #if (HAVE_FFTW3)
cannam@126: 
cannam@126: #include <fftw3.h>
cannam@126: 
cannam@126: #include <samplerate.h>
cannam@126: 
cannam@126: #include "util.h"
cannam@126: 
cannam@126: #define	BUFFER_LEN		50000
cannam@126: #define	MAX_FREQS		4
cannam@126: #define	MAX_RATIOS		6
cannam@126: #define	MAX_SPEC_LEN	(1<<15)
cannam@126: 
cannam@126: #ifndef	M_PI
cannam@126: #define	M_PI			3.14159265358979323846264338
cannam@126: #endif
cannam@126: 
cannam@126: enum
cannam@126: {	BOOLEAN_FALSE	= 0,
cannam@126: 	BOOLEAN_TRUE	= 1
cannam@126: } ;
cannam@126: 
cannam@126: typedef struct
cannam@126: {	int		freq_count ;
cannam@126: 	double	freqs [MAX_FREQS] ;
cannam@126: 
cannam@126: 	double	src_ratio ;
cannam@126: 	int		pass_band_peaks ;
cannam@126: 
cannam@126: 	double	snr ;
cannam@126: 	double	peak_value ;
cannam@126: } SINGLE_TEST ;
cannam@126: 
cannam@126: typedef struct
cannam@126: {	int			converter ;
cannam@126: 	int			tests ;
cannam@126: 	int			do_bandwidth_test ;
cannam@126: 	SINGLE_TEST	test_data [10] ;
cannam@126: } CONVERTER_TEST ;
cannam@126: 
cannam@126: static double snr_test (SINGLE_TEST *snr_test_data, int number, int converter, int verbose) ;
cannam@126: static double find_peak (float *output, int output_len) ;
cannam@126: static double bandwidth_test (int converter, int verbose) ;
cannam@126: 
cannam@126: int
cannam@126: main (int argc, char *argv [])
cannam@126: {	CONVERTER_TEST snr_test_data [] =
cannam@126: 	{
cannam@126: 		{	SRC_ZERO_ORDER_HOLD,
cannam@126: 			8,
cannam@126: 			BOOLEAN_FALSE,
cannam@126: 			{	{	1,	{ 0.01111111111 },		3.0,		1,	 28.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.6,		1,	 36.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.3,		1,	 36.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.0,		1,	150.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.001,		1,	 38.0,	1.0 },
cannam@126: 				{	2,	{ 0.011111, 0.324 },	1.9999,		2,	 14.0,	1.0 },
cannam@126: 				{	2,	{ 0.012345, 0.457 },	0.456789,	1,	 12.0,	1.0 },
cannam@126: 				{	1,	{ 0.3511111111 },		1.33,		1,	 10.0,	1.0 }
cannam@126: 				}
cannam@126: 			},
cannam@126: 
cannam@126: 		{	SRC_LINEAR,
cannam@126: 			8,
cannam@126: 			BOOLEAN_FALSE,
cannam@126: 			{	{	1,	{ 0.01111111111 },		3.0,		1,	 73.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.6,		1,	 73.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.3,		1,	 73.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.0,		1,	150.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.001,		1,	 77.0,	1.0 },
cannam@126: 				{	2,	{ 0.011111, 0.324 },	1.9999,		2,	 15.0,	0.94 },
cannam@126: 				{	2,	{ 0.012345, 0.457 },	0.456789,	1,	 25.0,	0.96 },
cannam@126: 				{	1,	{ 0.3511111111 },		1.33,		1,	 22.0,	0.99 }
cannam@126: 				}
cannam@126: 			},
cannam@126: 
cannam@126: 		{	SRC_SINC_FASTEST,
cannam@126: 			9,
cannam@126: 			BOOLEAN_TRUE,
cannam@126: 			{	{	1,	{ 0.01111111111 },		3.0,		1,	100.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.6,		1,	 99.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.3,		1,	100.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.0,		1,	150.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.001,		1,	100.0,	1.0 },
cannam@126: 				{	2,	{ 0.011111, 0.324 },	1.9999,		2,	 97.0,	1.0 },
cannam@126: 				{	2,	{ 0.012345, 0.457 },	0.456789,	1,	100.0,	0.5 },
cannam@126: 				{	2,	{ 0.011111, 0.45 },		0.6,		1,	 97.0,	0.5 },
cannam@126: 				{	1,	{ 0.3511111111 },		1.33,		1,	 97.0,	1.0 }
cannam@126: 				}
cannam@126: 			},
cannam@126: 
cannam@126: 		{	SRC_SINC_MEDIUM_QUALITY,
cannam@126: 			9,
cannam@126: 			BOOLEAN_TRUE,
cannam@126: 			{	{	1,	{ 0.01111111111 },		3.0,		1,	145.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.6,		1,	132.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.3,		1,	138.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.0,		1,	157.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.001,		1,	148.0,	1.0 },
cannam@126: 				{	2,	{ 0.011111, 0.324 },	1.9999,		2,	127.0,	1.0 },
cannam@126: 				{	2,	{ 0.012345, 0.457 },	0.456789,	1,	123.0,	0.5 },
cannam@126: 				{	2,	{ 0.011111, 0.45 },		0.6,		1,	126.0,	0.5 },
cannam@126: 				{	1,	{ 0.43111111111 },		1.33,		1,	121.0,	1.0 }
cannam@126: 				}
cannam@126: 			},
cannam@126: 
cannam@126: 		{	SRC_SINC_BEST_QUALITY,
cannam@126: 			9,
cannam@126: 			BOOLEAN_TRUE,
cannam@126: 			{	{	1,	{ 0.01111111111 },		3.0,		1,	147.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.6,		1,	147.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		0.3,		1,	148.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.0,		1,	155.0,	1.0 },
cannam@126: 				{	1,	{ 0.01111111111 },		1.001,		1,	148.0,	1.0 },
cannam@126: 				{	2,	{ 0.011111, 0.324 },	1.9999,		2,	146.0,	1.0 },
cannam@126: 				{	2,	{ 0.012345, 0.457 },	0.456789,	1,	147.0,	0.5 },
cannam@126: 				{	2,	{ 0.011111, 0.45 },		0.6,		1,	144.0,	0.5 },
cannam@126: 				{	1,	{ 0.43111111111 },		1.33,		1,	145.0,	1.0 }
cannam@126: 				}
cannam@126: 			},
cannam@126: 		} ; /* snr_test_data */
cannam@126: 
cannam@126: 	double	best_snr, snr, freq3dB ;
cannam@126: 	int 	j, k, converter, verbose = 0 ;
cannam@126: 
cannam@126: 	if (argc == 2 && strcmp (argv [1], "--verbose") == 0)
cannam@126: 		verbose = 1 ;
cannam@126: 
cannam@126: 	puts ("") ;
cannam@126: 
cannam@126: 	for (j = 0 ; j < ARRAY_LEN (snr_test_data) ; j++)
cannam@126: 	{	best_snr = 5000.0 ;
cannam@126: 
cannam@126: 		converter = snr_test_data [j].converter ;
cannam@126: 
cannam@126: 		printf ("    Converter %d : %s\n", converter, src_get_name (converter)) ;
cannam@126: 		printf ("    %s\n", src_get_description (converter)) ;
cannam@126: 
cannam@126: 		for (k = 0 ; k < snr_test_data [j].tests ; k++)
cannam@126: 		{	snr = snr_test (&(snr_test_data [j].test_data [k]), k, converter, verbose) ;
cannam@126: 			if (best_snr > snr)
cannam@126: 				best_snr = snr ;
cannam@126: 			} ;
cannam@126: 
cannam@126: 		printf ("    Worst case Signal-to-Noise Ratio : %.2f dB.\n", best_snr) ;
cannam@126: 
cannam@126: 		if (snr_test_data [j].do_bandwidth_test == BOOLEAN_FALSE)
cannam@126: 		{	puts ("    Bandwith test not performed on this converter.\n") ;
cannam@126: 			continue ;
cannam@126: 			}
cannam@126: 
cannam@126: 		freq3dB = bandwidth_test (converter, verbose) ;
cannam@126: 
cannam@126: 		printf ("    Measured -3dB rolloff point      : %5.2f %%.\n\n", freq3dB) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	fftw_cleanup () ;
cannam@126: 
cannam@126: 	return 0 ;
cannam@126: } /* main */
cannam@126: 
cannam@126: /*==============================================================================
cannam@126: */
cannam@126: 
cannam@126: static double
cannam@126: snr_test (SINGLE_TEST *test_data, int number, int converter, int verbose)
cannam@126: {	static float data [BUFFER_LEN + 1] ;
cannam@126: 	static float output [MAX_SPEC_LEN] ;
cannam@126: 
cannam@126: 	SRC_STATE	*src_state ;
cannam@126: 	SRC_DATA	src_data ;
cannam@126: 
cannam@126: 	double		output_peak, snr ;
cannam@126: 	int 		k, output_len, input_len, error ;
cannam@126: 
cannam@126: 	if (verbose != 0)
cannam@126: 	{	printf ("\tSignal-to-Noise Ratio Test %d.\n"
cannam@126: 				"\t=====================================\n", number) ;
cannam@126: 		printf ("\tFrequencies : [ ") ;
cannam@126: 		for (k = 0 ; k < test_data->freq_count ; k++)
cannam@126: 			printf ("%6.4f ", test_data->freqs [k]) ;
cannam@126: 
cannam@126: 		printf ("]\n\tSRC Ratio   : %8.4f\n", test_data->src_ratio) ;
cannam@126: 		}
cannam@126: 	else
cannam@126: 	{	printf ("\tSignal-to-Noise Ratio Test %d : ", number) ;
cannam@126: 		fflush (stdout) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	/* Set up the output array. */
cannam@126: 	if (test_data->src_ratio >= 1.0)
cannam@126: 	{	output_len = MAX_SPEC_LEN ;
cannam@126: 		input_len = (int) ceil (MAX_SPEC_LEN / test_data->src_ratio) ;
cannam@126: 		if (input_len > BUFFER_LEN)
cannam@126: 			input_len = BUFFER_LEN ;
cannam@126: 		}
cannam@126: 	else
cannam@126: 	{	input_len = BUFFER_LEN ;
cannam@126: 		output_len = (int) ceil (BUFFER_LEN * test_data->src_ratio) ;
cannam@126: 		output_len &= ((~0u) << 4) ;
cannam@126: 		if (output_len > MAX_SPEC_LEN)
cannam@126: 			output_len = MAX_SPEC_LEN ;
cannam@126: 		input_len = (int) ceil (output_len / test_data->src_ratio) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	memset (output, 0, sizeof (output)) ;
cannam@126: 
cannam@126: 	/* Generate input data array. */
cannam@126: 	gen_windowed_sines (test_data->freq_count, test_data->freqs, 1.0, data, input_len) ;
cannam@126: 
cannam@126: 	/* Perform sample rate conversion. */
cannam@126: 	if ((src_state = src_new (converter, 1, &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 = 1 ; /* Only one buffer worth of input. */
cannam@126: 
cannam@126: 	src_data.data_in = data ;
cannam@126: 	src_data.input_frames = input_len ;
cannam@126: 
cannam@126: 	src_data.src_ratio = test_data->src_ratio ;
cannam@126: 
cannam@126: 	src_data.data_out = output ;
cannam@126: 	src_data.output_frames = output_len ;
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: 	src_state = src_delete (src_state) ;
cannam@126: 
cannam@126: 	if (verbose != 0)
cannam@126: 		printf ("\tOutput Len  :   %ld\n", src_data.output_frames_gen) ;
cannam@126: 
cannam@126: 	if (abs (src_data.output_frames_gen - output_len) > 4)
cannam@126: 	{	printf ("\n\nLine %d : output data length should be %d.\n\n", __LINE__, output_len) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	/* Check output peak. */
cannam@126: 	output_peak = find_peak (output, src_data.output_frames_gen) ;
cannam@126: 
cannam@126: 	if (verbose != 0)
cannam@126: 		printf ("\tOutput Peak :   %6.4f\n", output_peak) ;
cannam@126: 
cannam@126: 	if (fabs (output_peak - test_data->peak_value) > 0.01)
cannam@126: 	{	printf ("\n\nLine %d : output peak (%6.4f) should be %6.4f\n\n", __LINE__, output_peak, test_data->peak_value) ;
cannam@126: 		save_oct_float ("snr_test.dat", data, BUFFER_LEN, output, output_len) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	/* Calculate signal-to-noise ratio. */
cannam@126: 	snr = calculate_snr (output, src_data.output_frames_gen, test_data->pass_band_peaks) ;
cannam@126: 
cannam@126: 	if (snr < 0.0)
cannam@126: 	{	/* An error occurred. */
cannam@126: 		save_oct_float ("snr_test.dat", data, BUFFER_LEN, output, src_data.output_frames_gen) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	if (verbose != 0)
cannam@126: 		printf ("\tSNR Ratio   :   %.2f dB\n", snr) ;
cannam@126: 
cannam@126: 	if (snr < test_data->snr)
cannam@126: 	{	printf ("\n\nLine %d : SNR (%5.2f) should be > %6.2f dB\n\n", __LINE__, snr, test_data->snr) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	if (verbose != 0)
cannam@126: 		puts ("\t-------------------------------------\n\tPass\n") ;
cannam@126: 	else
cannam@126: 		puts ("Pass") ;
cannam@126: 
cannam@126: 	return snr ;
cannam@126: } /* snr_test */
cannam@126: 
cannam@126: static double
cannam@126: find_peak (float *data, int len)
cannam@126: {	double 	peak = 0.0 ;
cannam@126: 	int		k = 0 ;
cannam@126: 
cannam@126: 	for (k = 0 ; k < len ; k++)
cannam@126: 		if (fabs (data [k]) > peak)
cannam@126: 			peak = fabs (data [k]) ;
cannam@126: 
cannam@126: 	return peak ;
cannam@126: } /* find_peak */
cannam@126: 
cannam@126: 
cannam@126: static double
cannam@126: find_attenuation (double freq, int converter, int verbose)
cannam@126: {	static float input	[BUFFER_LEN] ;
cannam@126: 	static float output [2 * BUFFER_LEN] ;
cannam@126: 
cannam@126: 	SRC_DATA	src_data ;
cannam@126: 	double 		output_peak ;
cannam@126: 	int			error ;
cannam@126: 
cannam@126: 	gen_windowed_sines (1, &freq, 1.0, input, BUFFER_LEN) ;
cannam@126: 
cannam@126: 	src_data.end_of_input = 1 ; /* Only one buffer worth of input. */
cannam@126: 
cannam@126: 	src_data.data_in = input ;
cannam@126: 	src_data.input_frames = BUFFER_LEN ;
cannam@126: 
cannam@126: 	src_data.src_ratio = 1.999 ;
cannam@126: 
cannam@126: 	src_data.data_out = output ;
cannam@126: 	src_data.output_frames = ARRAY_LEN (output) ;
cannam@126: 
cannam@126: 	if ((error = src_simple (&src_data, converter, 1)))
cannam@126: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	output_peak = find_peak (output, ARRAY_LEN (output)) ;
cannam@126: 
cannam@126: 	if (verbose)
cannam@126: 		printf ("\tFreq : %6f   InPeak : %6f    OutPeak : %6f   Atten : %6.2f dB\n",
cannam@126: 				freq, 1.0, output_peak, 20.0 * log10 (1.0 / output_peak)) ;
cannam@126: 
cannam@126: 	return 20.0 * log10 (1.0 / output_peak) ;
cannam@126: } /* find_attenuation */
cannam@126: 
cannam@126: static double
cannam@126: bandwidth_test (int converter, int verbose)
cannam@126: {	double	f1, f2, a1, a2 ;
cannam@126: 	double	freq, atten ;
cannam@126: 
cannam@126: 	f1 = 0.35 ;
cannam@126: 	a1 = find_attenuation (f1, converter, verbose) ;
cannam@126: 
cannam@126: 	f2 = 0.495 ;
cannam@126: 	a2 = find_attenuation (f2, converter, verbose) ;
cannam@126: 
cannam@126: 	if (a1 > 3.0 || a2 < 3.0)
cannam@126: 	{	printf ("\n\nLine %d : cannot bracket 3dB point.\n\n", __LINE__) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	while (a2 - a1 > 1.0)
cannam@126: 	{	freq = f1 + 0.5 * (f2 - f1) ;
cannam@126: 		atten = find_attenuation (freq, converter, verbose) ;
cannam@126: 
cannam@126: 		if (atten < 3.0)
cannam@126: 		{	f1 = freq ;
cannam@126: 			a1 = atten ;
cannam@126: 			}
cannam@126: 		else
cannam@126: 		{	f2 = freq ;
cannam@126: 			a2 = atten ;
cannam@126: 			} ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	freq = f1 + (3.0 - a1) * (f2 - f1) / (a2 - a1) ;
cannam@126: 
cannam@126: 	return 200.0 * freq ;
cannam@126: } /* bandwidth_test */
cannam@126: 
cannam@126: #else /* (HAVE_FFTW3) == 0 */
cannam@126: 
cannam@126: /* Alternative main function when librfftw is not available. */
cannam@126: 
cannam@126: int
cannam@126: main (void)
cannam@126: {	puts ("\n"
cannam@126: 		"****************************************************************\n"
cannam@126: 		" This test cannot be run without FFTW (http://www.fftw.org/).\n"
cannam@126: 		" Both the real and the complex versions of the library are\n"
cannam@126: 		" required.") ;
cannam@126: 	puts ("****************************************************************\n") ;
cannam@126: 
cannam@126: 	return 0 ;
cannam@126: } /* main */
cannam@126: 
cannam@126: #endif
cannam@126: