[sink_wavwrite] check fseek and fwrite return values

[aubio.git] / src / io / source.c

diff --git a/src/io/source.c b/src/io/source.c
index dd8e072..5a2e493 100644 (file)
--- a/src/io/source.c
+++ b/src/io/source.c
@@ -18,37 +18,144 @@
 
 */
 
-#include "config.h"
 #include "aubio_priv.h"
 #include "fvec.h"
+#include "fmat.h"
 #include "io/source.h"
-#ifdef __APPLE__
-#include "io/source_apple_audio_file.h"
-#endif /* __APPLE__ */
+#ifdef HAVE_LIBAV
+#include "io/source_avcodec.h"
+#endif /* HAVE_LIBAV */
+#ifdef HAVE_SOURCE_APPLE_AUDIO
+#include "io/source_apple_audio.h"
+#endif /* HAVE_SOURCE_APPLE_AUDIO */
+#ifdef HAVE_SNDFILE
+#include "io/source_sndfile.h"
+#endif /* HAVE_SNDFILE */
+#ifdef HAVE_WAVREAD
+#include "io/source_wavread.h"
+#endif /* HAVE_WAVREAD */
+
+typedef void (*aubio_source_do_t)(aubio_source_t * s, fvec_t * data, uint_t * read);
+typedef void (*aubio_source_do_multi_t)(aubio_source_t * s, fmat_t * data, uint_t * read);
+typedef uint_t (*aubio_source_get_samplerate_t)(aubio_source_t * s);
+typedef uint_t (*aubio_source_get_channels_t)(aubio_source_t * s);
+typedef uint_t (*aubio_source_get_duration_t)(aubio_source_t * s);
+typedef uint_t (*aubio_source_seek_t)(aubio_source_t * s, uint_t seek);
+typedef uint_t (*aubio_source_close_t)(aubio_source_t * s);
+typedef void (*del_aubio_source_t)(aubio_source_t * s);
 
 struct _aubio_source_t { 
   void *source;
+  aubio_source_do_t s_do;
+  aubio_source_do_multi_t s_do_multi;
+  aubio_source_get_samplerate_t s_get_samplerate;
+  aubio_source_get_channels_t s_get_channels;
+  aubio_source_get_duration_t s_get_duration;
+  aubio_source_seek_t s_seek;
+  aubio_source_close_t s_close;
+  del_aubio_source_t s_del;
 };
 
-aubio_source_t * new_aubio_source(char_t * uri, uint_t hop_size, uint_t samplerate) {
+aubio_source_t * new_aubio_source(const char_t * uri, uint_t samplerate, uint_t hop_size) {
   aubio_source_t * s = AUBIO_NEW(aubio_source_t);
-#ifdef __APPLE__
-  s->source= (void *)new_aubio_source_apple_audio(uri, hop_size, samplerate);
-#endif /* __APPLE__ */
-  if (s->source == NULL) return NULL;
-  return s;
+#ifdef HAVE_LIBAV
+  s->source = (void *)new_aubio_source_avcodec(uri, samplerate, hop_size);
+  if (s->source) {
+    s->s_do = (aubio_source_do_t)(aubio_source_avcodec_do);
+    s->s_do_multi = (aubio_source_do_multi_t)(aubio_source_avcodec_do_multi);
+    s->s_get_channels = (aubio_source_get_channels_t)(aubio_source_avcodec_get_channels);
+    s->s_get_samplerate = (aubio_source_get_samplerate_t)(aubio_source_avcodec_get_samplerate);
+    s->s_get_duration = (aubio_source_get_duration_t)(aubio_source_avcodec_get_duration);
+    s->s_seek = (aubio_source_seek_t)(aubio_source_avcodec_seek);
+    s->s_close = (aubio_source_close_t)(aubio_source_avcodec_close);
+    s->s_del = (del_aubio_source_t)(del_aubio_source_avcodec);
+    return s;
+  }
+#endif /* HAVE_LIBAV */
+#ifdef HAVE_SOURCE_APPLE_AUDIO
+  s->source = (void *)new_aubio_source_apple_audio(uri, samplerate, hop_size);
+  if (s->source) {
+    s->s_do = (aubio_source_do_t)(aubio_source_apple_audio_do);
+    s->s_do_multi = (aubio_source_do_multi_t)(aubio_source_apple_audio_do_multi);
+    s->s_get_channels = (aubio_source_get_channels_t)(aubio_source_apple_audio_get_channels);
+    s->s_get_samplerate = (aubio_source_get_samplerate_t)(aubio_source_apple_audio_get_samplerate);
+    s->s_get_duration = (aubio_source_get_duration_t)(aubio_source_apple_audio_get_duration);
+    s->s_seek = (aubio_source_seek_t)(aubio_source_apple_audio_seek);
+    s->s_close = (aubio_source_close_t)(aubio_source_apple_audio_close);
+    s->s_del = (del_aubio_source_t)(del_aubio_source_apple_audio);
+    return s;
+  }
+#endif /* HAVE_SOURCE_APPLE_AUDIO */
+#ifdef HAVE_SNDFILE
+  s->source = (void *)new_aubio_source_sndfile(uri, samplerate, hop_size);
+  if (s->source) {
+    s->s_do = (aubio_source_do_t)(aubio_source_sndfile_do);
+    s->s_do_multi = (aubio_source_do_multi_t)(aubio_source_sndfile_do_multi);
+    s->s_get_channels = (aubio_source_get_channels_t)(aubio_source_sndfile_get_channels);
+    s->s_get_samplerate = (aubio_source_get_samplerate_t)(aubio_source_sndfile_get_samplerate);
+    s->s_get_duration = (aubio_source_get_duration_t)(aubio_source_sndfile_get_duration);
+    s->s_seek = (aubio_source_seek_t)(aubio_source_sndfile_seek);
+    s->s_close = (aubio_source_close_t)(aubio_source_sndfile_close);
+    s->s_del = (del_aubio_source_t)(del_aubio_source_sndfile);
+    return s;
+  }
+#endif /* HAVE_SNDFILE */
+#ifdef HAVE_WAVREAD
+  s->source = (void *)new_aubio_source_wavread(uri, samplerate, hop_size);
+  if (s->source) {
+    s->s_do = (aubio_source_do_t)(aubio_source_wavread_do);
+    s->s_do_multi = (aubio_source_do_multi_t)(aubio_source_wavread_do_multi);
+    s->s_get_channels = (aubio_source_get_channels_t)(aubio_source_wavread_get_channels);
+    s->s_get_samplerate = (aubio_source_get_samplerate_t)(aubio_source_wavread_get_samplerate);
+    s->s_get_duration = (aubio_source_get_duration_t)(aubio_source_wavread_get_duration);
+    s->s_seek = (aubio_source_seek_t)(aubio_source_wavread_seek);
+    s->s_close = (aubio_source_close_t)(aubio_source_wavread_close);
+    s->s_del = (del_aubio_source_t)(del_aubio_source_wavread);
+    return s;
+  }
+#endif /* HAVE_WAVREAD */
+#if !defined(HAVE_WAVREAD) && \
+  !defined(HAVE_LIBAV) && \
+  !defined(HAVE_SOURCE_APPLE_AUDIO) && \
+  !defined(HAVE_SNDFILE)
+  AUBIO_ERROR("source: failed creating with %s at %dHz with hop size %d"
+     " (no source built-in)\n", uri, samplerate, hop_size);
+#endif
+  del_aubio_source(s);
+  return NULL;
 }
 
 void aubio_source_do(aubio_source_t * s, fvec_t * data, uint_t * read) {
-#ifdef __APPLE__
-  aubio_source_apple_audio_do((aubio_source_apple_audio_t *)s->source, data, read);
-#endif /* __APPLE__ */
+  s->s_do((void *)s->source, data, read);
+}
+
+void aubio_source_do_multi(aubio_source_t * s, fmat_t * data, uint_t * read) {
+  s->s_do_multi((void *)s->source, data, read);
+}
+
+uint_t aubio_source_close(aubio_source_t * s) {
+  return s->s_close((void *)s->source);
 }
 
 void del_aubio_source(aubio_source_t * s) {
-#ifdef __APPLE__
-  del_aubio_source_apple_audio((aubio_source_apple_audio_t *)s->source);
-#endif /* __APPLE__ */
+  AUBIO_ASSERT(s);
+  if (s->s_del && s->source)
+    s->s_del((void *)s->source);
   AUBIO_FREE(s);
 }
 
+uint_t aubio_source_get_samplerate(aubio_source_t * s) {
+  return s->s_get_samplerate((void *)s->source);
+}
+
+uint_t aubio_source_get_channels(aubio_source_t * s) {
+  return s->s_get_channels((void *)s->source);
+}
+
+uint_t aubio_source_get_duration(aubio_source_t *s) {
+  return s->s_get_duration((void *)s->source);
+}
+
+uint_t aubio_source_seek (aubio_source_t * s, uint_t seek ) {
+  return s->s_seek((void *)s->source, seek);
+}