mirror of https://github.com/PentHertz/srsLTE.git
120 lines
3.2 KiB
C++
120 lines
3.2 KiB
C++
/**
|
|
*
|
|
* \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.
|
|
*
|
|
*/
|
|
|
|
#ifndef SRSLTE_SCHED_TEST_COMMON_H
|
|
#define SRSLTE_SCHED_TEST_COMMON_H
|
|
|
|
#include "sched_sim_ue.h"
|
|
#include "sched_test_utils.h"
|
|
#include "srsenb/hdr/stack/mac/sched.h"
|
|
#include <random>
|
|
|
|
namespace srsenb {
|
|
|
|
/***************************
|
|
* Random Utils
|
|
**************************/
|
|
|
|
void set_randseed(uint64_t seed);
|
|
float randf();
|
|
std::default_random_engine& get_rand_gen();
|
|
|
|
/**************************
|
|
* Testers
|
|
*************************/
|
|
|
|
using dl_sched_res_list = std::vector<sched_interface::dl_sched_res_t>;
|
|
using ul_sched_res_list = std::vector<sched_interface::ul_sched_res_t>;
|
|
|
|
class sched_sim_random : public sched_sim_base
|
|
{
|
|
public:
|
|
using sched_sim_base::sched_sim_base;
|
|
void set_external_tti_events(const sim_ue_ctxt_t& ue_ctxt, ue_tti_events& pending_events) override;
|
|
std::map<uint16_t, ue_ctxt_test_cfg> ue_sim_cfg_map;
|
|
};
|
|
|
|
class sched_result_stats
|
|
{
|
|
public:
|
|
explicit sched_result_stats(std::vector<srsenb::sched::cell_cfg_t> cell_params_) :
|
|
cell_params(std::move(cell_params_))
|
|
{}
|
|
|
|
void process_results(tti_point tti_rx,
|
|
const std::vector<sched_interface::dl_sched_res_t>& dl_result,
|
|
const std::vector<sched_interface::ul_sched_res_t>& ul_result);
|
|
|
|
struct user_stats {
|
|
uint16_t rnti;
|
|
std::vector<uint64_t> tot_dl_sched_data; // includes retxs
|
|
std::vector<uint64_t> tot_ul_sched_data;
|
|
};
|
|
|
|
std::map<uint16_t, user_stats> users;
|
|
|
|
private:
|
|
user_stats* get_user(uint16_t rnti);
|
|
|
|
const std::vector<srsenb::sched::cell_cfg_t> cell_params;
|
|
};
|
|
|
|
// Intrusive Scheduler Tester
|
|
class common_sched_tester : public sched
|
|
{
|
|
public:
|
|
struct tti_info_t {
|
|
uint32_t nof_prachs = 0;
|
|
std::vector<sched_interface::dl_sched_res_t> dl_sched_result;
|
|
std::vector<sched_interface::ul_sched_res_t> ul_sched_result;
|
|
};
|
|
|
|
~common_sched_tester() override = default;
|
|
|
|
const ue_cfg_t* get_current_ue_cfg(uint16_t rnti) const;
|
|
|
|
int sim_cfg(sim_sched_args args);
|
|
virtual int add_user(uint16_t rnti, const ue_ctxt_test_cfg& ue_cfg_);
|
|
virtual int reconf_user(uint16_t rnti, const sched_interface::ue_cfg_t& ue_cfg_);
|
|
virtual int rem_user(uint16_t rnti);
|
|
virtual int process_results();
|
|
int process_tti_events(const tti_ev& tti_ev);
|
|
|
|
int test_next_ttis(const std::vector<tti_ev>& tti_events);
|
|
int run_tti(const tti_ev& tti_events);
|
|
|
|
int run_ue_ded_tests_and_update_ctxt(const sf_output_res_t& sf_out);
|
|
|
|
// args
|
|
sim_sched_args sim_args0; ///< arguments used to generate TTI events
|
|
srslte::log* tester_log = nullptr;
|
|
|
|
// tti specific params
|
|
tti_info_t tti_info;
|
|
tti_point tti_rx;
|
|
uint32_t tti_count = 0;
|
|
|
|
// eNB+UE state handlers
|
|
std::unique_ptr<sched_sim_random> sched_sim;
|
|
|
|
// statistics
|
|
std::unique_ptr<sched_result_stats> sched_stats;
|
|
|
|
protected:
|
|
virtual void new_test_tti();
|
|
virtual void before_sched() {}
|
|
};
|
|
|
|
} // namespace srsenb
|
|
|
|
#endif // SRSLTE_SCHED_TEST_COMMON_H
|