srsLTE: Radio benchmark plots all ports

lib/src/radio/test/benchmark_radio.cc

@@ -64,7 +64,7 @@ static pthread_t   plot_thread;
 static sem_t       plot_sem;
 static uint32_t    plot_sf_idx                          = 0;
 static plot_real_t fft_plot[SRSLTE_MAX_RADIOS]          = {};
-static cf_t*       fft_plot_buffer[SRSLTE_MAX_RADIOS] = {};
+static cf_t*       fft_plot_buffer[SRSLTE_MAX_CHANNELS] = {};
 static float*      fft_plot_temp                        = nullptr;
 static uint32_t    fft_plot_buffer_size;
 srslte_dft_plan_t  dft_spectrum = {};
@@ -174,14 +174,17 @@ static void* plot_thread_run(void* arg)
   sdrgui_init();
 
   for (uint32_t i = 0; i < nof_radios; i++) {
+    for (uint32_t j = 0; j < nof_ports; j++) {
+      uint32_t plot_idx    = i * nof_ports + j;
       char     str_buf[32] = {};
-    snprintf(str_buf, 32, "Radio %d spectrum", i);
-    plot_real_init(&fft_plot[i]);
-    plot_real_setTitle(&fft_plot[i], str_buf);
-    plot_real_setXAxisAutoScale(&fft_plot[i], true);
-    plot_real_setYAxisAutoScale(&fft_plot[i], true);
+      snprintf(str_buf, 32, "Radio %d Port %d spectrum", i, j);
+      plot_real_init(&fft_plot[plot_idx]);
+      plot_real_setTitle(&fft_plot[plot_idx], str_buf);
+      plot_real_setXAxisAutoScale(&fft_plot[plot_idx], true);
+      plot_real_setYAxisAutoScale(&fft_plot[plot_idx], true);
 
-    plot_scatter_addToWindowGrid(&fft_plot[i], (char*)"pdsch_ue", 0, i);
+      plot_scatter_addToWindowGrid(&fft_plot[plot_idx], (char*)"pdsch_ue", i, j);
+    }
   }
 
   while (fft_plot_enable) {
@@ -189,13 +192,17 @@ static void* plot_thread_run(void* arg)
 
     if (fft_plot_buffer_size) {
       for (uint32_t r = 0; r < nof_radios; r++) {
-        srslte_vec_abs_square_cf(fft_plot_buffer[r], fft_plot_temp, fft_plot_buffer_size);
+        for (uint32_t p = 0; p < nof_ports; p++) {
+          uint32_t plot_idx = r * nof_ports + p;
+
+          srslte_vec_abs_square_cf(fft_plot_buffer[plot_idx], fft_plot_temp, fft_plot_buffer_size);
 
           for (uint32_t j = 0; j < fft_plot_buffer_size; j++) {
             fft_plot_temp[j] = srslte_convert_power_to_dB(fft_plot_temp[j]);
           }
 
-        plot_real_setNewData(&fft_plot[r], fft_plot_temp, fft_plot_buffer_size);
+          plot_real_setNewData(&fft_plot[plot_idx], fft_plot_temp, fft_plot_buffer_size);
+        }
       }
     }
   }
@@ -212,12 +219,15 @@ static int init_plots(uint32_t frame_size)
   }
 
   for (uint32_t r = 0; r < nof_radios; r++) {
-    fft_plot_buffer[r] = srslte_vec_cf_malloc(frame_size);
-    if (!fft_plot_buffer[r]) {
+    for (uint32_t p = 0; p < nof_ports; p++) {
+      uint32_t plot_idx         = r * nof_ports + p;
+      fft_plot_buffer[plot_idx] = srslte_vec_cf_malloc(frame_size);
+      if (!fft_plot_buffer[plot_idx]) {
         ERROR("Error: Allocating buffer\n");
         return SRSLTE_ERROR;
       }
     }
+  }
 
   fft_plot_temp = srslte_vec_f_malloc(frame_size);
   if (!fft_plot_temp) {
@@ -439,7 +449,10 @@ int main(int argc, char** argv)
     if (fft_plot_enable) {
       if (frame_size != nof_samples) {
         for (uint32_t r = 0; r < nof_radios; r++) {
-          srslte_dft_run(&dft_spectrum, buffers[r][0], fft_plot_buffer[r]);
+          for (uint32_t p = 0; p < nof_ports; p++) {
+            uint32_t plot_idx = r * nof_ports + p;
+            srslte_dft_run(&dft_spectrum, buffers[r][p], fft_plot_buffer[plot_idx]);
+          }
         }
       } else {
         fft_plot_enable = false;
@@ -545,8 +558,11 @@ clean_exit:
   pthread_join(plot_thread, NULL);
   srslte_dft_plan_free(&dft_spectrum);
   for (uint32_t r = 0; r < nof_radios; r++) {
-    if (fft_plot_buffer[r]) {
-      free(fft_plot_buffer[r]);
+    for (uint32_t p = 0; p < nof_ports; p++) {
+      uint32_t plot_idx = r * nof_ports + p;
+      if (fft_plot_buffer[plot_idx]) {
+        free(fft_plot_buffer[plot_idx]);
+      }
     }
   }
   if (fft_plot_temp) {