From 6e0fd43c1fd1f865e74ceee3b4589f244a539855 Mon Sep 17 00:00:00 2001
From: Ismael Gomez <ismagom@gmail.com>
Date: Mon, 4 Sep 2017 10:45:13 +0200
Subject: [PATCH] Moved FFTW to WISDOM and save/load from file

---
 lib/include/srslte/phy/dft/dft.h |  6 ++++--
 lib/src/phy/dft/dft_fftw.c       | 18 ++++++++++++++++--
 srsenb/src/enb.cc                |  2 ++
 srsue/hdr/ue_base.h              |  2 +-
 srsue/src/ue.cc                  |  2 +-
 srsue/src/ue_base.cc             |  8 ++++++++
 6 files changed, 32 insertions(+), 6 deletions(-)

diff --git a/lib/include/srslte/phy/dft/dft.h b/lib/include/srslte/phy/dft/dft.h
index 03a04fcd5..b7fc663d8 100644
--- a/lib/include/srslte/phy/dft/dft.h
+++ b/lib/include/srslte/phy/dft/dft.h
@@ -72,9 +72,11 @@ typedef struct SRSLTE_API {
   srslte_dft_mode_t mode;   // Complex/Real
 }srslte_dft_plan_t;
 
-/* Create DFT plans */
+SRSLTE_API void srslte_dft_load();
 
-SRSLTE_API int srslte_dft_plan(srslte_dft_plan_t *plan, 
+SRSLTE_API void srslte_dft_exit();
+
+SRSLTE_API int srslte_dft_plan(srslte_dft_plan_t *plan,
                                int dft_points, 
                                srslte_dft_dir_t dir,                         
                                srslte_dft_mode_t type);
diff --git a/lib/src/phy/dft/dft_fftw.c b/lib/src/phy/dft/dft_fftw.c
index d841d7d33..134221c1a 100644
--- a/lib/src/phy/dft/dft_fftw.c
+++ b/lib/src/phy/dft/dft_fftw.c
@@ -37,6 +37,20 @@
 #define dft_ceil(a,b) ((a-1)/b+1)
 #define dft_floor(a,b) (a/b)
 
+#define FFTW_WISDOM_FILE ".fftw_wisdom"
+
+void srslte_dft_load() {
+  if (!fftwf_import_wisdom_from_filename(FFTW_WISDOM_FILE)) {
+    fprintf(stderr, "Error loading FFTW wisdom from file %s\n", FFTW_WISDOM_FILE);
+  }
+}
+
+void srslte_dft_exit() {
+  if (!fftwf_export_wisdom_to_filename(FFTW_WISDOM_FILE)) {
+    fprintf(stderr, "Error saving FFTW wisdom to file %s\n", FFTW_WISDOM_FILE);
+  }
+}
+
 int srslte_dft_plan(srslte_dft_plan_t *plan, const int dft_points, srslte_dft_dir_t dir,
              srslte_dft_mode_t mode) {
   if(mode == SRSLTE_DFT_COMPLEX){
@@ -85,7 +99,7 @@ int srslte_dft_replan_c(srslte_dft_plan_t *plan, const int new_dft_points) {
 int srslte_dft_plan_c(srslte_dft_plan_t *plan, const int dft_points, srslte_dft_dir_t dir) {
   allocate(plan,sizeof(fftwf_complex),sizeof(fftwf_complex), dft_points);
   int sign = (dir == SRSLTE_DFT_FORWARD) ? FFTW_FORWARD : FFTW_BACKWARD;
-  plan->p = fftwf_plan_dft_1d(dft_points, plan->in, plan->out, sign, 0U);
+  plan->p = fftwf_plan_dft_1d(dft_points, plan->in, plan->out, sign, FFTW_MEASURE);
   if (!plan->p) {
     return -1;
   }
@@ -108,7 +122,7 @@ int srslte_dft_replan_r(srslte_dft_plan_t *plan, const int new_dft_points) {
     fftwf_destroy_plan(plan->p);
     plan->p = NULL;
   }
-  plan->p = fftwf_plan_r2r_1d(new_dft_points, plan->in, plan->out, sign, 0U);
+  plan->p = fftwf_plan_r2r_1d(new_dft_points, plan->in, plan->out, sign, FFTW_MEASURE);
   if (!plan->p) {
     return -1;
   }
diff --git a/srsenb/src/enb.cc b/srsenb/src/enb.cc
index 9a747dba6..bc3e9d7af 100644
--- a/srsenb/src/enb.cc
+++ b/srsenb/src/enb.cc
@@ -54,11 +54,13 @@ void enb::cleanup(void)
 enb::enb()
     :started(false)
 {
+  srslte_dft_load();
   pool = srslte::byte_buffer_pool::get_instance();
 }
 
 enb::~enb()
 {
+  srslte_dft_exit();
   srslte::byte_buffer_pool::cleanup();
 }
 
diff --git a/srsue/hdr/ue_base.h b/srsue/hdr/ue_base.h
index a03afd5e6..7765d43bd 100644
--- a/srsue/hdr/ue_base.h
+++ b/srsue/hdr/ue_base.h
@@ -135,7 +135,7 @@ class ue_base
     ,public ue_metrics_interface
 {
 public:
-  ue_base() {}
+  ue_base();
   virtual ~ue_base() {}
 
   static ue_base* get_instance(srsue_instance_type_t type);
diff --git a/srsue/src/ue.cc b/srsue/src/ue.cc
index 216c4db23..135bb0591 100644
--- a/srsue/src/ue.cc
+++ b/srsue/src/ue.cc
@@ -179,7 +179,7 @@ bool ue::init(all_args_t *args_)
   phy.wait_initialize();
   phy.configure_ul_params();
 
-  printf("\n\nRequesting NAS Attach...\n");
+  printf("...\n");
   nas.attach_request();
 
   started = true;
diff --git a/srsue/src/ue_base.cc b/srsue/src/ue_base.cc
index 61cbafb2c..c84393ae4 100644
--- a/srsue/src/ue_base.cc
+++ b/srsue/src/ue_base.cc
@@ -57,8 +57,16 @@ ue_base* ue_base::get_instance(srsue_instance_type_t type)
   return(instance);
 }
 
+ue_base::ue_base() {
+  // load FFTW wisdom
+  srslte_dft_load();
+}
+
 void ue_base::cleanup(void)
 {
+  // save FFTW wisdom
+  srslte_dft_exit();
+
   pthread_mutex_lock(&ue_instance_mutex);
   if(NULL != instance) {
     delete instance;