Chris@40: /*
Chris@40: ** Copyright (C) 1999-2011 Erik de Castro Lopo <erikd@mega-nerd.com>
Chris@40: **
Chris@40: ** This program is free software; you can redistribute it and/or modify
Chris@40: ** it under the terms of the GNU General Public License as published by
Chris@40: ** the Free Software Foundation; either version 2 of the License, or
Chris@40: ** (at your option) any later version.
Chris@40: **
Chris@40: ** This program is distributed in the hope that it will be useful,
Chris@40: ** but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@40: ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@40: ** GNU General Public License for more details.
Chris@40: **
Chris@40: ** You should have received a copy of the GNU General Public License
Chris@40: ** along with this program; if not, write to the Free Software
Chris@40: ** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
Chris@40: */
Chris@40: 
Chris@40: #include "sfconfig.h"
Chris@40: 
Chris@40: #include <stdio.h>
Chris@40: #include <stdlib.h>
Chris@40: 
Chris@40: #include <fcntl.h>
Chris@40: #include <math.h>
Chris@40: #include <string.h>
Chris@40: #include <errno.h>
Chris@40: #include <sys/stat.h>
Chris@40: 
Chris@40: #include <sndfile.h>
Chris@40: 
Chris@40: #include "utils.h"
Chris@40: 
Chris@40: static void vio_test (const char *fname, int format) ;
Chris@40: 
Chris@40: int
Chris@40: main (void)
Chris@40: {
Chris@40: 	vio_test ("vio_pcm16.wav", SF_FORMAT_WAV | SF_FORMAT_PCM_16) ;
Chris@40: 	vio_test ("vio_pcm24.aiff", SF_FORMAT_AIFF | SF_FORMAT_PCM_24) ;
Chris@40: 	vio_test ("vio_float.au", SF_FORMAT_AU | SF_FORMAT_FLOAT) ;
Chris@40: 	vio_test ("vio_pcm24.paf", SF_FORMAT_PAF | SF_FORMAT_PCM_24) ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* main */
Chris@40: 
Chris@40: /*==============================================================================
Chris@40: */
Chris@40: 
Chris@40: typedef struct
Chris@40: {	sf_count_t offset, length ;
Chris@40: 	unsigned char data [16 * 1024] ;
Chris@40: } VIO_DATA ;
Chris@40: 
Chris@40: static sf_count_t
Chris@40: vfget_filelen (void *user_data)
Chris@40: {	VIO_DATA *vf = (VIO_DATA *) user_data ;
Chris@40: 
Chris@40: 	return vf->length ;
Chris@40: } /* vfget_filelen */
Chris@40: 
Chris@40: static sf_count_t
Chris@40: vfseek (sf_count_t offset, int whence, void *user_data)
Chris@40: {	VIO_DATA *vf = (VIO_DATA *) user_data ;
Chris@40: 
Chris@40: 	switch (whence)
Chris@40: 	{	case SEEK_SET :
Chris@40: 			vf->offset = offset ;
Chris@40: 			break ;
Chris@40: 
Chris@40: 		case SEEK_CUR :
Chris@40: 			vf->offset = vf->offset + offset ;
Chris@40: 			break ;
Chris@40: 
Chris@40: 		case SEEK_END :
Chris@40: 			vf->offset = vf->length + offset ;
Chris@40: 			break ;
Chris@40: 		default :
Chris@40: 			break ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	return vf->offset ;
Chris@40: } /* vfseek */
Chris@40: 
Chris@40: static sf_count_t
Chris@40: vfread (void *ptr, sf_count_t count, void *user_data)
Chris@40: {	VIO_DATA *vf = (VIO_DATA *) user_data ;
Chris@40: 
Chris@40: 	/*
Chris@40: 	**	This will brack badly for files over 2Gig in length, but
Chris@40: 	**	is sufficient for testing.
Chris@40: 	*/
Chris@40: 	if (vf->offset + count > vf->length)
Chris@40: 		count = vf->length - vf->offset ;
Chris@40: 
Chris@40: 	memcpy (ptr, vf->data + vf->offset, count) ;
Chris@40: 	vf->offset += count ;
Chris@40: 
Chris@40: 	return count ;
Chris@40: } /* vfread */
Chris@40: 
Chris@40: static sf_count_t
Chris@40: vfwrite (const void *ptr, sf_count_t count, void *user_data)
Chris@40: {	VIO_DATA *vf = (VIO_DATA *) user_data ;
Chris@40: 
Chris@40: 	/*
Chris@40: 	**	This will break badly for files over 2Gig in length, but
Chris@40: 	**	is sufficient for testing.
Chris@40: 	*/
Chris@40: 	if (vf->offset >= SIGNED_SIZEOF (vf->data))
Chris@40: 		return 0 ;
Chris@40: 
Chris@40: 	if (vf->offset + count > SIGNED_SIZEOF (vf->data))
Chris@40: 		count = sizeof (vf->data) - vf->offset ;
Chris@40: 
Chris@40: 	memcpy (vf->data + vf->offset, ptr, (size_t) count) ;
Chris@40: 	vf->offset += count ;
Chris@40: 
Chris@40: 	if (vf->offset > vf->length)
Chris@40: 		vf->length = vf->offset ;
Chris@40: 
Chris@40: 	return count ;
Chris@40: } /* vfwrite */
Chris@40: 
Chris@40: static sf_count_t
Chris@40: vftell (void *user_data)
Chris@40: {	VIO_DATA *vf = (VIO_DATA *) user_data ;
Chris@40: 
Chris@40: 	return vf->offset ;
Chris@40: } /* vftell */
Chris@40: 
Chris@40: 
Chris@40: /*==============================================================================
Chris@40: */
Chris@40: 
Chris@40: static void
Chris@40: gen_short_data (short * data, int len, int start)
Chris@40: {	int k ;
Chris@40: 
Chris@40: 	for (k = 0 ; k < len ; k++)
Chris@40: 		data [k] = start + k ;
Chris@40: } /* gen_short_data */
Chris@40: 
Chris@40: 
Chris@40: static void
Chris@40: check_short_data (short * data, int len, int start, int line)
Chris@40: {	int k ;
Chris@40: 
Chris@40: 	for (k = 0 ; k < len ; k++)
Chris@40: 		if (data [k] != start + k)
Chris@40: 		{	printf ("\n\nLine %d : data [%d] = %d (should be %d).\n\n", line, k, data [k], start + k) ;
Chris@40: 			exit (1) ;
Chris@40: 			} ;
Chris@40: } /* gen_short_data */
Chris@40: 
Chris@40: /*------------------------------------------------------------------------------
Chris@40: */
Chris@40: 
Chris@40: static void
Chris@40: vio_test (const char *fname, int format)
Chris@40: {	static VIO_DATA vio_data ;
Chris@40: 	static short data [256] ;
Chris@40: 
Chris@40: 	SF_VIRTUAL_IO vio ;
Chris@40: 	SNDFILE * file ;
Chris@40: 	SF_INFO sfinfo ;
Chris@40: 
Chris@40: 	print_test_name ("virtual i/o test", fname) ;
Chris@40: 
Chris@40: 	/* Set up pointers to the locally defined functions. */
Chris@40: 	vio.get_filelen = vfget_filelen ;
Chris@40: 	vio.seek = vfseek ;
Chris@40: 	vio.read = vfread ;
Chris@40: 	vio.write = vfwrite ;
Chris@40: 	vio.tell = vftell ;
Chris@40: 
Chris@40: 	/* Set virtual file offset and length to zero. */
Chris@40: 	vio_data.offset = 0 ;
Chris@40: 	vio_data.length = 0 ;
Chris@40: 
Chris@40: 	memset (&sfinfo, 0, sizeof (sfinfo)) ;
Chris@40: 	sfinfo.format = format ;
Chris@40: 	sfinfo.channels = 2 ;
Chris@40: 	sfinfo.samplerate = 44100 ;
Chris@40: 
Chris@40: 	if ((file = sf_open_virtual (&vio, SFM_WRITE, &sfinfo, &vio_data)) == NULL)
Chris@40: 	{	printf ("\n\nLine %d : sf_open_write failed with error : ", __LINE__) ;
Chris@40: 		fflush (stdout) ;
Chris@40: 		puts (sf_strerror (NULL)) ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	if (vfget_filelen (&vio_data) < 0)
Chris@40: 	{	printf ("\n\nLine %d : vfget_filelen returned negative length.\n\n", __LINE__) ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	gen_short_data (data, ARRAY_LEN (data), 0) ;
Chris@40: 	sf_write_short (file, data, ARRAY_LEN (data)) ;
Chris@40: 
Chris@40: 	gen_short_data (data, ARRAY_LEN (data), 1) ;
Chris@40: 	sf_write_short (file, data, ARRAY_LEN (data)) ;
Chris@40: 
Chris@40: 	gen_short_data (data, ARRAY_LEN (data), 2) ;
Chris@40: 	sf_write_short (file, data, ARRAY_LEN (data)) ;
Chris@40: 
Chris@40: 	sf_close (file) ;
Chris@40: 
Chris@40: 	/* Now test read. */
Chris@40: 	memset (&sfinfo, 0, sizeof (sfinfo)) ;
Chris@40: 
Chris@40: 	vio_data.offset = 0 ;
Chris@40: 
Chris@40: 	if ((file = sf_open_virtual (&vio, SFM_READ, &sfinfo, &vio_data)) == NULL)
Chris@40: 	{	printf ("\n\nLine %d : sf_open_write failed with error : ", __LINE__) ;
Chris@40: 		fflush (stdout) ;
Chris@40: 		puts (sf_strerror (NULL)) ;
Chris@40: 
Chris@40: 		dump_data_to_file (fname, vio_data.data, vio_data.length) ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 
Chris@40: 	sf_read_short (file, data, ARRAY_LEN (data)) ;
Chris@40: 	check_short_data (data, ARRAY_LEN (data), 0, __LINE__) ;
Chris@40: 
Chris@40: 	sf_read_short (file, data, ARRAY_LEN (data)) ;
Chris@40: 	check_short_data (data, ARRAY_LEN (data), 1, __LINE__) ;
Chris@40: 
Chris@40: 	sf_read_short (file, data, ARRAY_LEN (data)) ;
Chris@40: 	check_short_data (data, ARRAY_LEN (data), 2, __LINE__) ;
Chris@40: 
Chris@40: 	sf_close (file) ;
Chris@40: 
Chris@40: 	puts ("ok") ;
Chris@40: } /* vio_test */
Chris@40: