Merge branch 'feat/fft_cpp_interface' into 'master'

feat: dl_fft add C++ interface

See merge request ai/esp-dl!196

Author: BlueSkyB

test_apps/dl_fft/main/test_dl_fft.cpp

@@ -57,3 +57,14 @@ bool check_fft_results(const float *x, const float *gt, int size, float snr_thre
     }
     return pass;
 }
+
+bool check_is_same(const float *x, const float *gt, int size, float threshold)
+{
+    for (int i = 0; i < size; i++) {
+        if (fabs(x[i] - gt[i]) > threshold) {
+            return false;
+        }
+    }
+    printf("check is same pass\n");
+    return true;
+}

test_apps/dl_fft/main/test_fft.c

@@ -27,6 +27,8 @@ float get_snr(const float *x, const float *gt, int size);
 float get_rmse(const float *x, const float *gt, int size);
 
 bool check_fft_results(const float *x, const float *gt, int size, float snr_threshold, float rmse_threshold);
+
+bool check_is_same(const float *x, const float *gt, int size, float threshold);
 #ifdef __cplusplus
 }
 #endif

test_apps/dl_fft/main/test_fft.h

@@ -4,6 +4,7 @@ set(srcs        "dl_fft_f32.c"
                 "dl_fft_s16.c"
                 "dl_rfft_f32.c"
                 "dl_rfft_s16.c"
+                "dl_fft.hpp"
                 "base/dl_fft2r_fc32_ansi.c"
                 "base/dl_fft4r_fc32_ansi.c"
                 "base/dl_fft2r_sc16_ansi.c"

tools/dl_fft/CMakeLists.txt

@@ -6,28 +6,32 @@ The float FFT implementation is come from esp-dsp. And we further optimized the
 For int16 FFT, we recommend to use `dl_fft_s16_hp_run` or `dl_rfft_s16_hp_run` interface. `hp` means "high precision".
 
 ## Get Started
+
+### C interface
 ```
 
 #include "dl_fft.h"
 #include "dl_rfft.h"
 
 // float fft
 float  x[nfft*2];
+
+float *x = (float *)heap_caps_aligned_alloc(16, nfft * sizeof(float) *2, MALLOC_CAP_8BIT);
 dl_fft_f32_t *fft_handle = dl_fft_f32_init(nfft, MALLOC_CAP_8BIT);
 dl_fft_f32_run(fft_handle, x);
 dl_ifft_f32_run(fft_handle, x);
 dl_fft_f32_deinit(fft_handle);
 
 // float rfft
-float  x[nfft];
+float *x = (float *)heap_caps_aligned_alloc(16, nfft * sizeof(float), MALLOC_CAP_8BIT);
 dl_fft_f32_t *fft_handle = dl_rfft_f32_init(nfft, MALLOC_CAP_8BIT);
 dl_rfft_f32_run(fft_handle, x);
 dl_irfft_f32_run(fft_handle, x);
 dl_rfft_f32_deinit(fft_handle);
 
 // int16 fft
-int16_t  x[nfft*2];
-float  y[nfft*2];
+int16_t *x= (float *)heap_caps_aligned_alloc(16, nfft * sizeof(int16_t) * 2, MALLOC_CAP_8BIT);
+float *y = (float *)heap_caps_aligned_alloc(16, nfft * sizeof(float) *2, MALLOC_CAP_8BIT);
 int in_exponent = -15;  //  float y = x * 2^in_exponent;
 int fft_exponent;
 int ifft_exponent;
@@ -38,8 +42,8 @@ dl_short_to_float(x, nfft, ifft_exponent, y); // convert output from int16_t to
 dl_fft_s16_deinit(fft_handle);
 
 // int16 rfft
-int16_t  x[nfft];
-float  y[nfft];
+int16_t *x= (float *)heap_caps_aligned_alloc(16, nfft * sizeof(int16_t), MALLOC_CAP_8BIT);
+float *y = (float *)heap_caps_aligned_alloc(16, nfft * sizeof(float), MALLOC_CAP_8BIT);
 int in_exponent = -15;  //  float y = x * 2^in_exponent;
 int fft_exponent;
 int ifft_exponent;
@@ -53,6 +57,32 @@ dl_rfft_s16_deinit(fft_handle);
 ```
 Please refer to [dl_fft.h](./dl_fft.h) and [dl_rfft.h](./dl_rfft.h) for more details. 
 
+
+### C++ interface:
+```
+float *x1 = (float *)heap_caps_aligned_alloc(16, nfft * sizeof(float) *2, MALLOC_CAP_8BIT);
+int16_t *x2= (float *)heap_caps_aligned_alloc(16, nfft * sizeof(int16_t)*2, MALLOC_CAP_8BIT);
+FFT *fft = FFT::get_instance();
+
+# float 
+fft->fft(x1, nfft);
+fft->ifft(x1, nfft);
+fft->rfft(x1, nfft);
+fft->irfft(x1, nfft);
+
+#int16_t
+int in_exponent=-15;
+int out_exponent;
+fft->fft_hp(x2, nfft, in_exponent, &out_exponent);
+fft->ifft_hp(x2, nfft, in_exponent, &out_exponent);
+fft->rfft_hp(x2, nfft, in_exponent, &out_exponent);
+fft->irfft_hp(x2, nfft, in_exponent, &out_exponent);
+```
+Please refer to [dl_fft.hpp](./dl_fft.hpp) for more details.
+
+> Note: The input array x must be allocated with heap_caps_aligned_alloc and aligned to 16 bytes.
+
+
 ## FAQ:
 
 #### 1. Why not just use esp-dsp directly?