mirror of https://github.com/pvnis/srsRAN_4G.git
PDSCH NR: Added files
Xavier Arteaga
4 years ago
committed by
Xavier Arteaga
parent
30d8e05d50
commit
c8477483d9
@ -0,0 +1,99 @@
|
||||
/*
|
||||
* Copyright 2013-2020 Software Radio Systems Limited
|
||||
*
|
||||
* This file is part of srsLTE.
|
||||
*
|
||||
* srsLTE is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as
|
||||
* published by the Free Software Foundation, either version 3 of
|
||||
* the License, or (at your option) any later version.
|
||||
*
|
||||
* srsLTE is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* A copy of the GNU Affero General Public License can be found in
|
||||
* the LICENSE file in the top-level directory of this distribution
|
||||
* and at http://www.gnu.org/licenses/.
|
||||
*
|
||||
*/
|
||||
|
||||
/******************************************************************************
|
||||
* File: pdsch_nr.h
|
||||
*
|
||||
* Description: Physical downlink shared channel for NR
|
||||
*
|
||||
* Reference: 3GPP TS 38.211 V15.8.0 Sec. 7.3.1
|
||||
*****************************************************************************/
|
||||
|
||||
#ifndef srslte_pdsch_nr_H
|
||||
#define srslte_pdsch_nr_H
|
||||
|
||||
#include "srslte/config.h"
|
||||
#include "srslte/phy/phch/pdsch_cfg_nr.h"
|
||||
#include "srslte/phy/ch_estimation/dmrs_pdsch.h"
|
||||
#include "srslte/phy/phch/regs.h"
|
||||
#include "srslte/phy/phch/sch.h"
|
||||
#include "srslte/phy/scrambling/scrambling.h"
|
||||
|
||||
/**
|
||||
* @brief PDSCH NR object
|
||||
*/
|
||||
typedef struct SRSLTE_API {
|
||||
srslte_carrier_nr_t carrier;
|
||||
} srslte_pdsch_nr_t;
|
||||
|
||||
/**
|
||||
*
|
||||
*/
|
||||
typedef struct {
|
||||
uint8_t* payload;
|
||||
bool crc;
|
||||
float avg_iterations_block;
|
||||
float evm;
|
||||
} srslte_pdsch_res_nr_t;
|
||||
|
||||
SRSLTE_API int srslte_pdsch_nr_init_ue(srslte_pdsch_nr_t* q);
|
||||
|
||||
SRSLTE_API int srslte_pdsch_nr_init_enb(srslte_pdsch_nr_t* q);
|
||||
|
||||
SRSLTE_API void srslte_pdsch_nr_free(srslte_pdsch_nr_t* q);
|
||||
|
||||
SRSLTE_API int srslte_pdsch_nr_set_carrier(srslte_pdsch_nr_t* q, srslte_cell_t cell);
|
||||
|
||||
/* These functions do not modify the state and run in real-time */
|
||||
SRSLTE_API int srslte_pdsch_nr_encode(srslte_pdsch_nr_t* q,
|
||||
uint32_t slot_idx,
|
||||
srslte_pdsch_cfg_nr_t* cfg,
|
||||
uint8_t* data[SRSLTE_MAX_CODEWORDS],
|
||||
cf_t* sf_symbols[SRSLTE_MAX_PORTS]);
|
||||
|
||||
SRSLTE_API int srslte_pdsch_nr_decode(srslte_pdsch_nr_t* q,
|
||||
uint32_t slot_idx,
|
||||
srslte_pdsch_cfg_nr_t* cfg,
|
||||
srslte_chest_dl_res_t* channel,
|
||||
cf_t* sf_symbols[SRSLTE_MAX_PORTS],
|
||||
srslte_pdsch_res_nr_t data[SRSLTE_MAX_CODEWORDS]);
|
||||
|
||||
SRSLTE_API int srslte_pdsch_nr_select_pmi(srslte_pdsch_nr_t* q,
|
||||
srslte_chest_dl_res_t* channel,
|
||||
uint32_t nof_layers,
|
||||
uint32_t* best_pmi,
|
||||
float sinr[SRSLTE_MAX_CODEBOOKS]);
|
||||
|
||||
SRSLTE_API int srslte_pdsch_nr_compute_cn(srslte_pdsch_nr_t* q, srslte_chest_dl_res_t* channel, float* cn);
|
||||
|
||||
SRSLTE_API uint32_t srslte_pdsch_nr_grant_rx_info(srslte_pdsch_grant_nr_t* grant,
|
||||
srslte_pdsch_res_nr_t res[SRSLTE_MAX_CODEWORDS],
|
||||
char* str,
|
||||
uint32_t str_len);
|
||||
|
||||
SRSLTE_API uint32_t srslte_pdsch_nr_rx_info(srslte_pdsch_cfg_nr_t* cfg,
|
||||
srslte_pdsch_res_nr_t res[SRSLTE_MAX_CODEWORDS],
|
||||
char* str,
|
||||
uint32_t str_len);
|
||||
|
||||
SRSLTE_API uint32_t srslte_pdsch_nr_tx_info(srslte_pdsch_cfg_nr_t* cfg, char* str, uint32_t str_len);
|
||||
|
||||
#endif // srslte_pdsch_nr_H
|
||||
@ -0,0 +1,208 @@
|
||||
/*
|
||||
* Copyright 2013-2020 Software Radio Systems Limited
|
||||
*
|
||||
* This file is part of srsLTE.
|
||||
*
|
||||
* srsLTE is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as
|
||||
* published by the Free Software Foundation, either version 3 of
|
||||
* the License, or (at your option) any later version.
|
||||
*
|
||||
* srsLTE is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* A copy of the GNU Affero General Public License can be found in
|
||||
* the LICENSE file in the top-level directory of this distribution
|
||||
* and at http://www.gnu.org/licenses/.
|
||||
*
|
||||
*/
|
||||
#include "srslte/phy/phch/pdsch_nr.h"
|
||||
|
||||
/**
|
||||
* @brief copies a number of countiguous Resource Elements
|
||||
* @param sf_symbols slot symbols in frequency domain
|
||||
* @param symbols resource elements
|
||||
* @param count number of resource elements to copy
|
||||
* @param put Direction, symbols are copied into sf_symbols if put is true, otherwise sf_symbols are copied into symbols
|
||||
*/
|
||||
static void srslte_pdsch_re_cp(cf_t* sf_symbols, cf_t* symbols, uint32_t count, bool put)
|
||||
{
|
||||
if (put) {
|
||||
srslte_vec_cf_copy(sf_symbols, symbols, count);
|
||||
} else {
|
||||
srslte_vec_cf_copy(symbols, sf_symbols, count);
|
||||
}
|
||||
}
|
||||
|
||||
static uint32_t srslte_pdsch_nr_cp_dmrs_type1(const srslte_pdsch_nr_t* q,
|
||||
const srslte_pdsch_cfg_nr_t* cfg,
|
||||
cf_t* symbols,
|
||||
cf_t* sf_symbols,
|
||||
bool put)
|
||||
{
|
||||
uint32_t count = 0;
|
||||
uint32_t delta = 0;
|
||||
|
||||
for (uint32_t i = 0; i < q->carrier.nof_prb; i++) {
|
||||
if (cfg->grant.prb_idx[i]) {
|
||||
for (uint32_t j = 0; j < SRSLTE_NRE; j += 2) {
|
||||
if (put) {
|
||||
sf_symbols[i * SRSLTE_NRE + delta + j] = symbols[count++];
|
||||
} else {
|
||||
symbols[count++] = sf_symbols[i * SRSLTE_NRE + delta + j];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return count;
|
||||
}
|
||||
|
||||
static uint32_t srslte_pdsch_nr_cp_dmrs_type2(const srslte_pdsch_nr_t* q,
|
||||
const srslte_pdsch_cfg_nr_t* cfg,
|
||||
cf_t* symbols,
|
||||
cf_t* sf_symbols,
|
||||
bool put)
|
||||
{
|
||||
uint32_t count = 0;
|
||||
uint32_t delta = 0;
|
||||
|
||||
for (uint32_t i = 0; i < q->carrier.nof_prb; i++) {
|
||||
if (cfg->grant.prb_idx[i]) {
|
||||
// Copy RE before first pilot pair
|
||||
if (delta > 0) {
|
||||
srslte_pdsch_re_cp(&sf_symbols[i * SRSLTE_NRE], &symbols[count], delta, put);
|
||||
count += delta;
|
||||
}
|
||||
|
||||
// Copy RE between pilot pairs
|
||||
srslte_pdsch_re_cp(&sf_symbols[i * SRSLTE_NRE + delta + 2], &symbols[count], 4, put);
|
||||
count += 4;
|
||||
|
||||
// Copy RE after second pilot
|
||||
srslte_pdsch_re_cp(&sf_symbols[(i + 1) * SRSLTE_NRE - 4 + delta], &symbols[count], 4 - delta, put);
|
||||
count += 4 - delta;
|
||||
}
|
||||
}
|
||||
|
||||
return count;
|
||||
}
|
||||
|
||||
static uint32_t srslte_pdsch_nr_cp_dmrs(const srslte_pdsch_nr_t* q,
|
||||
const srslte_pdsch_cfg_nr_t* cfg,
|
||||
cf_t* symbols,
|
||||
cf_t* sf_symbols,
|
||||
bool put)
|
||||
{
|
||||
uint32_t count = 0;
|
||||
|
||||
switch (cfg->dmrs_cfg.type) {
|
||||
case srslte_dmrs_pdsch_type_1:
|
||||
count = srslte_pdsch_nr_cp_dmrs_type1(q, cfg, symbols, sf_symbols, put);
|
||||
break;
|
||||
case srslte_dmrs_pdsch_type_2:
|
||||
count = srslte_pdsch_nr_cp_dmrs_type2(q, cfg, symbols, sf_symbols, put);
|
||||
break;
|
||||
}
|
||||
|
||||
return count;
|
||||
}
|
||||
|
||||
static uint32_t srslte_pdsch_nr_cp_clean(const srslte_pdsch_nr_t* q,
|
||||
const srslte_pdsch_cfg_nr_t* cfg,
|
||||
cf_t* symbols,
|
||||
cf_t* sf_symbols,
|
||||
bool put)
|
||||
{
|
||||
uint32_t count = 0;
|
||||
uint32_t start = 0; // Index of the start of continuous data
|
||||
uint32_t length = 0; // End of continuous RE
|
||||
|
||||
for (uint32_t i = 0; i < q->carrier.nof_prb; i++) {
|
||||
if (cfg->grant.prb_idx[i]) {
|
||||
// If fist continuous block, save start
|
||||
if (length == 0) {
|
||||
start = i * SRSLTE_NRE;
|
||||
}
|
||||
length += SRSLTE_NRE;
|
||||
} else {
|
||||
// Consecutive block is finished
|
||||
if (put) {
|
||||
srslte_vec_cf_copy(&sf_symbols[start], &symbols[count], length);
|
||||
} else {
|
||||
srslte_vec_cf_copy(&symbols[count], &sf_symbols[start], length);
|
||||
}
|
||||
|
||||
// Increase RE count
|
||||
count += length;
|
||||
|
||||
// Reset consecutive block
|
||||
length = 0;
|
||||
}
|
||||
}
|
||||
|
||||
// Copy last contiguous block
|
||||
if (length > 0) {
|
||||
if (put) {
|
||||
srslte_vec_cf_copy(&sf_symbols[start], &symbols[count], length);
|
||||
} else {
|
||||
srslte_vec_cf_copy(&symbols[count], &sf_symbols[start], length);
|
||||
}
|
||||
count += length;
|
||||
}
|
||||
|
||||
return count;
|
||||
}
|
||||
|
||||
static int srslte_pdsch_nr_cp(const srslte_pdsch_nr_t* q,
|
||||
const srslte_pdsch_cfg_nr_t* cfg,
|
||||
cf_t* symbols,
|
||||
cf_t* sf_symbols,
|
||||
bool put)
|
||||
{
|
||||
uint32_t count = 0;
|
||||
uint32_t dmrs_l_idx[SRSLTE_DMRS_PDSCH_MAX_SYMBOLS] = {};
|
||||
uint32_t dmrs_l_count = 0;
|
||||
|
||||
// Get symbol indexes carrying DMRS
|
||||
int32_t nof_dmrs_symbols = srslte_dmrs_pdsch_get_symbols_idx(cfg, dmrs_l_idx);
|
||||
if (nof_dmrs_symbols < SRSLTE_SUCCESS) {
|
||||
return SRSLTE_ERROR;
|
||||
}
|
||||
|
||||
for (uint32_t l = cfg->grant.S; l < cfg->grant.L; l++) {
|
||||
// Advance DMRS symbol counter until:
|
||||
// - the current DMRS symbol index is greater or equal than current symbol l
|
||||
// - no more DMRS symbols
|
||||
while (dmrs_l_idx[dmrs_l_count] < l && dmrs_l_count < nof_dmrs_symbols) {
|
||||
dmrs_l_count++;
|
||||
}
|
||||
|
||||
if (l == dmrs_l_idx[dmrs_l_count]) {
|
||||
count += srslte_pdsch_nr_cp_dmrs(q, cfg, &symbols[count], &sf_symbols[l * q->carrier.nof_prb * SRSLTE_NRE], put);
|
||||
} else {
|
||||
count += srslte_pdsch_nr_cp_clean(q, cfg, &symbols[count], &sf_symbols[l * q->carrier.nof_prb * SRSLTE_NRE], put);
|
||||
}
|
||||
}
|
||||
|
||||
return count;
|
||||
}
|
||||
|
||||
int srslte_pdsch_nr_put(const srslte_pdsch_nr_t* q, const srslte_pdsch_cfg_nr_t* cfg, cf_t* symbols, cf_t* sf_symbols)
|
||||
{
|
||||
return srslte_pdsch_nr_cp(q, cfg, symbols, sf_symbols, true);
|
||||
}
|
||||
|
||||
|
||||
int srslte_pdsch_nr_encode(srslte_pdsch_nr_t* q,
|
||||
uint32_t slot_idx,
|
||||
srslte_pdsch_cfg_nr_t* cfg,
|
||||
uint8_t* data[SRSLTE_MAX_CODEWORDS],
|
||||
cf_t* sf_symbols[SRSLTE_MAX_PORTS]) {
|
||||
|
||||
|
||||
|
||||
return SRSLTE_SUCCESS;
|
||||
}
|
||||
Loading…
Reference in New Issue