Add gNB support for multiple UEs for SRS on PHY

openair1/PHY/INIT/nr_init.c

@@ -195,8 +195,14 @@ int phy_init_nr_gNB(PHY_VARS_gNB *gNB,
 
   nr_init_csi_rs(gNB, cfg->cell_config.phy_cell_id.value);
 
-  gNB->nr_srs_info = (nr_srs_info_t *)malloc16_clear(sizeof(nr_srs_info_t));
-  gNB->nr_srs_info->srs_generated_signal = (int32_t*)malloc16_clear(fp->samples_per_frame*sizeof(int32_t));
+  for (int id=0; id<NUMBER_OF_NR_SRS_MAX; id++) {
+    gNB->nr_srs_info[id] = (nr_srs_info_t *)malloc16_clear(sizeof(nr_srs_info_t));
+    gNB->nr_srs_info[id]->srs_generated_signal = (int32_t*)malloc16_clear(fp->samples_per_frame*sizeof(int32_t));
+    gNB->nr_srs_info[id]->srs_received_signal = (int32_t **)malloc16(Prx*sizeof(int32_t*));
+    for (i=0;i<Prx;i++){
+      gNB->nr_srs_info[id]->srs_received_signal[i] = (int32_t*)malloc16_clear(fp->samples_per_frame*sizeof(int32_t));
+    }
+  }
 
   generate_ul_reference_signal_sequences(SHRT_MAX);
 
@@ -212,7 +218,6 @@ int phy_init_nr_gNB(PHY_VARS_gNB *gNB,
   common_vars->txdataF = (int32_t **)malloc16(Ptx*sizeof(int32_t*));
   common_vars->rxdataF = (int32_t **)malloc16(Prx*sizeof(int32_t*));
   common_vars->beam_id = (uint8_t **)malloc16(Ptx*sizeof(uint8_t*));
-  gNB->nr_srs_info->srs_received_signal = (int32_t **)malloc16(Prx*sizeof(int32_t*));
 
   for (i=0;i<Ptx;i++){
       common_vars->txdataF[i] = (int32_t*)malloc16_clear(fp->samples_per_frame_wCP*sizeof(int32_t)); // [hna] samples_per_frame without CP
@@ -225,7 +230,6 @@ int phy_init_nr_gNB(PHY_VARS_gNB *gNB,
   for (i=0;i<Prx;i++){
     common_vars->rxdataF[i] = (int32_t*)malloc16_clear(fp->samples_per_frame_wCP*sizeof(int32_t));
     common_vars->rxdata[i] = (int32_t*)malloc16_clear(fp->samples_per_frame*sizeof(int32_t));
-    gNB->nr_srs_info->srs_received_signal[i] = (int32_t*)malloc16_clear(fp->samples_per_frame*sizeof(int32_t));
   }
   common_vars->debugBuff = (int32_t*)malloc16_clear(fp->samples_per_frame*sizeof(int32_t)*100);	
   common_vars->debugBuff_sample_offset = 0; 

openair1/PHY/NR_TRANSPORT/nr_transport_proto.h

@@ -312,6 +312,11 @@ int nr_find_pucch(uint16_t rnti,
 
 NR_gNB_SRS_t *new_gNB_srs(void);
 
+int nr_find_srs(uint16_t rnti,
+                int frame,
+                int slot,
+                PHY_VARS_gNB *gNB);
+
 void nr_fill_srs(PHY_VARS_gNB *gNB,
                  int frame,
                  int slot,

openair1/PHY/NR_TRANSPORT/srs_rx.c

@@ -59,7 +59,7 @@ int nr_find_srs(uint16_t rnti,
 
   for (int i=0; i<NUMBER_OF_NR_SRS_MAX; i++) {
     AssertFatal(gNB->srs[i]!=NULL,"gNB->srs[%d] is null\n",i);
-    if ((gNB->srs[i]->active >0) &&
+    if ((gNB->srs[i]->active>0) &&
         (gNB->srs[i]->srs_pdu.rnti==rnti) &&
         (gNB->srs[i]->frame==frame) &&
         (gNB->srs[i]->slot==slot)) return(i);

openair1/PHY/defs_gNB.h

@@ -795,7 +795,7 @@ typedef struct PHY_VARS_gNB_s {
   t_nrPolar_params    *uci_polarParams;
 
   /// SRS variables
-  nr_srs_info_t *nr_srs_info;
+  nr_srs_info_t *nr_srs_info[NUMBER_OF_NR_SRS_MAX];
 
   uint8_t pbch_configured;
   char gNB_generate_rar;

openair1/SCHED_NR/phy_procedures_nr_gNB.c

@@ -799,11 +799,11 @@ int phy_procedures_gNB_uespec_RX(PHY_VARS_gNB *gNB, int frame_rx, int slot_rx) {
         nfapi_nr_srs_pdu_t *srs_pdu = &srs->srs_pdu;
 
         // At least currently, the configuration is constant, so it is enough to generate the sequence just once.
-        if(gNB->nr_srs_info->n_symbs==0) {
-          generate_srs_nr(srs_pdu, &gNB->frame_parms, gNB->nr_srs_info->srs_generated_signal, gNB->nr_srs_info, AMP, frame_rx, slot_rx);
+        if(gNB->nr_srs_info[i]->n_symbs==0) {
+          generate_srs_nr(srs_pdu, &gNB->frame_parms, gNB->nr_srs_info[i]->srs_generated_signal, gNB->nr_srs_info[i], AMP, frame_rx, slot_rx);
         }
 
-        nr_get_srs_signal(gNB,frame_rx,slot_rx,srs_pdu, gNB->nr_srs_info, gNB->nr_srs_info->srs_received_signal);
+        nr_get_srs_signal(gNB,frame_rx,slot_rx,srs_pdu, gNB->nr_srs_info[i], gNB->nr_srs_info[i]->srs_received_signal);
         srs->active = 0;
       }
     }