ref: 4b1e7e79aaeafa5ee853c5202262b4ad87d56ba6
dir: /ext/sndfileio.c/
/*
Copyright (C) 2003 Paul Brossier
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include <string.h>
#include <sndfile.h>
#include "aubio_priv.h"
#include "sample.h"
#include "sndfileio.h"
#include "mathutils.h"
#define MAX_CHANNELS 6
#define MAX_SIZE 4096
struct _aubio_sndfile_t {
SNDFILE *handle;
int samplerate;
int channels;
int format;
float *tmpdata; /** scratch pad for interleaving/deinterleaving. */
int size; /** store the size to check if realloc needed */
};
aubio_sndfile_t * new_aubio_sndfile_ro(const char* outputname) {
aubio_sndfile_t * f = AUBIO_NEW(aubio_sndfile_t);
SF_INFO sfinfo;
AUBIO_MEMSET(&sfinfo, 0, sizeof (sfinfo));
if (! (f->handle = sf_open (outputname, SFM_READ, &sfinfo))) {
AUBIO_ERR("Unable to open input file %s.\n", outputname);
AUBIO_ERR("%s\n",sf_strerror (NULL)); /* libsndfile err msg */
return NULL;
}
if (sfinfo.channels > MAX_CHANNELS) {
AUBIO_ERR("Not able to process more than %d channels\n", MAX_CHANNELS);
return NULL;
}
f->size = MAX_SIZE*sfinfo.channels;
f->tmpdata = AUBIO_ARRAY(float,f->size);
/* get input specs */
f->samplerate = sfinfo.samplerate;
f->channels = sfinfo.channels;
f->format = sfinfo.format;
return f;
}
int aubio_sndfile_open_wo(aubio_sndfile_t * f, const char* inputname) {
SF_INFO sfinfo;
memset (&sfinfo, 0, sizeof (sfinfo));
/* define file output spec */
sfinfo.samplerate = f->samplerate;
sfinfo.channels = f->channels;
sfinfo.format = f->format;
if (! (f->handle = sf_open (inputname, SFM_WRITE, &sfinfo))) {
AUBIO_ERR("Not able to open output file %s.\n", inputname);
AUBIO_ERR("%s\n",sf_strerror (NULL)); /* libsndfile err msg */
AUBIO_QUIT(AUBIO_FAIL);
}
if (sfinfo.channels > MAX_CHANNELS) {
AUBIO_ERR("Not able to process more than %d channels\n", MAX_CHANNELS);
AUBIO_QUIT(AUBIO_FAIL);
}
f->size = MAX_SIZE*sfinfo.channels;
f->tmpdata = AUBIO_ARRAY(float,f->size);
return AUBIO_OK;
}
/* setup file struct from existing one */
aubio_sndfile_t * new_aubio_sndfile_wo(aubio_sndfile_t * fmodel, const char *outputname) {
aubio_sndfile_t * f = AUBIO_NEW(aubio_sndfile_t);
f->samplerate = fmodel->samplerate;
f->channels = fmodel->channels;
f->format = fmodel->format;
aubio_sndfile_open_wo(f, outputname);
return f;
}
/* return 0 if properly closed, 1 otherwise */
int del_aubio_sndfile(aubio_sndfile_t * f) {
if (sf_close(f->handle)) {
AUBIO_ERR("Error closing file.");
puts (sf_strerror (NULL));
return 1;
}
AUBIO_FREE(f->tmpdata);
AUBIO_FREE(f);
//AUBIO_DBG("File closed.\n");
return 0;
}
/**************************************************************
*
* Read write methods
*
*/
/* read frames from file in data
* return the number of frames actually read */
int aubio_sndfile_read(aubio_sndfile_t * f, int frames, fvec_t * read) {
sf_count_t read_frames;
int i,j, channels = f->channels;
int nsamples = frames*channels;
int aread;
float *pread;
/* allocate data for de/interleaving reallocated when needed. */
if (nsamples >= f->size) {
AUBIO_ERR("Maximum aubio_sndfile_read buffer size exceeded.");
return -1;
/*
AUBIO_FREE(f->tmpdata);
f->tmpdata = AUBIO_ARRAY(float,nsamples);
*/
}
//f->size = nsamples;
/* do actual reading */
read_frames = sf_read_float (f->handle, f->tmpdata, nsamples);
aread = (int)FLOOR(read_frames/(float)channels);
/* de-interleaving data */
for (i=0; i<channels; i++) {
pread = fvec_get_channel(read,i);
for (j=0; j<aread; j++) {
pread[j] = f->tmpdata[channels*j+i];
}
}
return aread;
}
/* write 'frames' samples to file from data
* return the number of frames actually written
*/
int aubio_sndfile_write(aubio_sndfile_t * f, int frames, fvec_t * write) {
sf_count_t written_frames = 0;
int i, j, channels = f->channels;
int nsamples = channels*frames;
float *pwrite;
/* allocate data for de/interleaving reallocated when needed. */
if (nsamples >= f->size) {
AUBIO_ERR("Maximum aubio_sndfile_write buffer size exceeded.");
return -1;
/*
AUBIO_FREE(f->tmpdata);
f->tmpdata = AUBIO_ARRAY(float,nsamples);
*/
}
//f->size = nsamples;
/* interleaving data */
for (i=0; i<channels; i++) {
pwrite = fvec_get_channel(write,i);
for (j=0; j<frames; j++) {
f->tmpdata[channels*j+i] = pwrite[j];
}
}
written_frames = sf_write_float (f->handle, f->tmpdata, nsamples);
return written_frames/channels;
}
/*******************************************************************
*
* Get object info
*
*/
uint_t aubio_sndfile_channels(aubio_sndfile_t * f) {
return f->channels;
}
uint_t aubio_sndfile_samplerate(aubio_sndfile_t * f) {
return f->samplerate;
}
void aubio_sndfile_info(aubio_sndfile_t * f) {
AUBIO_DBG("srate : %d\n", f->samplerate);
AUBIO_DBG("channels : %d\n", f->channels);
AUBIO_DBG("format : %d\n", f->format);
}