Chris@0: /*
Chris@0: ** Copyright (C) 2007-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 Lesser General Public License as published by
Chris@0: ** the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
Chris@0: **
Chris@0: ** You should have received a copy of the GNU Lesser 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 <octave/oct.h>
Chris@0: 
Chris@0: #include "sndfile.h"
Chris@0: 
Chris@0: #define FOUR_GIG 		(0x100000000LL)
Chris@0: #define	BUFFER_FRAMES	8192
Chris@0: 
Chris@0: 
Chris@0: static int format_of_str (const std::string & fmt) ;
Chris@0: static void string_of_format (std::string & fmt, int format) ;
Chris@0: 
Chris@0: 
Chris@0: DEFUN_DLD (sfversion, args, nargout ,
Chris@0: "-*- texinfo -*-\n\
Chris@0: @deftypefn {Loadable Function} {@var{version} =} sfversion ()\n\
Chris@0: @cindex Reading sound files\n\
Chris@0: Return a string containing the libsndfile version.\n\
Chris@0: @seealso{sfread, sfwrite}\n\
Chris@0: @end deftypefn")
Chris@0: {	char buffer [256] ;
Chris@0: 	octave_value_list retval ;
Chris@0: 
Chris@0: 	/* Bail out if the input parameters are bad. */
Chris@0: 	if (args.length () != 0 || nargout > 1)
Chris@0: 	{	print_usage () ;
Chris@0: 		return retval ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	sf_command (NULL, SFC_GET_LIB_VERSION, buffer, sizeof (buffer)) ;
Chris@0: 
Chris@0: 	std::string version (buffer) ;
Chris@0: 
Chris@0: 	retval.append (version) ;
Chris@0: 	return retval ;
Chris@0: } /* sfversion */
Chris@0: 
Chris@0: 
Chris@0: DEFUN_DLD (sfread, args, nargout ,
Chris@0: "-*- texinfo -*-\n\
Chris@0: @deftypefn {Loadable Function} {@var{data},@var{srate},@var{format} =} sfread (@var{filename})\n\
Chris@0: @cindex Reading sound files\n\
Chris@0: Read a sound file from disk using libsndfile.\n\
Chris@0: @seealso{sfversion, sfwrite}\n\
Chris@0: @end deftypefn")
Chris@0: {	SNDFILE * file ;
Chris@0: 	SF_INFO sfinfo ;
Chris@0: 
Chris@0: 	octave_value_list retval ;
Chris@0: 
Chris@0: 	int nargin  = args.length () ;
Chris@0: 
Chris@0: 	/* Bail out if the input parameters are bad. */
Chris@0: 	if ((nargin != 1) || !args (0) .is_string () || nargout < 1 || nargout > 3)
Chris@0: 	{	print_usage () ;
Chris@0: 		return retval ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	memset (&sfinfo, 0, sizeof (sfinfo)) ;
Chris@0: 
Chris@0: 	std::string filename = args (0).string_value () ;
Chris@0: 
Chris@0: 	if ((file = sf_open (filename.c_str (), SFM_READ, &sfinfo)) == NULL)
Chris@0: 	{	error ("sfread: couldn't open file %s : %s", filename.c_str (), sf_strerror (NULL)) ;
Chris@0: 		return retval ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	if (sfinfo.frames > FOUR_GIG)
Chris@0: 		printf ("This is a really huge file (%lld frames).\nYou may run out of memory trying to load it.\n", (long long) sfinfo.frames) ;
Chris@0: 
Chris@0: 	dim_vector dim = dim_vector () ;
Chris@0: 	dim.resize (2) ;
Chris@0: 	dim (0) = sfinfo.frames ;
Chris@0: 	dim (1) = sfinfo.channels ;
Chris@0: 
Chris@0: 	/* Should I be using Matrix instead? */
Chris@0: 	NDArray out (dim, 0.0) ;
Chris@0: 
Chris@0: 	float buffer [BUFFER_FRAMES * sfinfo.channels] ;
Chris@0: 	int readcount ;
Chris@0: 	sf_count_t total = 0 ;
Chris@0: 
Chris@0: 	do
Chris@0: 	{	readcount = sf_readf_float (file, buffer, BUFFER_FRAMES) ;
Chris@0: 
Chris@0: 		/* Make sure we don't read more frames than we allocated. */
Chris@0: 		if (total + readcount > sfinfo.frames)
Chris@0: 			readcount = sfinfo.frames - total ;
Chris@0: 
Chris@0: 		for (int ch = 0 ; ch < sfinfo.channels ; ch++)
Chris@0: 		{	for (int k = 0 ; k < readcount ; k++)
Chris@0: 				out (total + k, ch) = buffer [k * sfinfo.channels + ch] ;
Chris@0: 			} ;
Chris@0: 
Chris@0: 		total += readcount ;
Chris@0: 	} while (readcount > 0 && total < sfinfo.frames) ;
Chris@0: 
Chris@0: 	retval.append (out.squeeze ()) ;
Chris@0: 
Chris@0: 	if (nargout >= 2)
Chris@0: 		retval.append ((octave_uint32) sfinfo.samplerate) ;
Chris@0: 
Chris@0: 	if (nargout >= 3)
Chris@0: 	{	std::string fmt ("") ;
Chris@0: 		string_of_format (fmt, sfinfo.format) ;
Chris@0: 		retval.append (fmt) ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	/* Clean up. */
Chris@0: 	sf_close (file) ;
Chris@0: 
Chris@0: 	return retval ;
Chris@0: } /* sfread */
Chris@0: 
Chris@0: DEFUN_DLD (sfwrite, args, nargout ,
Chris@0: "-*- texinfo -*-\n\
Chris@0: @deftypefn {Function File} sfwrite (@var{filename},@var{data},@var{srate},@var{format})\n\
Chris@0: Write a sound file to disk using libsndfile.\n\
Chris@0: @seealso{sfread, sfversion}\n\
Chris@0: @end deftypefn\n\
Chris@0: ")
Chris@0: {	SNDFILE * file ;
Chris@0: 	SF_INFO sfinfo ;
Chris@0: 
Chris@0:     octave_value_list retval ;
Chris@0: 
Chris@0:     int nargin  = args.length () ;
Chris@0: 
Chris@0:     /* Bail out if the input parameters are bad. */
Chris@0:     if (nargin != 4 || !args (0).is_string () || !args (1).is_real_matrix ()
Chris@0: 			|| !args (2).is_real_scalar () || !args (3).is_string ()
Chris@0: 			|| nargout != 0)
Chris@0: 	{	print_usage () ;
Chris@0: 		return retval ;
Chris@0:     	} ;
Chris@0: 
Chris@0:     std::string filename = args (0).string_value () ;
Chris@0:     std::string format = args (3).string_value () ;
Chris@0: 
Chris@0: 	memset (&sfinfo, 0, sizeof (sfinfo)) ;
Chris@0: 
Chris@0: 	sfinfo.format = format_of_str (format) ;
Chris@0: 	if (sfinfo.format == 0)
Chris@0: 	{	error ("Bad format '%s'", format.c_str ()) ;
Chris@0: 		return retval ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	sfinfo.samplerate = lrint (args (2).scalar_value ()) ;
Chris@0: 	if (sfinfo.samplerate < 1)
Chris@0: 	{	error ("Bad sample rate : %d.\n", sfinfo.samplerate) ;
Chris@0: 		return retval ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	Matrix data = args (1).matrix_value () ;
Chris@0: 	long rows = args (1).rows () ;
Chris@0: 	long cols = args (1).columns () ;
Chris@0: 
Chris@0: 	if (cols > rows)
Chris@0: 	{	error ("Audio data should have one column per channel, but supplied data "
Chris@0: 				"has %ld rows and %ld columns.\n", rows, cols) ;
Chris@0: 		return retval ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	sfinfo.channels = cols ;
Chris@0: 
Chris@0:     if ((file = sf_open (filename.c_str (), SFM_WRITE, &sfinfo)) == NULL)
Chris@0: 	{	error ("Couldn't open file %s : %s", filename.c_str (), sf_strerror (NULL)) ;
Chris@0: 		return retval ;
Chris@0:     	} ;
Chris@0: 
Chris@0: 	float buffer [BUFFER_FRAMES * sfinfo.channels] ;
Chris@0: 	int writecount ;
Chris@0: 	long total = 0 ;
Chris@0: 
Chris@0: 	do
Chris@0: 	{
Chris@0: 		writecount = BUFFER_FRAMES ;
Chris@0: 
Chris@0: 		/* Make sure we don't read more frames than we allocated. */
Chris@0: 		if (total + writecount > rows)
Chris@0: 			writecount = rows - total ;
Chris@0: 
Chris@0: 		for (int ch = 0 ; ch < sfinfo.channels ; ch++)
Chris@0: 		{	for (int k = 0 ; k < writecount ; k++)
Chris@0: 				buffer [k * sfinfo.channels + ch] = data (total + k, ch) ;
Chris@0: 			} ;
Chris@0: 
Chris@0: 		if (writecount > 0)
Chris@0: 			sf_writef_float (file, buffer, writecount) ;
Chris@0: 
Chris@0: 		total += writecount ;
Chris@0: 	} while (writecount > 0 && total < rows) ;
Chris@0: 
Chris@0:     /* Clean up. */
Chris@0:     sf_close (file) ;
Chris@0: 
Chris@0:     return retval ;
Chris@0: } /* sfwrite */
Chris@0: 
Chris@0: 
Chris@0: static void
Chris@0: str_split (const std::string & str, const std::string & delim, std::vector <std::string> & output)
Chris@0: {
Chris@0:     unsigned int offset = 0 ;
Chris@0:     size_t delim_index = 0 ;
Chris@0: 
Chris@0:     delim_index = str.find (delim, offset) ;
Chris@0: 
Chris@0:     while (delim_index != std::string::npos)
Chris@0:     {
Chris@0:         output.push_back (str.substr(offset, delim_index - offset)) ;
Chris@0:         offset += delim_index - offset + delim.length () ;
Chris@0:         delim_index = str.find (delim, offset) ;
Chris@0:     }
Chris@0: 
Chris@0:     output.push_back (str.substr (offset)) ;
Chris@0: } /* str_split */
Chris@0: 
Chris@0: static int
Chris@0: hash_of_str (const std::string & str)
Chris@0: {
Chris@0: 	int hash = 0 ;
Chris@0: 
Chris@0: 	for (unsigned k = 0 ; k < str.length () ; k++)
Chris@0: 		hash = (hash * 3) + tolower (str [k]) ;
Chris@0: 
Chris@0: 	return hash ;
Chris@0: } /* hash_of_str */
Chris@0: 
Chris@0: static int
Chris@0: major_format_of_hash (const std::string & str)
Chris@0: {	int hash ;
Chris@0: 
Chris@0: 	hash = hash_of_str (str) ;
Chris@0: 
Chris@0: 	switch (hash)
Chris@0: 	{
Chris@0: 		case 0x5c8 : /* 'wav' */ return SF_FORMAT_WAV ;
Chris@0: 		case 0xf84 : /* 'aiff' */ return SF_FORMAT_AIFF ;
Chris@0: 		case 0x198 : /* 'au' */ return SF_FORMAT_AU ;
Chris@0: 		case 0x579 : /* 'paf' */ return SF_FORMAT_PAF ;
Chris@0: 		case 0x5e5 : /* 'svx' */ return SF_FORMAT_SVX ;
Chris@0: 		case 0x1118 : /* 'nist' */ return SF_FORMAT_NIST ;
Chris@0: 		case 0x5d6 : /* 'voc' */ return SF_FORMAT_VOC ;
Chris@0: 		case 0x324a : /* 'ircam' */ return SF_FORMAT_IRCAM ;
Chris@0: 		case 0x505 : /* 'w64' */ return SF_FORMAT_W64 ;
Chris@0: 		case 0x1078 : /* 'mat4' */ return SF_FORMAT_MAT4 ;
Chris@0: 		case 0x1079 : /* 'mat5' */ return SF_FORMAT_MAT5 ;
Chris@0: 		case 0x5b8 : /* 'pvf' */ return SF_FORMAT_PVF ;
Chris@0: 		case 0x1d1 : /* 'xi' */ return SF_FORMAT_XI ;
Chris@0: 		case 0x56f : /* 'htk' */ return SF_FORMAT_HTK ;
Chris@0: 		case 0x5aa : /* 'sds' */ return SF_FORMAT_SDS ;
Chris@0: 		case 0x53d : /* 'avr' */ return SF_FORMAT_AVR ;
Chris@0: 		case 0x11d0 : /* 'wavx' */ return SF_FORMAT_WAVEX ;
Chris@0: 		case 0x569 : /* 'sd2' */ return SF_FORMAT_SD2 ;
Chris@0: 		case 0x1014 : /* 'flac' */ return SF_FORMAT_FLAC ;
Chris@0: 		case 0x504 : /* 'caf' */ return SF_FORMAT_CAF ;
Chris@0: 		case 0x5f6 : /* 'wve' */ return SF_FORMAT_WVE ;
Chris@0: 		default : break ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	printf ("%s : hash '%s' -> 0x%x\n", __func__, str.c_str (), hash) ;
Chris@0: 
Chris@0: 	return 0 ;
Chris@0: } /* major_format_of_hash */
Chris@0: 
Chris@0: static int
Chris@0: minor_format_of_hash (const std::string & str)
Chris@0: {	int hash ;
Chris@0: 
Chris@0: 	hash = hash_of_str (str) ;
Chris@0: 
Chris@0: 	switch (hash)
Chris@0: 	{
Chris@0: 		case 0x1085 : /* 'int8' */ return SF_FORMAT_PCM_S8 ;
Chris@0: 		case 0x358a : /* 'uint8' */ return SF_FORMAT_PCM_U8 ;
Chris@0: 		case 0x31b0 : /* 'int16' */ return SF_FORMAT_PCM_16 ;
Chris@0: 		case 0x31b1 : /* 'int24' */ return SF_FORMAT_PCM_24 ;
Chris@0: 		case 0x31b2 : /* 'int32' */ return SF_FORMAT_PCM_32 ;
Chris@0: 		case 0x3128 : /* 'float' */ return SF_FORMAT_FLOAT ;
Chris@0: 		case 0x937d : /* 'double' */ return SF_FORMAT_DOUBLE ;
Chris@0: 		case 0x11bd : /* 'ulaw' */ return SF_FORMAT_ULAW ;
Chris@0: 		case 0xfa1 : /* 'alaw' */ return SF_FORMAT_ALAW ;
Chris@0: 		case 0xfc361 : /* 'ima_adpcm' */ return SF_FORMAT_IMA_ADPCM ;
Chris@0: 		case 0x5739a : /* 'ms_adpcm' */ return SF_FORMAT_MS_ADPCM ;
Chris@0: 		case 0x9450 : /* 'gsm610' */ return SF_FORMAT_GSM610 ;
Chris@0: 		case 0x172a3 : /* 'g721_32' */ return SF_FORMAT_G721_32 ;
Chris@0: 		case 0x172d8 : /* 'g723_24' */ return SF_FORMAT_G723_24 ;
Chris@0: 		case 0x172da : /* 'g723_40' */ return SF_FORMAT_G723_40 ;
Chris@0: 		default : break ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	printf ("%s : hash '%s' -> 0x%x\n", __func__, str.c_str (), hash) ;
Chris@0: 
Chris@0: 	return 0 ;
Chris@0: } /* minor_format_of_hash */
Chris@0: 
Chris@0: 
Chris@0: static const char *
Chris@0: string_of_major_format (int format)
Chris@0: {
Chris@0: 	switch (format & SF_FORMAT_TYPEMASK)
Chris@0: 	{
Chris@0: 		case SF_FORMAT_WAV : return "wav" ;
Chris@0: 		case SF_FORMAT_AIFF : return "aiff" ;
Chris@0: 		case SF_FORMAT_AU : return "au" ;
Chris@0: 		case SF_FORMAT_PAF : return "paf" ;
Chris@0: 		case SF_FORMAT_SVX : return "svx" ;
Chris@0: 		case SF_FORMAT_NIST : return "nist" ;
Chris@0: 		case SF_FORMAT_VOC : return "voc" ;
Chris@0: 		case SF_FORMAT_IRCAM : return "ircam" ;
Chris@0: 		case SF_FORMAT_W64 : return "w64" ;
Chris@0: 		case SF_FORMAT_MAT4 : return "mat4" ;
Chris@0: 		case SF_FORMAT_MAT5 : return "mat5" ;
Chris@0: 		case SF_FORMAT_PVF : return "pvf" ;
Chris@0: 		case SF_FORMAT_XI : return "xi" ;
Chris@0: 		case SF_FORMAT_HTK : return "htk" ;
Chris@0: 		case SF_FORMAT_SDS : return "sds" ;
Chris@0: 		case SF_FORMAT_AVR : return "avr" ;
Chris@0: 		case SF_FORMAT_WAVEX : return "wavx" ;
Chris@0: 		case SF_FORMAT_SD2 : return "sd2" ;
Chris@0: 		case SF_FORMAT_FLAC : return "flac" ;
Chris@0: 		case SF_FORMAT_CAF : return "caf" ;
Chris@0: 		case SF_FORMAT_WVE : return "wfe" ;
Chris@0: 		default : break ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	return "unknown" ;
Chris@0: } /* string_of_major_format */
Chris@0: 
Chris@0: static const char *
Chris@0: string_of_minor_format (int format)
Chris@0: {
Chris@0: 	switch (format & SF_FORMAT_SUBMASK)
Chris@0: 	{
Chris@0: 		case SF_FORMAT_PCM_S8 : return "int8" ;
Chris@0: 		case SF_FORMAT_PCM_U8 : return "uint8" ;
Chris@0: 		case SF_FORMAT_PCM_16 : return "int16" ;
Chris@0: 		case SF_FORMAT_PCM_24 : return "int24" ;
Chris@0: 		case SF_FORMAT_PCM_32 : return "int32" ;
Chris@0: 		case SF_FORMAT_FLOAT : return "float" ;
Chris@0: 		case SF_FORMAT_DOUBLE : return "double" ;
Chris@0: 		case SF_FORMAT_ULAW : return "ulaw" ;
Chris@0: 		case SF_FORMAT_ALAW : return "alaw" ;
Chris@0: 		case SF_FORMAT_IMA_ADPCM : return "ima_adpcm" ;
Chris@0: 		case SF_FORMAT_MS_ADPCM : return "ms_adpcm" ;
Chris@0: 		case SF_FORMAT_GSM610 : return "gsm610" ;
Chris@0: 		case SF_FORMAT_G721_32 : return "g721_32" ;
Chris@0: 		case SF_FORMAT_G723_24 : return "g723_24" ;
Chris@0: 		case SF_FORMAT_G723_40 : return "g723_40" ;
Chris@0: 		default : break ;
Chris@0: 		} ;
Chris@0: 
Chris@0: 	return "unknown" ;
Chris@0: } /* string_of_minor_format */
Chris@0: 
Chris@0: static int
Chris@0: format_of_str (const std::string & fmt)
Chris@0: {
Chris@0: 	std::vector <std::string> split ;
Chris@0: 
Chris@0: 	str_split (fmt, "-", split) ;
Chris@0: 
Chris@0: 	if (split.size () != 2)
Chris@0: 		return 0 ;
Chris@0: 
Chris@0: 	int major_fmt = major_format_of_hash (split.at (0)) ;
Chris@0: 	if (major_fmt == 0)
Chris@0: 		return 0 ;
Chris@0: 
Chris@0: 	int minor_fmt = minor_format_of_hash (split.at (1)) ;
Chris@0: 	if (minor_fmt == 0)
Chris@0: 		return 0 ;
Chris@0: 
Chris@0: 	return major_fmt | minor_fmt ;
Chris@0: } /* format_of_str */
Chris@0: 
Chris@0: static void
Chris@0: string_of_format (std::string & fmt, int format)
Chris@0: {
Chris@0: 	char buffer [64] ;
Chris@0: 
Chris@0: 	snprintf (buffer, sizeof (buffer), "%s-%s", string_of_major_format (format), string_of_minor_format (format)) ;
Chris@0: 
Chris@0: 	fmt = buffer ;
Chris@0: 
Chris@0: 	return ;
Chris@0: } /* string_of_format */