Update source to be compatible with emscripten

3 files changed, 27 insertions, 8 deletions

diff --git a/NaiveFFT/fft.c b/NaiveFFT/fft.c
index cfcf3fa..b4c7d33 100644
--- a/NaiveFFT/fft.c
+++ b/NaiveFFT/fft.c
@@ -11,7 +11,7 @@
 
 
 //calc the fft
-void fft_if(double *x_i, double *x_q, int n, int inv) {
+void KEEPALIVE fft_if(double *x_i, double *x_q, int n, int inv) {
     
     int i=0,j=0,k=0,m=0, irem=0, sign;
     double tq,ti;
@@ -109,7 +109,7 @@ void fft_if(double *x_i, double *x_q, int n, int inv) {
 #define complex_mul_re(a_re, a_im, b_re, b_im)  (a_re * b_re - a_im * b_im)
 #define complex_mul_im(a_re, a_im, b_re, b_im)  (a_re * b_im + a_im * b_re)
 //https://github.com/rshuston/FFT-C/blob/master/libfft/fft.c
-void ffti_shuffle_1(double *x_i, double *x_q, uint64_t n) {
+void KEEPALIVE ffti_shuffle_1(double *x_i, double *x_q, uint64_t n) {
     int Nd2 = n>>1;
     int Nm1 = n-1;
     int i,j;
@@ -156,7 +156,7 @@ void ffti_shuffle_1(double *x_i, double *x_q, uint64_t n) {
            j ^= bits;
        }
 }
-void fft_1(double *x_i, double *x_q, uint64_t n, uint64_t inv) {
+void KEEPALIVE fft_1(double *x_i, double *x_q, uint64_t n, uint64_t inv) {
     uint64_t n_log2;
     uint64_t r;
     uint64_t m, md2;
@@ -225,7 +225,7 @@ void fft_1(double *x_i, double *x_q, uint64_t n, uint64_t inv) {
 }
 
 //dft works fine
-void dft(double *x_i, double *x_q, int n, int inv) {
+void KEEPALIVE dft(double *x_i, double *x_q, int n, int inv) {
     double Wn,Wk;
     //static array
     //double Xi[DATA_SIZE],Xq[DATA_SIZE];
diff --git a/NaiveFFT/fft.h b/NaiveFFT/fft.h
index f1ee783..80ae6cc 100644
--- a/NaiveFFT/fft.h
+++ b/NaiveFFT/fft.h
@@ -11,10 +11,21 @@
 #include <stdlib.h>
 #include <stdio.h>
 #include <math.h>
+#include <stdint.h>
+#include <unistd.h>
+#if __EMSCRIPTEN__
+#include <emscripten/emscripten.h>
+#endif
 
-void fft_if(double *x_i, double *x_q, int n, int inv);
-void ffti_shuffle_1(double *x_i, double *x_q, uint64_t n);
-void fft_1(double *x_i, double *x_q, uint64_t n, uint64_t inv);
-void dft(double *x_i, double *x_q, int n, int inv);
+#ifdef __EMSCRIPTEN__
+#define KEEPALIVE EMSCRIPTEN_KEEPALIVE
+#else
+#define KEEPALIVE
+#endif
+
+void KEEPALIVE fft_if(double *x_i, double *x_q, int n, int inv);
+void KEEPALIVE ffti_shuffle_1(double *x_i, double *x_q, uint64_t n);
+void KEEPALIVE fft_1(double *x_i, double *x_q, uint64_t n, uint64_t inv);
+void KEEPALIVE dft(double *x_i, double *x_q, int n, int inv);
 
 #endif /* fft_h */
diff --git a/NaiveFFT/main.c b/NaiveFFT/main.c
index 036bb79..6372c79 100644
--- a/NaiveFFT/main.c
+++ b/NaiveFFT/main.c
@@ -14,6 +14,10 @@
 #include <time.h>
 #include <stdint.h>
 
+#if __EMSCRIPTEN__
+#include <emscripten/emscripten.h>
+#endif
+
 #include "fft.h"
 
 //test data array
@@ -26,6 +30,9 @@ double data_q[DATA_SIZE] = {0.0f,0.0f,0.0f,0.0f,0.0f,0.0f,0.0f,0.0f};
 
 
 int main(int argc, const char * argv[]) {
+#if __EMSCRIPTEN__
+    printf("Wasm NaiveFFT module loaded\n");
+#else 
     int i;
     
     // insert code here...
@@ -95,5 +102,6 @@ int main(int argc, const char * argv[]) {
         strftime(buffer, 32, "%H:%M:%S", tm_info);
         puts(buffer);
     }
+#endif
     return 0;
 }