cannam@126: /*
cannam@126: ** Copyright (c) 2003-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 <stdio.h>
cannam@126: #include <stdlib.h>
cannam@126: #include <string.h>
cannam@126: #include <math.h>
cannam@126: 
cannam@126: #include <samplerate.h>
cannam@126: 
cannam@126: #include "util.h"
cannam@126: 
cannam@126: #define	BUFFER_LEN		10000
cannam@126: #define CB_READ_LEN		256
cannam@126: 
cannam@126: static void callback_test (int converter, double ratio) ;
cannam@126: static void end_of_stream_test (int converter) ;
cannam@126: 
cannam@126: int
cannam@126: main (void)
cannam@126: {	static double src_ratios [] =
cannam@126: 	{	1.0, 0.099, 0.1, 0.33333333, 0.789, 1.0001, 1.9, 3.1, 9.9
cannam@126: 	} ;
cannam@126: 
cannam@126: 	int k ;
cannam@126: 
cannam@126: 	puts ("") ;
cannam@126: 
cannam@126: 	puts ("    Zero Order Hold interpolator :") ;
cannam@126: 	for (k = 0 ; k < ARRAY_LEN (src_ratios) ; k++)
cannam@126: 		callback_test (SRC_ZERO_ORDER_HOLD, src_ratios [k]) ;
cannam@126: 
cannam@126: 	puts ("    Linear interpolator :") ;
cannam@126: 	for (k = 0 ; k < ARRAY_LEN (src_ratios) ; k++)
cannam@126: 		callback_test (SRC_LINEAR, src_ratios [k]) ;
cannam@126: 
cannam@126: 	puts ("    Sinc interpolator :") ;
cannam@126: 	for (k = 0 ; k < ARRAY_LEN (src_ratios) ; k++)
cannam@126: 		callback_test (SRC_SINC_FASTEST, src_ratios [k]) ;
cannam@126: 
cannam@126: 	puts ("") ;
cannam@126: 
cannam@126: 	puts ("    End of stream test :") ;
cannam@126: 	end_of_stream_test (SRC_ZERO_ORDER_HOLD) ;
cannam@126: 	end_of_stream_test (SRC_LINEAR) ;
cannam@126: 	end_of_stream_test (SRC_SINC_FASTEST) ;
cannam@126: 
cannam@126: 	puts ("") ;
cannam@126: 	return 0 ;
cannam@126: } /* main */
cannam@126: 
cannam@126: /*=====================================================================================
cannam@126: */
cannam@126: 
cannam@126: typedef struct
cannam@126: {	int channels ;
cannam@126: 	long count, total ;
cannam@126: 	int end_of_data ;
cannam@126: 	float data [BUFFER_LEN] ;
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: 	if (pcb_data->total - pcb_data->count > CB_READ_LEN)
cannam@126: 		frames = CB_READ_LEN / pcb_data->channels ;
cannam@126: 	else
cannam@126: 		frames = (pcb_data->total - pcb_data->count) / pcb_data->channels ;
cannam@126: 
cannam@126: 	*data = pcb_data->data + pcb_data->count ;
cannam@126: 	pcb_data->count += frames ;
cannam@126: 
cannam@126: 	return frames ;
cannam@126: } /* test_callback_func */
cannam@126: 
cannam@126: 
cannam@126: static void
cannam@126: callback_test (int converter, double src_ratio)
cannam@126: {	static TEST_CB_DATA test_callback_data ;
cannam@126: 	static float output [BUFFER_LEN] ;
cannam@126: 
cannam@126: 	SRC_STATE	*src_state ;
cannam@126: 
cannam@126: 	long	read_count, read_total ;
cannam@126: 	int 	error ;
cannam@126: 
cannam@126: 	printf ("\tcallback_test    (SRC ratio = %6.4f) ........... ", src_ratio) ;
cannam@126: 	fflush (stdout) ;
cannam@126: 
cannam@126: 	test_callback_data.channels = 2 ;
cannam@126: 	test_callback_data.count = 0 ;
cannam@126: 	test_callback_data.end_of_data = 0 ;
cannam@126: 	test_callback_data.total = ARRAY_LEN (test_callback_data.data) ;
cannam@126: 
cannam@126: 	if ((src_state = src_callback_new (test_callback_func, converter, test_callback_data.channels, &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: 	do
cannam@126: 	{	/* We will be throwing away output data, so just grab as much as possible. */
cannam@126: 		read_count = ARRAY_LEN (output) / test_callback_data.channels ;
cannam@126: 		read_count = src_callback_read (src_state, src_ratio, read_count, output) ;
cannam@126: 		read_total += read_count ;
cannam@126: 		}
cannam@126: 	while (read_count > 0) ;
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 - ARRAY_LEN (test_callback_data.data)) > 2.0)
cannam@126: 	{	printf ("\n\nLine %d : input / output length mismatch.\n\n", __LINE__) ;
cannam@126: 		printf ("    input len  : %d\n", ARRAY_LEN (test_callback_data.data)) ;
cannam@126: 		printf ("    output len : %ld (should be %g +/- 2)\n\n", read_total,
cannam@126: 					floor (0.5 + src_ratio * ARRAY_LEN (test_callback_data.data))) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	puts ("ok") ;
cannam@126: 
cannam@126: 	return ;
cannam@126: } /* callback_test */
cannam@126: 
cannam@126: /*=====================================================================================
cannam@126: */
cannam@126: 
cannam@126: static long
cannam@126: eos_callback_func (void *cb_data, float **data)
cannam@126: {
cannam@126: 	TEST_CB_DATA *pcb_data ;
cannam@126: 	long frames ;
cannam@126: 
cannam@126: 	if (data == NULL)
cannam@126: 		return 0 ;
cannam@126: 
cannam@126: 	if ((pcb_data = cb_data) == NULL)
cannam@126: 		return 0 ;
cannam@126: 
cannam@126: 	/*
cannam@126: 	**	Return immediately if there is no more data.
cannam@126: 	**	In this case, the output pointer 'data' will not be set and
cannam@126: 	**	valgrind should not warn about it.
cannam@126: 	*/
cannam@126: 	if (pcb_data->end_of_data)
cannam@126: 		return 0 ;
cannam@126: 
cannam@126: 	if (pcb_data->total - pcb_data->count > CB_READ_LEN)
cannam@126: 		frames = CB_READ_LEN / pcb_data->channels ;
cannam@126: 	else
cannam@126: 		frames = (pcb_data->total - pcb_data->count) / pcb_data->channels ;
cannam@126: 
cannam@126: 	*data = pcb_data->data + pcb_data->count ;
cannam@126: 	pcb_data->count += frames ;
cannam@126: 
cannam@126: 	/*
cannam@126: 	**	Set end_of_data so that the next call to the callback function will
cannam@126: 	**	return zero ocunt without setting the 'data' pointer.
cannam@126: 	*/
cannam@126: 	if (pcb_data->total < 2 * pcb_data->count)
cannam@126: 		pcb_data->end_of_data = 1 ;
cannam@126: 
cannam@126: 	return frames ;
cannam@126: } /* eos_callback_data */
cannam@126: 
cannam@126: 
cannam@126: static void
cannam@126: end_of_stream_test (int converter)
cannam@126: {	static TEST_CB_DATA test_callback_data ;
cannam@126: 	static float output [BUFFER_LEN] ;
cannam@126: 
cannam@126: 	SRC_STATE	*src_state ;
cannam@126: 
cannam@126: 	double	src_ratio = 0.3 ;
cannam@126: 	long	read_count, read_total ;
cannam@126: 	int 	error ;
cannam@126: 
cannam@126: 	printf ("\t%-30s        ........... ", src_get_name (converter)) ;
cannam@126: 	fflush (stdout) ;
cannam@126: 
cannam@126: 	test_callback_data.channels = 2 ;
cannam@126: 	test_callback_data.count = 0 ;
cannam@126: 	test_callback_data.end_of_data = 0 ;
cannam@126: 	test_callback_data.total = ARRAY_LEN (test_callback_data.data) ;
cannam@126: 
cannam@126: 	if ((src_state = src_callback_new (eos_callback_func, converter, test_callback_data.channels, &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: 	do
cannam@126: 	{	/* We will be throwing away output data, so just grab as much as possible. */
cannam@126: 		read_count = ARRAY_LEN (output) / test_callback_data.channels ;
cannam@126: 		read_count = src_callback_read (src_state, src_ratio, read_count, output) ;
cannam@126: 		read_total += read_count ;
cannam@126: 		}
cannam@126: 	while (read_count > 0) ;
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 (test_callback_data.end_of_data == 0)
cannam@126: 	{	printf ("\n\nLine %d : test_callback_data.end_of_data should not be 0."
cannam@126: 				" This is a bug in the test.\n\n", __LINE__) ;
cannam@126: 		exit (1) ;
cannam@126: 		} ;
cannam@126: 
cannam@126: 	puts ("ok") ;
cannam@126: 	return ;
cannam@126: } /* end_of_stream_test */