srsLTE/test/phy/dummy_tx_harq_proc.h

102 lines
2.7 KiB
C++

/**
* Copyright 2013-2022 Software Radio Systems Limited
*
* This file is part of srsRAN.
*
* srsRAN 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.
*
* srsRAN 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/.
*
*/
#ifndef SRSRAN_TX_DUMMY_HARQ_PROC_H
#define SRSRAN_TX_DUMMY_HARQ_PROC_H
#include <mutex>
#include <srsenb/hdr/stack/mac/common/mac_metrics.h>
#include <srsran/adt/circular_array.h>
#include <srsran/common/buffer_pool.h>
#include <srsran/common/phy_cfg_nr.h>
#include <srsran/common/standard_streams.h>
class dummy_tx_harq_proc
{
private:
mutable std::mutex mutex;
srsran_random_t random_gen = nullptr;
srsran::byte_buffer_t data;
srsran_softbuffer_tx_t softbuffer = {};
uint32_t tbs = 0;
bool first = true;
uint32_t ndi = 0;
public:
dummy_tx_harq_proc()
{
// Initialise softbuffer
if (srsran_softbuffer_tx_init_guru(&softbuffer, SRSRAN_SCH_NR_MAX_NOF_CB_LDPC, SRSRAN_LDPC_MAX_LEN_ENCODED_CB) <
SRSRAN_SUCCESS) {
ERROR("Error Tx buffer");
}
}
void init(uint32_t pid) { random_gen = srsran_random_init(pid * 1234); }
~dummy_tx_harq_proc()
{
srsran_softbuffer_tx_free(&softbuffer);
srsran_random_free(random_gen);
}
srsran::byte_buffer_t* get_tb(uint32_t tbs_)
{
std::unique_lock<std::mutex> lock(mutex);
tbs = tbs_;
srsran_random_byte_vector(random_gen, data.msg, tbs / 8);
return &data;
}
srsran_softbuffer_tx_t& get_softbuffer(uint32_t ndi_)
{
std::unique_lock<std::mutex> lock(mutex);
if (ndi_ != ndi || first) {
srsran_softbuffer_tx_reset(&softbuffer);
ndi = ndi_;
first = false;
}
return softbuffer;
}
uint32_t get_tbs() const
{
std::unique_lock<std::mutex> lock(mutex);
return tbs;
}
};
class dummy_tx_harq_entity : public srsran::circular_array<dummy_tx_harq_proc, SRSRAN_MAX_HARQ_PROC_DL_NR>
{
public:
dummy_tx_harq_entity() : srsran::circular_array<dummy_tx_harq_proc, SRSRAN_MAX_HARQ_PROC_DL_NR>()
{
uint32_t pid = 0;
for (dummy_tx_harq_proc& proc : *this) {
proc.init(pid++);
}
}
};
#endif // SRSRAN_TX_DUMMY_HARQ_PROC_H