mirror of https://github.com/pvnis/srsRAN_4G.git
Added CSI-RS unit test and fix defects
Xavier Arteaga
4 years ago
committed by
Andre Puschmann
parent
6127681102
commit
d6b5229dbe
@ -0,0 +1,210 @@
|
||||
/**
|
||||
*
|
||||
* \section COPYRIGHT
|
||||
*
|
||||
* Copyright 2013-2020 Software Radio Systems Limited
|
||||
*
|
||||
* By using this file, you agree to the terms and conditions set
|
||||
* forth in the LICENSE file which can be found at the top level of
|
||||
* the distribution.
|
||||
*
|
||||
*/
|
||||
|
||||
#include "srslte/common/test_common.h"
|
||||
#include "srslte/phy/ch_estimation/csi_rs.h"
|
||||
#include "srslte/phy/utils/vector.h"
|
||||
#include <getopt.h>
|
||||
#include <srslte/srslte.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
static srslte_carrier_nr_t carrier = {
|
||||
0, // cell_id
|
||||
0, // numerology
|
||||
50, // nof_prb
|
||||
0, // start
|
||||
1 // max_mimo_layers
|
||||
};
|
||||
|
||||
static float snr_dB = 20.0;
|
||||
static float power_control_offset = NAN;
|
||||
static uint32_t start_rb = UINT32_MAX;
|
||||
static uint32_t nof_rb = UINT32_MAX;
|
||||
static uint32_t first_symbol = UINT32_MAX;
|
||||
|
||||
static int test(const srslte_dl_slot_cfg_t* slot_cfg,
|
||||
const srslte_csi_rs_nzp_resource_t* resource,
|
||||
srslte_channel_awgn_t* awgn,
|
||||
cf_t* grid)
|
||||
{
|
||||
srslte_csi_rs_measure_t measure = {};
|
||||
|
||||
// Put NZP-CSI-RS
|
||||
TESTASSERT(srslte_csi_rs_nzp_put(&carrier, slot_cfg, resource, grid) == SRSLTE_SUCCESS);
|
||||
|
||||
// Configure N0 and add Noise
|
||||
TESTASSERT(srslte_channel_awgn_set_n0(awgn, (float)resource->power_control_offset - snr_dB) == SRSLTE_SUCCESS);
|
||||
srslte_channel_awgn_run_c(awgn, grid, grid, SRSLTE_SLOT_LEN_RE_NR(carrier.nof_prb));
|
||||
|
||||
TESTASSERT(srslte_csi_rs_nzp_measure(&carrier, slot_cfg, resource, grid, &measure) == SRSLTE_SUCCESS);
|
||||
|
||||
const float rsrp_dB_gold = (float)resource->power_control_offset;
|
||||
const float epre_dB_gold =
|
||||
srslte_convert_power_to_dB(srslte_convert_dB_to_power(rsrp_dB_gold) + awgn->std_dev * awgn->std_dev);
|
||||
const float n0_dB_gold = srslte_convert_amplitude_to_dB(awgn->std_dev);
|
||||
|
||||
if (srslte_verbose >= SRSLTE_VERBOSE_INFO) {
|
||||
char str[128] = {};
|
||||
srslte_csi_rs_measure_info(&measure, str, sizeof(str));
|
||||
INFO("Measure: %s\n", str);
|
||||
}
|
||||
|
||||
TESTASSERT(fabsf(measure.rsrp_dB - rsrp_dB_gold) < 1.0f);
|
||||
TESTASSERT(fabsf(measure.epre_dB - epre_dB_gold) < 1.0f);
|
||||
TESTASSERT(fabsf(measure.n0_dB - n0_dB_gold) < 2.0f);
|
||||
TESTASSERT(fabsf(measure.snr_dB - snr_dB) < 2.0f);
|
||||
|
||||
return SRSLTE_SUCCESS;
|
||||
}
|
||||
|
||||
static void usage(char* prog)
|
||||
{
|
||||
printf("Usage: %s [recov]\n", prog);
|
||||
printf("\t-p nof_prb [Default %d]\n", carrier.nof_prb);
|
||||
printf("\t-c cell_id [Default %d]\n", carrier.id);
|
||||
printf("\t-s SNR in dB [Default %.2f]\n", snr_dB);
|
||||
printf("\t-S Start RB index [Default %d]\n", start_rb);
|
||||
printf("\t-L Number of RB [Default %d]\n", nof_rb);
|
||||
printf("\t-f First symbol index [Default %d]\n", first_symbol);
|
||||
printf("\t-o Power control offset [Default %.2f]\n", power_control_offset);
|
||||
printf("\t-v increase verbosity\n");
|
||||
}
|
||||
|
||||
static void parse_args(int argc, char** argv)
|
||||
{
|
||||
int opt;
|
||||
while ((opt = getopt(argc, argv, "pcosSLfv")) != -1) {
|
||||
switch (opt) {
|
||||
case 'p':
|
||||
carrier.nof_prb = (uint32_t)strtol(argv[optind], NULL, 10);
|
||||
break;
|
||||
case 'c':
|
||||
carrier.id = (uint32_t)strtol(argv[optind], NULL, 10);
|
||||
break;
|
||||
case 'o':
|
||||
power_control_offset = strtof(argv[optind], NULL);
|
||||
break;
|
||||
case 's':
|
||||
snr_dB = strtof(argv[optind], NULL);
|
||||
break;
|
||||
case 'S':
|
||||
start_rb = (uint32_t)strtol(argv[optind], NULL, 10);
|
||||
break;
|
||||
case 'L':
|
||||
nof_rb = (uint32_t)strtol(argv[optind], NULL, 10);
|
||||
break;
|
||||
case 'f':
|
||||
first_symbol = (uint32_t)strtol(argv[optind], NULL, 10);
|
||||
break;
|
||||
case 'v':
|
||||
srslte_verbose++;
|
||||
break;
|
||||
default:
|
||||
usage(argv[0]);
|
||||
exit(-1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
int ret = SRSLTE_ERROR;
|
||||
srslte_dl_slot_cfg_t slot_cfg = {};
|
||||
srslte_csi_rs_nzp_resource_t resource = {};
|
||||
srslte_channel_awgn_t awgn = {};
|
||||
|
||||
parse_args(argc, argv);
|
||||
|
||||
cf_t* grid = srslte_vec_cf_malloc(SRSLTE_SLOT_LEN_RE_NR(carrier.nof_prb));
|
||||
if (grid == NULL) {
|
||||
ERROR("Alloc\n");
|
||||
goto clean_exit;
|
||||
}
|
||||
|
||||
if (srslte_channel_awgn_init(&awgn, 1234) < SRSLTE_SUCCESS) {
|
||||
ERROR("AWGN Init\n");
|
||||
goto clean_exit;
|
||||
}
|
||||
|
||||
// Fixed parameters, other params are not implemented
|
||||
resource.resource_mapping.cdm = srslte_csi_rs_cdm_nocdm;
|
||||
resource.resource_mapping.density = srslte_csi_rs_resource_mapping_density_three;
|
||||
resource.resource_mapping.row = srslte_csi_rs_resource_mapping_row_1;
|
||||
resource.resource_mapping.ports = 1;
|
||||
|
||||
// Row 1 supported only!
|
||||
uint32_t nof_freq_dom_alloc = SRSLTE_CSI_RS_NOF_FREQ_DOMAIN_ALLOC_ROW1;
|
||||
|
||||
uint32_t first_symbol_begin = (first_symbol != UINT32_MAX) ? first_symbol : 0;
|
||||
uint32_t first_symbol_end = (first_symbol != UINT32_MAX) ? first_symbol : 13;
|
||||
for (resource.resource_mapping.first_symbol_idx = first_symbol_begin;
|
||||
resource.resource_mapping.first_symbol_idx <= first_symbol_end;
|
||||
resource.resource_mapping.first_symbol_idx++) {
|
||||
|
||||
// Iterate over possible power control offset
|
||||
float power_control_offset_begin = isnormal(power_control_offset) ? power_control_offset : -8.0f;
|
||||
float power_control_offset_end = isnormal(power_control_offset) ? power_control_offset : 15.0f;
|
||||
for (resource.power_control_offset = power_control_offset_begin;
|
||||
resource.power_control_offset <= power_control_offset_end;
|
||||
resource.power_control_offset += 1.0f) {
|
||||
|
||||
// Iterate over all possible starting number of PRB
|
||||
uint32_t start_rb_begin = (start_rb != UINT32_MAX) ? start_rb : 0;
|
||||
uint32_t start_rb_end = (start_rb != UINT32_MAX) ? start_rb : carrier.nof_prb - 24;
|
||||
for (resource.resource_mapping.freq_band.start_rb = start_rb_begin;
|
||||
resource.resource_mapping.freq_band.start_rb <= start_rb_end;
|
||||
resource.resource_mapping.freq_band.start_rb += 4) {
|
||||
|
||||
// Iterate over all possible number of PRB
|
||||
uint32_t nof_rb_begin = (nof_rb != UINT32_MAX) ? nof_rb : 24;
|
||||
uint32_t nof_rb_end =
|
||||
(nof_rb != UINT32_MAX) ? nof_rb : (carrier.nof_prb - resource.resource_mapping.freq_band.start_rb);
|
||||
for (resource.resource_mapping.freq_band.nof_rb = nof_rb_begin;
|
||||
resource.resource_mapping.freq_band.nof_rb <= nof_rb_end;
|
||||
resource.resource_mapping.freq_band.nof_rb += 4) {
|
||||
|
||||
// Iterate for all slot numbers
|
||||
for (slot_cfg.idx = 0; slot_cfg.idx < SRSLTE_NSLOTS_PER_FRAME_NR(carrier.numerology); slot_cfg.idx++) {
|
||||
// Steer Frequency allocation
|
||||
for (uint32_t freq_dom_alloc = 0; freq_dom_alloc < nof_freq_dom_alloc; freq_dom_alloc++) {
|
||||
for (uint32_t i = 0; i < nof_freq_dom_alloc; i++) {
|
||||
resource.resource_mapping.frequency_domain_alloc[i] = i == freq_dom_alloc;
|
||||
}
|
||||
|
||||
// Call actual test
|
||||
if (test(&slot_cfg, &resource, &awgn, grid) < SRSLTE_SUCCESS) {
|
||||
goto clean_exit;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
ret = SRSLTE_SUCCESS;
|
||||
|
||||
clean_exit:
|
||||
if (grid) {
|
||||
free(grid);
|
||||
}
|
||||
|
||||
srslte_channel_awgn_free(&awgn);
|
||||
|
||||
if (ret == SRSLTE_SUCCESS) {
|
||||
printf("Passed!\n");
|
||||
} else {
|
||||
printf("Failed!\n");
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
Loading…
Reference in New Issue