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