nr_dlsch_coding.c 19 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 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37
/*
 * 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.0  (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
 */

/*! \file PHY/LTE_TRANSPORT/dlsch_coding.c
* \brief Top-level routines for implementing LDPC-coded (DLSCH) transport channels from 38-212, 15.2
* \author H.Wang
* \date 2018
* \version 0.1
* \company Eurecom
* \email:
* \note
* \warning
*/

#include "PHY/defs_gNB.h"
#include "PHY/phy_extern.h"
#include "PHY/CODING/coding_extern.h"
#include "PHY/CODING/coding_defs.h"
#include "PHY/CODING/lte_interleaver_inline.h"
38
#include "PHY/CODING/nrLDPC_extern.h"
cig's avatar
cig committed
39
#include "PHY/NR_TRANSPORT/nr_transport_proto.h"
40
#include "PHY/NR_TRANSPORT/nr_transport_common_proto.h"
41
#include "PHY/NR_TRANSPORT/nr_dlsch.h"
42
#include "openair2/LAYER2/NR_MAC_gNB/mac_proto.h"
43
#include "SCHED_NR/sched_nr.h"
44 45
#include "common/utils/LOG/vcd_signal_dumper.h"
#include "common/utils/LOG/log.h"
46 47 48 49 50
#include <syscall.h>

//#define DEBUG_DLSCH_CODING
//#define DEBUG_DLSCH_FREE 1

51

cig's avatar
cig committed
52
void free_gNB_dlsch(NR_gNB_DLSCH_t **dlschptr, uint16_t N_RB)
53 54 55 56
{
  int i;
  int r;

57
  NR_gNB_DLSCH_t *dlsch = *dlschptr;
58

59
  uint16_t a_segments = MAX_NUM_NR_DLSCH_SEGMENTS;  //number of segments to be allocated
60
  if (dlsch) {
61 62 63

    if (N_RB != 273) {
      a_segments = a_segments*N_RB;
64
      a_segments = a_segments/273 +1;
65 66 67 68
    }  



69
#ifdef DEBUG_DLSCH_FREE
70
    LOG_D(PHY,"Freeing dlsch %p\n",dlsch);
71 72 73 74
#endif

    for (i=0; i<dlsch->Mdlharq; i++) {
#ifdef DEBUG_DLSCH_FREE
75
      LOG_D(PHY,"Freeing dlsch process %d\n",i);
76 77 78 79
#endif

      if (dlsch->harq_processes[i]) {
#ifdef DEBUG_DLSCH_FREE
80
        LOG_D(PHY,"Freeing dlsch process %d (%p)\n",i,dlsch->harq_processes[i]);
81 82 83
#endif

        if (dlsch->harq_processes[i]->b) {
laurent's avatar
laurent committed
84
          free16(dlsch->harq_processes[i]->b,a_segments*1056);
85 86
          dlsch->harq_processes[i]->b = NULL;
#ifdef DEBUG_DLSCH_FREE
87
          LOG_D(PHY,"Freeing dlsch process %d b (%p)\n",i,dlsch->harq_processes[i]->b);
88 89 90
#endif
        }

91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
        if (dlsch->harq_processes[i]->e) {
          free16(dlsch->harq_processes[i]->e,14*N_RB*12*8);
          dlsch->harq_processes[i]->e = NULL;
#ifdef DEBUG_DLSCH_FREE
          printf("Freeing dlsch process %d e (%p)\n",i,dlsch->harq_processes[i]->e);
#endif
        }

        if (dlsch->harq_processes[i]->f) {
          free16(dlsch->harq_processes[i]->f,14*N_RB*12*8);
          dlsch->harq_processes[i]->f = NULL;
#ifdef DEBUG_DLSCH_FREE
          printf("Freeing dlsch process %d f (%p)\n",i,dlsch->harq_processes[i]->f);
#endif
        }

107
#ifdef DEBUG_DLSCH_FREE
108
        LOG_D(PHY,"Freeing dlsch process %d c (%p)\n",i,dlsch->harq_processes[i]->c);
109 110
#endif

111
        for (r=0; r<a_segments; r++) {
112 113

#ifdef DEBUG_DLSCH_FREE
114
          LOG_D(PHY,"Freeing dlsch process %d c[%d] (%p)\n",i,r,dlsch->harq_processes[i]->c[r]);
115 116 117
#endif

          if (dlsch->harq_processes[i]->c[r]) {
118
            free16(dlsch->harq_processes[i]->c[r],1056);
119 120 121
            dlsch->harq_processes[i]->c[r] = NULL;
          }
          if (dlsch->harq_processes[i]->d[r]) {
122
            free16(dlsch->harq_processes[i]->d[r],3*8448);
123 124 125
            dlsch->harq_processes[i]->d[r] = NULL;
          }

yilmazt's avatar
yilmazt committed
126
	    }
127 128 129 130 131 132
	free16(dlsch->harq_processes[i],sizeof(NR_DL_gNB_HARQ_t));
	dlsch->harq_processes[i] = NULL;
      }
    }

    free16(dlsch,sizeof(NR_gNB_DLSCH_t));
cig's avatar
cig committed
133
    *dlschptr = NULL;
yilmazt's avatar
yilmazt committed
134
  }
135 136 137

}

138 139
NR_gNB_DLSCH_t *new_gNB_dlsch(NR_DL_FRAME_PARMS *frame_parms,
                              unsigned char Kmimo,
140 141
                              unsigned char Mdlharq,
                              uint32_t Nsoft,
142 143
                              uint8_t  abstraction_flag,
                              uint16_t N_RB)
144 145 146
{

  NR_gNB_DLSCH_t *dlsch;
147
  unsigned char exit_flag = 0,i,r,aa,layer;
148
  int re;
149
  uint16_t a_segments = MAX_NUM_NR_DLSCH_SEGMENTS;  //number of segments to be allocated
150

151 152
  if (N_RB != 273) {
    a_segments = a_segments*N_RB;
153
    a_segments = a_segments/273 +1;
154
  }  
155

156
  uint16_t dlsch_bytes = a_segments*1056;  // allocated bytes per segment
157

158
  
159 160 161 162 163 164 165 166 167
  dlsch = (NR_gNB_DLSCH_t *)malloc16(sizeof(NR_gNB_DLSCH_t));

  if (dlsch) {
    bzero(dlsch,sizeof(NR_gNB_DLSCH_t));
    dlsch->Kmimo = Kmimo;
    dlsch->Mdlharq = Mdlharq;
    dlsch->Mlimit = 4;
    dlsch->Nsoft = Nsoft;

168
    for (layer=0; layer<NR_MAX_NB_LAYERS; layer++) {
169
      dlsch->ue_spec_bf_weights[layer] = (int32_t**)malloc16(64*sizeof(int32_t*));
170

171
      for (aa=0; aa<64; aa++) {
172 173 174 175
         dlsch->ue_spec_bf_weights[layer][aa] = (int32_t *)malloc16(OFDM_SYMBOL_SIZE_COMPLEX_SAMPLES*sizeof(int32_t));
         for (re=0;re<OFDM_SYMBOL_SIZE_COMPLEX_SAMPLES; re++) {
           dlsch->ue_spec_bf_weights[layer][aa][re] = 0x00007fff;
         }
176
      }
177

178
      dlsch->txdataF[layer] = (int32_t *)malloc16((NR_MAX_PDSCH_ENCODED_LENGTH/NR_MAX_NB_LAYERS)*sizeof(int32_t)); // NR_MAX_NB_LAYERS is already included in NR_MAX_PDSCH_ENCODED_LENGTH
179
    }
180

181
    for (int q=0; q<NR_MAX_NB_CODEWORDS; q++)
182
      dlsch->mod_symbs[q] = (int32_t *)malloc16(NR_MAX_PDSCH_ENCODED_LENGTH*sizeof(int32_t));
183

184 185
     dlsch->calib_dl_ch_estimates = (int32_t**)malloc16(64*sizeof(int32_t*));
     for (aa=0; aa<64; aa++) {
186 187 188 189
       dlsch->calib_dl_ch_estimates[aa] = (int32_t *)malloc16(OFDM_SYMBOL_SIZE_COMPLEX_SAMPLES*sizeof(int32_t));

     }

190 191 192 193
    for (i=0; i<20; i++) {
      dlsch->harq_ids[0][i] = 0;
      dlsch->harq_ids[1][i] = 0;
    }
Guy De Souza's avatar
Guy De Souza committed
194

195 196
    for (i=0; i<Mdlharq; i++) {
      dlsch->harq_processes[i] = (NR_DL_gNB_HARQ_t *)malloc16(sizeof(NR_DL_gNB_HARQ_t));
197 198
      LOG_T(PHY, "Required mem size %d  dlsch->harq_processes[%d] %p\n",
    		  dlsch_bytes, i,dlsch->harq_processes[i]);
199 200 201 202

      if (dlsch->harq_processes[i]) {
        bzero(dlsch->harq_processes[i],sizeof(NR_DL_gNB_HARQ_t));
        //    dlsch->harq_processes[i]->first_tx=1;
203 204
        dlsch->harq_processes[i]->b = (unsigned char*)malloc16(dlsch_bytes);
        dlsch->harq_processes[i]->pdu = (uint8_t*)malloc16(dlsch_bytes);
Guy De Souza's avatar
Guy De Souza committed
205
        if (dlsch->harq_processes[i]->pdu) {
206 207
          bzero(dlsch->harq_processes[i]->pdu,dlsch_bytes);
          nr_emulate_dlsch_payload(dlsch->harq_processes[i]->pdu, (dlsch_bytes)>>3);
Guy De Souza's avatar
Guy De Souza committed
208
        } else {
209
          LOG_D(PHY,"Can't allocate PDU\n");
Guy De Souza's avatar
Guy De Souza committed
210 211
          exit_flag=1;
        }
212 213

        if (dlsch->harq_processes[i]->b) {
214
          bzero(dlsch->harq_processes[i]->b,dlsch_bytes);
215
        } else {
216
          LOG_D(PHY,"Can't get b\n");
217 218 219 220
          exit_flag=1;
        }

        if (abstraction_flag==0) {
221
          for (r=0; r<a_segments; r++) {
222
            // account for filler in first segment and CRCs for multiple segment case
Ahmed's avatar
Ahmed committed
223 224 225
            // [hna] 8448 is the maximum CB size in NR
            //       68*348 = 68*(maximum size of Zc)
            //       In section 5.3.2 in 38.212, the for loop is up to N + 2*Zc (maximum size of N is 66*Zc, therefore 68*Zc)
226
            dlsch->harq_processes[i]->c[r] = (uint8_t*)malloc16(8448);
227
            dlsch->harq_processes[i]->d[r] = (uint8_t*)malloc16(68*384); //max size for coded output
228 229 230
            if (dlsch->harq_processes[i]->c[r]) {
              bzero(dlsch->harq_processes[i]->c[r],8448);
            } else {
231
              LOG_D(PHY,"Can't get c\n");
232 233 234 235 236
              exit_flag=2;
            }
            if (dlsch->harq_processes[i]->d[r]) {
              bzero(dlsch->harq_processes[i]->d[r],(3*8448));
            } else {
237
              LOG_D(PHY,"Can't get d\n");
238 239 240
              exit_flag=2;
            }
          }
241 242 243 244 245 246 247 248 249 250 251 252 253 254
          dlsch->harq_processes[i]->e = (uint8_t*)malloc16(14*N_RB*12*8);
          if (dlsch->harq_processes[i]->e) {
            bzero(dlsch->harq_processes[i]->e,14*N_RB*12*8);
          } else {
            printf("Can't get e\n");
            exit_flag=1;
          }
          dlsch->harq_processes[i]->f = (uint8_t*)malloc16(14*N_RB*12*8);
          if (dlsch->harq_processes[i]->f) {
            bzero(dlsch->harq_processes[i]->f,14*N_RB*12*8);
          } else {
            printf("Can't get f\n");
            exit_flag=1;
          }
255 256
        }
      } else {
257
        LOG_D(PHY,"Can't get harq_p %d\n",i);
258 259 260 261 262 263 264 265 266 267 268 269 270
        exit_flag=3;
      }
    }

    if (exit_flag==0) {
      for (i=0; i<Mdlharq; i++) {
        dlsch->harq_processes[i]->round=0;
      }

      return(dlsch);
    }
  }

271
  LOG_D(PHY,"new_gNB_dlsch exit flag %d, size of  %ld\n",
272
	exit_flag, sizeof(NR_gNB_DLSCH_t));
273

Francesco Mani's avatar
Francesco Mani committed
274
  free_gNB_dlsch(&dlsch,N_RB);
275

276
  return(NULL);
277 278 279 280 281 282 283 284 285 286 287 288 289 290 291


}

void clean_gNB_dlsch(NR_gNB_DLSCH_t *dlsch)
{

  unsigned char Mdlharq;
  unsigned char i,j,r;

  if (dlsch) {
    Mdlharq = dlsch->Mdlharq;
    dlsch->rnti = 0;
    dlsch->active = 0;

292 293 294 295
    for (i=0; i<10; i++) {
      dlsch->harq_ids[0][i] = Mdlharq;
      dlsch->harq_ids[1][i] = Mdlharq;
    }
296 297 298
    for (i=0; i<Mdlharq; i++) {
      if (dlsch->harq_processes[i]) {
        //  dlsch->harq_processes[i]->Ndi    = 0;
299
        //dlsch->harq_processes[i]->status = 0;
300 301 302
        dlsch->harq_processes[i]->round  = 0;

	for (j=0; j<96; j++)
303
	  for (r=0; r<MAX_NUM_NR_DLSCH_SEGMENTS; r++)
304 305 306 307 308 309 310 311
	    if (dlsch->harq_processes[i]->d[r])
	      dlsch->harq_processes[i]->d[r][j] = NR_NULL;

      }
    }
  }
}

312 313
int nr_dlsch_encoding(PHY_VARS_gNB *gNB,
		      unsigned char *a,
yilmazt's avatar
yilmazt committed
314 315 316
                      int frame,
                      uint8_t slot,
                      NR_gNB_DLSCH_t *dlsch,
317 318 319 320
                      NR_DL_FRAME_PARMS* frame_parms,
		      time_stats_t *tinput,time_stats_t *tprep,time_stats_t *tparity,time_stats_t *toutput,
		      time_stats_t *dlsch_rate_matching_stats,time_stats_t *dlsch_interleaving_stats,
		      time_stats_t *dlsch_segmentation_stats)
321 322 323 324
{

  unsigned int G;
  unsigned int crc=1;
325
  uint8_t harq_pid = dlsch->harq_ids[frame%2][slot];
326
  AssertFatal(harq_pid<8 && harq_pid>=0,"illegal harq_pid %d\b",harq_pid);
Raymond Knopp's avatar
Raymond Knopp committed
327
  nfapi_nr_dl_tti_pdsch_pdu_rel15_t *rel15 = &dlsch->harq_processes[harq_pid]->pdsch_pdu.pdsch_pdu_rel15;
Raymond Knopp's avatar
Raymond Knopp committed
328 329
  uint16_t nb_rb = rel15->rbSize;
  uint8_t nb_symb_sch = rel15->NrOfSymbols;
Sakthivel Velumani's avatar
Sakthivel Velumani committed
330 331
  uint32_t A, Kb, F=0;
  static uint32_t Z = 0;
Francesco Mani's avatar
Francesco Mani committed
332
  uint32_t *Zc = &Z;
Raymond Knopp's avatar
Raymond Knopp committed
333
  uint8_t mod_order = rel15->qamModOrder[0];
Francesco Mani's avatar
Francesco Mani committed
334 335
  uint16_t Kr=0,r;
  uint32_t r_offset=0;
336
  uint8_t BG=1;
337
  uint32_t E;
338
  uint8_t Ilbrm = 1;
339
  uint32_t Tbslbrm = 950984; //max tbs
340
  uint8_t nb_re_dmrs;
341 342 343

  if (rel15->dmrsConfigType==NFAPI_NR_DMRS_TYPE1)
    nb_re_dmrs = 6*rel15->numDmrsCdmGrpsNoData;
344
  else
345 346
    nb_re_dmrs = 4*rel15->numDmrsCdmGrpsNoData;

Raymond Knopp's avatar
Raymond Knopp committed
347 348
  uint16_t length_dmrs = get_num_dmrs(rel15->dlDmrsSymbPos);
  uint16_t R=rel15->targetCodeRate[0];
349
  float Coderate = 0.0;
350
  uint8_t Nl = 4;
351

352 353
  dlsch->harq_processes[harq_pid]->round = nr_rv_round_map[rel15->rvIndex[0]];

Rakesh's avatar
Rakesh committed
354
  VCD_SIGNAL_DUMPER_DUMP_FUNCTION_BY_NAME(VCD_SIGNAL_DUMPER_FUNCTIONS_gNB_DLSCH_ENCODING, VCD_FUNCTION_IN);
355

Raymond Knopp's avatar
Raymond Knopp committed
356
  A = rel15->TBSize[0]<<3;
357

358 359 360 361 362 363 364 365 366 367 368
  NR_gNB_SCH_STATS_t *stats=NULL;
  int first_free=-1;
  for (int i=0;i<NUMBER_OF_NR_SCH_STATS_MAX;i++) {
    if (gNB->dlsch_stats[i].rnti == 0 && first_free == -1) {
      first_free = i;
      stats=&gNB->dlsch_stats[i];
    }
    if (gNB->dlsch_stats[i].rnti == dlsch->rnti) {
      stats=&gNB->dlsch_stats[i];
      break;
    }
369 370
  }

371 372 373 374 375 376 377 378 379
  if (stats) {
    stats->round_trials[dlsch->harq_processes[harq_pid]->round]++;
    stats->rnti = dlsch->rnti;
    if (dlsch->harq_processes[harq_pid]->round == 0){
      stats->total_bytes_tx += rel15->TBSize[0];
      stats->current_RI   = rel15->nrOfLayers;
      stats->current_Qm   = rel15->qamModOrder[0];
    }
  }
Raymond Knopp's avatar
Raymond Knopp committed
380
  G = nr_get_G(nb_rb, nb_symb_sch, nb_re_dmrs, length_dmrs,mod_order,rel15->nrOfLayers);
381

382
  LOG_D(PHY,"dlsch coding A %d G %d mod_order %d\n", A,G, mod_order);
383 384 385 386

  //  if (dlsch->harq_processes[harq_pid]->Ndi == 1) {  // this is a new packet
  if (dlsch->harq_processes[harq_pid]->round == 0) {  // this is a new packet
#ifdef DEBUG_DLSCH_CODING
387
  LOG_D(PHY,"encoding thinks this is a new packet \n");
388
#endif
389
  /*    
390
    int i;
391
    LOG_D(PHY,"dlsch (tx): \n");
392
    for (i=0;i<(A>>3);i++)
393
      LOG_D(PHY,"%02x\n",a[i]);
394
    LOG_D(PHY,"\n");
395
  */
396

397 398 399 400 401 402 403 404 405 406 407
    if (A > 3824) {
      // Add 24-bit crc (polynomial A) to payload
      crc = crc24a(a,A)>>8;
      a[A>>3] = ((uint8_t*)&crc)[2];
      a[1+(A>>3)] = ((uint8_t*)&crc)[1];
      a[2+(A>>3)] = ((uint8_t*)&crc)[0];
      //printf("CRC %x (A %d)\n",crc,A);
      //printf("a0 %d a1 %d a2 %d\n", a[A>>3], a[1+(A>>3)], a[2+(A>>3)]);
  
      dlsch->harq_processes[harq_pid]->B = A+24;
      //    dlsch->harq_processes[harq_pid]->b = a;
408
   
409
      AssertFatal((A/8)+4 <= MAX_NR_DLSCH_PAYLOAD_BYTES,"A %d is too big (A/8+4 = %d > %d)\n",A,(A/8)+4,MAX_NR_DLSCH_PAYLOAD_BYTES);
410

411 412 413 414 415 416 417 418 419 420 421 422 423
      memcpy(dlsch->harq_processes[harq_pid]->b,a,(A/8)+4);  // why is this +4 if the CRC is only 3 bytes?
    }
    else {
      // Add 16-bit crc (polynomial A) to payload
      crc = crc16(a,A)>>16;
      a[A>>3] = ((uint8_t*)&crc)[1];
      a[1+(A>>3)] = ((uint8_t*)&crc)[0];
      //printf("CRC %x (A %d)\n",crc,A);
      //printf("a0 %d a1 %d \n", a[A>>3], a[1+(A>>3)]);
  
      dlsch->harq_processes[harq_pid]->B = A+16;
      //    dlsch->harq_processes[harq_pid]->b = a;
   
424
      AssertFatal((A/8)+3 <= MAX_NR_DLSCH_PAYLOAD_BYTES,"A %d is too big (A/8+3 = %d > %d)\n",A,(A/8)+3,MAX_NR_DLSCH_PAYLOAD_BYTES);
425

426 427
      memcpy(dlsch->harq_processes[harq_pid]->b,a,(A/8)+3);  // using 3 bytes to mimic the case of 24 bit crc
    }
428 429 430 431
    if (R<1000)
      Coderate = (float) R /(float) 1024;
    else  // to scale for mcs 20 and 26 in table 5.1.3.1-2 which are decimal and input 2* in nr_tbs_tools
      Coderate = (float) R /(float) 2048;
432

433
    if ((A <=292) || ((A<=3824) && (Coderate <= 0.6667)) || Coderate <= 0.25)
434
		BG = 2;
435
    else
436
		BG = 1;
437

438
    start_meas(dlsch_segmentation_stats);
Francesco Mani's avatar
Francesco Mani committed
439 440 441 442 443 444 445 446
    Kb = nr_segmentation(dlsch->harq_processes[harq_pid]->b,
		         dlsch->harq_processes[harq_pid]->c,
		         dlsch->harq_processes[harq_pid]->B,
		         &dlsch->harq_processes[harq_pid]->C,
		         &dlsch->harq_processes[harq_pid]->K,
		         Zc, 
		         &dlsch->harq_processes[harq_pid]->F,
                         BG);
447
    stop_meas(dlsch_segmentation_stats);
448
    F = dlsch->harq_processes[harq_pid]->F;
449 450

    Kr = dlsch->harq_processes[harq_pid]->K;
451 452
#ifdef DEBUG_DLSCH_CODING
    uint16_t Kr_bytes;
453
    Kr_bytes = Kr>>3;
454
#endif
455

Francesco Mani's avatar
Francesco Mani committed
456
    //printf("segment Z %d k %d Kr %d BG %d C %d\n", *Zc,dlsch->harq_processes[harq_pid]->K,Kr,BG,dlsch->harq_processes[harq_pid]->C);
Hongzhi Wang's avatar
Hongzhi Wang committed
457

458
    for (r=0; r<dlsch->harq_processes[harq_pid]->C; r++) {
459
      //d_tmp[r] = &dlsch->harq_processes[harq_pid]->d[r][0];
460
      //channel_input[r] = &dlsch->harq_processes[harq_pid]->d[r][0];
461
#ifdef DEBUG_DLSCH_CODING
462 463 464
      LOG_D(PHY,"Encoder: B %d F %d \n",dlsch->harq_processes[harq_pid]->B, dlsch->harq_processes[harq_pid]->F);
      LOG_D(PHY,"start ldpc encoder segment %d/%d\n",r,dlsch->harq_processes[harq_pid]->C);
      LOG_D(PHY,"input %d %d %d %d %d \n", dlsch->harq_processes[harq_pid]->c[r][0], dlsch->harq_processes[harq_pid]->c[r][1], dlsch->harq_processes[harq_pid]->c[r][2],dlsch->harq_processes[harq_pid]->c[r][3], dlsch->harq_processes[harq_pid]->c[r][4]);
465
      for (int cnt =0 ; cnt < 22*(*Zc)/8; cnt ++){
466
      LOG_D(PHY,"%d ", dlsch->harq_processes[harq_pid]->c[r][cnt]);
467
      }
468
      LOG_D(PHY,"\n");
469 470

#endif
Francesco Mani's avatar
Francesco Mani committed
471 472
      //ldpc_encoder_orig((unsigned char*)dlsch->harq_processes[harq_pid]->c[r],dlsch->harq_processes[harq_pid]->d[r],*Zc,Kb,Kr,BG,0);
      //ldpc_encoder_optim((unsigned char*)dlsch->harq_processes[harq_pid]->c[r],(unsigned char*)&dlsch->harq_processes[harq_pid]->d[r][0],*Zc,Kb,Kr,BG,NULL,NULL,NULL,NULL);
473
    }
474 475
    encoder_implemparams_t impp;
    impp.n_segments=dlsch->harq_processes[harq_pid]->C;
476 477 478 479 480
    impp.tprep = tprep;
    impp.tinput = tinput;
    impp.tparity = tparity;
    impp.toutput = toutput;

481
    for(int j=0;j<(dlsch->harq_processes[harq_pid]->C/8+1);j++) {
482 483
      impp.macro_num=j;
      nrLDPC_encoder(dlsch->harq_processes[harq_pid]->c,dlsch->harq_processes[harq_pid]->d,*Zc,Kb,Kr,BG,&impp);
484
    }
485

486

487 488 489 490 491 492 493
#ifdef DEBUG_DLSCH_CODING
      write_output("enc_input0.m","enc_in0",&dlsch->harq_processes[harq_pid]->c[0][0],Kr_bytes,1,4);
      write_output("enc_output0.m","enc0",&dlsch->harq_processes[harq_pid]->d[0][0],(3*8*Kr_bytes)+12,1,4);
#endif

  }

494 495 496
    F = dlsch->harq_processes[harq_pid]->F;

    Kr = dlsch->harq_processes[harq_pid]->K;
497
  for (r=0; r<dlsch->harq_processes[harq_pid]->C; r++) {
498

499
    if (F>0) {
Francesco Mani's avatar
Francesco Mani committed
500
      for (int k=(Kr-F-2*(*Zc)); k<Kr-2*(*Zc); k++) {
501
	// writing into positions d[r][k-2Zc] as in clause 5.3.2 step 2) in 38.212
502 503 504 505 506 507
        dlsch->harq_processes[harq_pid]->d[r][k] = NR_NULL;
	//if (k<(Kr-F+8))
	//printf("r %d filler bits [%d] = %d \n", r,k, dlsch->harq_processes[harq_pid]->d[r][k]);
      }
    }

508

509 510

#ifdef DEBUG_DLSCH_CODING
Raymond Knopp's avatar
Raymond Knopp committed
511
  LOG_D(PHY,"rvidx in encoding = %d\n", rel15->rvIndex[0]);
512 513
#endif

Raymond Knopp's avatar
Raymond Knopp committed
514
    E = nr_get_E(G, dlsch->harq_processes[harq_pid]->C, mod_order, rel15->nrOfLayers, r);
515

516 517 518 519 520 521 522 523 524
    //#ifdef DEBUG_DLSCH_CODING
    LOG_D(PHY,"Rate Matching, Code segment %d/%d (coded bits (G) %u, E %d, Filler bits %d, Filler offset %d mod_order %d, nb_rb %d)...\n",
	  r,
	  dlsch->harq_processes[harq_pid]->C,
	  G,
	  E,
	  F,
	  Kr-F-2*(*Zc),
	  mod_order,nb_rb);
525

526
    // for tbslbrm calculation according to 5.4.2.1 of 38.212
Raymond Knopp's avatar
Raymond Knopp committed
527 528
    if (rel15->nrOfLayers < Nl)
      Nl = rel15->nrOfLayers;
529

Raymond Knopp's avatar
Raymond Knopp committed
530
    Tbslbrm = nr_compute_tbslbrm(rel15->mcsTable[0],nb_rb,Nl,dlsch->harq_processes[harq_pid]->C);
531

532
    start_meas(dlsch_rate_matching_stats);
533 534 535
    nr_rate_matching_ldpc(Ilbrm,
                          Tbslbrm,
                          BG,
Francesco Mani's avatar
Francesco Mani committed
536
                          *Zc,
537 538 539
                          dlsch->harq_processes[harq_pid]->d[r],
                          dlsch->harq_processes[harq_pid]->e+r_offset,
                          dlsch->harq_processes[harq_pid]->C,
540 541
                          F,
                          Kr-F-2*(*Zc),
Raymond Knopp's avatar
Raymond Knopp committed
542
                          rel15->rvIndex[0],
543
                          E);
544
    stop_meas(dlsch_rate_matching_stats);
545 546
#ifdef DEBUG_DLSCH_CODING
    for (int i =0; i<16; i++)
547
      printf("output ratematching e[%d]= %d r_offset %u\n", i,dlsch->harq_processes[harq_pid]->e[i+r_offset], r_offset);
548
#endif
549

550 551 552 553 554 555
    start_meas(dlsch_interleaving_stats);
    nr_interleaving_ldpc(E,
			 mod_order,
			 dlsch->harq_processes[harq_pid]->e+r_offset,
			 dlsch->harq_processes[harq_pid]->f+r_offset);
    stop_meas(dlsch_interleaving_stats);
556

557 558
#ifdef DEBUG_DLSCH_CODING
    for (int i =0; i<16; i++)
559
      printf("output interleaving f[%d]= %d r_offset %u\n", i,dlsch->harq_processes[harq_pid]->f[i+r_offset], r_offset);
560 561

    if (r==dlsch->harq_processes[harq_pid]->C-1)
562
      write_output("enc_output.m","enc",dlsch->harq_processes[harq_pid]->f,G,1,4);
563
#endif
564 565

    r_offset += E;
566 567
  }

Rakesh's avatar
Rakesh committed
568
  VCD_SIGNAL_DUMPER_DUMP_FUNCTION_BY_NAME(VCD_SIGNAL_DUMPER_FUNCTIONS_gNB_DLSCH_ENCODING, VCD_FUNCTION_OUT);
569

Khalid Ahmed's avatar
Khalid Ahmed committed
570
  return 0;
571
}