PentHertz
/
srsLTE
mirror of https://github.com/PentHertz/srsLTE.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added NR layer mapper and demapper

This commit is contained in:
Xavier Arteaga 2020-11-10 19:56:59 +01:00 committed by Andre Puschmann
parent 3060972514
commit 6df83c32b3
2 changed files with 71 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/include/srslte/phy/mimo/layermap.h
View File

@ -75,4 +75,8 @@ SRSLTE_API int srslte_layerdemap_type(cf_t* x[SRSLTE_MAX_LAYERS],
int nof_symbols[SRSLTE_MAX_CODEWORDS],
srslte_tx_scheme_t type);
SRSLTE_API int srslte_layermap_nr(cf_t** d, int nof_cw, cf_t** x, int nof_layers, uint32_t nof_re);
SRSLTE_API int srslte_layerdemap_nr(cf_t** d, int nof_cw, const cf_t** x, int nof_layers, uint32_t nof_re);
#endif // SRSLTE_LAYERMAP_H

67
lib/src/phy/mimo/layermap.c
View File

@ -24,6 +24,7 @@
#include <string.h>
#include "srslte/phy/common/phy_common.h"
#include "srslte/phy/common/phy_common_nr.h"
#include "srslte/phy/mimo/layermap.h"
#include "srslte/phy/utils/debug.h"
#include "srslte/phy/utils/vector.h"
@ -227,3 +228,69 @@ int srslte_layerdemap_type(cf_t* x[SRSLTE_MAX_LAYERS],
}
return 0;
}
int srslte_layermap_nr(cf_t** d, int nof_cw, cf_t** x, int nof_layers, uint32_t nof_re)
{
if (nof_cw == 1 && nof_layers > 0 && nof_layers < 5) {
for (uint32_t i = 0; i < nof_re / nof_layers; i++) {
for (uint32_t j = 0; j < nof_layers; j++) {
x[j][i] = d[0][nof_layers * i + j];
}
}
return SRSLTE_SUCCESS;
}
if (nof_cw == 2 && nof_layers >= 5 && nof_layers <= SRSLTE_MAX_LAYERS_NR) {
uint32_t nof_layers_cw1 = nof_layers / 2;
for (uint32_t i = 0; i < nof_re / nof_layers_cw1; i++) {
for (uint32_t j = 0; j < nof_layers_cw1; j++) {
x[j][i] = d[0][nof_layers_cw1 * i + j];
}
}
uint32_t nof_layers_cw2 = nof_layers - nof_layers_cw1;
for (uint32_t i = 0; i < nof_re / nof_layers_cw2; i++) {
for (uint32_t j = 0; j < nof_layers_cw2; j++) {
x[j + nof_layers_cw1][i] = d[0][nof_layers_cw2 * i + j];
}
}
return SRSLTE_SUCCESS;
}
ERROR("Error. Incompatible number of layers (%d) and codewords (%d)\n", nof_layers, nof_cw);
return SRSLTE_ERROR;
}
int srslte_layerdemap_nr(cf_t** d, int nof_cw, const cf_t** x, int nof_layers, uint32_t nof_re)
{
if (nof_cw == 1 && nof_layers > 0 && nof_layers < 5) {
for (uint32_t i = 0; i < nof_re / nof_layers; i++) {
for (uint32_t j = 0; j < nof_layers; j++) {
d[0][nof_layers * i + j] = x[j][i];
}
}
return SRSLTE_SUCCESS;
}
if (nof_cw == 2 && nof_layers >= 5 && nof_layers <= SRSLTE_MAX_LAYERS_NR) {
uint32_t nof_layers_cw1 = nof_layers / 2;
for (uint32_t i = 0; i < nof_re / nof_layers_cw1; i++) {
for (uint32_t j = 0; j < nof_layers_cw1; j++) {
d[0][nof_layers_cw1 * i + j] = x[j][i];
}
}
uint32_t nof_layers_cw2 = nof_layers - nof_layers_cw1;
for (uint32_t i = 0; i < nof_re / nof_layers_cw2; i++) {
for (uint32_t j = 0; j < nof_layers_cw2; j++) {
d[0][nof_layers_cw2 * i + j] = x[j + nof_layers_cw1][i];
}
}
return SRSLTE_SUCCESS;
}
ERROR("Error. Incompatible number of layers (%d) and codewords (%d)\n", nof_layers, nof_cw);
return SRSLTE_ERROR;
}