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   SF_INFO sfinfo;
  50
  51   if (path == NULL) {
  52     AUBIO_ERR("Aborted opening null path\n");
  53     return NULL;
  54   }
  55
  56   s->samplerate = samplerate;
  57   s->max_size = MAX_SIZE;
  58   s->channels = 1;
  59   s->path = path;
  60
  61   /* set output format */
  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     AUBIO_FREE(s);
  74     return NULL;
  75   }
  76
  77   s->scratch_size = s->max_size*s->channels;
  78   /* allocate data for de/interleaving reallocated when needed. */
  79   if (s->scratch_size >= MAX_SIZE * MAX_CHANNELS) {
  80     AUBIO_ERR("%d x %d exceeds maximum aubio_sink_sndfile buffer size %d\n",
  81         s->max_size, s->channels, MAX_CHANNELS * MAX_CHANNELS);
  82     AUBIO_FREE(s);
  83     return NULL;
  84   }
  85   s->scratch_data = AUBIO_ARRAY(float,s->scratch_size);
  86
  87   return s;
  88 }
  89
  90 void aubio_sink_sndfile_do(aubio_sink_sndfile_t *s, fvec_t * write_data, uint_t write){
  91   uint_t i, j,  channels = s->channels;
  92   int nsamples = channels*write;
  93   smpl_t *pwrite;
  94   sf_count_t written_frames;
  95
  96   if (write > s->max_size) {
  97     AUBIO_WRN("trying to write %d frames, but only %d can be written at a time",
  98       write, s->max_size);
  99     write = s->max_size;
 100   }
 101
 102   /* interleaving data  */
 103   for ( i = 0; i < channels; i++) {
 104     pwrite = (smpl_t *)write_data->data;
 105     for (j = 0; j < write; j++) {
 106       s->scratch_data[channels*j+i] = pwrite[j];
 107     }
 108   }
 109
 110   written_frames = sf_write_float (s->handle, s->scratch_data, nsamples);
 111   if (written_frames/channels != write) {
 112     AUBIO_WRN("trying to write %d frames to %s, but only %d could be written",
 113       write, s->path, (uint_t)written_frames);
 114   }
 115   return;
 116 }
 117
 118 void del_aubio_sink_sndfile(aubio_sink_sndfile_t * s){
 119   if (!s) return;
 120   if (sf_close(s->handle)) {
 121     AUBIO_ERR("Error closing file %s: %s", s->path, sf_strerror (NULL));
 122   }
 123   AUBIO_FREE(s->scratch_data);
 124   AUBIO_FREE(s);
 125 }
 126
 127 #endif /* HAVE_SNDFILE */