add aubio_cleanup to clean fftw cached memory

add aubio_cleanup to clean fftw cached memory

diff --git a/src/mathutils.c b/src/mathutils.c
index 79b4def5ddc17fa006798170a3a8e1f4fbc30bfb..dfb401954ea09daec7d42c37d643d9583a79a830 100644 (file)
--- a/src/mathutils.c
+++ b/src/mathutils.c
@@ -22,6 +22,7 @@
 #include "aubio_priv.h"
 #include "sample.h"
 #include "mathutils.h"
+#include "config.h"
 
 void aubio_window(smpl_t *w, uint_t size, aubio_window_type wintype) {
   uint_t i;
@@ -453,3 +454,13 @@ void aubio_autocorr(fvec_t * input, fvec_t * output){
         }
 }
 
+void aubio_cleanup(void)
+{
+#if FFTW3_SUPPORT
+        fftw_cleanup();
+#else
+#if FFTW3F_SUPPORT
+        fftwf_cleanup();
+#endif
+#endif
+}

diff --git a/src/mathutils.h b/src/mathutils.h
index a66780dd5da2d8a7d4471f32e3ceb8b34dbde5e3..ef2def44ccdda5fda1ecc2eac913f1af8cd0a7f3 100644 (file)
--- a/src/mathutils.h
+++ b/src/mathutils.h
@@ -245,6 +245,14 @@ smpl_t aubio_level_detection(fvec_t * ibuf, smpl_t threshold);
  * calculate normalised autocorrelation function
  */
 void aubio_autocorr(fvec_t * input, fvec_t * output);
+/**
+ * clean up cached memory at the end of program
+ *
+ * use this function at the end of programs to purge all
+ * cached memory. so far this function is only used to clean
+ * fftw cache.
+ */
+void aubio_cleanup(void);
 
 #ifdef __cplusplus
 }