cannam@85: /*
cannam@85: ** Copyright (C) 2002-2011 Erik de Castro Lopo <erikd@mega-nerd.com>
cannam@85: **
cannam@85: ** This program is free software; you can redistribute it and/or modify
cannam@85: ** it under the terms of the GNU General Public License as published by
cannam@85: ** the Free Software Foundation; either version 2 of the License, or
cannam@85: ** (at your option) any later version.
cannam@85: **
cannam@85: ** This program is distributed in the hope that it will be useful,
cannam@85: ** but WITHOUT ANY WARRANTY; without even the implied warranty of
cannam@85: ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
cannam@85: ** GNU General Public License for more details.
cannam@85: **
cannam@85: ** You should have received a copy of the GNU General Public License
cannam@85: ** along with this program; if not, write to the Free Software
cannam@85: ** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
cannam@85: */
cannam@85: 
cannam@85: #include <stdio.h>
cannam@85: #include <stdlib.h>
cannam@85: #include <math.h>
cannam@85: 
cannam@85: #include <samplerate.h>
cannam@85: 
cannam@85: #include "util.h"
cannam@85: 
cannam@85: #define	SHORT_BUFFER_LEN	2048
cannam@85: #define	LONG_BUFFER_LEN		((1 << 16) - 20)
cannam@85: 
cannam@85: static void simple_test (int converter) ;
cannam@85: static void stream_test (int converter, double ratio) ;
cannam@85: static void init_term_test (int converter, double ratio) ;
cannam@85: 
cannam@85: static int	next_block_length (int reset) ;
cannam@85: 
cannam@85: int
cannam@85: main (void)
cannam@85: {	static double src_ratios [] =
cannam@85: 	{	0.999900, 1.000100, 0.789012, 1.200000, 0.333333, 3.100000,
cannam@85: 		0.125000, 8.000000, 0.099900, 9.990000, 0.100000, 10.00000
cannam@85: 	} ;
cannam@85: 
cannam@85: 	int k ;
cannam@85: 
cannam@85: 	puts ("\n    Zero Order Hold interpolator:") ;
cannam@85: 
cannam@85: 	for (k = 0 ; k < ARRAY_LEN (src_ratios) ; k++)
cannam@85: 		init_term_test (SRC_ZERO_ORDER_HOLD, src_ratios [k]) ;
cannam@85: 	puts ("") ;
cannam@85: 	for (k = 0 ; k < ARRAY_LEN (src_ratios) ; k++)
cannam@85: 		stream_test (SRC_ZERO_ORDER_HOLD, src_ratios [k]) ;
cannam@85: 
cannam@85: 
cannam@85: 	puts ("\n    Linear interpolator:") ;
cannam@85: 	for (k = 0 ; k < ARRAY_LEN (src_ratios) ; k++)
cannam@85: 		init_term_test (SRC_LINEAR, src_ratios [k]) ;
cannam@85: 	puts ("") ;
cannam@85: 	for (k = 0 ; k < ARRAY_LEN (src_ratios) ; k++)
cannam@85: 		stream_test (SRC_LINEAR, src_ratios [k]) ;
cannam@85: 
cannam@85: 
cannam@85: 	puts ("\n    Sinc interpolator:") ;
cannam@85: 	for (k = 0 ; k < ARRAY_LEN (src_ratios) ; k++)
cannam@85: 		init_term_test (SRC_SINC_FASTEST, src_ratios [k]) ;
cannam@85: 	puts ("") ;
cannam@85: 	for (k = 0 ; k < ARRAY_LEN (src_ratios) ; k++)
cannam@85: 		stream_test (SRC_SINC_FASTEST, src_ratios [k]) ;
cannam@85: 
cannam@85: 	puts ("") ;
cannam@85: 
cannam@85: 	simple_test (SRC_SINC_FASTEST) ;
cannam@85: 
cannam@85: 	return 0 ;
cannam@85: } /* main */
cannam@85: 
cannam@85: static void
cannam@85: simple_test (int converter)
cannam@85: {
cannam@85: 	int ilen = 199030, olen = 1000, error ;
cannam@85: 
cannam@85: 	{
cannam@85: 		float in [ilen] ;
cannam@85: 		float out [olen] ;
cannam@85: 		double ratio = (1.0 * olen) / ilen ;
cannam@85: 		SRC_DATA src_data =
cannam@85: 		{	in, out,
cannam@85: 			ilen, olen,
cannam@85: 			0, 0, 0,
cannam@85: 			ratio
cannam@85: 		} ;
cannam@85: 
cannam@85: 		error = src_simple (&src_data, converter, 1) ;
cannam@85: 		if (error)
cannam@85: 		{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 	} ;
cannam@85: 
cannam@85:     return ;
cannam@85: } /* simple_test */
cannam@85: 
cannam@85: static void
cannam@85: init_term_test (int converter, double src_ratio)
cannam@85: {	static float input [SHORT_BUFFER_LEN], output [SHORT_BUFFER_LEN] ;
cannam@85: 
cannam@85: 	SRC_DATA	src_data ;
cannam@85: 
cannam@85: 	int k, input_len, output_len, error, terminate ;
cannam@85: 
cannam@85: 	printf ("\tinit_term_test   (SRC ratio = %7.4f) .......... ", src_ratio) ;
cannam@85: 	fflush (stdout) ;
cannam@85: 
cannam@85: 	/* Calculate maximun input and output lengths. */
cannam@85: 	if (src_ratio >= 1.0)
cannam@85: 	{	output_len = SHORT_BUFFER_LEN ;
cannam@85: 		input_len = (int) floor (SHORT_BUFFER_LEN / src_ratio) ;
cannam@85: 		}
cannam@85: 	else
cannam@85: 	{	input_len = SHORT_BUFFER_LEN ;
cannam@85: 		output_len = (int) floor (SHORT_BUFFER_LEN * src_ratio) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	/* Reduce input_len by 10 so output is longer than necessary. */
cannam@85: 	input_len -= 10 ;
cannam@85: 
cannam@85: 	for (k = 0 ; k < ARRAY_LEN (input) ; k++)
cannam@85: 		input [k] = 1.0 ;
cannam@85: 
cannam@85: 	if (output_len > SHORT_BUFFER_LEN)
cannam@85: 	{	printf ("\n\nLine %d : output_len > SHORT_BUFFER_LEN\n\n", __LINE__) ;
cannam@85: 		exit (1) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	src_data.data_in = input ;
cannam@85: 	src_data.input_frames = input_len ;
cannam@85: 
cannam@85: 	src_data.src_ratio = src_ratio ;
cannam@85: 
cannam@85: 	src_data.data_out = output ;
cannam@85: 	src_data.output_frames = SHORT_BUFFER_LEN ;
cannam@85: 
cannam@85: 	if ((error = src_simple (&src_data, converter, 1)))
cannam@85: 	{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@85: 		exit (1) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	terminate = (int) ceil ((src_ratio >= 1.0) ? 1 : 1.0 / src_ratio) ;
cannam@85: 
cannam@85: 	if (fabs (src_ratio * input_len - src_data.output_frames_gen) > terminate)
cannam@85: 	{	printf ("\n\nLine %d : Bad output frame count.\n\n", __LINE__) ;
cannam@85: 		printf ("\tterminate             : %d\n", terminate) ;
cannam@85: 		printf ("\tsrc_ratio             : %.4f\n", src_ratio) ;
cannam@85: 		printf ("\tinput_len             : %d\n"
cannam@85: 				"\tinput_len * src_ratio : %f\n", input_len, input_len * src_ratio) ;
cannam@85: 		printf ("\toutput_frames_gen     : %ld\n\n", src_data.output_frames_gen) ;
cannam@85: 		exit (1) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (abs (src_data.input_frames_used - input_len) > 1)
cannam@85: 	{	printf ("\n\nLine %d : input_frames_used should be %d, is %ld.\n\n",
cannam@85: 					 __LINE__, input_len, src_data.input_frames_used) ;
cannam@85: 		printf ("\tsrc_ratio  : %.4f\n", src_ratio) ;
cannam@85: 		printf ("\tinput_len  : %d\n\tinput_used : %ld\n\n", input_len, src_data.input_frames_used) ;
cannam@85: 		exit (1) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (fabs (output [0]) < 0.1)
cannam@85: 	{	printf ("\n\nLine %d : First output sample is bad.\n\n", __LINE__) ;
cannam@85: 		printf ("\toutput [0] == %f\n\n", output [0]) ;
cannam@85: 		exit (1) ;
cannam@85: 		}
cannam@85: 
cannam@85: 	puts ("ok") ;
cannam@85: 
cannam@85: 	return ;
cannam@85: } /* init_term_test */
cannam@85: 
cannam@85: static void
cannam@85: stream_test (int converter, double src_ratio)
cannam@85: {	static float input [LONG_BUFFER_LEN], output [LONG_BUFFER_LEN] ;
cannam@85: 
cannam@85: 	SRC_STATE	*src_state ;
cannam@85: 	SRC_DATA	src_data ;
cannam@85: 
cannam@85: 	int input_len, output_len, current_in, current_out ;
cannam@85: 	int k, error, terminate ;
cannam@85: 
cannam@85: 	printf ("\tstream_test      (SRC ratio = %7.4f) .......... ", src_ratio) ;
cannam@85: 	fflush (stdout) ;
cannam@85: 
cannam@85: /* Erik */
cannam@85: for (k = 0 ; k < LONG_BUFFER_LEN ; k++) input [k] = k * 1.0 ;
cannam@85: 
cannam@85: 	/* Calculate maximun input and output lengths. */
cannam@85: 	if (src_ratio >= 1.0)
cannam@85: 	{	output_len = LONG_BUFFER_LEN ;
cannam@85: 		input_len = (int) floor (LONG_BUFFER_LEN / src_ratio) ;
cannam@85: 		}
cannam@85: 	else
cannam@85: 	{	input_len = LONG_BUFFER_LEN ;
cannam@85: 		output_len = (int) floor (LONG_BUFFER_LEN * src_ratio) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	/* Reduce input_len by 10 so output is longer than necessary. */
cannam@85: 	input_len -= 20 ;
cannam@85: 
cannam@85: 	if (output_len > LONG_BUFFER_LEN)
cannam@85: 	{	printf ("\n\nLine %d : output_len > LONG_BUFFER_LEN\n\n", __LINE__) ;
cannam@85: 		exit (1) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	current_in = current_out = 0 ;
cannam@85: 
cannam@85: 	/* Perform sample rate conversion. */
cannam@85: 	if ((src_state = src_new (converter, 1, &error)) == NULL)
cannam@85: 	{	printf ("\n\nLine %d : src_new() failed : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@85: 		exit (1) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	src_data.end_of_input = 0 ; /* Set this later. */
cannam@85: 
cannam@85: 	src_data.data_in = input ;
cannam@85: 
cannam@85: 	src_data.src_ratio = src_ratio ;
cannam@85: 
cannam@85: 	src_data.data_out = output ;
cannam@85: 	src_data.output_frames = ARRAY_LEN (output) / 10 ;
cannam@85: 
cannam@85: 	terminate = 1 + (int) ceil ((src_ratio >= 1.0) ? src_ratio : 1.0 / src_ratio) ;
cannam@85: 
cannam@85: 	while (1)
cannam@85: 	{
cannam@85: 		src_data.input_frames = next_block_length (0) ;
cannam@85: 		src_data.input_frames = MIN (src_data.input_frames, input_len - current_in) ;
cannam@85: 
cannam@85: 		src_data.output_frames = ARRAY_LEN (output) - current_out ;
cannam@85: 		/*-Erik MIN (src_data.output_frames, output_len - current_out) ;-*/
cannam@85: 
cannam@85: 		src_data.end_of_input = (current_in >= input_len) ? 1 : 0 ;
cannam@85: 
cannam@85: 		if ((error = src_process (src_state, &src_data)))
cannam@85: 		{	printf ("\n\nLine %d : %s\n\n", __LINE__, src_strerror (error)) ;
cannam@85: 			printf ("  src_data.input_frames  : %ld\n", src_data.input_frames) ;
cannam@85: 			printf ("  src_data.output_frames : %ld\n\n", src_data.output_frames) ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		if (src_data.end_of_input && src_data.output_frames_gen == 0)
cannam@85: 			break ;
cannam@85: 
cannam@85: 		if (src_data.input_frames_used > src_data.input_frames)
cannam@85: 		{	printf ("\n\nLine %d : input_frames_used > input_frames\n\n", __LINE__) ;
cannam@85: 			printf ("  src_data.input_frames      : %ld\n", src_data.input_frames) ;
cannam@85: 			printf ("  src_data.input_frames_used : %ld\n", src_data.input_frames_used) ;
cannam@85: 			printf ("  src_data.output_frames     : %ld\n", src_data.output_frames) ;
cannam@85: 			printf ("  src_data.output_frames_gen : %ld\n\n", src_data.output_frames_gen) ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		if (src_data.input_frames_used < 0)
cannam@85: 		{	printf ("\n\nLine %d : input_frames_used (%ld) < 0\n\n", __LINE__, src_data.input_frames_used) ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		if (src_data.output_frames_gen < 0)
cannam@85: 		{	printf ("\n\nLine %d : output_frames_gen (%ld) < 0\n\n", __LINE__, src_data.output_frames_gen) ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		current_in	+= src_data.input_frames_used ;
cannam@85: 		current_out += src_data.output_frames_gen ;
cannam@85: 
cannam@85: 		if (current_in > input_len + terminate)
cannam@85: 		{	printf ("\n\nLine %d : current_in (%d) > input_len (%d + %d)\n\n", __LINE__, current_in, input_len, terminate) ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		if (current_out > output_len)
cannam@85: 		{	printf ("\n\nLine %d : current_out (%d) > output_len (%d)\n\n", __LINE__, current_out, output_len) ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		if (src_data.input_frames_used > input_len)
cannam@85: 		{	printf ("\n\nLine %d : input_frames_used (%ld) > %d\n\n", __LINE__, src_data.input_frames_used, input_len) ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		if (src_data.output_frames_gen > output_len)
cannam@85: 		{	printf ("\n\nLine %d : output_frames_gen (%ld) > %d\n\n", __LINE__, src_data.output_frames_gen, output_len) ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		if (src_data.data_in == NULL && src_data.output_frames_gen == 0)
cannam@85: 			break ;
cannam@85: 
cannam@85: 
cannam@85: 		src_data.data_in	+= src_data.input_frames_used ;
cannam@85: 		src_data.data_out	+= src_data.output_frames_gen ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	src_state = src_delete (src_state) ;
cannam@85: 
cannam@85: 	if (fabs (current_out - src_ratio * input_len) > terminate)
cannam@85: 	{	printf ("\n\nLine %d : bad output data length %d should be %2.1f +/- %d.\n", __LINE__,
cannam@85: 					current_out, src_ratio * input_len, terminate) ;
cannam@85: 		printf ("\tsrc_ratio  : %.4f\n", src_ratio) ;
cannam@85: 		printf ("\tinput_len  : %d\n\tinput_used : %d\n", input_len, current_in) ;
cannam@85: 		printf ("\toutput_len : %d\n\toutput_gen : %d\n\n", output_len, current_out) ;
cannam@85: 		exit (1) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (current_in != input_len)
cannam@85: 	{	printf ("\n\nLine %d : unused input.\n", __LINE__) ;
cannam@85: 		printf ("\tinput_len         : %d\n", input_len) ;
cannam@85: 		printf ("\tinput_frames_used : %d\n\n", current_in) ;
cannam@85: 		exit (1) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	puts ("ok") ;
cannam@85: 
cannam@85: 	return ;
cannam@85: } /* stream_test */
cannam@85: 
cannam@85: static int
cannam@85: next_block_length (int reset)
cannam@85: {	static int block_lengths [] = /* Should be an odd length. */
cannam@85: 	{	/*-2, 500, 5, 400, 10, 300, 20, 200, 50, 100, 70 -*/
cannam@85: 		5, 400, 10, 300, 20, 200, 50, 100, 70
cannam@85: 		} ;
cannam@85: 	static int block_len_index = 0 ;
cannam@85: 
cannam@85: 	if (reset)
cannam@85: 		block_len_index = 0 ;
cannam@85: 	else
cannam@85: 		block_len_index = (block_len_index + 1) % ARRAY_LEN (block_lengths) ;
cannam@85: 
cannam@85: 	return block_lengths [block_len_index] ;
cannam@85: } /* next_block_length */
cannam@85: