dlsim.c 56.6 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
/*
 * Licensed to the OpenAirInterface (OAI) Software Alliance under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The OpenAirInterface Software Alliance licenses this file to You under
 * the OAI Public License, Version 1.1  (the "License"); you may not use this file
 * except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.openairinterface.org/?page_id=698
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *-------------------------------------------------------------------------------
 * For more information about the OpenAirInterface (OAI) Software Alliance:
 *      contact@openairinterface.org
 */

#include <fcntl.h>
23 24
#include <math.h>
#include <string.h>
25 26
#include <sys/ioctl.h>
#include <sys/mman.h>
27 28
#include <unistd.h>
#include "common/ran_context.h"
29
#include "common/config/config_userapi.h"
francescomani's avatar
francescomani committed
30
#include "common/utils/nr/nr_common.h"
31
#include "common/utils/LOG/log.h"
32 33 34
#include "LAYER2/NR_MAC_gNB/nr_mac_gNB.h"
#include "LAYER2/NR_MAC_UE/mac_defs.h"
#include "LAYER2/NR_MAC_UE/mac_extern.h"
35
#include "PHY/defs_gNB.h"
36 37
#include "PHY/defs_nr_common.h"
#include "PHY/defs_nr_UE.h"
38
#include "PHY/phy_vars_nr_ue.h"
39 40
#include "PHY/types.h"
#include "PHY/INIT/phy_init.h"
41
#include "PHY/MODULATION/modulation_eNB.h"
42
#include "PHY/MODULATION/nr_modulation.h"
43
#include "PHY/MODULATION/modulation_UE.h"
44
#include "PHY/NR_REFSIG/refsig_defs_ue.h"
45
#include "PHY/NR_TRANSPORT/nr_dlsch.h"
cig's avatar
cig committed
46
#include "PHY/NR_TRANSPORT/nr_transport_proto.h"
47
#include "PHY/NR_UE_TRANSPORT/nr_transport_proto_ue.h"
48
#include "SCHED_NR/fapi_nr_l1.h"
49
#include "SCHED_NR/sched_nr.h"
50
#include "SCHED_NR_UE/defs.h"
51 52 53 54
#include "SCHED_NR_UE/fapi_nr_ue_l1.h"
#include "NR_PHY_INTERFACE/NR_IF_Module.h"
#include "NR_UE_PHY_INTERFACE/NR_IF_Module.h"

55
#include "LAYER2/NR_MAC_UE/mac_proto.h"
yilmazt's avatar
yilmazt committed
56 57
//#include "LAYER2/NR_MAC_gNB/mac_proto.h"
//#include "openair2/LAYER2/NR_MAC_UE/mac_proto.h"
58
#include "LAYER2/NR_MAC_gNB/mac_proto.h"
59
#include "NR_asn_constant.h"
60
#include "RRC/NR/MESSAGES/asn1_msg.h"
61 62 63
#include "openair1/SIMULATION/RF/rf.h"
#include "openair1/SIMULATION/TOOLS/sim.h"
#include "openair1/SIMULATION/NR_PHY/nr_unitary_defs.h"
Raymond Knopp's avatar
Raymond Knopp committed
64
//#include "openair1/SIMULATION/NR_PHY/nr_dummy_functions.c"
65
#include "PHY/NR_REFSIG/ptrs_nr.h"
Raymond Knopp's avatar
Raymond Knopp committed
66
#include "NR_RRCReconfiguration.h"
67
#define inMicroS(a) (((double)(a))/(get_cpu_freq_GHz()*1000.0))
68 69
#include "SIMULATION/LTE_PHY/common_sim.h"

Laurent's avatar
Laurent committed
70 71
#include <openair2/LAYER2/MAC/mac_vars.h>
#include <openair2/RRC/LTE/rrc_vars.h>
72

Raymond Knopp's avatar
Raymond Knopp committed
73 74
#include <executables/softmodem-common.h>

75 76
LCHAN_DESC DCCH_LCHAN_DESC,DTCH_DL_LCHAN_DESC,DTCH_UL_LCHAN_DESC;
rlc_info_t Rlc_info_um,Rlc_info_am_config;
Raymond Knopp's avatar
Raymond Knopp committed
77

78 79 80
PHY_VARS_gNB *gNB;
PHY_VARS_NR_UE *UE;
RAN_CONTEXT_t RC;
81 82
int32_t uplink_frequency_offset[MAX_NUM_CCs][4];

83
double cpuf;
84

Laurent THOMAS's avatar
Laurent THOMAS committed
85 86
uint16_t sf_ahead=4 ;
uint16_t sl_ahead=0;
87
//uint8_t nfapi_mode = 0;
88
uint64_t downlink_frequency[MAX_NUM_CCs][4];
89 90
THREAD_STRUCT thread_struct;
nfapi_ue_release_request_body_t release_rntis;
rmagueta's avatar
rmagueta committed
91
uint32_t N_RB_DL = 106;
92 93

// dummy functions
Raymond Knopp's avatar
Raymond Knopp committed
94 95
int dummy_nr_ue_ul_indication(nr_uplink_indication_t *ul_info)              { return(0);  }

96 97 98 99 100 101 102 103 104
int8_t nr_mac_rrc_data_ind_ue(const module_id_t module_id,
                              const int CC_id,
                              const uint8_t gNB_index,
                              const frame_t frame,
                              const sub_frame_t sub_frame,
                              const rnti_t rnti,
                              const channel_t channel,
                              const uint8_t* pduP,
                              const sdu_size_t pdu_len)
105 106 107 108
{
  return 0;
}

109 110 111 112 113
void nr_rrc_ue_generate_RRCSetupRequest(module_id_t module_id, const uint8_t gNB_index)
{
  return;
}

114 115 116 117 118 119 120 121 122 123
int8_t nr_mac_rrc_data_req_ue(const module_id_t Mod_idP,
                              const int         CC_id,
                              const uint8_t     gNB_id,
                              const frame_t     frameP,
                              const rb_id_t     Srb_id,
                              uint8_t           *buffer_pP)
{
  return 0;
}

124 125
void
rrc_data_ind(
126
  const protocol_ctxt_t *const ctxt_pP,
127 128 129 130 131 132 133
  const rb_id_t                Srb_id,
  const sdu_size_t             sdu_sizeP,
  const uint8_t   *const       buffer_pP
)
{
}

rmagueta's avatar
rmagueta committed
134 135 136 137 138 139
int ocp_gtpv1u_create_s1u_tunnel(instance_t instance,
                                 const gtpv1u_enb_create_tunnel_req_t  *create_tunnel_req,
                                 gtpv1u_enb_create_tunnel_resp_t *create_tunnel_resp) {
    return 0;
}

140 141 142 143 144 145 146 147
int
gtpv1u_create_s1u_tunnel(
  const instance_t                              instanceP,
  const gtpv1u_enb_create_tunnel_req_t *const  create_tunnel_req_pP,
  gtpv1u_enb_create_tunnel_resp_t *const create_tunnel_resp_pP
) {
  return 0;
}
148

149 150 151 152 153 154 155 156
int
rrc_gNB_process_GTPV1U_CREATE_TUNNEL_RESP(
  const protocol_ctxt_t *const ctxt_pP,
  const gtpv1u_enb_create_tunnel_resp_t *const create_tunnel_resp_pP,
  uint8_t                         *inde_list
) {
  return 0;
}
157

heshanyun's avatar
heshanyun committed
158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174
int
gtpv1u_create_ngu_tunnel(
  const instance_t instanceP,
  const gtpv1u_gnb_create_tunnel_req_t *  const create_tunnel_req_pP,
        gtpv1u_gnb_create_tunnel_resp_t * const create_tunnel_resp_pP){
  return 0;
}

int
gtpv1u_update_ngu_tunnel(
  const instance_t                              instanceP,
  const gtpv1u_gnb_create_tunnel_req_t *const  create_tunnel_req_pP,
  const rnti_t                                  prior_rnti
){
  return 0;
}

rmagueta's avatar
rmagueta committed
175 176 177 178
int ocp_gtpv1u_delete_s1u_tunnel(const instance_t instance, const gtpv1u_enb_delete_tunnel_req_t *const req_pP) {
  return 0;
}

heshanyun's avatar
heshanyun committed
179 180 181 182 183 184 185 186 187
int
nr_rrc_gNB_process_GTPV1U_CREATE_TUNNEL_RESP(
  const protocol_ctxt_t *const ctxt_pP,
  const gtpv1u_gnb_create_tunnel_resp_t *const create_tunnel_resp_pP,
  uint8_t                         *inde_list
){
  return 0;
}

188 189 190 191 192 193
int nr_derive_key(int alg_type, uint8_t alg_id,
               const uint8_t key[32], uint8_t **out)
{
  return 0;
}

194
void config_common(int Mod_idP,
195
                   int ssb_SubcarrierOffset,
196 197
                   int pdsch_AntennaPorts,
                   int pusch_AntennaPorts,
198
		   NR_ServingCellConfigCommon_t *scc
Raymond Knopp's avatar
Raymond Knopp committed
199
		   );
200

201 202 203 204 205 206 207 208 209
int generate_dlsch_header(unsigned char *mac_header,
                          unsigned char num_sdus,
                          unsigned short *sdu_lengths,
                          unsigned char *sdu_lcids,
                          unsigned char drx_cmd,
                          unsigned short timing_advance_cmd,
                          unsigned char *ue_cont_res_id,
                          unsigned char short_padding,
                          unsigned short post_padding){return 0;}
210

211
// Dummy function to avoid linking error at compilation of nr-dlsim
212 213 214 215 216
int is_x2ap_enabled(void)
{
  return 0;
}

rmagueta's avatar
rmagueta committed
217 218 219 220 221 222 223 224 225 226 227
int DU_send_INITIAL_UL_RRC_MESSAGE_TRANSFER(module_id_t     module_idP,
                                            int             CC_idP,
                                            int             UE_id,
                                            rnti_t          rntiP,
                                            const uint8_t   *sduP,
                                            sdu_size_t      sdu_lenP,
                                            const uint8_t   *sdu2P,
                                            sdu_size_t      sdu2_lenP) {
  return 0;
}

228 229
void processSlotTX(void *arg) {}

Mahesh's avatar
Mahesh committed
230
//nFAPI P7 dummy functions to avoid linking errors 
Mahesh's avatar
Mahesh committed
231 232 233 234 235

int oai_nfapi_dl_tti_req(nfapi_nr_dl_tti_request_t *dl_config_req) { return(0);  }
int oai_nfapi_tx_data_req(nfapi_nr_tx_data_request_t *tx_data_req){ return(0);  }
int oai_nfapi_ul_dci_req(nfapi_nr_ul_dci_request_t *ul_dci_req){ return(0);  }
int oai_nfapi_ul_tti_req(nfapi_nr_ul_tti_request_t *ul_tti_req){ return(0);  }
Mahesh's avatar
Mahesh committed
236 237 238 239 240
int oai_nfapi_nr_rx_data_indication(nfapi_nr_rx_data_indication_t *ind) { return(0);  }
int oai_nfapi_nr_crc_indication(nfapi_nr_crc_indication_t *ind) { return(0);  }
int oai_nfapi_nr_srs_indication(nfapi_nr_srs_indication_t *ind) { return(0);  }
int oai_nfapi_nr_uci_indication(nfapi_nr_uci_indication_t *ind) { return(0);  }
int oai_nfapi_nr_rach_indication(nfapi_nr_rach_indication_t *ind) { return(0);  }
rmagueta's avatar
rmagueta committed
241

242
// needed for some functions
243
openair0_config_t openair0_cfg[MAX_CARDS];
244
void update_ptrs_config(NR_CellGroupConfig_t *secondaryCellGroup, uint16_t *rbSize, uint8_t *mcsIndex,int8_t *ptrs_arg);
245
void update_dmrs_config(NR_CellGroupConfig_t *scg, int8_t* dmrs_arg);
rmagueta's avatar
rmagueta committed
246
extern void fix_scd(NR_ServingCellConfig_t *scd);// forward declaration
247

248
/* specific dlsim DL preprocessor: uses rbStart/rbSize/mcs/nrOfLayers from command line of
249
   dlsim, does not search for CCE/PUCCH occasion but simply sets to 0 */
250
int g_mcsIndex = -1, g_mcsTableIdx = 0, g_rbStart = -1, g_rbSize = -1, g_nrOfLayers = 1;
251 252
void nr_dlsim_preprocessor(module_id_t module_id,
                           frame_t frame,
253
                           sub_frame_t slot) {
254 255 256
  NR_UE_info_t *UE_info = &RC.nrmac[module_id]->UE_info;
  AssertFatal(UE_info->num_UEs == 1, "can have only a single UE\n");
  NR_UE_sched_ctrl_t *sched_ctrl = &UE_info->UE_sched_ctrl[0];
rmagueta's avatar
rmagueta committed
257
  NR_ServingCellConfigCommon_t *scc = RC.nrmac[0]->common_channels[0].ServingCellConfigCommon;
258 259 260

  /* manually set free CCE to 0 */
  const int target_ss = NR_SearchSpace__searchSpaceType_PR_ue_Specific;
rmagueta's avatar
rmagueta committed
261
  sched_ctrl->search_space = get_searchspace(scc, sched_ctrl->active_bwp ? sched_ctrl->active_bwp->bwp_Dedicated : NULL, target_ss);
262 263 264
  uint8_t nr_of_candidates;
  find_aggregation_candidates(&sched_ctrl->aggregation_level,
                              &nr_of_candidates,
Eurecom's avatar
Eurecom committed
265
                              sched_ctrl->search_space,4);
266
  sched_ctrl->coreset = get_coreset(module_id, scc, sched_ctrl->active_bwp->bwp_Dedicated, sched_ctrl->search_space, target_ss);
267 268
  sched_ctrl->cce_index = 0;

269
  NR_pdsch_semi_static_t *ps = &sched_ctrl->pdsch_semi_static;
rmagueta's avatar
rmagueta committed
270

271 272
  ps->nrOfLayers = g_nrOfLayers;

rmagueta's avatar
rmagueta committed
273 274 275
  int dci_format = sched_ctrl->search_space && sched_ctrl->search_space->searchSpaceType->choice.ue_Specific->dci_Formats ?
      NR_DL_DCI_FORMAT_1_1 : NR_DL_DCI_FORMAT_1_0;

276
  nr_set_pdsch_semi_static(scc,
rmagueta's avatar
rmagueta committed
277
                           UE_info->CellGroup[0],
278
                           sched_ctrl->active_bwp,
rmagueta's avatar
rmagueta committed
279
                           NULL,
280
                           /* tda = */ 2,
rmagueta's avatar
rmagueta committed
281
                           dci_format,
282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300
                           ps);

  NR_sched_pdsch_t *sched_pdsch = &sched_ctrl->sched_pdsch;
  sched_pdsch->rbStart = g_rbStart;
  sched_pdsch->rbSize = g_rbSize;
  sched_pdsch->mcs = g_mcsIndex;
  /* the following might override the table that is mandated by RRC
   * configuration */
  ps->mcsTableIdx = g_mcsTableIdx;

  sched_pdsch->Qm = nr_get_Qm_dl(sched_pdsch->mcs, ps->mcsTableIdx);
  sched_pdsch->R = nr_get_code_rate_dl(sched_pdsch->mcs, ps->mcsTableIdx);
  sched_pdsch->tb_size = nr_compute_tbs(sched_pdsch->Qm,
                                        sched_pdsch->R,
                                        sched_pdsch->rbSize,
                                        ps->nrOfSymbols,
                                        ps->N_PRB_DMRS * ps->N_DMRS_SLOT,
                                        0 /* N_PRB_oh, 0 for initialBWP */,
                                        0 /* tb_scaling */,
301
                                        ps->nrOfLayers)
302 303
                         >> 3;

304
  /* the simulator assumes the HARQ PID is equal to the slot number */
305 306
  sched_pdsch->dl_harq_pid = slot;

307 308 309 310 311 312 313 314 315 316 317 318
  /* The scheduler uses lists to track whether a HARQ process is
   * free/busy/awaiting retransmission, and updates the HARQ process states.
   * However, in the simulation, we never get ack or nack for any HARQ process,
   * thus the list and HARQ states don't match what the scheduler expects.
   * Therefore, below lines just "repair" everything so that the scheduler
   * won't remark that there is no HARQ feedback */
  sched_ctrl->feedback_dl_harq.head = -1; // always overwrite feedback HARQ process
  if (sched_ctrl->harq_processes[slot].round == 0) // depending on round set in simulation ...
    add_front_nr_list(&sched_ctrl->available_dl_harq, slot); // ... make PID available
  else
    add_front_nr_list(&sched_ctrl->retrans_dl_harq, slot);   // ... make PID retransmission
  sched_ctrl->harq_processes[slot].is_waiting = false;
319 320 321 322
  AssertFatal(sched_pdsch->rbStart >= 0, "invalid rbStart %d\n", sched_pdsch->rbStart);
  AssertFatal(sched_pdsch->rbSize > 0, "invalid rbSize %d\n", sched_pdsch->rbSize);
  AssertFatal(sched_pdsch->mcs >= 0, "invalid mcs %d\n", sched_pdsch->mcs);
  AssertFatal(ps->mcsTableIdx >= 0 && ps->mcsTableIdx <= 2, "invalid mcsTableIdx %d\n", ps->mcsTableIdx);
323
}
324

Sakthivel Velumani's avatar
Sakthivel Velumani committed
325 326 327
typedef struct {
  uint64_t       optmask;   //mask to store boolean config options
  uint8_t        nr_dlsch_parallel; // number of threads for dlsch decoding, 0 means no parallelization
328
  tpool_t        Tpool;             // thread pool
Sakthivel Velumani's avatar
Sakthivel Velumani committed
329 330 331 332 333 334 335
} nrUE_params_t;

nrUE_params_t nrUE_params;

nrUE_params_t *get_nrUE_params(void) {
  return &nrUE_params;
}
336

Sakthivel Velumani's avatar
Sakthivel Velumani committed
337 338
void do_nothing(void *args) {
}
339

340 341 342
int main(int argc, char **argv)
{
  char c;
343
  int i,aa;//,l;
344
  double sigma2, sigma2_dB=10, SNR, snr0=-2.0, snr1=2.0;
345
  uint8_t snr1set=0;
346 347 348 349
  double roundStats[500] = {0};
  double blerStats[500] = {0};
  double berStats[500] = {0};
  double snrStats[500] = {0};
350
  float effRate;
Francesco Mani's avatar
Francesco Mani committed
351
  //float psnr;
352
  float eff_tp_check = 0.7;
353
  uint8_t snrRun;
354
  uint32_t TBS = 0;
355 356
  int **txdata;
  double **s_re,**s_im,**r_re,**r_im;
357 358
  //double iqim = 0.0;
  //unsigned char pbch_pdu[6];
359 360 361
  //  int sync_pos, sync_pos_slot;
  //  FILE *rx_frame_file;
  FILE *output_fd = NULL;
362
  //uint8_t write_output_file=0;
363
  //int result;
364
  //int freq_offset;
365
  //  int subframe_offset;
366
  //  char fname[40], vname[40];
Ahmed Hussein's avatar
Ahmed Hussein committed
367
  int trial, n_trials = 1, n_errors = 0, n_false_positive = 0;
368
  //int n_errors2, n_alamouti;
369
  uint8_t n_tx=1,n_rx=1;
370 371
  uint8_t round;
  uint8_t num_rounds = 4;
372 373

  channel_desc_t *gNB2UE;
374 375 376
  //uint32_t nsymb,tx_lev,tx_lev1 = 0,tx_lev2 = 0;
  //uint8_t extended_prefix_flag=0;
  //int8_t interf1=-21,interf2=-21;
377 378

  FILE *input_fd=NULL,*pbch_file_fd=NULL;
379
  //char input_val_str[50],input_val_str2[50];
380

381
  //uint8_t frame_mod4,num_pdcch_symbols = 0;
382

383
  SCM_t channel_model = AWGN; // AWGN Rayleigh1 Rayleigh1_anticorr;
384

Laurent's avatar
Laurent committed
385
  NB_UE_INST = 1;
386 387
  //double pbch_sinr;
  //int pbch_tx_ant;
Ahmed Hussein's avatar
Ahmed Hussein committed
388
  int N_RB_DL=106,mu=1;
389

390
  //unsigned char frame_type = 0;
391

392
  int frame=1,slot=1;
393
  int frame_length_complex_samples;
394
  //int frame_length_complex_samples_no_prefix;
395 396 397 398 399
  NR_DL_FRAME_PARMS *frame_parms;
  UE_nr_rxtx_proc_t UE_proc;
  NR_Sched_Rsp_t Sched_INFO;
  gNB_MAC_INST *gNB_mac;
  NR_UE_MAC_INST_t *UE_mac;
400
  int cyclic_prefix_type = NFAPI_CP_NORMAL;
401
  int run_initial_sync=0;
Raymond Knopp's avatar
Raymond Knopp committed
402
  int loglvl=OAILOG_INFO;
403

404
  //float target_error_rate = 0.01;
405
  int css_flag=0;
406 407

  cpuf = get_cpu_freq_GHz();
408 409 410
  int8_t enable_ptrs = 0;
  int8_t modify_dmrs = 0;

411
  int8_t dmrs_arg[3] = {-1,-1,-1};// Invalid values
412
  /* L_PTRS = ptrs_arg[0], K_PTRS = ptrs_arg[1] */
413 414
  int8_t ptrs_arg[2] = {-1,-1};// Invalid values

415 416 417 418 419 420
  uint16_t ptrsRePerSymb = 0;
  uint16_t pdu_bit_map = 0x0;
  uint16_t dlPtrsSymPos = 0;
  uint16_t ptrsSymbPerSlot = 0;
  uint16_t rbSize = 106;
  uint8_t  mcsIndex = 9;
421
  uint8_t  dlsch_threads = 0;
422
  int      prb_inter = 0;
423
  if ( load_configmodule(argc,argv,CONFIG_ENABLECMDLINEONLY) == 0) {
yilmazt's avatar
yilmazt committed
424
    exit_fun("[NR_DLSIM] Error, configuration module init failed\n");
425 426 427 428
  }

  randominit(0);

429
  int print_perf             = 0;
430

431 432
  FILE *scg_fd=NULL;
  
frtabu's avatar
frtabu committed
433 434 435

  while ((c = getopt (argc, argv, "f:hA:pf:g:in:s:S:t:x:y:z:M:N:F:GR:dPIL:Ea:b:D:e:m:w:T:U:q")) != -1) {

436
    switch (c) {
437 438
    case 'f':
      scg_fd = fopen(optarg,"r");
439

440
      if (scg_fd==NULL) {
441 442 443
        printf("Error opening %s\n",optarg);
        exit(-1);
      }
444
      break;
445

446
    /*case 'd':
447
      frame_type = 1;
448
      break;*/
449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479

    case 'g':
      switch((char)*optarg) {
      case 'A':
        channel_model=SCM_A;
        break;

      case 'B':
        channel_model=SCM_B;
        break;

      case 'C':
        channel_model=SCM_C;
        break;

      case 'D':
        channel_model=SCM_D;
        break;

      case 'E':
        channel_model=EPA;
        break;

      case 'F':
        channel_model=EVA;
        break;

      case 'G':
        channel_model=ETU;
        break;

sfn's avatar
sfn committed
480 481 482 483
      case 'R':
        channel_model=Rayleigh1;
        break;

484
      default:
485
        printf("Unsupported channel model!\n");
486 487 488 489 490
        exit(-1);
      }

      break;

491
    case 'i':
492
      prb_inter=1;
493 494 495 496 497 498 499 500
      break;

    case 'n':
      n_trials = atoi(optarg);
      break;

    case 's':
      snr0 = atof(optarg);
501
      printf("Setting SNR0 to %f\n",snr0);
502 503 504 505 506
      break;

    case 'S':
      snr1 = atof(optarg);
      snr1set=1;
507
      printf("Setting SNR1 to %f\n",snr1);
508 509 510 511 512 513 514
      break;

      /*
      case 't':
      Td= atof(optarg);
      break;
      */
515
    /*case 'p':
516
      extended_prefix_flag=1;
517
      break;*/
518 519 520 521 522 523 524 525 526 527 528

      /*
      case 'r':
      ricean_factor = pow(10,-.1*atof(optarg));
      if (ricean_factor>1) {
        printf("Ricean factor must be between 0 and 1\n");
        exit(-1);
      }
      break;
      */
    case 'x':
529
      g_nrOfLayers=atoi(optarg);
530

531 532 533
      if ((g_nrOfLayers!=1) &&
          (g_nrOfLayers!=2)) {
        printf("Unsupported nr Of Layers %d\n",g_nrOfLayers);
534 535 536 537 538 539 540 541
        exit(-1);
      }

      break;

    case 'y':
      n_tx=atoi(optarg);

542
      if ((n_tx==0) || (n_tx>4)) {//extend gNB to support n_tx = 4
543
        printf("Unsupported number of tx antennas %d\n",n_tx);
544 545 546 547 548 549 550 551
        exit(-1);
      }

      break;

    case 'z':
      n_rx=atoi(optarg);

552
      if ((n_rx==0) || (n_rx>4)) {//extend UE to support n_tx = 4
553
        printf("Unsupported number of rx antennas %d\n",n_rx);
554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573
        exit(-1);
      }

      break;

    case 'R':
      N_RB_DL = atoi(optarg);
      break;

    case 'F':
      input_fd = fopen(optarg,"r");

      if (input_fd==NULL) {
        printf("Problem with filename %s\n",optarg);
        exit(-1);
      }

      break;

    case 'P':
574 575
      print_perf=1;
      opp_enabled=1;
576 577 578 579
      break;
      
    case 'I':
      run_initial_sync=1;
580
      //target_error_rate=0.1;
581
      slot = 0;
582 583 584 585 586 587
      break;

    case 'L':
      loglvl = atoi(optarg);
      break;

588

589 590 591
    case 'E':
	css_flag=1;
	break;
592

593

594
    case 'a':
595
      g_rbStart = atoi(optarg);
596 597 598
      break;

    case 'b':
599
      g_rbSize = atoi(optarg);
600
      break;
frtabu's avatar
frtabu committed
601
    case 'D':
602 603
      dlsch_threads = atoi(optarg);
      break;    
604
    case 'e':
605
      g_mcsIndex = atoi(optarg);
606 607
      break;

608 609 610 611 612
    case 'q':
      g_mcsTableIdx = 1;
      get_softmodem_params()->use_256qam_table = 1;
      break;

613 614 615
    case 'm':
      mu = atoi(optarg);
      break;
616

617 618 619 620
    case 't':
      eff_tp_check = (float)atoi(optarg)/100;
      break;

621 622 623
    case 'w':
      output_fd = fopen("txdata.dat", "w+");
      break;
Sakthivel Velumani's avatar
Sakthivel Velumani committed
624

625 626 627 628 629 630 631
    case 'T':
      enable_ptrs=1;
      for(i=0; i < atoi(optarg); i++) {
        ptrs_arg[i] = atoi(argv[optind++]);
      }
      break;

632 633
    case 'U':
      modify_dmrs = 1;
634
      for(i=0; i < atoi(optarg); i++) {
635 636 637
        dmrs_arg[i] = atoi(argv[optind++]);
      }
      break;
Sakthivel Velumani's avatar
Sakthivel Velumani committed
638

639 640 641 642 643
    default:
    case 'h':
      printf("%s -h(elp) -p(extended_prefix) -N cell_id -f output_filename -F input_filename -g channel_model -n n_frames -t Delayspread -s snr0 -S snr1 -x transmission_mode -y TXant -z RXant -i Intefrence0 -j Interference1 -A interpolation_file -C(alibration offset dB) -N CellId\n",
             argv[0]);
      printf("-h This message\n");
644 645
      //printf("-p Use extended prefix mode\n");
      //printf("-d Use TDD\n");
646 647 648 649
      printf("-n Number of frames to simulate\n");
      printf("-s Starting SNR, runs from SNR0 to SNR0 + 5 dB.  If n_frames is 1 then just SNR is simulated\n");
      printf("-S Ending SNR, runs from SNR0 to SNR1\n");
      printf("-t Delay spread for multipath channel\n");
sfn's avatar
sfn committed
650
      printf("-g [A,B,C,D,E,F,G,R] Use 3GPP SCM (A,B,C,D) or 36-101 (E-EPA,F-EVA,G-ETU) models or R for MIMO model (ignores delay spread and Ricean factor)\n");
651
      printf("-y Number of TX antennas used in gNB\n");
652
      printf("-z Number of RX antennas used in UE\n");
653
      printf("-i Activate PRB based averaging for channel estimation. Frequncy domain interpolation by default.\n");
654
      //printf("-j Relative strength of second intefering gNB (in dB) - cell_id mod 3 = 2\n");
655 656 657
      printf("-R N_RB_DL\n");
      printf("-O oversampling factor (1,2,4,8,16)\n");
      printf("-A Interpolation_filname Run with Abstraction to generate Scatter plot using interpolation polynomial in file\n");
658
      //printf("-C Generate Calibration information for Abstraction (effective SNR adjustment to remove Pe bias w.r.t. AWGN)\n");
659
      printf("-f raw file containing RRC configuration (generated by gNB)\n");
660
      printf("-F Input filename (.txt format) for RX conformance testing\n");
661
      printf("-E used CSS scheduler\n");
662
      printf("-o CORESET offset\n");
663 664
      printf("-a Start PRB for PDSCH\n");
      printf("-b Number of PRB for PDSCH\n");
665 666
      printf("-c Start symbol for PDSCH (fixed for now)\n");
      printf("-j Number of symbols for PDSCH (fixed for now)\n");
667
      printf("-e MSC index\n");
668
      printf("-q Use 2nd MCS table (256 QAM table) for PDSCH\n");
669
      printf("-t Acceptable effective throughput (in percentage)\n");
670
      printf("-T Enable PTRS, arguments list L_PTRS{0,1,2} K_PTRS{2,4}, e.g. -T 2 0 2 \n");
671
      printf("-U Change DMRS Config, arguments list DMRS TYPE{0=A,1=B} DMRS AddPos{0:2} DMRS ConfType{1:2}, e.g. -U 3 0 2 1 \n");
672
      printf("-P Print DLSCH performances\n");
Sakthivel Velumani's avatar
Sakthivel Velumani committed
673
      printf("-w Write txdata to binary file (one frame)\n");
frtabu's avatar
frtabu committed
674
      printf("-D number of dlsch threads, 0: no dlsch parallelization\n");
675 676 677 678
      exit (-1);
      break;
    }
  }
679

680 681 682
  logInit();
  set_glog(loglvl);
  T_stdout = 1;
683 684
  /* initialize the sin table */
  InitSinLUT();
685

686
  get_softmodem_params()->phy_test = 1;
687 688
  get_softmodem_params()->do_ra = 0;

689 690
  if (snr1set==0)
    snr1 = snr0+10;
691
  init_dlsch_tpool(dlsch_threads);
692 693


694 695 696
  RC.gNB = (PHY_VARS_gNB**) malloc(sizeof(PHY_VARS_gNB *));
  RC.gNB[0] = (PHY_VARS_gNB*) malloc(sizeof(PHY_VARS_gNB ));
  memset(RC.gNB[0],0,sizeof(PHY_VARS_gNB));
Raymond Knopp's avatar
Raymond Knopp committed
697

698
  gNB = RC.gNB[0];
699
  gNB->ofdm_offset_divisor = UINT_MAX;
700 701 702 703 704 705
  frame_parms = &gNB->frame_parms; //to be initialized I suppose (maybe not necessary for PBCH)
  frame_parms->nb_antennas_tx = n_tx;
  frame_parms->nb_antennas_rx = n_rx;
  frame_parms->N_RB_DL = N_RB_DL;
  frame_parms->N_RB_UL = N_RB_DL;

Raymond Knopp's avatar
Raymond Knopp committed
706
  RC.nb_nr_macrlc_inst = 1;
cig's avatar
cig committed
707 708 709
  RC.nb_nr_mac_CC = (int*)malloc(RC.nb_nr_macrlc_inst*sizeof(int));
  for (i = 0; i < RC.nb_nr_macrlc_inst; i++)
    RC.nb_nr_mac_CC[i] = 1;
Raymond Knopp's avatar
Raymond Knopp committed
710 711 712
  mac_top_init_gNB();
  gNB_mac = RC.nrmac[0];
  gNB_RRC_INST rrc;
Raymond Knopp's avatar
Raymond Knopp committed
713
  memset((void*)&rrc,0,sizeof(rrc));
714

715
  /*
Raymond Knopp's avatar
Raymond Knopp committed
716
  // read in SCGroupConfig
Raymond Knopp's avatar
Raymond Knopp committed
717
  AssertFatal(scg_fd != NULL,"no reconfig.raw file\n");
Raymond Knopp's avatar
Raymond Knopp committed
718 719 720 721
  char buffer[1024];
  int msg_len=fread(buffer,1,1024,scg_fd);
  NR_RRCReconfiguration_t *NR_RRCReconfiguration;

Raymond Knopp's avatar
Raymond Knopp committed
722
  printf("Decoding NR_RRCReconfiguration (%d bytes)\n",msg_len);
Raymond Knopp's avatar
Raymond Knopp committed
723 724 725 726 727 728 729 730 731 732 733 734 735 736
  asn_dec_rval_t dec_rval = uper_decode_complete( NULL,
						  &asn_DEF_NR_RRCReconfiguration,
						  (void **)&NR_RRCReconfiguration,
						  (uint8_t *)buffer,
						  msg_len); 
  
  if ((dec_rval.code != RC_OK) && (dec_rval.consumed == 0)) {
    AssertFatal(1==0,"NR_RRCReConfiguration decode error\n");
    // free the memory
    SEQUENCE_free( &asn_DEF_NR_RRCReconfiguration, NR_RRCReconfiguration, 1 );
    exit(-1);
  }      
  fclose(scg_fd);

Raymond Knopp's avatar
Raymond Knopp committed
737 738
  AssertFatal(NR_RRCReconfiguration->criticalExtensions.present == NR_RRCReconfiguration__criticalExtensions_PR_rrcReconfiguration,"wrong NR_RRCReconfiguration->criticalExstions.present type\n");

Raymond Knopp's avatar
Raymond Knopp committed
739 740 741 742 743 744 745 746 747 748 749 750 751 752 753
  NR_RRCReconfiguration_IEs_t *reconfig_ies = NR_RRCReconfiguration->criticalExtensions.choice.rrcReconfiguration;
  NR_CellGroupConfig_t *secondaryCellGroup;
  dec_rval = uper_decode_complete( NULL,
				   &asn_DEF_NR_CellGroupConfig,
				   (void **)&secondaryCellGroup,
				   (uint8_t *)reconfig_ies->secondaryCellGroup->buf,
				   reconfig_ies->secondaryCellGroup->size); 
  
  if ((dec_rval.code != RC_OK) && (dec_rval.consumed == 0)) {
    AssertFatal(1==0,"NR_CellGroupConfig decode error\n");
    // free the memory
    SEQUENCE_free( &asn_DEF_NR_CellGroupConfig, secondaryCellGroup, 1 );
    exit(-1);
  }      
  
754 755
  NR_ServingCellConfigCommon_t *scc = secondaryCellGroup->spCellConfig->reconfigurationWithSync->spCellConfigCommon;
  */
Raymond Knopp's avatar
Raymond Knopp committed
756

757 758 759 760

  rrc.carrier.servingcellconfigcommon = calloc(1,sizeof(*rrc.carrier.servingcellconfigcommon));

  NR_ServingCellConfigCommon_t *scc = rrc.carrier.servingcellconfigcommon;
761
  NR_ServingCellConfig_t *scd = calloc(1,sizeof(NR_ServingCellConfig_t));
762 763
  NR_CellGroupConfig_t *secondaryCellGroup=calloc(1,sizeof(*secondaryCellGroup));
  prepare_scc(rrc.carrier.servingcellconfigcommon);
764
  uint64_t ssb_bitmap = 1;
765
  fill_scc(rrc.carrier.servingcellconfigcommon,&ssb_bitmap,N_RB_DL,N_RB_DL,mu,mu);
766
  ssb_bitmap = 1;// Enable only first SSB with index ssb_indx=0
Francesco Mani's avatar
Francesco Mani committed
767 768
  fix_scc(scc,ssb_bitmap);

769
  prepare_scd(scd);
770

771
  fill_default_secondaryCellGroup(scc, scd, secondaryCellGroup, 0, 1, n_tx, 0, 0, 0);
772

773 774
  /* RRC parameter validation for secondaryCellGroup */
  fix_scd(scd);
775
  /* -U option modify DMRS */
776
  if(modify_dmrs) {
777
    update_dmrs_config(secondaryCellGroup, dmrs_arg);
778
  }
779
  /* -T option enable PTRS */
780
  if(enable_ptrs) {
781
    update_ptrs_config(secondaryCellGroup, &rbSize, &mcsIndex, ptrs_arg);
782 783
  }

784

785
  //xer_fprint(stdout, &asn_DEF_NR_CellGroupConfig, (const void*)secondaryCellGroup);
Raymond Knopp's avatar
Raymond Knopp committed
786

Raymond Knopp's avatar
Raymond Knopp committed
787 788
  AssertFatal((gNB->if_inst         = NR_IF_Module_init(0))!=NULL,"Cannot register interface");
  gNB->if_inst->NR_PHY_config_req      = nr_phy_config_request;
Raymond Knopp's avatar
Raymond Knopp committed
789
  // common configuration
Eurecom's avatar
Eurecom committed
790
  rrc_mac_config_req_gNB(0,0, n_tx, n_tx, 0, scc, NULL, 0, 0, NULL);
Raymond Knopp's avatar
Raymond Knopp committed
791
  // UE dedicated configuration
Eurecom's avatar
Eurecom committed
792
  rrc_mac_config_req_gNB(0,0, n_tx, n_tx, 0, scc, NULL, 1, secondaryCellGroup->spCellConfig->reconfigurationWithSync->newUE_Identity,secondaryCellGroup);
793 794 795
  // reset preprocessor to the one of DLSIM after it has been set during
  // rrc_mac_config_req_gNB
  gNB_mac->pre_processor_dl = nr_dlsim_preprocessor;
796
  phy_init_nr_gNB(gNB,0,1);
797
  N_RB_DL = gNB->frame_parms.N_RB_DL;
798 799 800
  NR_UE_info_t *UE_info = &RC.nrmac[0]->UE_info;
  UE_info->num_UEs=1;

Raymond Knopp's avatar
Raymond Knopp committed
801
  // stub to configure frame_parms
Raymond Knopp's avatar
Raymond Knopp committed
802
  //  nr_phy_config_request_sim(gNB,N_RB_DL,N_RB_DL,mu,Nid_cell,SSB_positions);
Raymond Knopp's avatar
Raymond Knopp committed
803 804
  // call MAC to configure common parameters

805 806 807 808 809 810
  /* rrc_mac_config_req_gNB() has created one user, so set the scheduling
   * parameters from command line in global variables that will be picked up by
   * scheduling preprocessor */
  if (g_mcsIndex < 0) g_mcsIndex = 9;
  if (g_rbStart < 0) g_rbStart=0;
  if (g_rbSize < 0) g_rbSize = N_RB_DL - g_rbStart;
Raymond Knopp's avatar
Raymond Knopp committed
811

812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829
  double fs,bw;

  if (mu == 1 && N_RB_DL == 217) { 
    fs = 122.88e6;
    bw = 80e6;
  }					       
  else if (mu == 1 && N_RB_DL == 245) {
    fs = 122.88e6;
    bw = 90e6;
  }
  else if (mu == 1 && N_RB_DL == 273) {
    fs = 122.88e6;
    bw = 100e6;
  }
  else if (mu == 1 && N_RB_DL == 106) { 
    fs = 61.44e6;
    bw = 40e6;
  }
830 831 832 833
  else if (mu == 3 && N_RB_DL == 66) {
    fs = 122.88e6;
    bw = 100e6;
  }
834 835 836 837
  else if (mu == 3 && N_RB_DL == 32) {
    fs = 61.44e6;
    bw = 50e6;
  }
838 839 840 841 842
  else AssertFatal(1==0,"Unsupported numerology for mu %d, N_RB %d\n",mu, N_RB_DL);

  gNB2UE = new_channel_desc_scm(n_tx,
                                n_rx,
                                channel_model,
843
                                fs/1e6,//sampling frequency in MHz
844
				bw,
845
				30e-9,
846 847
                                0,
                                0,
848
                                0, 0);
849 850

  if (gNB2UE==NULL) {
851
    printf("Problem generating channel model. Exiting.\n");
852 853 854 855
    exit(-1);
  }

  frame_length_complex_samples = frame_parms->samples_per_subframe*NR_NUMBER_OF_SUBFRAMES_PER_FRAME;
856
  //frame_length_complex_samples_no_prefix = frame_parms->samples_per_subframe_wCP*NR_NUMBER_OF_SUBFRAMES_PER_FRAME;
857

858 859 860 861 862
  s_re = malloc(n_tx*sizeof(double*));
  s_im = malloc(n_tx*sizeof(double*));
  r_re = malloc(n_rx*sizeof(double*));
  r_im = malloc(n_rx*sizeof(double*));
  txdata = malloc(n_tx*sizeof(int*));
863

864
  for (i=0; i<n_tx; i++) {
865 866 867 868 869
    s_re[i] = malloc(frame_length_complex_samples*sizeof(double));
    bzero(s_re[i],frame_length_complex_samples*sizeof(double));
    s_im[i] = malloc(frame_length_complex_samples*sizeof(double));
    bzero(s_im[i],frame_length_complex_samples*sizeof(double));

870 871 872 873 874 875
    printf("Allocating %d samples for txdata\n",frame_length_complex_samples);
    txdata[i] = malloc(frame_length_complex_samples*sizeof(int));
    bzero(txdata[i],frame_length_complex_samples*sizeof(int));
  }

  for (i=0; i<n_rx; i++) {
876 877 878 879 880 881 882 883 884 885 886 887 888
    r_re[i] = malloc(frame_length_complex_samples*sizeof(double));
    bzero(r_re[i],frame_length_complex_samples*sizeof(double));
    r_im[i] = malloc(frame_length_complex_samples*sizeof(double));
    bzero(r_im[i],frame_length_complex_samples*sizeof(double));
  }

  if (pbch_file_fd!=NULL) {
    load_pbch_desc(pbch_file_fd);
  }


  //configure UE
  UE = malloc(sizeof(PHY_VARS_NR_UE));
Raymond Knopp's avatar
Raymond Knopp committed
889
  memset((void*)UE,0,sizeof(PHY_VARS_NR_UE));
890 891 892 893
  PHY_vars_UE_g = malloc(sizeof(PHY_VARS_NR_UE**));
  PHY_vars_UE_g[0] = malloc(sizeof(PHY_VARS_NR_UE*));
  PHY_vars_UE_g[0][0] = UE;
  memcpy(&UE->frame_parms,frame_parms,sizeof(NR_DL_FRAME_PARMS));
894
  UE->frame_parms.nb_antennas_rx = n_rx;
Raymond Knopp's avatar
Raymond Knopp committed
895

896
  if (run_initial_sync==1)  UE->is_synchronized = 0;
897
  else                      {UE->is_synchronized = 1; UE->UE_mode[0]=PUSCH;}
898 899 900 901 902 903 904 905 906
                      
  UE->perfect_ce = 0;

  if (init_nr_ue_signal(UE, 1, 0) != 0)
  {
    printf("Error at UE NR initialisation\n");
    exit(-1);
  }

907 908
  init_nr_ue_transport(UE,0);

909
  nr_gold_pbch(UE);
francescomani's avatar
francescomani committed
910
  nr_gold_pdcch(UE,0);
911

912
  nr_l2_init_ue(NULL);
913
  UE_mac = get_mac_inst(0);
914

915 916 917
  UE->if_inst = nr_ue_if_module_init(0);
  UE->if_inst->scheduled_response = nr_ue_scheduled_response;
  UE->if_inst->phy_config_request = nr_ue_phy_config_request;
918
  UE->if_inst->dl_indication = nr_ue_dl_indication;
919
  UE->if_inst->ul_indication = dummy_nr_ue_ul_indication;
920
  UE->prb_interpolation = prb_inter;
921

922

923
  UE_mac->if_module = nr_ue_if_module_init(0);
924

925
  unsigned int available_bits=0;
Ahmed Hussein's avatar
Ahmed Hussein committed
926 927 928 929 930
  unsigned char *estimated_output_bit;
  unsigned char *test_input_bit;
  unsigned int errors_bit    = 0;
  uint32_t errors_scrambling = 0;

Sakthivel Velumani's avatar
Sakthivel Velumani committed
931
  initTpool("N", &(nrUE_params.Tpool), false);
932

Ahmed Hussein's avatar
Ahmed Hussein committed
933 934
  test_input_bit       = (unsigned char *) malloc16(sizeof(unsigned char) * 16 * 68 * 384);
  estimated_output_bit = (unsigned char *) malloc16(sizeof(unsigned char) * 16 * 68 * 384);
935 936
  
  // generate signal
937 938
  AssertFatal(input_fd==NULL,"Not ready for input signal file\n");
  gNB->pbch_configured = 1;
Raymond Knopp's avatar
Raymond Knopp committed
939

940
  //Configure UE
941 942
  rrc.carrier.MIB = (uint8_t*) malloc(4);
  rrc.carrier.sizeof_MIB = do_MIB_NR(&rrc,0);
943

rmagueta's avatar
rmagueta committed
944
  nr_rrc_mac_config_req_ue(0,0,0,rrc.carrier.mib.message.choice.mib, NULL, NULL, secondaryCellGroup);
945

946 947 948 949 950 951 952 953

  nr_dcireq_t dcireq;
  nr_scheduled_response_t scheduled_response;
  memset((void*)&dcireq,0,sizeof(dcireq));
  memset((void*)&scheduled_response,0,sizeof(scheduled_response));
  dcireq.module_id = 0;
  dcireq.gNB_index = 0;
  dcireq.cc_id     = 0;
954
  
955 956 957 958 959 960 961
  scheduled_response.dl_config = &dcireq.dl_config_req;
  scheduled_response.ul_config = &dcireq.ul_config_req;
  scheduled_response.tx_request = NULL;
  scheduled_response.module_id = 0;
  scheduled_response.CC_id     = 0;
  scheduled_response.frame = frame;
  scheduled_response.slot  = slot;
962
  scheduled_response.thread_id = 0;
Francesco Mani's avatar
Francesco Mani committed
963

964
  nr_ue_phy_config_request(&UE_mac->phy_config);
Francesco Mani's avatar
Francesco Mani committed
965
  //NR_COMMON_channels_t *cc = RC.nrmac[0]->common_channels;
966
  snrRun = 0;
967

Sakthivel Velumani's avatar
Sakthivel Velumani committed
968 969 970 971 972 973 974 975 976 977 978 979 980 981 982
  gNB->threadPool = (tpool_t*)malloc(sizeof(tpool_t));
  char tp_param[] = "n";
  initTpool(tp_param, gNB->threadPool, true);
  gNB->resp_L1_tx = (notifiedFIFO_t*) malloc(sizeof(notifiedFIFO_t));
  initNotifiedFIFO(gNB->resp_L1_tx);
  // we create 2 threads for L1 tx processing
  notifiedFIFO_elt_t *msgL1Tx = newNotifiedFIFO_elt(sizeof(processingData_L1tx_t),0,gNB->resp_L1_tx,processSlotTX);
  processingData_L1tx_t *msgDataTx = (processingData_L1tx_t *)NotifiedFifoData(msgL1Tx);
  init_DLSCH_struct(gNB, msgDataTx);
  msgDataTx->slot = slot;
  msgDataTx->frame = frame;
  memset(msgDataTx->ssb, 0, 64*sizeof(NR_gNB_SSB_t));
  reset_meas(&msgDataTx->phy_proc_tx);
  gNB->phy_proc_tx_0 = &msgDataTx->phy_proc_tx;
  pushTpool(gNB->threadPool,msgL1Tx);
983

Ahmed Hussein's avatar
Ahmed Hussein committed
984
  for (SNR = snr0; SNR < snr1; SNR += .2) {
985

986 987 988 989 990 991 992 993 994 995
    varArray_t *table_tx=initVarArray(1000,sizeof(double));
    reset_meas(&gNB->dlsch_scrambling_stats);
    reset_meas(&gNB->dlsch_interleaving_stats);
    reset_meas(&gNB->dlsch_rate_matching_stats);
    reset_meas(&gNB->dlsch_segmentation_stats);
    reset_meas(&gNB->dlsch_modulation_stats);
    reset_meas(&gNB->dlsch_encoding_stats);
    reset_meas(&gNB->tinput);
    reset_meas(&gNB->tprep);
    reset_meas(&gNB->tparity);
996
    reset_meas(&gNB->toutput);
997

998 999
    clear_pdsch_stats(gNB);

1000
    n_errors = 0;
1001
    effRate = 0;
1002 1003
    //n_errors2 = 0;
    //n_alamouti = 0;
1004
    errors_scrambling=0;
1005
    n_false_positive = 0;
1006 1007
    if (n_trials== 1) num_rounds = 1;

Ahmed Hussein's avatar
Ahmed Hussein committed
1008
    for (trial = 0; trial < n_trials; trial++) {
1009

1010
      errors_bit = 0;
Ahmed Hussein's avatar
Ahmed Hussein committed
1011
      //multipath channel
1012
      //multipath_channel(gNB2UE,s_re,s_im,r_re,r_im,frame_length_complex_samples,0);
1013

1014
      UE->rx_offset=0;
1015 1016 1017
      UE_proc.thread_id  = 0;
      UE_proc.frame_rx   = frame;
      UE_proc.nr_slot_rx = slot;
1018 1019 1020
      
      dcireq.frame     = frame;
      dcireq.slot      = slot;
1021

1022
      NR_UE_DLSCH_t *dlsch0 = UE->dlsch[UE_proc.thread_id][0][0];
1023

1024
      int harq_pid = slot;
1025 1026
      NR_DL_UE_HARQ_t *UE_harq_process = dlsch0->harq_processes[harq_pid];

Sakthivel Velumani's avatar
Sakthivel Velumani committed
1027
      NR_gNB_DLSCH_t *gNB_dlsch = msgDataTx->dlsch[0][0];
1028
      nfapi_nr_dl_tti_pdsch_pdu_rel15_t *rel15 = &gNB_dlsch->harq_process.pdsch_pdu.pdsch_pdu_rel15;
1029
      
1030
      UE_harq_process->ack = 0;
1031
      round = 0;
1032
      UE_harq_process->round = round;
francescomani's avatar
fixes  
francescomani committed
1033
      UE_harq_process->first_rx = 1;
1034
        
1035
      while ((round<num_rounds) && (UE_harq_process->ack==0)) {
francescomani's avatar
fixes  
francescomani committed
1036

1037 1038
        memset(RC.nrmac[0]->cce_list[1][0],0,MAX_NUM_CCE*sizeof(int));
        memset(RC.nrmac[0]->cce_list[1][1],0,MAX_NUM_CCE*sizeof(int));
1039
        clear_nr_nfapi_information(RC.nrmac[0], 0, frame, slot);
1040

1041
        UE_info->UE_sched_ctrl[0].harq_processes[harq_pid].ndi = !(trial&1);
1042

1043

1044
        UE_info->UE_sched_ctrl[0].harq_processes[harq_pid].round = round;
Sakthivel Velumani's avatar
Sakthivel Velumani committed
1045
        for (int i=0; i<MAX_NUM_CORESET; i++)
1046
          gNB_mac->UE_info.num_pdcch_cand[0][i] = 0;
1047
      
1048
        if (css_flag == 0) {
1049
          nr_schedule_ue_spec(0, frame, slot);
1050 1051 1052
        } else {
          nr_schedule_css_dlsch_phytest(0,frame,slot);
        }
1053 1054 1055 1056 1057
        Sched_INFO.module_id = 0;
        Sched_INFO.CC_id     = 0;
        Sched_INFO.frame     = frame;
        Sched_INFO.slot      = slot;
        Sched_INFO.DL_req    = &gNB_mac->DL_req[0];
1058
        Sched_INFO.UL_tti_req    = gNB_mac->UL_tti_req_ahead[slot];
1059 1060 1061
        Sched_INFO.UL_dci_req  = NULL;
        Sched_INFO.TX_req    = &gNB_mac->TX_req[0];
        nr_schedule_response(&Sched_INFO);
1062 1063 1064 1065 1066 1067

        /* PTRS values for DLSIM calculations   */
        nfapi_nr_dl_tti_request_body_t *dl_req = &gNB_mac->DL_req[Sched_INFO.CC_id].dl_tti_request_body;
        nfapi_nr_dl_tti_request_pdu_t  *dl_tti_pdsch_pdu = &dl_req->dl_tti_pdu_list[1];
        nfapi_nr_dl_tti_pdsch_pdu_rel15_t *pdsch_pdu_rel15 = &dl_tti_pdsch_pdu->pdsch_pdu.pdsch_pdu_rel15;
        pdu_bit_map = pdsch_pdu_rel15->pduBitmap;
1068
        if(pdu_bit_map & 0x1) {
1069 1070 1071 1072 1073 1074 1075
          set_ptrs_symb_idx(&dlPtrsSymPos,
                            pdsch_pdu_rel15->NrOfSymbols,
                            pdsch_pdu_rel15->StartSymbolIndex,
                            1<<pdsch_pdu_rel15->PTRSTimeDensity,
                            pdsch_pdu_rel15->dlDmrsSymbPos);
          ptrsSymbPerSlot = get_ptrs_symbols_in_slot(dlPtrsSymPos, pdsch_pdu_rel15->StartSymbolIndex, pdsch_pdu_rel15->NrOfSymbols);
          ptrsRePerSymb = ((rel15->rbSize + rel15->PTRSFreqDensity - 1)/rel15->PTRSFreqDensity);
1076
          printf("[DLSIM] PTRS Symbols in a slot: %2u, RE per Symbol: %3u, RE in a slot %4d\n", ptrsSymbPerSlot,ptrsRePerSymb, ptrsSymbPerSlot*ptrsRePerSymb );
1077
        }
Sakthivel Velumani's avatar
Sakthivel Velumani committed
1078 1079 1080 1081

        msgDataTx->ssb[0].ssb_pdu.ssb_pdu_rel15.bchPayload=0x001234;
        msgDataTx->ssb[0].ssb_pdu.ssb_pdu_rel15.SsbBlockIndex = 0;
        msgDataTx->gNB = gNB;
1082
        if (run_initial_sync)
Sakthivel Velumani's avatar
Sakthivel Velumani committed
1083
          nr_common_signal_procedures(gNB,frame,slot,msgDataTx->ssb[0].ssb_pdu);
1084
        else
Sakthivel Velumani's avatar
Sakthivel Velumani committed
1085
          phy_procedures_gNB_TX(msgDataTx,frame,slot,1);
1086 1087 1088 1089
            
        int txdataF_offset = (slot%2) * frame_parms->samples_per_slot_wCP;
        
        if (n_trials==1) {
1090
          LOG_M("txsigF0.m","txsF0=", &gNB->common_vars.txdataF[0][txdataF_offset+2*frame_parms->ofdm_symbol_size],frame_parms->ofdm_symbol_size,1,1);
1091
          if (gNB->frame_parms.nb_antennas_tx>1)
1092
            LOG_M("txsigF1.m","txsF1=", &gNB->common_vars.txdataF[1][txdataF_offset+2*frame_parms->ofdm_symbol_size],frame_parms->ofdm_symbol_size,1,1);
1093 1094
        }
        int tx_offset = frame_parms->get_samples_slot_timestamp(slot,frame_parms,0);
1095
        if (n_trials==1) printf("tx_offset %d, txdataF_offset %d \n", tx_offset,txdataF_offset);
1096

1097 1098 1099
        //TODO: loop over slots
        for (aa=0; aa<gNB->frame_parms.nb_antennas_tx; aa++) {
    
1100 1101 1102 1103 1104 1105 1106
          if (cyclic_prefix_type == 1) {
            PHY_ofdm_mod(&gNB->common_vars.txdataF[aa][txdataF_offset],
                         &txdata[aa][tx_offset],
                         frame_parms->ofdm_symbol_size,
                         12,
                         frame_parms->nb_prefix_samples,
                         CYCLIC_PREFIX);
1107
          } else {
1108 1109 1110
            nr_normal_prefix_mod(&gNB->common_vars.txdataF[aa][txdataF_offset],
                                 &txdata[aa][tx_offset],
                                 14,
1111 1112
                                 frame_parms,
                                 slot);
1113
          }
1114 1115 1116
        }
       
        if (n_trials==1) {
1117 1118 1119 1120 1121
          char filename[100];//LOG_M
          for (aa=0;aa<n_tx;aa++) {
            sprintf(filename,"txsig%d.m", aa);//LOG_M
            LOG_M(filename,"txs", &txdata[aa][tx_offset+frame_parms->ofdm_symbol_size+frame_parms->nb_prefix_samples0],6*(frame_parms->ofdm_symbol_size+frame_parms->nb_prefix_samples),1,1);
          }
1122
        }
Sakthivel Velumani's avatar
Sakthivel Velumani committed
1123 1124
        if (output_fd) {
          printf("writing txdata to binary file\n");
1125
          fwrite(txdata[0],sizeof(int32_t),frame_length_complex_samples,output_fd);
Sakthivel Velumani's avatar
Sakthivel Velumani committed
1126
        }
1127

1128 1129 1130 1131 1132 1133 1134 1135 1136
        int txlev[n_tx];
        int txlev_sum = 0;
        int l_ofdm = 6;
        for (aa=0; aa<n_tx; aa++) {
          txlev[aa] = signal_energy(&txdata[aa][tx_offset+l_ofdm*frame_parms->ofdm_symbol_size + (l_ofdm-1)*frame_parms->nb_prefix_samples + frame_parms->nb_prefix_samples0],
          frame_parms->ofdm_symbol_size + frame_parms->nb_prefix_samples);
          txlev_sum += txlev[aa];
          if (n_trials==1) printf("txlev[%d] = %d (%f dB) txlev_sum %d\n",aa,txlev[aa],10*log10((double)txlev[aa]),txlev_sum);
        }
1137 1138
        
        for (i=(frame_parms->get_samples_slot_timestamp(slot,frame_parms,0)); 
1139 1140 1141 1142
             i<(frame_parms->get_samples_slot_timestamp(slot+1,frame_parms,0)); 
             i++) {
    
          for (aa=0; aa<frame_parms->nb_antennas_tx; aa++) {
1143 1144
            s_re[aa][i] = ((double)(((short *)txdata[aa]))[(i<<1)]);
            s_im[aa][i] = ((double)(((short *)txdata[aa]))[(i<<1)+1]);
1145
          }
1146
        }
1147

1148
        double ts = 1.0/(frame_parms->subcarrier_spacing * frame_parms->ofdm_symbol_size); 
1149
        //Compute AWGN variance
1150
        sigma2_dB = 10 * log10((double)txlev_sum * ((double)UE->frame_parms.ofdm_symbol_size/(12*rel15->rbSize))) - SNR;
1151
        sigma2    = pow(10, sigma2_dB/10);
1152
        if (n_trials==1) printf("sigma2 %f (%f dB), txlev_sum %f (factor %f)\n",sigma2,sigma2_dB,10*log10((double)txlev_sum),(double)(double)UE->frame_parms.ofdm_symbol_size/(12*rel15->rbSize));
1153

1154 1155 1156 1157
        for (aa=0; aa<n_rx; aa++) {
          bzero(r_re[aa],frame_length_complex_samples*sizeof(double));
          bzero(r_im[aa],frame_length_complex_samples*sizeof(double));
        }
1158
        
1159 1160 1161 1162 1163 1164 1165 1166 1167
        // Apply MIMO Channel
        if (channel_model != AWGN) multipath_tv_channel(gNB2UE,
                             s_re,
                             s_im,
                             r_re,
                             r_im,
                             frame_length_complex_samples,
                             0);

1168 1169 1170 1171 1172
        double H_awgn_mimo[4][4] ={{1.0, 0.5, 0.25, 0.125},//rx 0
                                   {0.5, 1.0, 0.5, 0.25},  //rx 1
                                   {0.25, 0.5, 1.0, 0.5},  //rx 2
                                   {0.125, 0.25, 0.5, 1.0}};//rx 3

1173
        for (i=frame_parms->get_samples_slot_timestamp(slot,frame_parms,0); 
1174 1175 1176
             i<frame_parms->get_samples_slot_timestamp(slot+1,frame_parms,0);
             i++) {

1177
          for (int aa_rx=0; aa_rx<n_rx; aa_rx++) {
1178 1179 1180 1181 1182

            if (channel_model == AWGN) {
              // sum up signals from different Tx antennas
              r_re[aa_rx][i] = 0;
              r_im[aa_rx][i] = 0;
1183 1184 1185
             for (aa=0; aa<n_tx; aa++) {
                r_re[aa_rx][i] += s_re[aa][i]*H_awgn_mimo[aa_rx][aa];
                r_im[aa_rx][i] += s_im[aa][i]*H_awgn_mimo[aa_rx][aa];
1186
              }
1187
            }
1188
            // Add Gaussian noise
1189 1190
            ((short*) UE->common_vars.rxdata[aa_rx])[2*i]   = (short) ((r_re[aa_rx][i] + sqrt(sigma2/2)*gaussdouble(0.0,1.0)));
            ((short*) UE->common_vars.rxdata[aa_rx])[2*i+1] = (short) ((r_im[aa_rx][i] + sqrt(sigma2/2)*gaussdouble(0.0,1.0)));
1191
            /* Add phase noise if enabled */
1192
            if (pdu_bit_map & 0x1) {
1193 1194
              phase_noise(ts, &((short*) UE->common_vars.rxdata[aa_rx])[2*i],
                          &((short*) UE->common_vars.rxdata[aa_rx])[2*i+1]);
1195
            }
1196
          }
1197
        }
1198

1199 1200 1201 1202
        nr_ue_dcireq(&dcireq); //to be replaced with function pointer later
        nr_ue_scheduled_response(&scheduled_response);
        
        phy_procedures_nrUE_RX(UE,
1203 1204
                               &UE_proc,
                               0,
1205 1206
                               dlsch_threads,
                               NULL);
1207
        
1208
        //printf("dlsim round %d ends\n",round);
1209 1210
        round++;
      } // round
1211

1212 1213 1214
      //----------------------------------------------------------
      //---------------------- count errors ----------------------
      //----------------------------------------------------------
1215 1216 1217

      if (UE->dlsch[UE_proc.thread_id][0][0]->last_iteration_cnt >=
        UE->dlsch[UE_proc.thread_id][0][0]->max_ldpc_iterations+1)
1218
        n_errors++;
1219 1220

      NR_UE_PDSCH **pdsch_vars = UE->pdsch_vars[UE_proc.thread_id];
1221
      int16_t *UE_llr = pdsch_vars[0]->llr[0];
1222

1223
      TBS                  = UE_harq_process->TBS;//rel15->TBSize[0];
1224
      uint16_t length_dmrs = get_num_dmrs(rel15->dlDmrsSymbPos);
1225 1226 1227 1228
      uint16_t nb_rb       = rel15->rbSize;
      uint8_t  nb_re_dmrs  = rel15->dmrsConfigType == NFAPI_NR_DMRS_TYPE1 ? 6 : 4;
      uint8_t  mod_order   = rel15->qamModOrder[0];
      uint8_t  nb_symb_sch = rel15->NrOfSymbols;
1229

1230
      available_bits = nr_get_G(nb_rb, nb_symb_sch, nb_re_dmrs, length_dmrs, mod_order, rel15->nrOfLayers);
1231
      if(pdu_bit_map & 0x1) {
1232
        available_bits-= (ptrsSymbPerSlot * ptrsRePerSymb *rel15->nrOfLayers* 2);
1233
        printf("[DLSIM][PTRS] Available bits are: %5u, removed PTRS bits are: %5u \n",available_bits, (ptrsSymbPerSlot * ptrsRePerSymb *rel15->nrOfLayers* 2) );
1234
      }
1235

1236
      for (i = 0; i < available_bits; i++) {
1237 1238

	if(((gNB_dlsch->harq_process.f[i] == 0) && (UE_llr[i] <= 0)) ||
1239
	   ((gNB_dlsch->harq_process.f[i] == 1) && (UE_llr[i] >= 0)))
1240 1241 1242 1243
	  {
	    if(errors_scrambling == 0) {
	      LOG_D(PHY,"\n");
	      LOG_D(PHY,"First bit in error in unscrambling = %d\n",i);
Raymond Knopp's avatar
Raymond Knopp committed
1244
	    }
1245
	    errors_scrambling++;
Raymond Knopp's avatar
Raymond Knopp committed
1246
	  }
1247

1248 1249
      }
      for (i = 0; i < TBS; i++) {
1250

1251
	estimated_output_bit[i] = (UE_harq_process->b[i/8] & (1 << (i & 7))) >> (i & 7);
1252
	test_input_bit[i]       = (gNB_dlsch->harq_process.b[i / 8] & (1 << (i & 7))) >> (i & 7); // Further correct for multiple segments
Raymond Knopp's avatar
Raymond Knopp committed
1253
	
1254 1255 1256 1257
	if (estimated_output_bit[i] != test_input_bit[i]) {
	  if(errors_bit == 0)
	    LOG_D(PHY,"First bit in error in decoding = %d (errors scrambling %d)\n",i,errors_scrambling);
	  errors_bit++;
Raymond Knopp's avatar
Raymond Knopp committed
1258 1259
	}
	
1260 1261 1262 1263 1264 1265
      }
      
      ////////////////////////////////////////////////////////////
      
      if (errors_scrambling > 0) {
	if (n_trials == 1)
1266
	  printf("errors_scrambling = %u/%u (trial %d)\n", errors_scrambling, available_bits,trial);
1267 1268 1269 1270 1271 1272 1273
      }
      
      if (errors_bit > 0) {
	n_false_positive++;
	if (n_trials == 1)
	  printf("errors_bit = %u (trial %d)\n", errors_bit, trial);
      }
1274
      roundStats[snrRun]+=((float)round); 
1275
      if (UE_harq_process->ack==1) effRate += ((float)TBS)/round;
Ahmed Hussein's avatar
Ahmed Hussein committed
1276
    } // noise trials
1277

1278
    blerStats[snrRun] = (float) n_errors / (float) n_trials;
1279
    roundStats[snrRun]/=((float)n_trials);
1280
    berStats[snrRun] = (double)errors_scrambling/available_bits/n_trials;
1281
    effRate /= n_trials;
Ahmed Hussein's avatar
Ahmed Hussein committed
1282 1283
    printf("*****************************************\n");
    printf("SNR %f, (false positive %f)\n", SNR,
Sakthivel Velumani's avatar
Sakthivel Velumani committed
1284
           (float) n_errors / (float) n_trials);
Ahmed Hussein's avatar
Ahmed Hussein committed
1285 1286
    printf("*****************************************\n");
    printf("\n");
Raymond Knopp's avatar
Raymond Knopp committed
1287
    dump_pdsch_stats(stdout,gNB);
1288
    printf("SNR %f : n_errors (negative CRC) = %d/%d, Avg round %.2f, Channel BER %e, BLER %.2f, Eff Rate %.4f bits/slot, Eff Throughput %.2f, TBS %u bits/slot\n", SNR, n_errors, n_trials,roundStats[snrRun],berStats[snrRun],blerStats[snrRun],effRate,effRate/TBS*100,TBS);
Ahmed Hussein's avatar
Ahmed Hussein committed
1289
    printf("\n");
1290

1291
    if (print_perf==1) {
1292
      printf("\ngNB TX function statistics (per %d us slot, NPRB %d, mcs %d, TBS %d, Kr %d (Zc %d))\n",
1293
	     1000>>*scc->ssbSubcarrierSpacing, g_rbSize, g_mcsIndex,
Sakthivel Velumani's avatar
Sakthivel Velumani committed
1294 1295 1296 1297
	     msgDataTx->dlsch[0][0]->harq_process.pdsch_pdu.pdsch_pdu_rel15.TBSize[0]<<3,
	     msgDataTx->dlsch[0][0]->harq_process.K,
	     msgDataTx->dlsch[0][0]->harq_process.K/((msgDataTx->dlsch[0][0]->harq_process.pdsch_pdu.pdsch_pdu_rel15.TBSize[0]<<3)>3824?22:10));
      printDistribution(gNB->phy_proc_tx_0,table_tx,"PHY proc tx");
1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324
      printStatIndent2(&gNB->dlsch_encoding_stats,"DLSCH encoding time");
      printStatIndent3(&gNB->dlsch_segmentation_stats,"DLSCH segmentation time");
      printStatIndent3(&gNB->tinput,"DLSCH LDPC input processing time");
      printStatIndent3(&gNB->tprep,"DLSCH LDPC input preparation time");
      printStatIndent3(&gNB->tparity,"DLSCH LDPC parity generation time");
      printStatIndent3(&gNB->toutput,"DLSCH LDPC output generation time");
      printStatIndent3(&gNB->dlsch_rate_matching_stats,"DLSCH Rate Mataching time");
      printStatIndent3(&gNB->dlsch_interleaving_stats,  "DLSCH Interleaving time");
      printStatIndent2(&gNB->dlsch_modulation_stats,"DLSCH modulation time");
      printStatIndent2(&gNB->dlsch_scrambling_stats,  "DLSCH scrambling time");


      printf("\nUE RX function statistics (per %d us slot)\n",1000>>*scc->ssbSubcarrierSpacing);
      /*
      printDistribution(&phy_proc_rx_tot, table_rx,"Total PHY proc rx");
      printStatIndent(&ue_front_end_tot,"Front end processing");
      printStatIndent(&dlsch_llr_tot,"rx_pdsch processing");
      printStatIndent2(&pdsch_procedures_tot,"pdsch processing");
      printStatIndent2(&dlsch_procedures_tot,"dlsch processing");
      printStatIndent2(&UE->crnti_procedures_stats,"C-RNTI processing");
      printStatIndent(&UE->ofdm_demod_stats,"ofdm demodulation");
      printStatIndent(&UE->dlsch_channel_estimation_stats,"DLSCH channel estimation time");
      printStatIndent(&UE->dlsch_freq_offset_estimation_stats,"DLSCH frequency offset estimation time");
      printStatIndent(&dlsch_decoding_tot, "DLSCH Decoding time ");
      printStatIndent(&UE->dlsch_unscrambling_stats,"DLSCH unscrambling time");
      printStatIndent(&UE->dlsch_rate_unmatching_stats,"DLSCH Rate Unmatching");
      printf("|__ DLSCH Turbo Decoding(%d bits), avg iterations: %.1f       %.2f us (%d cycles, %d trials)\n",
1325 1326 1327
	     UE->dlsch[UE_proc.thread_id][0][0]->harq_processes[0]->Cminus ?
	     UE->dlsch[UE_proc.thread_id][0][0]->harq_processes[0]->Kminus :
	     UE->dlsch[UE_proc.thread_id][0][0]->harq_processes[0]->Kplus,
1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340
	     UE->dlsch_tc_intl1_stats.trials/(double)UE->dlsch_tc_init_stats.trials,
	     (double)UE->dlsch_turbo_decoding_stats.diff/UE->dlsch_turbo_decoding_stats.trials*timeBase,
	     (int)((double)UE->dlsch_turbo_decoding_stats.diff/UE->dlsch_turbo_decoding_stats.trials),
	     UE->dlsch_turbo_decoding_stats.trials);
      printStatIndent2(&UE->dlsch_tc_init_stats,"init");
      printStatIndent2(&UE->dlsch_tc_alpha_stats,"alpha");
      printStatIndent2(&UE->dlsch_tc_beta_stats,"beta");
      printStatIndent2(&UE->dlsch_tc_gamma_stats,"gamma");
      printStatIndent2(&UE->dlsch_tc_ext_stats,"ext");
      printStatIndent2(&UE->dlsch_tc_intl1_stats,"turbo internal interleaver");
      printStatIndent2(&UE->dlsch_tc_intl2_stats,"intl2+HardDecode+CRC");
      */
    }
1341 1342 1343 1344 1345 1346

    if (n_trials == 1) {
      
      LOG_M("rxsig0.m","rxs0", UE->common_vars.rxdata[0], frame_length_complex_samples, 1, 1);
      if (UE->frame_parms.nb_antennas_rx>1)
	LOG_M("rxsig1.m","rxs1", UE->common_vars.rxdata[1], frame_length_complex_samples, 1, 1);
1347
      LOG_M("chestF0.m","chF0",&UE->pdsch_vars[0][0]->dl_ch_estimates_ext[0][0],g_rbSize*12*14,1,1);
1348
      write_output("rxF_comp.m","rxFc",&UE->pdsch_vars[0][0]->rxdataF_comp0[0][0],N_RB_DL*12*14,1,1);
1349
      LOG_M("rxF_llr.m","rxFllr",UE->pdsch_vars[UE_proc.thread_id][0]->llr[0],available_bits,1,0);
Eurecom's avatar
Eurecom committed
1350 1351
      LOG_M("pdcch_rxFcomp.m","pdcch_rxFcomp",&UE->pdcch_vars[0][0]->rxdataF_comp[0][0],96*12,1,1);
      LOG_M("pdcch_rxFllr.m","pdcch_rxFllr",UE->pdcch_vars[0][0]->llr,96*12,1,1);
1352 1353 1354
      break;
    }

1355
    if (effRate > (eff_tp_check*TBS)) {
1356 1357 1358 1359
      printf("PDSCH test OK\n");
      break;
    }

1360
    snrStats[snrRun] = SNR;
1361
    snrRun++;
1362 1363
  } // NSR

1364 1365 1366 1367
  LOG_M("dlsimStats.m","SNR",snrStats,snrRun,1,7);
  LOG_MM("dlsimStats.m","BLER",blerStats,snrRun,1,7);
  LOG_MM("dlsimStats.m","BER",berStats,snrRun,1,7);
  LOG_MM("dlsimStats.m","rounds",roundStats,snrRun,1,7);
1368
  /*if (n_trials>1) {
1369 1370 1371 1372 1373 1374
    printf("HARQ stats:\nSNR\tRounds\n");
    psnr = snr0;
    for (uint8_t i=0; i<snrRun; i++) {
      printf("%.1f\t%.2f\n",psnr,roundStats[i]);
      psnr+=0.2;
    }
1375
  }*/
1376

1377 1378
  free_channel_desc_scm(gNB2UE);

1379
  for (i = 0; i < n_tx; i++) {
1380 1381
    free(s_re[i]);
    free(s_im[i]);
1382 1383 1384
    free(txdata[i]);
  }
  for (i = 0; i < n_rx; i++) {
1385 1386 1387 1388 1389 1390 1391 1392 1393
    free(r_re[i]);
    free(r_im[i]);
  }

  free(s_re);
  free(s_im);
  free(r_re);
  free(r_im);
  free(txdata);
Ahmed Hussein's avatar
Ahmed Hussein committed
1394 1395 1396
  free(test_input_bit);
  free(estimated_output_bit);
  
1397 1398 1399 1400 1401 1402
  if (output_fd)
    fclose(output_fd);

  if (input_fd)
    fclose(input_fd);

1403 1404
  if (scg_fd)
    fclose(scg_fd);
1405
  return(n_errors);
Raymond Knopp's avatar
Raymond Knopp committed
1406
  
1407
}
1408 1409


1410
void update_ptrs_config(NR_CellGroupConfig_t *secondaryCellGroup, uint16_t *rbSize, uint8_t *mcsIndex, int8_t *ptrs_arg)
1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423
{
  NR_BWP_Downlink_t *bwp=secondaryCellGroup->spCellConfig->spCellConfigDedicated->downlinkBWP_ToAddModList->list.array[0];
  int *ptrsFreqDenst = calloc(2, sizeof(long));
  ptrsFreqDenst[0]= 25;
  ptrsFreqDenst[1]= 115;
  int *ptrsTimeDenst = calloc(3, sizeof(long));
  ptrsTimeDenst[0]= 2;
  ptrsTimeDenst[1]= 4;
  ptrsTimeDenst[2]= 10;

  int epre_Ratio = 0;
  int reOffset = 0;

1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435
  if(ptrs_arg[0] ==0) {
    ptrsTimeDenst[2]= *mcsIndex -1;
  }
  else if(ptrs_arg[0] == 1) {
    ptrsTimeDenst[1]= *mcsIndex - 1;
    ptrsTimeDenst[2]= *mcsIndex + 1;
  }
  else if(ptrs_arg[0] ==2) {
    ptrsTimeDenst[0]= *mcsIndex - 1;
    ptrsTimeDenst[1]= *mcsIndex + 1;
  }
  else {
1436 1437 1438
    printf("[DLSIM] Wrong L_PTRS value, using default values 1\n");
  }
  /* L = 4 if Imcs < MCS4 */
1439 1440 1441 1442 1443 1444 1445 1446 1447 1448
  if(ptrs_arg[1] ==2) {
    ptrsFreqDenst[0]= *rbSize - 1;
    ptrsFreqDenst[1]= *rbSize + 1;
  }
  else if(ptrs_arg[1] == 4) {
    ptrsFreqDenst[1]= *rbSize - 1;
  }
  else {
    printf("[DLSIM] Wrong K_PTRS value, using default values 2\n");
  }
1449
  printf("[DLSIM] PTRS Enabled with L %d, K %d \n", 1<<ptrs_arg[0], ptrs_arg[1] );
1450 1451 1452 1453
  /* overwrite the values */
  rrc_config_dl_ptrs_params(bwp, ptrsFreqDenst, ptrsTimeDenst, &epre_Ratio, &reOffset);
}

1454
void update_dmrs_config(NR_CellGroupConfig_t *scg, int8_t* dmrs_arg)
1455 1456 1457
{
  int8_t  mapping_type = typeA;//default value
  int8_t  add_pos = pdsch_dmrs_pos0;//default value
1458 1459
  int8_t  dmrs_config_type = NFAPI_NR_DMRS_TYPE1;//default value

1460
  if(dmrs_arg[0] == 0) {
1461 1462
    mapping_type = typeA;
  }
1463
  else if (dmrs_arg[0] == 1) {
1464
    mapping_type = typeB;
1465 1466
  } else {
    AssertFatal(1==0,"Incorrect Mappingtype, valid options 0-typeA, 1-typeB\n");
1467
  }
1468 1469 1470

  /* Additional DMRS positions 0 ,1 ,2 and 3 */
  if(dmrs_arg[1] >= 0 && dmrs_arg[1] <4 ) {
1471
    add_pos = dmrs_arg[1];
1472 1473
  } else {
    AssertFatal(1==0,"Incorrect Additional Position, valid options 0-pos1, 1-pos1, 2-pos2, 3-pos3\n");
1474
  }
1475

1476
  /* DMRS Conf Type 1 or 2 */
1477 1478 1479 1480
  if(dmrs_arg[2] == 1) {
    dmrs_config_type = NFAPI_NR_DMRS_TYPE1;
  } else if(dmrs_arg[2] == 2) {
    dmrs_config_type = NFAPI_NR_DMRS_TYPE2;
1481
  }
1482

1483
  NR_BWP_Downlink_t *bwp = scg->spCellConfig->spCellConfigDedicated->downlinkBWP_ToAddModList->list.array[0];
1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501

  AssertFatal((bwp->bwp_Dedicated->pdsch_Config != NULL && bwp->bwp_Dedicated->pdsch_Config->choice.setup != NULL), "Base RRC reconfig structures are not allocated.\n");

  if(mapping_type == typeA) {
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA = calloc(1,sizeof(*bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA));
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->present= NR_SetupRelease_DMRS_DownlinkConfig_PR_setup;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup = calloc(1,sizeof(*bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup));
    if (dmrs_config_type == NFAPI_NR_DMRS_TYPE2)
      bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup->dmrs_Type = calloc(1,sizeof(*bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup->dmrs_Type));
    else
      bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup->dmrs_Type = NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup->maxLength=NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup->scramblingID0=NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup->scramblingID1=NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup->phaseTrackingRS=NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA->choice.setup->dmrs_AdditionalPosition = NULL;
    printf("DLSIM: Allocated Mapping TypeA in RRC reconfig message\n");
  }
1502

1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517
  if(mapping_type == typeB) {
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB = calloc(1,sizeof(*bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB));
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->present= NR_SetupRelease_DMRS_DownlinkConfig_PR_setup;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup = calloc(1,sizeof(*bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup));
    if (dmrs_config_type == NFAPI_NR_DMRS_TYPE2)
      bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup->dmrs_Type = calloc(1,sizeof(*bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup->dmrs_Type));
    else
      bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup->dmrs_Type = NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup->maxLength=NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup->scramblingID0=NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup->scramblingID1=NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup->phaseTrackingRS=NULL;
    bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB->choice.setup->dmrs_AdditionalPosition = NULL;
    printf("DLSIM: Allocated Mapping TypeB in RRC reconfig message\n");
  }
1518

1519 1520
  struct NR_SetupRelease_DMRS_DownlinkConfig	*dmrs_MappingtypeA = bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeA;
  struct NR_SetupRelease_DMRS_DownlinkConfig	*dmrs_MappingtypeB = bwp->bwp_Dedicated->pdsch_Config->choice.setup->dmrs_DownlinkForPDSCH_MappingTypeB;
1521 1522


1523 1524 1525 1526 1527
  NR_DMRS_DownlinkConfig_t *dmrs_config = (mapping_type == typeA) ? dmrs_MappingtypeA->choice.setup : dmrs_MappingtypeB->choice.setup;

  if (add_pos != 2) { // pos0,pos1,pos3
    if (dmrs_config->dmrs_AdditionalPosition == NULL) {
      dmrs_config->dmrs_AdditionalPosition = calloc(1,sizeof(*dmrs_MappingtypeA->choice.setup->dmrs_AdditionalPosition));
1528
    }
1529 1530 1531 1532
    *dmrs_config->dmrs_AdditionalPosition = add_pos;
  } else { // if NULL, Value pos2
    free(dmrs_config->dmrs_AdditionalPosition);
    dmrs_config->dmrs_AdditionalPosition = NULL;
1533
  }
1534

1535
  for (int i=0;i<bwp->bwp_Common->pdsch_ConfigCommon->choice.setup->pdsch_TimeDomainAllocationList->list.count;i++) {
1536
    bwp->bwp_Common->pdsch_ConfigCommon->choice.setup->pdsch_TimeDomainAllocationList->list.array[i]->mappingType = mapping_type;
1537
  }
1538

1539
  printf("[DLSIM] DMRS Config is modified with Mapping Type %d, Additional Positions %d Config. Type %d \n", mapping_type, add_pos, dmrs_config_type);
1540
}