Chris@41: /*
Chris@41: ** Copyright (c) 2006-2016, Erik de Castro Lopo <erikd@mega-nerd.com>
Chris@41: ** All rights reserved.
Chris@41: **
Chris@41: ** This code is released under 2-clause BSD license. Please see the
Chris@41: ** file at : https://github.com/erikd/libsamplerate/blob/master/COPYING
Chris@41: */
Chris@41: 
Chris@41: #include "config.h"
Chris@41: 
Chris@41: #include <stdio.h>
Chris@41: #include <stdlib.h>
Chris@41: #include <math.h>
Chris@41: #include <string.h>
Chris@41: 
Chris@41: #include <samplerate.h>
Chris@41: 
Chris@41: #include "util.h"
Chris@41: 
Chris@41: #if HAVE_FFTW3
Chris@41: #include <fftw3.h>
Chris@41: #else
Chris@41: #define fftw_cleanup()
Chris@41: #endif
Chris@41: 
Chris@41: #define	BUFFER_LEN		(1 << 16)
Chris@41: 
Chris@41: static void varispeed_test (int converter, double target_snr) ;
Chris@41: 
Chris@41: int
Chris@41: main (void)
Chris@41: {
Chris@41: 	puts ("") ;
Chris@41: 	printf ("    Zero Order Hold interpolator    : ") ;
Chris@41: 	varispeed_test (SRC_ZERO_ORDER_HOLD, 10.0) ;
Chris@41: 
Chris@41: 	printf ("    Linear interpolator             : ") ;
Chris@41: 	varispeed_test (SRC_LINEAR, 10.0) ;
Chris@41: 
Chris@41: 	printf ("    Sinc interpolator               : ") ;
Chris@41: 	varispeed_test (SRC_SINC_FASTEST, 115.0) ;
Chris@41: 
Chris@41: 	fftw_cleanup () ;
Chris@41: 	puts ("") ;
Chris@41: 
Chris@41: 	return 0 ;
Chris@41: } /* main */
Chris@41: 
Chris@41: static void
Chris@41: varispeed_test (int converter, double target_snr)
Chris@41: {	static float input [BUFFER_LEN], output [BUFFER_LEN] ;
Chris@41: 	double sine_freq, snr ;
Chris@41: 
Chris@41: 	SRC_STATE	*src_state ;
Chris@41: 	SRC_DATA	src_data ;
Chris@41: 
Chris@41: 	int input_len, error ;
Chris@41: 
Chris@41: 	memset (input, 0, sizeof (input)) ;
Chris@41: 
Chris@41: 	input_len = ARRAY_LEN (input) / 2 ;
Chris@41: 
Chris@41: 	sine_freq = 0.0111 ;
Chris@41: 	gen_windowed_sines (1, &sine_freq, 1.0, input, input_len) ;
Chris@41: 
Chris@41: 	/* Perform sample rate conversion. */
Chris@41: 	if ((src_state = src_new (converter, 1, &error)) == NULL)
Chris@41: 	{	printf ("\n\nLine %d : src_new() failed : %s\n\n", __LINE__, src_strerror (error)) ;
Chris@41: 		exit (1) ;
Chris@41: 		} ;
Chris@41: 
Chris@41: 	src_data.end_of_input = 1 ;
Chris@41: 
Chris@41: 	src_data.data_in = input ;
Chris@41: 	src_data.input_frames = input_len ;
Chris@41: 
Chris@41: 	src_data.src_ratio = 3.0 ;
Chris@41: 
Chris@41: 	src_data.data_out = output ;
Chris@41: 	src_data.output_frames = ARRAY_LEN (output) ;
Chris@41: 
Chris@41: 	if ((error = src_set_ratio (src_state, 1.0 / src_data.src_ratio)))
Chris@41: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
Chris@41: 		exit (1) ;
Chris@41: 		} ;
Chris@41: 
Chris@41: 	if ((error = src_process (src_state, &src_data)))
Chris@41: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
Chris@41: 		printf ("  src_data.input_frames  : %ld\n", src_data.input_frames) ;
Chris@41: 		printf ("  src_data.output_frames : %ld\n\n", src_data.output_frames) ;
Chris@41: 		exit (1) ;
Chris@41: 		} ;
Chris@41: 
Chris@41: 	if (src_data.input_frames_used != input_len)
Chris@41: 	{	printf ("\n\nLine %d : unused input.\n", __LINE__) ;
Chris@41: 		printf ("\tinput_len         : %d\n", input_len) ;
Chris@41: 		printf ("\tinput_frames_used : %ld\n\n", src_data.input_frames_used) ;
Chris@41: 		exit (1) ;
Chris@41: 		} ;
Chris@41: 
Chris@41: 	/* Copy the last output to the input. */
Chris@41: 	memcpy (input, output, sizeof (input)) ;
Chris@41: 	reverse_data (input, src_data.output_frames_gen) ;
Chris@41: 
Chris@41: 	if ((error = src_reset (src_state)))
Chris@41: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
Chris@41: 		exit (1) ;
Chris@41: 		} ;
Chris@41: 
Chris@41: 	src_data.end_of_input = 1 ;
Chris@41: 
Chris@41: 	src_data.data_in = input ;
Chris@41: 	input_len = src_data.input_frames = src_data.output_frames_gen ;
Chris@41: 
Chris@41: 	src_data.data_out = output ;
Chris@41: 	src_data.output_frames = ARRAY_LEN (output) ;
Chris@41: 
Chris@41: 	if ((error = src_set_ratio (src_state, 1.0 / src_data.src_ratio)))
Chris@41: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
Chris@41: 		exit (1) ;
Chris@41: 		} ;
Chris@41: 
Chris@41: 	if ((error = src_process (src_state, &src_data)))
Chris@41: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
Chris@41: 		printf ("  src_data.input_frames  : %ld\n", src_data.input_frames) ;
Chris@41: 		printf ("  src_data.output_frames : %ld\n\n", src_data.output_frames) ;
Chris@41: 		exit (1) ;
Chris@41: 		} ;
Chris@41: 
Chris@41: 	if (src_data.input_frames_used != input_len)
Chris@41: 	{	printf ("\n\nLine %d : unused input.\n", __LINE__) ;
Chris@41: 		printf ("\tinput_len         : %d\n", input_len) ;
Chris@41: 		printf ("\tinput_frames_used : %ld\n\n", src_data.input_frames_used) ;
Chris@41: 		exit (1) ;
Chris@41: 		} ;
Chris@41: 
Chris@41: 	src_state = src_delete (src_state) ;
Chris@41: 
Chris@41: 	snr = calculate_snr (output, src_data.output_frames_gen, 1) ;
Chris@41: 
Chris@41: 	if (target_snr > snr)
Chris@41: 	{	printf ("\n\nLine %d : snr (%3.1f) does not meet target (%3.1f)\n\n", __LINE__, snr, target_snr) ;
Chris@41: 		save_oct_float ("varispeed.mat", input, src_data.input_frames, output, src_data.output_frames_gen) ;
Chris@41: 		exit (1) ;
Chris@41: 		} ;
Chris@41: 
Chris@41: 	puts ("ok") ;
Chris@41: 
Chris@41: 	return ;
Chris@41: } /* varispeed_test */
Chris@41: