From f5e5e946819c73d97323acbde8ee6b004729f7ba Mon Sep 17 00:00:00 2001
From: Xavier Arteaga <xavier@softwareradiosystems.com>
Date: Fri, 1 Sep 2017 14:27:37 +0200
Subject: [PATCH]       if (cell.nof_ports > 1) {

---
 srsue/src/phy/phch_worker.cc | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/srsue/src/phy/phch_worker.cc b/srsue/src/phy/phch_worker.cc
index 78c5b2f71..265d3e005 100644
--- a/srsue/src/phy/phch_worker.cc
+++ b/srsue/src/phy/phch_worker.cc
@@ -40,6 +40,8 @@
 #ifdef ENABLE_GUI
 #include "srsgui/srsgui.h"
 #include <semaphore.h>
+#include <srslte/srslte.h>
+
 void init_plots(srsue::phch_worker *worker);
 pthread_t plot_thread; 
 sem_t plot_sem; 
@@ -451,7 +453,11 @@ int phch_worker::decode_pdsch_multi(srslte_ra_dl_grant_t *grant, uint8_t *payloa
       mimo_type = SRSLTE_MIMO_TYPE_SINGLE_ANTENNA;
       break;
     case LIBLTE_RRC_TRANSMISSION_MODE_2:
-      mimo_type = SRSLTE_MIMO_TYPE_TX_DIVERSITY;
+      if (cell.nof_ports > 1) {
+        mimo_type = SRSLTE_MIMO_TYPE_TX_DIVERSITY;
+      } else {
+        mimo_type = SRSLTE_MIMO_TYPE_SINGLE_ANTENNA;
+      }
       break;
     case LIBLTE_RRC_TRANSMISSION_MODE_3:
       if (grant->nof_tb == 1) {