[io] add helpers to pad source output

diff --git a/src/io/ioutils.c b/src/io/ioutils.c
index f785b36..943731d 100644 (file)
--- a/src/io/ioutils.c
+++ b/src/io/ioutils.c
@@ -19,6 +19,7 @@
 */
 
 #include "aubio_priv.h"
+#include "fmat.h"
 
 uint_t
 aubio_io_validate_samplerate(const char_t *kind, const char_t *path, uint_t samplerate)
@@ -91,6 +92,30 @@ aubio_source_validate_input_channels(const char_t *kind, const char_t *path,
   return channels;
 }
 
+void
+aubio_source_pad_output (fvec_t *read_data, uint_t source_read)
+{
+  uint_t i = 0;
+  if (source_read < read_data->length) {
+    for (i = source_read; i < read_data->length; i++) {
+      read_data->data[i] = 0.;
+    }
+  }
+}
+
+void
+aubio_source_pad_multi_output (fmat_t *read_data,
+    uint_t source_channels, uint_t source_read) {
+  uint_t i, j;
+  if (source_read < read_data->length) {
+    for (i = 0; i < read_data->height; i++) {
+      for (j = source_read; j < read_data->length; j++) {
+        read_data->data[i][j] = 0.;
+      }
+    }
+  }
+}
+
 uint_t
 aubio_sink_validate_input_length(const char_t *kind, const char_t *path,
     uint_t max_size, uint_t write_data_length, uint_t write)

diff --git a/src/io/ioutils.h b/src/io/ioutils.h
index 995f2d2..f3cdf92 100644 (file)
--- a/src/io/ioutils.h
+++ b/src/io/ioutils.h
@@ -79,6 +79,26 @@ uint_t
 aubio_source_validate_input_channels(const char_t *kind, const char_t *path,
     uint_t source_channels, uint_t read_data_height);
 
+/** pad end of source output vector with zeroes
+
+  \param read_data   output vector to pad
+  \param source_read number of frames read
+
+*/
+void
+aubio_source_pad_output (fvec_t *read_data, uint_t source_read);
+
+/** pad end of source output matrix with zeroes
+
+  \param read_data   output matrix to pad
+  \param source_channels number of channels in the source
+  \param source_read number of frames read
+
+*/
+void
+aubio_source_pad_multi_output (fmat_t *read_data, uint_t source_channels,
+        uint_t source_read);
+
 /** validate length of sink input
 
   \param kind       the object kind to report on