Changeset 87dfc3f for src/io

Timestamp:

Jul 16, 2012, 2:55:13 AM (12 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch, yinfft+

Children:

7839c61

Parents:

e3b5962

Message:

src/io/source_sndfile.c: add support for resampling

File:

• src/io/source_sndfile.c (modified) (6 diffs)

src/io/source_sndfile.c

@@ -30 +30 @@
 #include "fvec.h"
 
+#include "temporal/resampler.h"
+
 #define MAX_CHANNELS 6
 #define MAX_SIZE 4096
@@ -44 +46 @@
   uint_t scratch_size;
   smpl_t *scratch_data;
+
+  smpl_t ratio;
+
+#ifdef HAVE_SAMPLERATE
+  aubio_resampler_t *resampler;
+  fvec_t *resampled_data;
+#endif /* HAVE_SAMPLERATE */
 };
 
@@ -70 +79 @@
   }     
 
-  if (sfinfo.channels > MAX_CHANNELS) { 
+  if (sfinfo.channels > MAX_CHANNELS) {
     AUBIO_ERR("Not able to process more than %d channels\n", MAX_CHANNELS);
     return NULL;
@@ -80 +89 @@
   s->input_format     = sfinfo.format;
 
-  if (s->samplerate != s->input_samplerate) {
-    AUBIO_ERR("resampling not implemented yet\n");
+  s->ratio = s->samplerate/(float)s->input_samplerate;
+
+  if (s->ratio != 1) {
+#ifdef HAVE_SAMPLERATE
+    s->resampler = new_aubio_resampler(s->ratio, 0);
+    s->resampled_data = new_fvec(s->hop_size / s->ratio);
+    if (s-> ratio > 1) {
+      AUBIO_WRN("upsampling from %d to % d\n", s->input_samplerate, s->samplerate);
+    }
+#else
+    AUBIO_ERR("can not read %s at samplerate %dHz\n", s->path, s->samplerate);
+    AUBIO_ERR("aubio was compiled without aubio_resampler\n");
     return NULL;
+#endif /* HAVE_SAMPLERATE */
   }
-  
-  s->scratch_size = s->hop_size*s->input_channels;
+
+  s->scratch_size = s->hop_size * s->input_channels / s->ratio;
   /* allocate data for de/interleaving reallocated when needed. */
   if (s->scratch_size >= MAX_SIZE * MAX_CHANNELS) {
@@ -104 +124 @@
   read_frames = sf_read_float (s->handle, s->scratch_data, s->scratch_size);
 
-  aread = (int)FLOOR(read_frames/(float)input_channels);
+  smpl_t *data;
+
+#ifdef HAVE_SAMPLERATE
+  if (s->resampler) {
+    data = s->resampled_data->data;
+  } else
+#endif /* HAVE_SAMPLERATE */
+  {
+    data = read_data->data;
+  }
+  aread = (int)FLOOR(s->ratio * read_frames / (float)input_channels + .5);
 
   /* de-interleaving and down-mixing data  */
-  for (j = 0; j < aread; j++) {
-    read_data->data[j] = 0;
+  for (j = 0; j < read_frames; j++) {
+    data[j] = 0;
     for (i = 0; i < input_channels; i++) {
-      read_data->data[j] += (smpl_t)s->scratch_data[input_channels*j+i];
+      data[j] += (smpl_t)s->scratch_data[input_channels*j+i];
     }
-    read_data->data[j] /= (smpl_t)input_channels;
+    data[j] /= (smpl_t)input_channels;
   }
+
+#ifdef HAVE_SAMPLERATE
+  if (s->resampler) {
+    aubio_resampler_do(s->resampler, s->resampled_data, read_data);
+    data = s->resampled_data->data;
+  }
+#endif /* HAVE_SAMPLERATE */
+
   *read = aread;
 }
@@ -122 +160 @@
     AUBIO_ERR("Error closing file %s: %s", s->path, sf_strerror (NULL));
   }
+#ifdef HAVE_SAMPLERATE
+  if (s->resampler) {
+    del_aubio_resampler(s->resampler);
+  }
+  if (s->resampled_data) {
+    del_fvec(s->resampled_data);
+  }
+#endif /* HAVE_SAMPLERATE */
   AUBIO_FREE(s->scratch_data);
   AUBIO_FREE(s);