cannam@85: [+ AutoGen5 template c +]
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 "sfconfig.h"
cannam@85: 
cannam@85: #include <stdio.h>
cannam@85: #include <stdlib.h>
cannam@85: 
cannam@85: #if HAVE_UNISTD_H
cannam@85: #include <unistd.h>
cannam@85: #endif
cannam@85: 
cannam@85: #if (HAVE_DECL_S_IRGRP == 0)
cannam@85: #include <sf_unistd.h>
cannam@85: #endif
cannam@85: 
cannam@85: #include <string.h>
cannam@85: #include <math.h>
cannam@85: #include <time.h>
cannam@85: #include <fcntl.h>
cannam@85: #include <sys/stat.h>
cannam@85: 
cannam@85: #include <sndfile.h>
cannam@85: 
cannam@85: #ifndef		M_PI
cannam@85: #define		M_PI		3.14159265358979323846264338
cannam@85: #endif
cannam@85: 
cannam@85: /*
cannam@85: **	Neat solution to the Win32/OS2 binary file flage requirement.
cannam@85: **	If O_BINARY isn't already defined by the inclusion of the system
cannam@85: **	headers, set it to zero.
cannam@85: */
cannam@85: #ifndef O_BINARY
cannam@85: #define O_BINARY 0
cannam@85: #endif
cannam@85: 
cannam@85: #define	WRITE_FLAGS	(O_WRONLY | O_CREAT | O_TRUNC | O_BINARY)
cannam@85: #define	READ_FLAGS	(O_RDONLY | O_BINARY)
cannam@85: 
cannam@85: #if (defined (WIN32) || defined (_WIN32) || defined (__OS2__))
cannam@85: 	#define WRITE_PERMS	0777
cannam@85: #else
cannam@85: 	#define WRITE_PERMS	(S_IRUSR | S_IWUSR | S_IRGRP)
cannam@85: #endif
cannam@85: 
cannam@85: #define	BUFFER_SIZE		(1<<18)
cannam@85: #define	BLOCK_COUNT		(30)
cannam@85: #define	TEST_DURATION	(5)		/* 5 Seconds. */
cannam@85: 
cannam@85: typedef struct
cannam@85: {	double	write_rate ;
cannam@85: 	double	read_rate ;
cannam@85: } PERF_STATS ;
cannam@85: 
cannam@85: static void	*data = NULL ;
cannam@85: 
cannam@85: static void calc_raw_performance (PERF_STATS *stats) ;
cannam@85: 
cannam@85: [+ FOR data_type
cannam@85: +]static void	calc_[+ (get "type_name") +]_performance (int format, double read_rate, double write_rate) ;
cannam@85: [+ ENDFOR data_type
cannam@85: +]
cannam@85: 
cannam@85: static int cpu_is_big_endian (void) ;
cannam@85: 
cannam@85: static const char* get_subtype_str (int subtype) ;
cannam@85: 
cannam@85: int
cannam@85: main (int argc, char *argv [])
cannam@85: {	PERF_STATS	stats ;
cannam@85: 	char		buffer [256] = "Benchmarking " ;
cannam@85: 	int			format_major ;
cannam@85: 
cannam@85: 	if (! (data = malloc (BUFFER_SIZE * sizeof (double))))
cannam@85: 	{	perror ("Error : malloc failed") ;
cannam@85: 		exit (1) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	sf_command (NULL, SFC_GET_LIB_VERSION, buffer + strlen (buffer), sizeof (buffer) - strlen (buffer)) ;
cannam@85: 
cannam@85: 	puts (buffer) ;
cannam@85: 	memset (buffer, '-', strlen (buffer)) ;
cannam@85: 	puts (buffer) ;
cannam@85: 	printf ("Each test takes a little over %d seconds.\n\n", TEST_DURATION) ;
cannam@85: 
cannam@85: 	calc_raw_performance (&stats) ;
cannam@85: 
cannam@85: 	if (argc < 2 || strcmp ("--native-only", argv [1]) == 0)
cannam@85: 	{	puts ("\nNative endian I/O :") ;
cannam@85: 		format_major = cpu_is_big_endian () ? SF_FORMAT_AIFF : SF_FORMAT_WAV ;
cannam@85: 
cannam@85: 		calc_short_performance	(format_major | SF_FORMAT_PCM_16, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_int_performance	(format_major | SF_FORMAT_PCM_24, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_int_performance	(format_major | SF_FORMAT_PCM_32, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_float_performance	(format_major | SF_FORMAT_PCM_16, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_float_performance	(format_major | SF_FORMAT_PCM_24, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_float_performance	(format_major | SF_FORMAT_PCM_32, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_float_performance	(format_major | SF_FORMAT_FLOAT , stats.read_rate, stats.write_rate) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (argc < 2 || strcmp ("--swap-only", argv [1]) == 0)
cannam@85: 	{	puts ("\nEndian swapped I/O :") ;
cannam@85: 		format_major = cpu_is_big_endian () ? SF_FORMAT_WAV : SF_FORMAT_AIFF ;
cannam@85: 
cannam@85: 		calc_short_performance	(format_major | SF_FORMAT_PCM_16, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_int_performance	(format_major | SF_FORMAT_PCM_24, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_int_performance	(format_major | SF_FORMAT_PCM_32, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_float_performance	(format_major | SF_FORMAT_PCM_16, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_float_performance	(format_major | SF_FORMAT_PCM_24, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_float_performance	(format_major | SF_FORMAT_PCM_32, stats.read_rate, stats.write_rate) ;
cannam@85: 		calc_float_performance	(format_major | SF_FORMAT_FLOAT , stats.read_rate, stats.write_rate) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	puts ("") ;
cannam@85: 
cannam@85: 	free (data) ;
cannam@85: 
cannam@85: 	return 0 ;
cannam@85: } /* main */
cannam@85: 
cannam@85: /*==============================================================================
cannam@85: */
cannam@85: 
cannam@85: static void
cannam@85: calc_raw_performance (PERF_STATS *stats)
cannam@85: {	clock_t start_clock, clock_time ;
cannam@85: 	int fd, k, byte_count, retval, op_count ;
cannam@85: 	const char *filename ;
cannam@85: 
cannam@85: 	filename = "benchmark.dat" ;
cannam@85: 
cannam@85: 	byte_count = BUFFER_SIZE * sizeof (short) ;
cannam@85: 
cannam@85: 	/* Collect write stats */
cannam@85: 	printf ("    Raw write PCM_16  : ") ;
cannam@85: 	fflush (stdout) ;
cannam@85: 
cannam@85: 	clock_time = 0 ;
cannam@85: 	op_count = 0 ;
cannam@85: 	start_clock = clock () ;
cannam@85: 
cannam@85: 	while (clock_time < (CLOCKS_PER_SEC * TEST_DURATION))
cannam@85: 	{	if ((fd = open (filename, WRITE_FLAGS, WRITE_PERMS)) < 0)
cannam@85: 		{	printf ("Error : not able to open file : %s\n", filename) ;
cannam@85: 			perror ("") ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		for (k = 0 ; k < BLOCK_COUNT ; k++)
cannam@85: 		{	if ((retval = write (fd, data, byte_count)) != byte_count)
cannam@85: 			{	printf ("Error : write returned %d (should have been %d)\n", retval, byte_count) ;
cannam@85: 				exit (1) ;
cannam@85: 				} ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		close (fd) ;
cannam@85: 
cannam@85: 		clock_time = clock () - start_clock ;
cannam@85: 		op_count ++ ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	stats->write_rate = (1.0 * BUFFER_SIZE) * BLOCK_COUNT * op_count ;
cannam@85: 	stats->write_rate *= (1.0 * CLOCKS_PER_SEC) / clock_time ;
cannam@85: 	printf ("%10.0f samples per sec\n", stats->write_rate) ;
cannam@85: 
cannam@85: 	/* Collect read stats */
cannam@85: 	printf ("    Raw read  PCM_16  : ") ;
cannam@85: 	fflush (stdout) ;
cannam@85: 
cannam@85: 	clock_time = 0 ;
cannam@85: 	op_count = 0 ;
cannam@85: 	start_clock = clock () ;
cannam@85: 
cannam@85: 	while (clock_time < (CLOCKS_PER_SEC * TEST_DURATION))
cannam@85: 	{	if ((fd = open (filename, READ_FLAGS)) < 0)
cannam@85: 		{	printf ("Error : not able to open file : %s\n", filename) ;
cannam@85: 			perror ("") ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		for (k = 0 ; k < BLOCK_COUNT ; k++)
cannam@85: 		{	if ((retval = read (fd, data, byte_count)) != byte_count)
cannam@85: 			{	printf ("Error : write returned %d (should have been %d)\n", retval, byte_count) ;
cannam@85: 				exit (1) ;
cannam@85: 				} ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		close (fd) ;
cannam@85: 
cannam@85: 		clock_time = clock () - start_clock ;
cannam@85: 		op_count ++ ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	stats->read_rate = (1.0 * BUFFER_SIZE) * BLOCK_COUNT * op_count ;
cannam@85: 	stats->read_rate *= (1.0 * CLOCKS_PER_SEC) / clock_time ;
cannam@85: 	printf ("%10.0f samples per sec\n", stats->read_rate) ;
cannam@85: 
cannam@85: 	unlink (filename) ;
cannam@85: } /* calc_raw_performance */
cannam@85: 
cannam@85: /*------------------------------------------------------------------------------
cannam@85: */
cannam@85: 
cannam@85: [+ FOR data_type
cannam@85: +]static void
cannam@85: calc_[+ (get "type_name") +]_performance (int format, double read_rate, double write_rate)
cannam@85: {	SNDFILE *file ;
cannam@85: 	SF_INFO	sfinfo ;
cannam@85: 	clock_t start_clock, clock_time ;
cannam@85: 	double	performance ;
cannam@85: 	int k, item_count, retval, op_count ;
cannam@85: 	const char* subtype ;
cannam@85: 	[+ (get "type_name") +] *[+ (get "type_name") +]_data ;
cannam@85: 	const char *filename ;
cannam@85: 
cannam@85: 	filename = "benchmark.dat" ;
cannam@85: 	subtype = get_subtype_str (format & SF_FORMAT_SUBMASK) ;
cannam@85: 
cannam@85: 	[+ (get "type_name") +]_data = data ;
cannam@85: 	item_count = BUFFER_SIZE ;
cannam@85: 	for (k = 0 ; k < item_count ; k++)
cannam@85: 		[+ (get "type_name") +]_data [k] = [+ (get "multiplier") +] * sin (2 * M_PI * k / 32000.0) ;
cannam@85: 
cannam@85: 	/* Collect write stats */
cannam@85: 	printf ("    Write %-5s   to  %s : ", "[+ (get "type_name") +]", subtype) ;
cannam@85: 	fflush (stdout) ;
cannam@85: 
cannam@85: 	sfinfo.channels = 1 ;
cannam@85: 	sfinfo.format = format ;
cannam@85: 	sfinfo.frames = 1 ;
cannam@85: 	sfinfo.samplerate = 32000 ;
cannam@85: 
cannam@85: 	clock_time = 0 ;
cannam@85: 	op_count = 0 ;
cannam@85: 	start_clock = clock () ;
cannam@85: 
cannam@85: 	while (clock_time < (CLOCKS_PER_SEC * TEST_DURATION))
cannam@85: 	{	if (! (file = sf_open (filename, SFM_WRITE, &sfinfo)))
cannam@85: 		{	printf ("Error : not able to open file : %s\n", filename) ;
cannam@85: 			perror ("") ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		/* Turn off the addition of a PEAK chunk. */
cannam@85: 		sf_command (file, SFC_SET_ADD_PEAK_CHUNK, NULL, SF_FALSE) ;
cannam@85: 
cannam@85: 		for (k = 0 ; k < BLOCK_COUNT ; k++)
cannam@85: 		{	if ((retval = sf_write_[+ (get "type_name") +] (file, [+ (get "type_name") +]_data, item_count)) != item_count)
cannam@85: 			{	printf ("Error : sf_write_short returned %d (should have been %d)\n", retval, item_count) ;
cannam@85: 				exit (1) ;
cannam@85: 				} ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		sf_close (file) ;
cannam@85: 
cannam@85: 		clock_time = clock () - start_clock ;
cannam@85: 		op_count ++ ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	performance = (1.0 * BUFFER_SIZE) * BLOCK_COUNT * op_count ;
cannam@85: 	performance *= (1.0 * CLOCKS_PER_SEC) / clock_time ;
cannam@85: 	printf ("%6.2f%% of raw write\n", 100.0 * performance / write_rate) ;
cannam@85: 
cannam@85: 	/* Collect read stats */
cannam@85: 	printf ("    Read  %-5s  from %s : ", "[+ (get "type_name") +]", subtype) ;
cannam@85: 	fflush (stdout) ;
cannam@85: 
cannam@85: 	clock_time = 0 ;
cannam@85: 	op_count = 0 ;
cannam@85: 	start_clock = clock () ;
cannam@85: 
cannam@85: 	while (clock_time < (CLOCKS_PER_SEC * TEST_DURATION))
cannam@85: 	{	if (! (file = sf_open (filename, SFM_READ, &sfinfo)))
cannam@85: 		{	printf ("Error : not able to open file : %s\n", filename) ;
cannam@85: 			perror ("") ;
cannam@85: 			exit (1) ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		for (k = 0 ; k < BLOCK_COUNT ; k++)
cannam@85: 		{	if ((retval = sf_read_[+ (get "type_name") +] (file, [+ (get "type_name") +]_data, item_count)) != item_count)
cannam@85: 			{	printf ("Error : write returned %d (should have been %d)\n", retval, item_count) ;
cannam@85: 				exit (1) ;
cannam@85: 				} ;
cannam@85: 			} ;
cannam@85: 
cannam@85: 		sf_close (file) ;
cannam@85: 
cannam@85: 		clock_time = clock () - start_clock ;
cannam@85: 		op_count ++ ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	performance = (1.0 * item_count) * BLOCK_COUNT * op_count ;
cannam@85: 	performance *= (1.0 * CLOCKS_PER_SEC) / clock_time ;
cannam@85: 	printf ("%6.2f%% of raw read\n", 100.0 * performance / read_rate) ;
cannam@85: 
cannam@85: 	unlink (filename) ;
cannam@85: 
cannam@85: } /* calc_[+ (get "type_name") +]_performance */
cannam@85: [+ ENDFOR data_type
cannam@85: +]
cannam@85: 
cannam@85: /*==============================================================================
cannam@85: */
cannam@85: 
cannam@85: static int
cannam@85: cpu_is_big_endian (void)
cannam@85: {	unsigned char 	*cptr ;
cannam@85: 	int 			endtest ;
cannam@85: 
cannam@85: 	endtest = 0x12345678 ;
cannam@85: 
cannam@85: 	cptr = (unsigned char*) (&endtest) ;
cannam@85: 
cannam@85: 	if (cptr [0] == 0x12 && cptr [1] == 0x34 && cptr [3] == 0x78)
cannam@85: 		return SF_TRUE ;
cannam@85: 
cannam@85: 	return SF_FALSE ;
cannam@85: } /* cpu_is_big_endian */
cannam@85: 
cannam@85: static const char*
cannam@85: get_subtype_str (int subtype)
cannam@85: {	switch (subtype)
cannam@85: 	{	case SF_FORMAT_PCM_16 :
cannam@85: 				return "PCM_16" ;
cannam@85: 
cannam@85: 		case SF_FORMAT_PCM_24 :
cannam@85: 				return "PCM_24" ;
cannam@85: 
cannam@85: 		case SF_FORMAT_PCM_32 :
cannam@85: 				return "PCM_32" ;
cannam@85: 
cannam@85: 		case SF_FORMAT_FLOAT :
cannam@85: 				return "FLOAT " ;
cannam@85: 
cannam@85: 		case SF_FORMAT_DOUBLE :
cannam@85: 				return "DOUBLE" ;
cannam@85: 
cannam@85: 		default : break ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	return "UNKNOWN" ;
cannam@85: } /* get_subtype_str */
cannam@85: