mirror of https://github.com/pvnis/srsRAN_4G.git
Implemented high performance AWGN generator
Xavier Arteaga
5 years ago
committed by
Xavier Arteaga
parent
b5be0b94b8
commit
c107b04f5a
@ -0,0 +1,198 @@
|
||||
/*
|
||||
* 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/utils/vector.h"
|
||||
#include <complex.h>
|
||||
#include <srslte/phy/channel/ch_awgn.h>
|
||||
#include <srslte/phy/dft/dft.h>
|
||||
#include <srslte/phy/utils/debug.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#undef ENABLE_GUI
|
||||
#ifdef ENABLE_GUI
|
||||
#include "srsgui/srsgui.h"
|
||||
#endif /* ENABLE_GUI */
|
||||
|
||||
static srslte_channel_awgn_t awgn = {};
|
||||
|
||||
static uint32_t nof_samples = 1920 * 8;
|
||||
static float n0_min = 0.0f;
|
||||
static float n0_max = 6.0f;
|
||||
static float n0_step = 0.25f;
|
||||
static float tolerance = 0.05f;
|
||||
|
||||
static void usage(char* prog)
|
||||
{
|
||||
printf("Usage: %s [nmMst]\n", prog);
|
||||
printf("\t-n number of samples to simulate [Default %d]\n", nof_samples);
|
||||
printf("\t-m Minimum n0 (in dBfs) to simulate [Default %.3f]\n", n0_min);
|
||||
printf("\t-M Maximum n0 (in dBfs) to simulate: [Default %.3f]\n", n0_max);
|
||||
printf("\t-s n0 step size: [Default %.3f]\n", n0_step);
|
||||
printf("\t-t tolerance: [Default %.3f]\n", tolerance);
|
||||
}
|
||||
|
||||
static void parse_args(int argc, char** argv)
|
||||
{
|
||||
int opt;
|
||||
while ((opt = getopt(argc, argv, "nmMst")) != -1) {
|
||||
switch (opt) {
|
||||
case 'm':
|
||||
n0_min = strtof(argv[optind], NULL);
|
||||
break;
|
||||
case 'M':
|
||||
n0_max = strtof(argv[optind], NULL);
|
||||
break;
|
||||
case 's':
|
||||
n0_step = strtof(argv[optind], NULL);
|
||||
break;
|
||||
case 't':
|
||||
tolerance = strtof(argv[optind], NULL);
|
||||
break;
|
||||
case 'n':
|
||||
nof_samples = (uint32_t)strtol(argv[optind], NULL, 10);
|
||||
break;
|
||||
default:
|
||||
usage(argv[0]);
|
||||
exit(-1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
int ret = SRSLTE_SUCCESS;
|
||||
cf_t* input_buffer = NULL;
|
||||
cf_t* output_buffer = NULL;
|
||||
uint64_t count_samples = 0;
|
||||
uint64_t count_us = 0;
|
||||
|
||||
// Parse arguments
|
||||
parse_args(argc, argv);
|
||||
|
||||
// Initialise buffers
|
||||
input_buffer = srslte_vec_cf_malloc(nof_samples);
|
||||
output_buffer = srslte_vec_cf_malloc(nof_samples);
|
||||
|
||||
if (!input_buffer || !output_buffer) {
|
||||
ERROR("Error: Allocating memory\n");
|
||||
ret = SRSLTE_ERROR;
|
||||
}
|
||||
|
||||
// Initialise input
|
||||
srslte_vec_cf_zero(input_buffer, nof_samples);
|
||||
|
||||
#ifdef ENABLE_GUI
|
||||
sdrgui_init();
|
||||
sdrgui_init_title("SRS Fading channel");
|
||||
plot_real_t plot_fft = NULL;
|
||||
|
||||
plot_real_init(&plot_fft);
|
||||
plot_real_setTitle(&plot_fft, "AWGN");
|
||||
plot_real_addToWindowGrid(&plot_fft, (char*)"Spectrum", 0, 0);
|
||||
|
||||
cf_t* fft_out = srslte_vec_cf_malloc(nof_samples);
|
||||
srslte_dft_plan_t fft = {};
|
||||
if (srslte_dft_plan_c(&fft, nof_samples, SRSLTE_DFT_FORWARD)) {
|
||||
ERROR("Error: init DFT\n");
|
||||
ret = SRSLTE_ERROR;
|
||||
}
|
||||
#endif /* ENABLE_GUI */
|
||||
|
||||
// Initialise AWGN channel
|
||||
if (ret == SRSLTE_SUCCESS) {
|
||||
ret = srslte_channel_awgn_init(&awgn, 0);
|
||||
}
|
||||
|
||||
float n0 = n0_min;
|
||||
while (!isnan(n0) && !isinf(n0) && n0 < n0_max) {
|
||||
struct timeval t[3] = {};
|
||||
|
||||
srslte_channel_awgn_set_n0(&awgn, n0);
|
||||
|
||||
// Run actual test
|
||||
gettimeofday(&t[1], NULL);
|
||||
srslte_channel_awgn_run_c(&awgn, input_buffer, output_buffer, nof_samples);
|
||||
gettimeofday(&t[2], NULL);
|
||||
get_time_interval(t);
|
||||
|
||||
float power = srslte_vec_avg_power_cf(output_buffer, nof_samples);
|
||||
float power_dB = srslte_convert_power_to_dB(power);
|
||||
|
||||
if ((n0 + tolerance) < power_dB || (n0 - tolerance) > power_dB) {
|
||||
printf("-- failed: %.3f<%.3f<%.3f\n", n0 - tolerance, power_dB, n0 + tolerance);
|
||||
ret = SRSLTE_ERROR;
|
||||
}
|
||||
|
||||
#ifdef ENABLE_GUI
|
||||
srslte_dft_run_c(&fft, output_buffer, fft_out);
|
||||
|
||||
float min = +INFINITY;
|
||||
float max = -INFINITY;
|
||||
float* fft_mag = (float*)fft_out;
|
||||
for (uint32_t i = 0; i < nof_samples; i++) {
|
||||
float mag = srslte_convert_amplitude_to_dB(cabsf(fft_out[i]));
|
||||
|
||||
min = SRSLTE_MIN(min, mag);
|
||||
max = SRSLTE_MAX(max, mag);
|
||||
|
||||
fft_mag[i] = srslte_convert_amplitude_to_dB(mag);
|
||||
}
|
||||
|
||||
plot_real_setYAxisScale(&plot_fft, min, max);
|
||||
|
||||
plot_real_setNewData(&plot_fft, fft_mag, nof_samples);
|
||||
sleep(5);
|
||||
#endif /* ENABLE_GUI */
|
||||
|
||||
count_samples += nof_samples;
|
||||
count_us += t->tv_usec + t->tv_sec * 1000000;
|
||||
|
||||
n0 += n0_step;
|
||||
}
|
||||
|
||||
// Free
|
||||
srslte_channel_awgn_free(&awgn);
|
||||
|
||||
if (input_buffer) {
|
||||
free(input_buffer);
|
||||
}
|
||||
|
||||
if (output_buffer) {
|
||||
free(output_buffer);
|
||||
}
|
||||
|
||||
#ifdef ENABLE_GUI
|
||||
if (fft_out) {
|
||||
free(fft_out);
|
||||
}
|
||||
srslte_dft_plan_free(&fft);
|
||||
#endif /* ENABLE_GUI */
|
||||
|
||||
// Print result and exit
|
||||
printf("Test n0_min=%.3f; n0_max=%.3f; n0_step=%.3f; nof_samples=%d; %s ... %.1f MSps\n",
|
||||
n0_min,
|
||||
n0_max,
|
||||
n0_step,
|
||||
nof_samples,
|
||||
(ret == SRSLTE_SUCCESS) ? "Passed" : "Failed",
|
||||
(double)nof_samples / (double)count_us);
|
||||
return ret;
|
||||
}
|
||||
Loading…
Reference in New Issue