1 files changed, 11 insertions, 0 deletions
diff --git a/src/simulation/Gravity.h b/src/simulation/Gravity.h
index 72b387d..e706d6d 100644
--- a/src/simulation/Gravity.h
+++ b/src/simulation/Gravity.h
@@ -5,6 +5,10 @@
 #include "Config.h"
 #include "Simulation.h"
 
+#ifdef GRAVFFT
+#include <fftw3.h>
+#endif
+
 class Simulation;
 
 struct mask_el {
@@ -55,6 +59,13 @@ private:
 	int grav_ready;
 	int gravthread_done;
 
+#ifdef GRAVFFT
+	bool grav_fft_status;
+	float *th_ptgravx, *th_ptgravy, *th_gravmapbig, *th_gravxbig, *th_gravybig;
+	fftwf_complex *th_ptgravxt, *th_ptgravyt, *th_gravmapbigt, *th_gravxbigt, *th_gravybigt;
+	fftwf_plan plan_gravmap, plan_gravx_inverse, plan_gravy_inverse;
+#endif
+
 	//Simulation * sim;
 public:
 	float *gravmap;