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