From ee20e21351ed1bd56f5cf1675097bff3092c3693 Mon Sep 17 00:00:00 2001
From: Francesco Mani <francesco.mani@orange.com>
Date: Tue, 24 Sep 2019 17:54:16 +0200
Subject: [PATCH] adding an if for dmrs offset config

---
 openair1/PHY/NR_TRANSPORT/nr_dci.c | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/openair1/PHY/NR_TRANSPORT/nr_dci.c b/openair1/PHY/NR_TRANSPORT/nr_dci.c
index a680fc3e4d..2ac7823d45 100644
--- a/openair1/PHY/NR_TRANSPORT/nr_dci.c
+++ b/openair1/PHY/NR_TRANSPORT/nr_dci.c
@@ -266,7 +266,13 @@ uint8_t nr_generate_dci_top(NR_gNB_PDCCH pdcch_vars,
   if (pdcch_params.precoder_granularity == NFAPI_NR_CSET_ALL_CONTIGUOUS_RBS) {
     /*in this case the DMRS are mapped on all the coreset*/
     for (l=cset_start_symb; l<cset_start_symb+ cset_nsymb; l++) {
-      dmrs_idx = pdcch_params.rb_offset*3;
+
+      // dmrs index depends on reference point for k according to 38.211 7.4.1.3.2
+      if (pdcch_params.config_type == NFAPI_NR_CSET_CONFIG_PDCCH_CONFIG)
+        dmrs_idx = 0;
+      else
+        dmrs_idx = pdcch_params.rb_offset*3;
+
       k = cset_start_sc + 1;
 
       while (dmrs_idx<3*pdcch_params.n_rb) {
@@ -296,7 +302,13 @@ uint8_t nr_generate_dci_top(NR_gNB_PDCCH pdcch_vars,
       k -= frame_parms.ofdm_symbol_size;
 
     l = cset_start_symb + reg.symb_idx;
-    dmrs_idx = ((reg.reg_idx/cset_nsymb)+pdcch_params.rb_offset)*3;
+
+    // dmrs index depends on reference point for k according to 38.211 7.4.1.3.2
+    if (pdcch_params.config_type == NFAPI_NR_CSET_CONFIG_PDCCH_CONFIG)
+      dmrs_idx = (reg.reg_idx/cset_nsymb)*3;
+    else
+      dmrs_idx = ((reg.reg_idx/cset_nsymb)+pdcch_params.rb_offset)*3;
+
     k_prime = 0;
 
     for (int m=0; m<NR_NB_SC_PER_RB; m++) {