src/io/sink_sndfile.c

   1 /*
   2   Copyright (C) 2012 Paul Brossier <piem@aubio.org>
   3
   4   This file is part of aubio.
   5
   6   aubio is free software: you can redistribute it and/or modify
   7   it under the terms of the GNU General Public License as published by
   8   the Free Software Foundation, either version 3 of the License, or
   9   (at your option) any later version.
  10
  11   aubio is distributed in the hope that it will be useful,
  12   but WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14   GNU General Public License for more details.
  15
  16   You should have received a copy of the GNU General Public License
  17   along with aubio.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21
  22 #include "config.h"
  23
  24 #ifdef HAVE_SNDFILE
  25
  26 #include <sndfile.h>
  27
  28 #include "aubio_priv.h"
  29 #include "sink_sndfile.h"
  30 #include "fvec.h"
  31
  32 #define MAX_CHANNELS 6
  33 #define MAX_SIZE 4096
  34
  35 struct _aubio_sink_sndfile_t {
  36   uint_t samplerate;
  37   uint_t channels;
  38   char_t *path;
  39
  40   uint_t max_size;
  41
  42   SNDFILE *handle;
  43   uint_t scratch_size;
  44   smpl_t *scratch_data;
  45 };
  46
  47 aubio_sink_sndfile_t * new_aubio_sink_sndfile(char_t * path, uint_t samplerate) {
  48   aubio_sink_sndfile_t * s = AUBIO_NEW(aubio_sink_sndfile_t);
  49
  50   if (path == NULL) {
  51     AUBIO_ERR("Aborted opening null path\n");
  52     return NULL;
  53   }
  54
  55   s->samplerate = samplerate;
  56   s->max_size = MAX_SIZE;
  57   s->channels = 1;
  58   s->path = path;
  59
  60   /* set output format */
  61   SF_INFO sfinfo;
  62   AUBIO_MEMSET(&sfinfo, 0, sizeof (sfinfo));
  63   sfinfo.samplerate = s->samplerate;
  64   sfinfo.channels   = s->channels;
  65   sfinfo.format     = SF_FORMAT_WAV | SF_FORMAT_PCM_16;
  66
  67   /* try creating the file */
  68   s->handle = sf_open (s->path, SFM_WRITE, &sfinfo);
  69
  70   if (s->handle == NULL) {
  71     /* show libsndfile err msg */
  72     AUBIO_ERR("Failed opening %s. %s\n", s->path, sf_strerror (NULL));
  73     return NULL;
  74   }
  75
  76   s->scratch_size = s->max_size*s->channels;
  77   /* allocate data for de/interleaving reallocated when needed. */
  78   if (s->scratch_size >= MAX_SIZE * MAX_CHANNELS) {
  79     AUBIO_ERR("%d x %d exceeds maximum aubio_sink_sndfile buffer size %d\n",
  80         s->max_size, s->channels, MAX_CHANNELS * MAX_CHANNELS);
  81     return NULL;
  82   }
  83   s->scratch_data = AUBIO_ARRAY(float,s->scratch_size);
  84
  85   return s;
  86 }
  87
  88 void aubio_sink_sndfile_do(aubio_sink_sndfile_t *s, fvec_t * write_data, uint_t write){
  89   sf_count_t written_frames = 0;
  90   int i, j,     channels = s->channels;
  91   int nsamples = channels*write;
  92   smpl_t *pwrite;
  93
  94   if (write > s->max_size) {
  95     AUBIO_WRN("trying to write %d frames, but only %d can be written at a time",
  96       write, s->max_size);
  97     write = s->max_size;
  98   }
  99
 100   /* interleaving data  */
 101   for ( i = 0; i < channels; i++) {
 102     pwrite = (smpl_t *)write_data->data;
 103     for (j=0; j < write; j++) {
 104       s->scratch_data[channels*j+i] = pwrite[j];
 105     }
 106   }
 107
 108   uint_t written = sf_write_float (s->handle, s->scratch_data, nsamples);
 109   if (written/channels != write) {
 110     AUBIO_WRN("trying to write %d frames to %s, but only %d could be written",
 111       write, s->path, written);
 112   }
 113   return;
 114 }
 115
 116 void del_aubio_sink_sndfile(aubio_sink_sndfile_t * s){
 117   if (!s) return;
 118   if (sf_close(s->handle)) {
 119     AUBIO_ERR("Error closing file %s: %s", s->path, sf_strerror (NULL));
 120   }
 121   AUBIO_FREE(s->scratch_data);
 122   AUBIO_FREE(s);
 123 }
 124
 125 #endif /* HAVE_SNDFILE */