From 4254d59a960e9db5e07de60b62076de94bd8159b Mon Sep 17 00:00:00 2001
From: rmagueta <rmagueta@allbesmart.pt>
Date: Mon, 4 Apr 2022 20:13:33 +0100
Subject: [PATCH] Create nr_csi_rs_pmi_estimation function for PMI computation
 (still almost empty)

---
 openair1/PHY/INIT/nr_init_ue.c        |  1 +
 openair1/PHY/NR_UE_TRANSPORT/csi_rx.c | 36 +++++++++++++++++++++++++++
 openair1/PHY/defs_nr_common.h         |  1 +
 3 files changed, 38 insertions(+)

diff --git a/openair1/PHY/INIT/nr_init_ue.c b/openair1/PHY/INIT/nr_init_ue.c
index 59c526b818a..07988f96650 100644
--- a/openair1/PHY/INIT/nr_init_ue.c
+++ b/openair1/PHY/INIT/nr_init_ue.c
@@ -347,6 +347,7 @@ int init_nr_ue_signal(PHY_VARS_NR_UE *ue, int nb_connected_gNB)
     }
     ue->nr_csi_rs_info->noise_power = (uint32_t*)malloc16_clear(sizeof(uint32_t));
     ue->nr_csi_rs_info->rank_indicator = (uint8_t*)malloc16_clear(sizeof(uint8_t));
+    ue->nr_csi_rs_info->pmi = (uint8_t*)malloc16_clear(sizeof(uint8_t));
     ue->nr_csi_rs_info->csi_rs_generated_signal = (int32_t **)malloc16(NR_MAX_NB_PORTS * sizeof(int32_t *) );
     for (i=0; i<NR_MAX_NB_PORTS; i++) {
       ue->nr_csi_rs_info->csi_rs_generated_signal[i] = (int32_t *) malloc16_clear(fp->samples_per_frame_wCP * sizeof(int32_t));
diff --git a/openair1/PHY/NR_UE_TRANSPORT/csi_rx.c b/openair1/PHY/NR_UE_TRANSPORT/csi_rx.c
index 4d34cf7d8c4..f5ecc874eb1 100644
--- a/openair1/PHY/NR_UE_TRANSPORT/csi_rx.c
+++ b/openair1/PHY/NR_UE_TRANSPORT/csi_rx.c
@@ -462,6 +462,35 @@ int nr_csi_rs_ri_estimation(PHY_VARS_NR_UE *ue,
   return 0;
 }
 
+int nr_csi_rs_pmi_estimation(PHY_VARS_NR_UE *ue,
+                             fapi_nr_dl_config_csirs_pdu_rel15_t *csirs_config_pdu,
+                             nr_csi_rs_info_t *nr_csi_rs_info,
+                             int32_t ***csi_rs_estimated_channel_freq,
+                             uint8_t rank_indicator,
+                             uint8_t *pmi) {
+
+  // TS 38.214: For 2 antenna ports {3000, 3001} and the UE configured with higher layer parameter codebookType set to
+  // 'typeISinglePanel' each PMI value corresponds to a codebook index given in Table 5.2.2.2.1-1.
+  // The first column is applicable if the UE is reporting a Rank = 1, whereas the second column is applicable if the
+  // UE is reporting a Rank = 2.
+
+  *pmi = 0;
+
+  if(ue->nr_csi_rs_info->N_ports == 1) {
+    return 0;
+  }
+
+  if(rank_indicator == 0) {
+
+  } else if(rank_indicator == 1) {
+
+  } else {
+    LOG_W(NR_PHY, "PMI computation is not implemented for rank indicator %i\n", rank_indicator);
+  }
+
+  return 0;
+}
+
 int nr_ue_csi_im_procedures(PHY_VARS_NR_UE *ue, UE_nr_rxtx_proc_t *proc, uint8_t gNB_id) {
   return 0;
 }
@@ -520,5 +549,12 @@ int nr_ue_csi_rs_procedures(PHY_VARS_NR_UE *ue, UE_nr_rxtx_proc_t *proc, uint8_t
                           ue->nr_csi_rs_info->csi_rs_estimated_channel_freq,
                           ue->nr_csi_rs_info->rank_indicator);
 
+  nr_csi_rs_pmi_estimation(ue,
+                           csirs_config_pdu,
+                           ue->nr_csi_rs_info,
+                           ue->nr_csi_rs_info->csi_rs_estimated_channel_freq,
+                           *ue->nr_csi_rs_info->rank_indicator,
+                           ue->nr_csi_rs_info->pmi);
+
   return 0;
 }
diff --git a/openair1/PHY/defs_nr_common.h b/openair1/PHY/defs_nr_common.h
index 7d4557e340c..6ae2608a16d 100644
--- a/openair1/PHY/defs_nr_common.h
+++ b/openair1/PHY/defs_nr_common.h
@@ -269,6 +269,7 @@ typedef struct {
   int32_t csi_rs_estimated_numer_fin[NR_MAX_OFDM_SYMBOL_SIZE] __attribute__((aligned(32)));
   uint32_t *noise_power;
   uint8_t *rank_indicator;
+  uint8_t *pmi;
 } nr_csi_rs_info_t;
 
 typedef struct NR_DL_FRAME_PARMS NR_DL_FRAME_PARMS;
-- 
GitLab