mirror of https://github.com/pvnis/srsRAN_4G.git
Initial real-time Tx gain setting (#2976)
* Make filename const in filesink * Sine generation returns the next phase * Avoid malloc/free in radio class * Implement Tx gain in ZMQ * Initial ratio RT gain test * UHD: use timed Tx gain commands to align changes to subframes * Minor improvement in test_radio_rt_gain * Fix compilation * Check RF gain thread id before joining * Remove redundant zero initialization. Co-authored-by: Fabian Eckermann <fabian@srs.io>master
Xavier Arteaga
3 years ago
committed by
GitHub
parent
ebab12403f
commit
322f57a952
@ -0,0 +1,300 @@
|
||||
/**
|
||||
*
|
||||
* \section COPYRIGHT
|
||||
*
|
||||
* Copyright 2013-2021 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 "srsran/common/test_common.h"
|
||||
#include "srsran/radio/radio.h"
|
||||
#include <boost/program_options.hpp>
|
||||
#include <iostream>
|
||||
|
||||
// shorten boost program options namespace
|
||||
namespace bpo = boost::program_options;
|
||||
|
||||
// Test arguments
|
||||
struct test_args_s {
|
||||
bool valid = false;
|
||||
double srate_hz = 3.84e6;
|
||||
double freq_hz = 3.5e9;
|
||||
float rx_gain_db = 20.0f;
|
||||
float tx_gain_db_begin = 0.0f;
|
||||
float tx_gain_db_end = 30.0f;
|
||||
float tx_gain_db_step = 1.0f;
|
||||
uint32_t tx_delay_ms = 4;
|
||||
uint32_t step_period_ms = 1;
|
||||
uint32_t nof_repetitions = 3;
|
||||
uint32_t power_ramping_ms = 50;
|
||||
uint32_t pre_tx_ms = 50; // After main loop acquire a few more ms
|
||||
uint32_t post_tx_ms = 50; // After main loop acquire a few more ms
|
||||
std::string filename = "/tmp/baseband.iq.dat";
|
||||
std::string device_name = "zmq";
|
||||
std::string device_args = "tx_port=tcp://*:5555,rx_port=tcp://localhost:5555,base_srate=3.84e6";
|
||||
|
||||
test_args_s(int argc, char** argv)
|
||||
{
|
||||
bpo::options_description options;
|
||||
|
||||
// clang-format off
|
||||
options.add_options()
|
||||
("srate", bpo::value<double>(&srate_hz)->default_value(srate_hz), "Sampling rate in Hz")
|
||||
("freq", bpo::value<double>(&freq_hz)->default_value(freq_hz), "Center frequency in Hz")
|
||||
("rx_gain", bpo::value<float>(&rx_gain_db)->default_value(rx_gain_db), "Receiver gain in dB")
|
||||
("tx_gain_begin", bpo::value<float>(&tx_gain_db_begin)->default_value(tx_gain_db_begin), "Initial transmitter gain in dB")
|
||||
("tx_gain_end", bpo::value<float>(&tx_gain_db_end)->default_value(tx_gain_db_end), "Final transmitter gain in dB")
|
||||
("tx_gain_step", bpo::value<float>(&tx_gain_db_step)->default_value(tx_gain_db_step), "Step transmitter gain in dB")
|
||||
("tx_delay", bpo::value<uint32_t>(&tx_delay_ms)->default_value(tx_delay_ms), "Delay between Rx and Tx in milliseconds")
|
||||
("step_period", bpo::value<uint32_t>(&step_period_ms)->default_value(step_period_ms), "Transmitter gain step period in milliseconds")
|
||||
("repetitions", bpo::value<uint32_t>(&nof_repetitions)->default_value(nof_repetitions), "Number of transmit gain steering repetitions")
|
||||
("power_ramping", bpo::value<uint32_t>(&power_ramping_ms)->default_value(power_ramping_ms), "Transmitter initial power ramping in milliseconds")
|
||||
("pre_tx", bpo::value<uint32_t>(&pre_tx_ms)->default_value(pre_tx_ms), "Initial acquisition time before start transmission in milliseconds")
|
||||
("post_tx", bpo::value<uint32_t>(&pre_tx_ms)->default_value(pre_tx_ms), "Initial acquisition time after ending transmission in milliseconds")
|
||||
("filename", bpo::value<std::string>(&filename)->default_value(filename), "File sink filename")
|
||||
("dev_name", bpo::value<std::string>(&device_name)->default_value(device_name), "RF Device name")
|
||||
("dev_args", bpo::value<std::string>(&device_args)->default_value(device_args), "RF Device arguments")
|
||||
("help", "Show this message")
|
||||
;
|
||||
// clang-format on
|
||||
|
||||
bpo::variables_map vm;
|
||||
try {
|
||||
bpo::store(bpo::command_line_parser(argc, argv).options(options).run(), vm);
|
||||
bpo::notify(vm);
|
||||
valid = true;
|
||||
} catch (bpo::error& e) {
|
||||
std::cerr << e.what() << std::endl;
|
||||
}
|
||||
|
||||
// help option was given or error - print usage and exit
|
||||
if (vm.count("help") > 0 or not valid) {
|
||||
std::cout << "Usage: " << argv[0] << " [OPTIONS] config_file" << std::endl << std::endl;
|
||||
std::cout << options << std::endl << std::endl;
|
||||
valid = false;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
class phy_radio_listener : public srsran::phy_interface_radio
|
||||
{
|
||||
private:
|
||||
srslog::basic_logger& logger = srslog::fetch_basic_logger("TEST", false);
|
||||
uint32_t overflow_count = 0;
|
||||
uint32_t failure_count = 0;
|
||||
|
||||
public:
|
||||
phy_radio_listener() {}
|
||||
void radio_overflow() override
|
||||
{
|
||||
overflow_count++;
|
||||
logger.error("Overflow");
|
||||
}
|
||||
void radio_failure() override
|
||||
{
|
||||
failure_count++;
|
||||
logger.error("Failure");
|
||||
}
|
||||
};
|
||||
|
||||
class test_sink
|
||||
{
|
||||
private:
|
||||
srslog::basic_logger& logger = srslog::fetch_basic_logger("SINK", false);
|
||||
srsran_filesink_t filesink = {};
|
||||
std::vector<float> meas_power_dB;
|
||||
|
||||
public:
|
||||
test_sink(const std::string& filename)
|
||||
{
|
||||
if (srsran_filesink_init(&filesink, filename.c_str(), SRSRAN_COMPLEX_FLOAT_BIN) < SRSRAN_SUCCESS) {
|
||||
ERROR("Error initiating filesink");
|
||||
filesink = {};
|
||||
}
|
||||
}
|
||||
|
||||
~test_sink()
|
||||
{
|
||||
// Free filesink
|
||||
srsran_filesink_free(&filesink);
|
||||
|
||||
// Print measure power
|
||||
printf("Measured power: ");
|
||||
srsran_vec_fprint_f(stdout, meas_power_dB.data(), (int)meas_power_dB.size());
|
||||
}
|
||||
|
||||
void write(std::vector<cf_t>& buffer, uint32_t nsamp)
|
||||
{
|
||||
// Measure average power
|
||||
float avg_pwr = srsran_vec_avg_power_cf(buffer.data(), nsamp);
|
||||
|
||||
// Push measurement in dB
|
||||
meas_power_dB.push_back(srsran_convert_power_to_dB(avg_pwr));
|
||||
|
||||
// Write signal in file
|
||||
srsran_filesink_write(&filesink, (void*)buffer.data(), (int)nsamp);
|
||||
}
|
||||
};
|
||||
|
||||
class test_source
|
||||
{
|
||||
private:
|
||||
// cf_t phase = 1.0f;
|
||||
cf_t phase = 0.01;
|
||||
float freq_norm = 0.125;
|
||||
|
||||
public:
|
||||
test_source() {}
|
||||
|
||||
~test_source() {}
|
||||
|
||||
void generate(std::vector<cf_t>& buffer, uint32_t nsamp)
|
||||
{
|
||||
phase *= srsran_vec_gen_sine(phase, freq_norm, buffer.data(), (int)nsamp);
|
||||
}
|
||||
};
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
srslog::basic_logger& logger = srslog::fetch_basic_logger("TEST", false);
|
||||
srslog::init();
|
||||
srsran::radio radio;
|
||||
phy_radio_listener radio_listener;
|
||||
|
||||
enum { BEFORE_RAMPING, RAMPING, GAIN_STEERING, POST_GAIN_STEERING, END } state = BEFORE_RAMPING;
|
||||
|
||||
test_args_s args(argc, argv);
|
||||
TESTASSERT(args.valid);
|
||||
|
||||
// Calculate number of steps
|
||||
TESTASSERT(std::isnormal(args.tx_gain_db_step));
|
||||
uint32_t nof_steps = (args.tx_gain_db_end - args.tx_gain_db_begin) / args.tx_gain_db_step;
|
||||
|
||||
// Calculate subframe size in 1ms
|
||||
TESTASSERT(std::isnormal(args.srate_hz));
|
||||
uint32_t sf_sz = (uint32_t)std::round(1e-3 * args.srate_hz);
|
||||
|
||||
// Allocate baseband buffer
|
||||
std::vector<cf_t> buffer(sf_sz);
|
||||
|
||||
// Prepare radio arguments
|
||||
srsran::rf_args_t rf_args = {};
|
||||
rf_args.log_level = "info";
|
||||
rf_args.srate_hz = args.srate_hz;
|
||||
rf_args.dl_freq = args.freq_hz;
|
||||
rf_args.ul_freq = args.freq_hz;
|
||||
rf_args.rx_gain = args.rx_gain_db;
|
||||
rf_args.tx_gain = args.tx_gain_db_begin;
|
||||
rf_args.nof_carriers = 1;
|
||||
rf_args.nof_antennas = 1;
|
||||
rf_args.device_name = args.device_name;
|
||||
rf_args.device_args = args.device_args;
|
||||
|
||||
// Initialise radio
|
||||
TESTASSERT(radio.init(rf_args, &radio_listener) == SRSRAN_SUCCESS);
|
||||
|
||||
// Setup LO frequencies
|
||||
radio.set_tx_freq(0, args.freq_hz);
|
||||
radio.set_rx_freq(0, args.freq_hz);
|
||||
|
||||
// Setup sampling rate
|
||||
radio.set_tx_srate(args.srate_hz);
|
||||
radio.set_rx_srate(args.srate_hz);
|
||||
|
||||
// Setup initial gains
|
||||
radio.set_tx_gain(args.tx_gain_db_begin);
|
||||
radio.set_rx_gain(args.rx_gain_db);
|
||||
|
||||
// Create signal sink
|
||||
test_sink sink(args.filename);
|
||||
|
||||
// Create signal source
|
||||
test_source source;
|
||||
|
||||
// Perform Tx/Rx
|
||||
uint32_t sf_count = 0;
|
||||
uint32_t repetition = 0;
|
||||
while (state != END) {
|
||||
switch (state) {
|
||||
case BEFORE_RAMPING:
|
||||
if (sf_count >= args.pre_tx_ms) {
|
||||
logger.info("-- Starting power ramping stage");
|
||||
state = RAMPING;
|
||||
sf_count = 0;
|
||||
continue;
|
||||
}
|
||||
break;
|
||||
case RAMPING:
|
||||
if (sf_count >= args.power_ramping_ms) {
|
||||
logger.info("-- Starting gain steering stage");
|
||||
state = GAIN_STEERING;
|
||||
sf_count = 0;
|
||||
continue;
|
||||
}
|
||||
break;
|
||||
case GAIN_STEERING:
|
||||
if (sf_count >= nof_steps * args.step_period_ms) {
|
||||
repetition++;
|
||||
sf_count = 0;
|
||||
logger.info("-- Finished repetition %d of %d", repetition, args.nof_repetitions);
|
||||
}
|
||||
|
||||
if (repetition >= args.nof_repetitions) {
|
||||
logger.info("-- Starting post gain steering stage");
|
||||
state = POST_GAIN_STEERING;
|
||||
radio.tx_end();
|
||||
continue;
|
||||
}
|
||||
if (sf_count % args.step_period_ms == 0 and sf_count != 0) {
|
||||
float tx_gain_db = args.tx_gain_db_begin + args.tx_gain_db_step * (sf_count / args.step_period_ms);
|
||||
logger.info("-- New Tx gain %+.2f dB", tx_gain_db);
|
||||
radio.set_tx_gain(tx_gain_db);
|
||||
}
|
||||
break;
|
||||
case POST_GAIN_STEERING:
|
||||
if (sf_count >= args.post_tx_ms) {
|
||||
logger.info("-- Ending...");
|
||||
state = END;
|
||||
sf_count = 0;
|
||||
continue;
|
||||
}
|
||||
break;
|
||||
case END:
|
||||
continue;
|
||||
}
|
||||
|
||||
// Prepare reception buffers
|
||||
srsran::rf_buffer_t rf_buffer = {};
|
||||
rf_buffer.set(0, buffer.data());
|
||||
rf_buffer.set_nof_samples(sf_sz);
|
||||
|
||||
// Receive
|
||||
srsran::rf_timestamp_t ts = {};
|
||||
TESTASSERT(radio.rx_now(rf_buffer, ts));
|
||||
|
||||
// Save signal, including the time before transmission
|
||||
sink.write(buffer, sf_sz);
|
||||
|
||||
if (state == RAMPING or state == GAIN_STEERING) {
|
||||
// Generate transmit signal
|
||||
source.generate(buffer, sf_sz);
|
||||
|
||||
// Update timestamp for Tx
|
||||
ts.add(1e-3 * (double)args.tx_delay_ms);
|
||||
|
||||
// Transmit
|
||||
radio.tx(rf_buffer, ts);
|
||||
}
|
||||
|
||||
// Increase SF counting
|
||||
sf_count++;
|
||||
}
|
||||
|
||||
// Tear down radio
|
||||
radio.stop();
|
||||
|
||||
return SRSRAN_SUCCESS;
|
||||
}
|
||||
Loading…
Reference in New Issue