mirror of https://github.com/pvnis/srsRAN_4G.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
137 lines
4.1 KiB
C++
137 lines
4.1 KiB
C++
/**
|
|
*
|
|
* \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.
|
|
*
|
|
*/
|
|
|
|
#ifndef SRSRAN_TEST_BENCH_H
|
|
#define SRSRAN_TEST_BENCH_H
|
|
|
|
#include "dummy_phy_common.h"
|
|
#include "srsenb/hdr/phy/nr/worker_pool.h"
|
|
#include "srsue/hdr/phy/nr/worker_pool.h"
|
|
|
|
class test_bench
|
|
{
|
|
private:
|
|
const std::string UE_PHY_COM_LOG_NAME = "UE /PHY/COM";
|
|
const std::string GNB_PHY_COM_LOG_NAME = "GNB/PHY/COM";
|
|
uint32_t tti = 0;
|
|
srsenb::nr::worker_pool gnb_phy;
|
|
phy_common gnb_phy_com;
|
|
srsue::nr::worker_pool ue_phy;
|
|
phy_common ue_phy_com;
|
|
bool initialised = false;
|
|
uint32_t sf_sz = 0;
|
|
|
|
public:
|
|
struct args_t {
|
|
double srate_hz = 11.52e6;
|
|
uint32_t nof_channels = 1;
|
|
uint32_t buffer_sz_ms = 10;
|
|
bool valid = false;
|
|
srsran::phy_cfg_nr_t phy_cfg = {};
|
|
srsenb::phy_cell_cfg_list_nr_t cell_list = {};
|
|
srsenb::nr::worker_pool::args_t gnb_args;
|
|
srsue::phy_args_nr_t ue_args;
|
|
uint16_t rnti = 0x1234;
|
|
std::string phy_com_log_level = "info";
|
|
|
|
args_t(int argc, char** argv);
|
|
};
|
|
|
|
test_bench(const args_t& args, srsenb::stack_interface_phy_nr& gnb_stack, srsue::stack_interface_phy_nr& ue_stack) :
|
|
ue_phy(args.ue_args.nof_phy_threads),
|
|
gnb_phy(gnb_phy_com, gnb_stack, srslog::get_default_sink(), args.gnb_args.nof_phy_threads),
|
|
ue_phy_com(phy_common::args_t(args.srate_hz, args.buffer_sz_ms, args.nof_channels),
|
|
srslog::fetch_basic_logger(UE_PHY_COM_LOG_NAME, srslog::get_default_sink(), false)),
|
|
gnb_phy_com(phy_common::args_t(args.srate_hz, args.buffer_sz_ms, args.nof_channels),
|
|
srslog::fetch_basic_logger(GNB_PHY_COM_LOG_NAME, srslog::get_default_sink(), false)),
|
|
sf_sz((uint32_t)std::round(args.srate_hz * 1e-3))
|
|
{
|
|
srslog::fetch_basic_logger(UE_PHY_COM_LOG_NAME).set_level(srslog::str_to_basic_level(args.phy_com_log_level));
|
|
srslog::fetch_basic_logger(GNB_PHY_COM_LOG_NAME).set_level(srslog::str_to_basic_level(args.phy_com_log_level));
|
|
|
|
if (not gnb_phy.init(args.gnb_args, args.cell_list)) {
|
|
return;
|
|
}
|
|
|
|
// Initialise UE PHY
|
|
if (not ue_phy.init(args.ue_args, ue_phy_com, &ue_stack, 31)) {
|
|
return;
|
|
}
|
|
|
|
// Set UE configuration
|
|
if (not ue_phy.set_config(args.phy_cfg)) {
|
|
return;
|
|
}
|
|
|
|
initialised = true;
|
|
}
|
|
|
|
~test_bench()
|
|
{
|
|
ue_phy_com.stop();
|
|
gnb_phy_com.stop();
|
|
gnb_phy.stop();
|
|
ue_phy.stop();
|
|
}
|
|
|
|
bool is_initialised() const { return initialised; }
|
|
|
|
bool run_tti()
|
|
{
|
|
// Get gNb worker
|
|
srsenb::nr::slot_worker* gnb_worker = gnb_phy.wait_worker(tti);
|
|
if (gnb_worker == nullptr) {
|
|
return false;
|
|
}
|
|
|
|
// Feed gNb the UE transmitted signal
|
|
srsran::rf_timestamp_t gnb_time = {};
|
|
std::vector<cf_t*> gnb_rx_buffers(1);
|
|
gnb_rx_buffers[0] = gnb_worker->get_buffer_rx(0);
|
|
ue_phy_com.read(gnb_rx_buffers, sf_sz, gnb_time);
|
|
|
|
// Set gNb time
|
|
gnb_time.add(TX_ENB_DELAY * 1e-3);
|
|
gnb_worker->set_time(tti, gnb_time);
|
|
|
|
// Start gNb work
|
|
gnb_phy_com.push_semaphore(gnb_worker);
|
|
gnb_phy.start_worker(gnb_worker);
|
|
|
|
// Get UE worker
|
|
srsue::nr::sf_worker* ue_worker = ue_phy.wait_worker(tti);
|
|
if (ue_worker == nullptr) {
|
|
return false;
|
|
}
|
|
|
|
// Feed UE the gNb transmitted signal
|
|
srsran::rf_timestamp_t ue_time = {};
|
|
std::vector<cf_t*> ue_rx_buffers(1);
|
|
ue_rx_buffers[0] = ue_worker->get_buffer(0, 0);
|
|
gnb_phy_com.read(ue_rx_buffers, sf_sz, ue_time);
|
|
|
|
// Set UE time
|
|
ue_time.add(TX_ENB_DELAY * 1e-3);
|
|
ue_worker->set_tti(tti);
|
|
ue_worker->set_tx_time(ue_time);
|
|
|
|
// Start gNb work
|
|
ue_phy_com.push_semaphore(ue_worker);
|
|
ue_phy.start_worker(ue_worker);
|
|
|
|
tti++;
|
|
return true;
|
|
}
|
|
};
|
|
|
|
#endif // SRSRAN_TEST_BENCH_H
|