fix memory leaks

applied fastlib#55

Author: egorsmkv

.gitignore

@@ -1,2 +1,6 @@
 *.DS_Store
 __pycache__
+.venv/
+*.egg-info
+*.so
+*.wis

src/fcwt/fcwt.cpp

@@ -37,6 +37,8 @@ limitations under the License.
 */
 
 #include "fcwt.h"
+#include <cassert>
+#include <cstring>
 
 Morlet::Morlet(float bandwidth) {
     four_wavelen = 0.9876f;
@@ -86,8 +88,8 @@ void Morlet::generate(float* real, float* imag, int size, float scale) {
 void Morlet::getWavelet(float scale, complex<float>* pwav, int pn) {
     int w = getSupport(scale);
 
-    float *real = (float*)malloc(sizeof(float)*max(w*2+1,pn));
-    float *imag = (float*)malloc(sizeof(float)*max(w*2+1,pn));
+    float *real = (float*)fftwf_malloc(sizeof(float)*max(w*2+1,pn));
+    float *imag = (float*)fftwf_malloc(sizeof(float)*max(w*2+1,pn));
     for(int t=0; t < max(w*2+1,pn); t++) {
         real[t] = 0;
         imag[t] = 0;
@@ -100,8 +102,8 @@ void Morlet::getWavelet(float scale, complex<float>* pwav, int pn) {
         pwav[t].imag(imag[t]);
     }
 
-	delete real;
-	delete imag;
+    fftwf_free(real);
+    fftwf_free(imag);
 };
 
 //==============================================================//
@@ -124,6 +126,10 @@ Scales::Scales(Wavelet *wav, SCALETYPE st, int afs, float af0, float af1, int af
 
 }
 
+Scales::~Scales() {
+    free(scales);
+}
+
 void Scales::getScales(float *pfreqs, int pnf) { 
     for(int i=0;i<pnf;i++) { 
         pfreqs[i]=scales[i]; 
@@ -321,7 +327,7 @@ void FCWT::create_FFT_optimization_plan(int maxsize, int flags) {
     for(int i=11; i<=nt; i++) {
         int n = 1 << i;
 
-        float *dat = (float*)malloc(sizeof(float)*n);
+        float *dat = (float*)fftwf_malloc(sizeof(float)*n);
         fftwf_complex *O1 = fftwf_alloc_complex(n);
         fftwf_complex *out = fftwf_alloc_complex(n);
 
@@ -347,9 +353,11 @@ void FCWT::create_FFT_optimization_plan(int maxsize, int flags) {
 
         fftwf_export_wisdom_to_filename(file_for);
 
-        free(dat);
+        fftwf_free(dat);
         fftwf_free(O1);
         fftwf_free(out);
+        fftwf_destroy_plan(p_for);
+        fftwf_destroy_plan(p_back);
 
         std::cout << "Optimization schemes for N: " << n << " have been calculated. Next time you use fCWT it will automatically choose the right optimization scheme based on number of threads and signal length." << std::endl;
     }
@@ -406,6 +414,11 @@ void FCWT::convolve(fftwf_plan p, fftwf_complex *Ihat, fftwf_complex *O1, comple
         fftbased(p, Ihat, O1, (float*)lastscalemem, wav->mother, newsize, scale, wav->imag_frequency, wav->doublesided);
         if(use_normalization) fft_normalize((complex<float>*)lastscalemem, newsize);
         memcpy(out, (complex<float>*)lastscalemem, sizeof(complex<float>)*size);
+        #ifdef _WIN32
+            _aligned_free(lastscalemem);
+        #else
+            free(lastscalemem);
+        #endif
     } else {
         if(!out) {
             std::cout << "OUT NOT A POINTER" << std::endl;

src/fcwt/fcwt.h

@@ -96,6 +96,7 @@ class Morlet : public Wavelet {
 class Scales {
 public:
     FCWT_LIBRARY_API Scales(Wavelet *pwav, SCALETYPE st, int fs, float f0, float f1, int fn);
+    FCWT_LIBRARY_API ~Scales();
 
     void FCWT_LIBRARY_API getScales(float *pfreqs, int pnf);
     void FCWT_LIBRARY_API getFrequencies(float *pfreqs, int pnf);
@@ -164,4 +165,4 @@ inline int find2power(int n)
     return(m);
 }
 
-#endif
+#endif