nr-gnb-phy.cc

// Copyright (c) 2019 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#define NS_LOG_APPEND_CONTEXT                                                                      \
    do                                                                                             \
    {                                                                                              \
        std::clog << " [ CellId " << GetCellId() << ", bwpId " << GetBwpId() << "] ";              \
    } while (false);
 
#include "nr-gnb-phy.h"
 
#include "beam-manager.h"
#include "nr-ch-access-manager.h"
#include "nr-gnb-net-device.h"
#include "nr-net-device.h"
#include "nr-ue-net-device.h"
#include "nr-ue-phy.h"
 
#include "ns3/double.h"
#include "ns3/enum.h"
#include "ns3/log.h"
#include "ns3/node-list.h"
#include "ns3/node.h"
#include "ns3/object-vector.h"
#include "ns3/pointer.h"
#include "ns3/uinteger.h"
 
#include <algorithm>
#include <functional>
#include <random>
#include <string>
#include <unordered_set>
#include <vector>
 
namespace ns3
{
 
NS_LOG_COMPONENT_DEFINE("NrGnbPhy");
 
NS_OBJECT_ENSURE_REGISTERED(NrGnbPhy);
 
NrGnbPhy::NrGnbPhy()
    : m_n0Delay(0),
      m_n1Delay(4)
{
    NS_LOG_FUNCTION(this);
    m_gnbCphySapProvider = new MemberNrGnbCphySapProvider<NrGnbPhy>(this);
    m_nrFhPhySapUser = new MemberNrFhPhySapUser<NrGnbPhy>(this);
}
 
NrGnbPhy::~NrGnbPhy()
{
}
 
void
NrGnbPhy::DoDispose()
{
    NS_LOG_FUNCTION(this);
    delete m_gnbCphySapProvider;
    delete m_nrFhPhySapUser;
    m_nrFhPhySapUser = nullptr;
    m_nrFhPhySapProvider = nullptr;
    NrPhy::DoDispose();
}
 
void
NrGnbPhy::EnableCsiRs()
{
    m_enableCsiRs = true;
}
 
TypeId
NrGnbPhy::GetTypeId()
{
    static TypeId tid =
        TypeId("ns3::NrGnbPhy")
            .SetParent<NrPhy>()
            .AddConstructor<NrGnbPhy>()
            .AddAttribute("RbOverhead",
                          "Overhead when calculating the usable RB number",
                          DoubleValue(0.04),
                          MakeDoubleAccessor(&NrGnbPhy::SetRbOverhead, &NrGnbPhy::GetRbOverhead),
                          MakeDoubleChecker<double>(0, 0.5))
            .AddAttribute("TxPower",
                          "Transmission power in dBm",
                          DoubleValue(4.0),
                          MakeDoubleAccessor(&NrGnbPhy::SetTxPower, &NrGnbPhy::GetTxPower),
                          MakeDoubleChecker<double>())
            .AddAttribute(
                "NoiseFigure",
                "Loss (dB) in the Signal-to-Noise-Ratio due to non-idealities in the receiver."
                " According to Wikipedia (http://en.wikipedia.org/wiki/Noise_figure), this is "
                "\"the difference in decibels (dB) between"
                " the noise output of the actual receiver to the noise output of an "
                " ideal receiver with the same overall gain and bandwidth when the receivers "
                " are connected to sources at the standard noise temperature T0.\" "
                "In this model, we consider T0 = 290K.",
                DoubleValue(5.0),
                MakeDoubleAccessor(&NrPhy::SetNoiseFigure, &NrPhy::GetNoiseFigure),
                MakeDoubleChecker<double>())
            .AddAttribute(
                "PowerAllocationType",
                "Defines the type of the power allocation. Currently are supported "
                "two types: \"UniformPowerAllocBw\", which is a uniform power allocation over all "
                "bandwidth (over all RBs), and \"UniformPowerAllocUsed\", which is a uniform "
                "power allocation over used (active) RBs. By default is set a uniform power "
                "allocation over used RBs .",
                EnumValue(NrSpectrumValueHelper::UNIFORM_POWER_ALLOCATION_USED),
                MakeEnumAccessor<NrSpectrumValueHelper::PowerAllocationType>(
                    &NrPhy::SetPowerAllocationType,
                    &NrPhy::GetPowerAllocationType),
                MakeEnumChecker(NrSpectrumValueHelper::UNIFORM_POWER_ALLOCATION_BW,
                                "UniformPowerAllocBw",
                                NrSpectrumValueHelper::UNIFORM_POWER_ALLOCATION_USED,
                                "UniformPowerAllocUsed"))
            .AddAttribute("SpectrumPhy",
                          "The downlink NrSpectrumPhy associated to this NrPhy",
                          TypeId::ATTR_GET,
                          PointerValue(),
                          MakePointerAccessor(&NrPhy::GetSpectrumPhy),
                          MakePointerChecker<NrSpectrumPhy>())
            .AddTraceSource("UlSinrTrace",
                            "UL SINR statistics.",
                            MakeTraceSourceAccessor(&NrGnbPhy::m_ulSinrTrace),
                            "ns3::UlSinr::TracedCallback")
            .AddTraceSource("GnbPhyRxedCtrlMsgsTrace",
                            "Gnb PHY Rxed Control Messages Traces.",
                            MakeTraceSourceAccessor(&NrGnbPhy::m_phyRxedCtrlMsgsTrace),
                            "ns3::NrPhyRxTrace::RxedGnbPhyCtrlMsgsTracedCallback")
            .AddTraceSource("GnbPhyTxedCtrlMsgsTrace",
                            "Gnb PHY Txed Control Messages Traces.",
                            MakeTraceSourceAccessor(&NrGnbPhy::m_phyTxedCtrlMsgsTrace),
                            "ns3::NrPhyRxTrace::TxedGnbPhyCtrlMsgsTracedCallback")
            .AddAttribute("N0Delay",
                          "Minimum processing delay needed to decode DL DCI and decode DL data",
                          UintegerValue(0),
                          MakeUintegerAccessor(&NrGnbPhy::SetN0Delay, &NrGnbPhy::GetN0Delay),
                          MakeUintegerChecker<uint32_t>(0, 1))
            .AddAttribute("N1Delay",
                          "Minimum processing delay (UE side) from the end of DL Data reception to "
                          "the earliest possible start of the corresponding ACK/NACK transmission",
                          UintegerValue(2),
                          MakeUintegerAccessor(&NrGnbPhy::SetN1Delay, &NrGnbPhy::GetN1Delay),
                          MakeUintegerChecker<uint32_t>(0, 4))
            .AddAttribute("N2Delay",
                          "Minimum processing delay needed to decode UL DCI and prepare UL data",
                          UintegerValue(2),
                          MakeUintegerAccessor(&NrGnbPhy::SetN2Delay, &NrGnbPhy::GetN2Delay),
                          MakeUintegerChecker<uint32_t>(0, 4))
            .AddAttribute("TbDecodeLatency",
                          "Transport block decode latency",
                          TimeValue(MicroSeconds(100)),
                          MakeTimeAccessor(&NrPhy::SetTbDecodeLatency, &NrPhy::GetTbDecodeLatency),
                          MakeTimeChecker())
            .AddAttribute("Numerology",
                          "The 3GPP numerology to be used",
                          UintegerValue(0),
                          MakeUintegerAccessor(&NrPhy::SetNumerology, &NrPhy::GetNumerology),
                          MakeUintegerChecker<uint16_t>())
            .AddAttribute(
                "SymbolsPerSlot",
                "Number of symbols in one slot",
                UintegerValue(14),
                MakeUintegerAccessor(&NrPhy::SetSymbolsPerSlot, &NrPhy::GetSymbolsPerSlot),
                MakeUintegerChecker<uint16_t>())
            .AddAttribute("Pattern",
                          "The slot pattern",
                          StringValue("F|F|F|F|F|F|F|F|F|F|"),
                          MakeStringAccessor(&NrGnbPhy::SetPattern, &NrGnbPhy::GetPattern),
                          MakeStringChecker())
            .AddAttribute(
                "CsiRsModel",
                "Defines the type of the CSI-RS model to use. Currently the user can select "
                "either: CsiRsPerUe or CsiRsPerBeam. CsiRsPerUe means that CSI-RS signals will be "
                "transmitted towards a specific "
                "UE periodically. CsiRsPerBeam means that the CSI-RS will be transmitted using a "
                "predefined set of beams.",
                EnumValue(NrGnbPhy::CSI_RS_PER_UE),
                MakeEnumAccessor<NrGnbPhy::CsiRsModel>(&NrGnbPhy::SetCsiRsModel,
                                                       &NrGnbPhy::GetCsiRsModel),
                MakeEnumChecker(NrGnbPhy::CSI_RS_PER_UE,
                                "CsiRsPerUe",
                                NrGnbPhy::CSI_RS_PER_BEAM,
                                "CsiRsPerBeam"))
            .AddAttribute("CsiRsPeriodicity",
                          "Default CSI periodicity in the number of slots",
                          UintegerValue(10),
                          MakeUintegerAccessor(&NrGnbPhy::SetCsiRsPeriodicity,
                                               &NrGnbPhy::GetCsiRsPeriodicity),
                          MakeUintegerChecker<uint16_t>())
            .AddTraceSource("SlotDataStats",
                            "Data statistics for the current slot: SfnSf, active UE, used RE, "
                            "used symbols, available RBs, available symbols, bwp ID, cell ID",
                            MakeTraceSourceAccessor(&NrGnbPhy::m_phySlotDataStats),
                            "ns3::NrGnbPhy::SlotStatsTracedCallback")
            .AddTraceSource("SlotCtrlStats",
                            "Ctrl statistics for the current slot: SfnSf, active UE, used RE, "
                            "used symbols, available RBs, available symbols, bwp ID, cell ID",
                            MakeTraceSourceAccessor(&NrGnbPhy::m_phySlotCtrlStats),
                            "ns3::NrGnbPhy::SlotStatsTracedCallback")
            .AddTraceSource(
                "RBDataStats",
                "Resource Block used for data: SfnSf, symbol, RB PHY map, bwp ID, cell ID",
                MakeTraceSourceAccessor(&NrGnbPhy::m_rbStatistics),
                "ns3::NrGnbPhy::RBStatsTracedCallback");
    return tid;
}
 
uint32_t
NrGnbPhy::GetNumRbPerRbg() const
{
    return m_phySapUser->GetNumRbPerRbg();
}
 
const SfnSf&
NrGnbPhy::GetCurrentSfnSf() const
{
    return m_currentSlot;
}
 
static uint32_t
modulo(int n, uint32_t m)
{
    if (n >= 0)
    {
        return static_cast<uint32_t>(n) % m;
    }
    else
    {
        while (n < 0)
        {
            n += m;
        }
        return static_cast<uint32_t>(n);
    }
}
 
static int32_t
ReturnHarqSlot(const std::vector<LteNrTddSlotType>& pattern, uint32_t pos, uint32_t n1)
{
    int32_t k1 = static_cast<int32_t>(n1);
 
    uint32_t index = modulo(static_cast<int>(pos) + k1, static_cast<uint32_t>(pattern.size()));
 
    while (pattern[index] < LteNrTddSlotType::S)
    {
        k1++;
        index = modulo(static_cast<int>(pos) + k1, static_cast<uint32_t>(pattern.size()));
        NS_ASSERT(index < pattern.size());
    }
 
    return k1;
}
 
struct DciKPair
{
    uint32_t indexDci{0};
    uint32_t k{0};
};
 
static DciKPair
ReturnDciSlot(const std::vector<LteNrTddSlotType>& pattern, uint32_t pos, uint32_t n)
{
    DciKPair ret;
    ret.k = n;
    ret.indexDci = modulo(static_cast<int>(pos) - static_cast<int>(ret.k),
                          static_cast<uint32_t>(pattern.size()));
 
    while (pattern[ret.indexDci] > LteNrTddSlotType::F)
    {
        ret.k++;
        ret.indexDci = modulo(static_cast<int>(pos) - static_cast<int>(ret.k),
                              static_cast<uint32_t>(pattern.size()));
        NS_ASSERT(ret.indexDci < pattern.size());
    }
 
    return ret;
}
 
static void
GenerateDciMaps(const std::vector<LteNrTddSlotType>& pattern,
                std::map<uint32_t, std::vector<uint32_t>>* toSend,
                std::map<uint32_t, std::vector<uint32_t>>* generate,
                uint32_t pos,
                uint32_t n,
                uint32_t l1l2CtrlLatency)
{
    auto dciSlot = ReturnDciSlot(pattern, pos, n);
    uint32_t indexGen =
        modulo(static_cast<int>(dciSlot.indexDci) - static_cast<int>(l1l2CtrlLatency),
               static_cast<uint32_t>(pattern.size()));
    uint32_t kWithCtrlLatency = static_cast<uint32_t>(dciSlot.k) + l1l2CtrlLatency;
 
    (*toSend)[dciSlot.indexDci].push_back(static_cast<uint32_t>(dciSlot.k));
    (*generate)[indexGen].push_back(kWithCtrlLatency);
}
 
void
NrGnbPhy::GenerateStructuresFromPattern(const std::vector<LteNrTddSlotType>& pattern,
                                        std::map<uint32_t, std::vector<uint32_t>>* toSendDl,
                                        std::map<uint32_t, std::vector<uint32_t>>* toSendUl,
                                        std::map<uint32_t, std::vector<uint32_t>>* generateDl,
                                        std::map<uint32_t, std::vector<uint32_t>>* generateUl,
                                        std::map<uint32_t, uint32_t>* dlHarqfbPosition,
                                        uint32_t n0,
                                        uint32_t n2,
                                        uint32_t n1,
                                        uint32_t l1l2CtrlLatency)
{
    const uint32_t n = static_cast<uint32_t>(pattern.size());
 
    // Create a pattern that is all F.
    std::vector<LteNrTddSlotType> fddGenerationPattern;
    fddGenerationPattern.resize(pattern.size(), LteNrTddSlotType::F);
 
    /* if we have to generate structs for a TDD pattern, then use the input pattern.
     * Otherwise, pass to the gen functions a pattern which is all F (therefore, the
     * the function will think that they will be able to transmit or
     * receive things following n0, n1, n2, that is what happen in FDD, just in
     * another band..
     */
 
    const std::vector<LteNrTddSlotType>* generationPattern;
 
    if (IsTdd(pattern))
    {
        generationPattern = &pattern;
    }
    else
    {
        generationPattern = &fddGenerationPattern;
    }
 
    for (uint32_t i = 0; i < n; i++)
    {
        if ((*generationPattern)[i] == LteNrTddSlotType::UL)
        {
            GenerateDciMaps(*generationPattern, toSendUl, generateUl, i, n2, l1l2CtrlLatency);
        }
        else if ((*generationPattern)[i] == LteNrTddSlotType::DL ||
                 pattern[i] == LteNrTddSlotType::S)
        {
            GenerateDciMaps(*generationPattern, toSendDl, generateDl, i, n0, l1l2CtrlLatency);
 
            int32_t k1 = ReturnHarqSlot(*generationPattern, i, n1);
            (*dlHarqfbPosition).insert(std::make_pair(i, k1));
        }
        else if ((*generationPattern)[i] == LteNrTddSlotType::F)
        {
            GenerateDciMaps(*generationPattern, toSendDl, generateDl, i, n0, l1l2CtrlLatency);
            GenerateDciMaps(*generationPattern, toSendUl, generateUl, i, n2, l1l2CtrlLatency);
 
            int32_t k1 = ReturnHarqSlot(*generationPattern, i, n1);
            (*dlHarqfbPosition).insert(std::make_pair(i, k1));
        }
    }
 
    /*
     * Now, if the input pattern is for FDD, remove the elements in the
     * opposite generate* structures: in the end, we don't want to generate DL
     * for a FDD-UL band, right?
     *
     * But.. maintain the toSend structures, as they will be used to send
     * feedback or other messages, like DCI.
     */
 
    if (!IsTdd(pattern))
    {
        if (HasUlSlot(pattern))
        {
            generateDl->clear();
        }
        else
        {
            generateUl->clear();
        }
    }
 
    for (auto& list : (*generateUl))
    {
        std::stable_sort(list.second.begin(), list.second.end());
    }
 
    for (auto& list : (*generateDl))
    {
        std::stable_sort(list.second.begin(), list.second.end());
    }
}
 
void
NrGnbPhy::PushDlAllocation(const SfnSf& sfnSf) const
{
    NS_LOG_FUNCTION(this);
    NS_ASSERT(m_phySapUser);
 
    auto dci = m_phySapUser->GetDlCtrlDci();
    VarTtiAllocInfo dlCtrlVarTti(dci);
 
    SlotAllocInfo slotAllocInfo = SlotAllocInfo(sfnSf);
 
    slotAllocInfo.m_numSymAlloc = dlCtrlVarTti.m_dci->m_numSym;
    slotAllocInfo.m_type = SlotAllocInfo::DL;
    slotAllocInfo.m_varTtiAllocInfo.emplace_back(dlCtrlVarTti);
 
    m_phySapProvider->SetSlotAllocInfo(slotAllocInfo);
}
 
void
NrGnbPhy::PushUlAllocation(const SfnSf& sfnSf) const
{
    NS_LOG_FUNCTION(this);
    NS_ASSERT(m_phySapUser);
 
    auto dci = m_phySapUser->GetUlCtrlDci();
    VarTtiAllocInfo ulCtrlVarTti(dci);
 
    SlotAllocInfo slotAllocInfo = SlotAllocInfo(sfnSf);
 
    slotAllocInfo.m_numSymAlloc = ulCtrlVarTti.m_dci->m_numSym;
    slotAllocInfo.m_type = SlotAllocInfo::UL;
    slotAllocInfo.m_varTtiAllocInfo.emplace_back(ulCtrlVarTti);
 
    m_phySapProvider->SetSlotAllocInfo(slotAllocInfo);
}
 
void
NrGnbPhy::SetTddPattern(const std::vector<LteNrTddSlotType>& pattern)
{
    NS_LOG_FUNCTION(this);
 
    std::stringstream ss;
 
    for (const auto& v : pattern)
    {
        ss << v << "|";
    }
    NS_LOG_INFO("Set pattern : " << ss.str());
 
    m_tddPattern = pattern;
 
    m_generateDl.clear();
    m_generateUl.clear();
    m_toSendDl.clear();
    m_toSendUl.clear();
    m_dlHarqfbPosition.clear();
 
    GenerateStructuresFromPattern(pattern,
                                  &m_toSendDl,
                                  &m_toSendUl,
                                  &m_generateDl,
                                  &m_generateUl,
                                  &m_dlHarqfbPosition,
                                  0,
                                  GetN2Delay(),
                                  GetN1Delay(),
                                  GetL1L2CtrlLatency());
}
 
void
NrGnbPhy::ScheduleStartEventLoop(uint32_t nodeId, uint16_t frame, uint8_t subframe, uint16_t slot)
{
    NS_LOG_FUNCTION(this);
    Simulator::ScheduleWithContext(nodeId,
                                   MilliSeconds(0),
                                   &NrGnbPhy::StartEventLoop,
                                   this,
                                   frame,
                                   subframe,
                                   slot);
}
 
void
NrGnbPhy::StartEventLoop(uint16_t frame, uint8_t subframe, uint16_t slot)
{
    NS_LOG_FUNCTION(this);
    NS_LOG_DEBUG("PHY starting. Configuration: "
                 << std::endl
                 << "\t TxPower: " << m_txPower << " dBm" << std::endl
                 << "\t NoiseFigure: " << m_noiseFigure << std::endl
                 << "\t N0: " << m_n0Delay << std::endl
                 << "\t N1: " << m_n1Delay << std::endl
                 << "\t N2: " << m_n2Delay << std::endl
                 << "\t TbDecodeLatency: " << GetTbDecodeLatency().GetMicroSeconds() << " us "
                 << std::endl
                 << "\t Numerology: " << GetNumerology() << std::endl
                 << "\t SymbolsPerSlot: " << GetSymbolsPerSlot() << std::endl
                 << "\t Pattern: " << GetPattern() << std::endl
                 << "Attached to physical channel: " << std::endl
                 << "\t Channel bandwidth: " << GetChannelBandwidth() << " Hz" << std::endl
                 << "\t Channel central freq: " << GetCentralFrequency() << " Hz" << std::endl
                 << "\t Num. RB: " << GetRbNum());
    SfnSf startSlot(frame, subframe, slot, GetNumerology());
    InitializeMessageList();
    StartSlot(startSlot);
}
 
void
NrGnbPhy::SetGnbCphySapUser(NrGnbCphySapUser* s)
{
    NS_LOG_FUNCTION(this);
    m_gnbCphySapUser = s;
}
 
NrGnbCphySapProvider*
NrGnbPhy::GetGnbCphySapProvider()
{
    NS_LOG_FUNCTION(this);
    return m_gnbCphySapProvider;
}
 
void
NrGnbPhy::SetNrFhPhySapProvider(NrFhPhySapProvider* s)
{
    m_nrFhPhySapProvider = s;
}
 
NrFhPhySapUser*
NrGnbPhy::GetNrFhPhySapUser()
{
    return m_nrFhPhySapUser;
}
 
uint32_t
NrGnbPhy::GetN0Delay() const
{
    return m_n0Delay;
}
 
uint32_t
NrGnbPhy::GetN1Delay() const
{
    return m_n1Delay;
}
 
uint32_t
NrGnbPhy::GetN2Delay() const
{
    return m_n2Delay;
}
 
void
NrGnbPhy::SetN0Delay(uint32_t delay)
{
    m_n0Delay = delay;
    SetTddPattern(m_tddPattern); // Update the generate/send structures
}
 
void
NrGnbPhy::SetN1Delay(uint32_t delay)
{
    m_n1Delay = delay;
    SetTddPattern(m_tddPattern); // Update the generate/send structures
}
 
void
NrGnbPhy::SetN2Delay(uint32_t delay)
{
    m_n2Delay = delay;
    SetTddPattern(m_tddPattern); // Update the generate/send structures
}
 
bool
NrGnbPhy::DoesFhAllocationFit(uint16_t bwpId, uint32_t mcs, uint32_t nRegs, uint8_t dlRank) const
{
    NS_LOG_FUNCTION(this);
    NS_ASSERT(m_nrFhPhySapProvider);
    return m_nrFhPhySapProvider->DoesAllocationFit(bwpId, mcs, nRegs, dlRank);
}
 
BeamId
NrGnbPhy::GetBeamId(uint16_t rnti) const
{
    NS_LOG_FUNCTION(this);
 
    for (const auto& i : m_deviceMap)
    {
        Ptr<NrUeNetDevice> ueDev = DynamicCast<NrUeNetDevice>(i);
        uint64_t ueRnti = (DynamicCast<NrUePhy>(ueDev->GetPhy(GetBwpId())))->GetRnti();
 
        if (ueRnti == rnti && DynamicCast<UniformPlanarArray>(m_spectrumPhy->GetAntenna()))
        {
            NS_ASSERT(m_spectrumPhy->GetBeamManager());
            return m_spectrumPhy->GetBeamManager()->GetBeamId(i);
        }
    }
    return BeamId(0, 0);
}
 
void
NrGnbPhy::SetCam(const Ptr<NrChAccessManager>& cam)
{
    NS_LOG_FUNCTION(this);
    NS_ASSERT(cam != nullptr);
    m_cam = cam;
    m_cam->SetAccessGrantedCallback(
        std::bind(&NrGnbPhy::ChannelAccessGranted, this, std::placeholders::_1));
    m_cam->SetAccessDeniedCallback(std::bind(&NrGnbPhy::ChannelAccessLost, this));
}
 
Ptr<NrChAccessManager>
NrGnbPhy::GetCam() const
{
    NS_LOG_FUNCTION(this);
    return m_cam;
}
 
void
NrGnbPhy::SetTxPower(double pow)
{
    m_txPower = pow;
}
 
double
NrGnbPhy::GetTxPower() const
{
    return m_txPower;
}
 
void
NrGnbPhy::SetSubChannels(const std::vector<int>& rbIndexVector, size_t nTotalAllocRbs)
{
    Ptr<SpectrumValue> txPsd = GetTxPowerSpectralDensity(rbIndexVector);
    NS_ASSERT(txPsd);
 
    // In case of UNIFORM_POWER_ALLOCATION_USED, the txPsd created by GetTxPowerSpectralDensity
    // assumed that the transmit power would be split only among RBs allocated to this signal/UE.
    // This assumption is false when there are concurrent transmissions on other RBs to other UEs
    // (OFDMA DL). To correct this, use the combined number of used RBs to scale down txPsd.
    if (GetPowerAllocationType() == NrSpectrumValueHelper::UNIFORM_POWER_ALLOCATION_USED)
    {
        auto scaling = double(rbIndexVector.size()) / double(nTotalAllocRbs);
        for (auto it = txPsd->ValuesBegin(); it != txPsd->ValuesEnd(); it++)
        {
            *it *= scaling;
        }
    }
    else
    {
        // UNIFORM_POWER_ALLOCATION_BW: no scaling required
    }
 
    m_spectrumPhy->SetTxPowerSpectralDensity(txPsd);
}
 
void
NrGnbPhy::QueueMib()
{
    NS_LOG_FUNCTION(this);
    NrRrcSap::MasterInformationBlock mib;
    mib.numerology = GetNumerology();
    mib.dlBandwidth = GetChannelBandwidth() / (1000 * 100);
    mib.systemFrameNumber = 1;
    Ptr<NrMibMessage> mibMsg = Create<NrMibMessage>();
    mibMsg->SetSourceBwp(GetBwpId());
    mibMsg->SetMib(mib);
    EnqueueCtrlMsgNow(mibMsg);
}
 
void
NrGnbPhy::QueueSib()
{
    NS_LOG_FUNCTION(this);
    Ptr<NrSib1Message> msg = Create<NrSib1Message>();
    msg->SetSib1(m_sib1);
    msg->SetSourceBwp(GetBwpId());
    EnqueueCtrlMsgNow(msg);
}
 
void
NrGnbPhy::CallMacForSlotIndication(const SfnSf& currentSlot)
{
    NS_LOG_FUNCTION(this);
    NS_ASSERT(!m_generateDl.empty() || !m_generateUl.empty());
 
    m_phySapUser->SetCurrentSfn(currentSlot);
 
    uint64_t currentSlotN = currentSlot.Normalize() % m_tddPattern.size();
 
    NS_LOG_DEBUG("Start Slot " << currentSlot << ". In position " << currentSlotN
                               << " there is a slot of type " << m_tddPattern[currentSlotN]);
 
    for (const auto& k2WithLatency : m_generateUl[currentSlotN])
    {
        SfnSf targetSlot = currentSlot;
        targetSlot.Add(k2WithLatency);
 
        uint64_t pos = targetSlot.Normalize() % m_tddPattern.size();
 
        NS_LOG_DEBUG(" in slot " << currentSlot << " generate UL for " << targetSlot
                                 << " which is of type " << m_tddPattern[pos]);
 
        m_phySapUser->SlotUlIndication(targetSlot, m_tddPattern[pos]);
    }
 
    for (const auto& k0WithLatency : m_generateDl[currentSlotN])
    {
        SfnSf targetSlot = currentSlot;
        targetSlot.Add(k0WithLatency);
 
        uint64_t pos = targetSlot.Normalize() % m_tddPattern.size();
 
        NS_LOG_DEBUG(" in slot " << currentSlot << " generate DL for " << targetSlot
                                 << " which is of type " << m_tddPattern[pos]);
 
        m_phySapUser->SlotDlIndication(targetSlot, m_tddPattern[pos]);
    }
}
 
void
NrGnbPhy::StartSlot(const SfnSf& startSlot)
{
    NS_LOG_FUNCTION(this);
    NS_ASSERT(m_channelStatus != TO_LOSE);
 
    m_currentSlot = startSlot;
    m_lastSlotStart = Simulator::Now();
 
    Simulator::Schedule(GetSlotPeriod(), &NrGnbPhy::EndSlot, this);
 
    // update the current slot allocation; if empty (e.g., at the beginning of simu)
    // then insert a dummy allocation, without anything.
    if (SlotAllocInfoExists(m_currentSlot))
    {
        m_currSlotAllocInfo = RetrieveSlotAllocInfo(m_currentSlot);
    }
    else
    {
        NS_LOG_WARN("No allocation for the current slot. Using an empty one");
        m_currSlotAllocInfo = SlotAllocInfo(m_currentSlot);
    }
 
    if (m_isPrimary)
    {
        if (m_currentSlot.GetSlot() == 0)
        {
            bool mibOrSib = false;
            if (m_currentSlot.GetSubframe() == 0) // send MIB at the beginning of each frame
            {
                QueueMib();
                mibOrSib = true;
            }
            else if (m_currentSlot.GetSubframe() == 5) // send SIB at beginning of second half-frame
            {
                QueueSib();
                mibOrSib = true;
            }
            if (mibOrSib && !m_currSlotAllocInfo.ContainsDlCtrlAllocation())
            {
                VarTtiAllocInfo dlCtrlSlot(m_phySapUser->GetDlCtrlDci());
                m_currSlotAllocInfo.m_varTtiAllocInfo.push_front(dlCtrlSlot);
                m_currSlotAllocInfo.m_numSymAlloc += m_phySapUser->GetDlCtrlSymbols();
            }
        }
    }
 
    if (m_channelStatus == GRANTED)
    {
        NS_LOG_INFO("Channel granted");
        CallMacForSlotIndication(m_currentSlot);
        DoStartSlot();
    }
    else
    {
        bool hasUlDci = false;
        SfnSf ulSfn = m_currentSlot;
        ulSfn.Add(GetN2Delay());
 
        if (GetN2Delay() > 0)
        {
            if (SlotAllocInfoExists(ulSfn))
            {
                SlotAllocInfo& ulSlot = PeekSlotAllocInfo(ulSfn);
                hasUlDci = ulSlot.ContainsDataAllocation() || ulSlot.ContainsUlCtrlAllocation() ||
                           ulSlot.ContainsUlMsg3Allocation();
            }
        }
        // If there is a DL CTRL, try to obtain the channel to transmit it;
        // because, even if right now there isn't any message, maybe they
        // will come from another BWP.
        if (m_currSlotAllocInfo.ContainsDataAllocation() ||
            m_currSlotAllocInfo.ContainsDlCtrlAllocation() ||
            m_currSlotAllocInfo.ContainsUlMsg3Allocation() || hasUlDci)
        {
            // Request the channel access
            if (m_channelStatus == NONE)
            {
                NS_LOG_INFO("Channel not granted, request the channel");
                m_channelStatus = REQUESTED; // This goes always before RequestAccess()
                m_cam->RequestAccess();
                if (m_channelStatus == GRANTED)
                {
                    // Repetition but we can have a CAM that gives the channel
                    // instantaneously
                    NS_LOG_INFO("Channel granted; asking MAC for SlotIndication for the future and "
                                "then start the slot");
                    CallMacForSlotIndication(m_currentSlot);
                    DoStartSlot();
                    return; // Exit without calling anything else
                }
            }
            // If the channel was not granted, queue back the allocation,
            // without calling the MAC for a new slot
            auto slotAllocCopy = m_currSlotAllocInfo;
            auto newSfnSf = slotAllocCopy.m_sfnSf;
            newSfnSf.Add(1);
            NS_LOG_INFO("Queueing allocation in front for " << newSfnSf);
            if (m_currSlotAllocInfo.ContainsDataAllocation())
            {
                NS_LOG_INFO("Reason: Current slot allocation has data");
            }
            else
            {
                NS_LOG_INFO("Reason: CTRL message list is not empty");
            }
 
            PushFrontSlotAllocInfo(newSfnSf, slotAllocCopy);
        }
        else
        {
            // It's an empty slot; ask the MAC for a new one (maybe a new data will arrive..)
            // and just let the current one go away
            NS_LOG_INFO("Empty slot, but asking MAC for SlotIndication for the future, maybe there "
                        "will be data");
            CallMacForSlotIndication(m_currentSlot);
        }
        // If we have the UL CTRL, then schedule it (we are listening, so
        // we don't need the channel.
 
        if (!m_currSlotAllocInfo.m_varTtiAllocInfo.empty())
        {
            for (const auto& alloc : m_currSlotAllocInfo.m_varTtiAllocInfo)
            {
                if (alloc.m_dci->m_type == DciInfoElementTdma::CTRL &&
                    alloc.m_dci->m_format == DciInfoElementTdma::UL)
                {
                    Time start = GetSymbolPeriod() * alloc.m_dci->m_symStart;
                    NS_LOG_INFO("Schedule UL CTRL at " << start);
                    Simulator::Schedule(start, &NrGnbPhy::UlCtrl, this, alloc.m_dci);
                }
                else if (alloc.m_dci->m_type == DciInfoElementTdma::SRS &&
                         alloc.m_dci->m_format == DciInfoElementTdma::UL)
                {
                    Time start = GetSymbolPeriod() * alloc.m_dci->m_symStart;
                    NS_LOG_INFO("Schedule UL SRS at " << start);
                    Simulator::Schedule(start, &NrGnbPhy::UlSrs, this, alloc.m_dci);
                }
            }
        }
    }
}
 
void
NrGnbPhy::DoCheckOrReleaseChannel()
{
    NS_LOG_FUNCTION(this);
 
    NS_ASSERT(m_channelStatus == GRANTED);
    // The channel is granted, we have to check if we maintain it for the next
    // slot or we have to release it.
 
    // Assuming the scheduler assign contiguous symbol
    uint8_t lastDlSymbol = 0;
    for (auto& dci : m_currSlotAllocInfo.m_varTtiAllocInfo)
    {
        if (dci.m_dci->m_type == DciInfoElementTdma::DATA &&
            dci.m_dci->m_format == DciInfoElementTdma::DL)
        {
            lastDlSymbol =
                std::max(lastDlSymbol,
                         static_cast<uint8_t>(dci.m_dci->m_symStart + dci.m_dci->m_numSym));
        }
    }
 
    Time lastDataTime = GetSymbolPeriod() * lastDlSymbol;
 
    if (GetSlotPeriod() - lastDataTime > MicroSeconds(25))
    {
        NS_LOG_LOGIC("Last symbol of data: " << +lastDlSymbol
                                             << ", to the end of slot we still have "
                                             << (GetSlotPeriod() - lastDataTime).GetMicroSeconds()
                                             << " us, so we're going to lose the channel");
        m_channelStatus = TO_LOSE;
    }
    else
    {
        NS_LOG_LOGIC("Last symbol of data: " << +lastDlSymbol
                                             << ", to the end of slot we still have "
                                             << (GetSlotPeriod() - lastDataTime).GetMicroSeconds()
                                             << " us, so we're NOT going to lose the channel");
    }
}
 
void
NrGnbPhy::RetrievePrepareEncodeCtrlMsgs()
{
    NS_LOG_FUNCTION(this);
    auto ctrlMsgs = PopCurrentSlotCtrlMsgs();
    ctrlMsgs.sort();
    ctrlMsgs.merge(RetrieveMsgsFromDCIs(m_currentSlot));
 
    if (m_netDevice != nullptr)
    {
        DynamicCast<NrGnbNetDevice>(m_netDevice)->RouteOutgoingCtrlMsgs(ctrlMsgs, GetBwpId());
    }
    else
    {
        // No netDevice (that could happen in tests) so just redirect them to us
        for (const auto& msg : ctrlMsgs)
        {
            EncodeCtrlMsg(msg);
        }
    }
}
 
void
NrGnbPhy::GenerateAllocationStatistics(const SlotAllocInfo& allocInfo) const
{
    NS_LOG_FUNCTION(this);
    std::unordered_set<uint16_t> activeUe;
    uint32_t availRb = GetRbNum();
    uint32_t dataReg = 0;
    uint32_t ctrlReg = 0;
    uint32_t dataSym = 0;
    uint32_t ctrlSym = 0;
 
    int lastSymStart = -1;
    uint32_t symUsed = 0;
 
    for (const auto& allocation : allocInfo.m_varTtiAllocInfo)
    {
        uint32_t rbg = std::count(allocation.m_dci->m_rbgBitmask.begin(),
                                  allocation.m_dci->m_rbgBitmask.end(),
                                  1);
 
        // First: Store the RNTI of the UE in the active list
        if (allocation.m_dci->m_rnti != 0)
        {
            activeUe.insert(allocation.m_dci->m_rnti);
        }
 
        NS_ASSERT(lastSymStart <= allocation.m_dci->m_symStart);
 
        auto rbgUsed = (rbg * GetNumRbPerRbg()) * allocation.m_dci->m_numSym;
        if (allocation.m_dci->m_type == DciInfoElementTdma::DATA ||
            allocation.m_dci->m_type == DciInfoElementTdma::MSG3)
        {
            dataReg += rbgUsed;
        }
        else
        {
            ctrlReg += rbgUsed;
        }
 
        if (lastSymStart != allocation.m_dci->m_symStart)
        {
            symUsed += allocation.m_dci->m_numSym;
 
            if (allocation.m_dci->m_type == DciInfoElementTdma::DATA ||
                allocation.m_dci->m_type == DciInfoElementTdma::MSG3)
            {
                dataSym += allocation.m_dci->m_numSym;
            }
            else
            {
                ctrlSym += allocation.m_dci->m_numSym;
            }
        }
 
        lastSymStart = allocation.m_dci->m_symStart;
    }
 
    NS_ASSERT_MSG(symUsed == allocInfo.m_numSymAlloc,
                  "Allocated " << +allocInfo.m_numSymAlloc << " but only " << symUsed
                               << " written in stats");
 
    m_phySlotDataStats(allocInfo.m_sfnSf,
                       activeUe.size(),
                       dataReg,
                       dataSym,
                       availRb,
                       GetSymbolsPerSlot() - ctrlSym,
                       GetBwpId(),
                       GetCellId());
    m_phySlotCtrlStats(allocInfo.m_sfnSf,
                       activeUe.size(),
                       ctrlReg,
                       ctrlSym,
                       availRb,
                       GetSymbolsPerSlot() - dataSym,
                       GetBwpId(),
                       GetCellId());
}
 
void
NrGnbPhy::DoStartSlot()
{
    NS_LOG_FUNCTION(this);
    NS_ASSERT(m_ctrlMsgs.empty()); // This assert has to be re-evaluated for NR-U.
                                   // We can have messages before we weren't able to tx them before.
 
    uint64_t currentSlotN = m_currentSlot.Normalize() % m_tddPattern.size();
 
    NS_LOG_DEBUG("Start Slot " << m_currentSlot << " of type " << m_tddPattern[currentSlotN]);
 
    GenerateAllocationStatistics(m_currSlotAllocInfo);
 
    if (m_currSlotAllocInfo.m_varTtiAllocInfo.empty())
    {
        return;
    }
 
    NS_LOG_DEBUG("Allocations of the current slot: " << std::endl << m_currSlotAllocInfo);
 
    DoCheckOrReleaseChannel();
 
    RetrievePrepareEncodeCtrlMsgs();
 
    PrepareRbgAllocationMap(m_currSlotAllocInfo.m_varTtiAllocInfo);
 
    FillTheEvent();
}
 
void
NrGnbPhy::PrepareRbgAllocationMap(const std::deque<VarTtiAllocInfo>& allocations)
{
    NS_LOG_FUNCTION(this);
 
    // Start with a clean RBG allocation bitmask
    m_rbgAllocationPerSym.clear();
 
    // Create RBG map to know where to put power in DL
    for (const auto& allocation : allocations)
    {
        if (allocation.m_dci->m_type != DciInfoElementTdma::CTRL)
        {
            if (allocation.m_dci->m_format == DciInfoElementTdma::DL)
            {
                // In m_rbgAllocationPerSym, store only the DL RBG set to 1:
                // these will used to put power
                StoreRBGAllocation(&m_rbgAllocationPerSym, allocation.m_dci);
            }
 
            // For statistics, store UL/DL allocations
            StoreRBGAllocation(&m_rbgAllocationPerSymDataStat, allocation.m_dci);
        }
    }
 
    for (const auto& s : m_rbgAllocationPerSymDataStat)
    {
        auto& rbgAllocation = s.second;
        m_rbStatistics(m_currentSlot,
                       s.first,
                       FromRBGBitmaskToRBAssignment(rbgAllocation),
                       GetBwpId(),
                       GetCellId());
    }
 
    m_rbgAllocationPerSymDataStat.clear();
}
 
void
NrGnbPhy::FillTheEvent()
{
    NS_LOG_FUNCTION(this);
 
    uint8_t lastSymStart = 0;
    for (const auto& allocation : m_currSlotAllocInfo.m_varTtiAllocInfo)
    {
        NS_ASSERT(lastSymStart <= allocation.m_dci->m_symStart);
 
        auto varTtiStart = GetSymbolPeriod() * allocation.m_dci->m_symStart;
        Simulator::Schedule(varTtiStart, &NrGnbPhy::StartVarTti, this, allocation.m_dci);
        lastSymStart = allocation.m_dci->m_symStart;
 
        NS_LOG_INFO("Scheduled allocation " << *(allocation.m_dci) << " at " << varTtiStart);
    }
 
    m_currSlotAllocInfo.m_varTtiAllocInfo.clear();
}
 
void
NrGnbPhy::StoreRBGAllocation(std::unordered_map<uint8_t, std::vector<bool>>* map,
                             const std::shared_ptr<DciInfoElementTdma>& dci) const
{
    NS_LOG_FUNCTION(this);
 
    auto itAlloc = map->find(dci->m_symStart);
    if (itAlloc == map->end())
    {
        itAlloc = map->insert(std::make_pair(dci->m_symStart, dci->m_rbgBitmask)).first;
    }
    else
    {
        auto& existingRBGBitmask = itAlloc->second;
        NS_ASSERT(existingRBGBitmask.size() == dci->m_rbgBitmask.size());
        for (uint32_t i = 0; i < existingRBGBitmask.size(); ++i)
        {
            existingRBGBitmask.at(i) = existingRBGBitmask.at(i) || dci->m_rbgBitmask.at(i);
        }
    }
}
 
std::list<Ptr<NrControlMessage>>
NrGnbPhy::RetrieveDciFromAllocation(const SlotAllocInfo& alloc,
                                    const DciInfoElementTdma::DciFormat& format,
                                    uint32_t kDelay,
                                    uint32_t k1Delay)
{
    NS_LOG_FUNCTION(this);
    std::list<Ptr<NrControlMessage>> ctrlMsgs;
 
    if (!alloc.m_buildRarList.empty())
    {
        Ptr<NrRarMessage> ulMsg3DciMsg = Create<NrRarMessage>();
        for (const auto& rarIt : alloc.m_buildRarList)
        {
            NrRarMessage::Rar rar{};
            // RA preamble and RNTI should be set before by MAC/scheduler
            NS_ASSERT(rarIt.raPreambleId != 255);
            rar.rarPayload = rarIt;
            rar.rarPayload.k2Delay = kDelay;
            ulMsg3DciMsg->AddRar(rar);
 
            NS_LOG_INFO("In slot " << m_currentSlot << " PHY retrieves the RAR message for RNTI "
                                   << rar.rarPayload.ulMsg3Dci->m_rnti << " RA preamble Id "
                                   << +rar.rarPayload.raPreambleId << " at:" << Simulator::Now()
                                   << " for slot:" << alloc.m_sfnSf << " kDelay:" << kDelay
                                   << "k1Delay:" << k1Delay);
            ulMsg3DciMsg->SetSourceBwp(GetBwpId());
        }
        if (kDelay != 0)
        {
            ctrlMsgs.push_back(ulMsg3DciMsg);
        }
    }
 
    for (const auto& dlAlloc : alloc.m_varTtiAllocInfo)
    {
        if (dlAlloc.m_dci->m_type != DciInfoElementTdma::CTRL &&
            dlAlloc.m_dci->m_type != DciInfoElementTdma::MSG3 // we are sending MSG3 grant via RAR
                                                              // message, we cannot also send UL DCI
            && dlAlloc.m_dci->m_format == format)
        {
            auto& dciElem = dlAlloc.m_dci;
            NS_ASSERT(dciElem->m_format == format);
            NS_ASSERT_MSG(dciElem->m_symStart + dciElem->m_numSym <= GetSymbolsPerSlot(),
                          "symStart: " << static_cast<uint32_t>(dciElem->m_symStart)
                                       << " numSym: " << static_cast<uint32_t>(dciElem->m_numSym)
                                       << " symPerSlot: "
                                       << static_cast<uint32_t>(GetSymbolsPerSlot()));
 
            NS_LOG_INFO("Send DCI to RNTI " << dciElem->m_rnti << " from sym "
                                            << +dciElem->m_symStart << " to "
                                            << +dciElem->m_symStart + dciElem->m_numSym);
 
            Ptr<NrControlMessage> msg;
 
            if (dciElem->m_format == DciInfoElementTdma::DL)
            {
                Ptr<NrDlDciMessage> dciMsg = Create<NrDlDciMessage>(dciElem);
 
                dciMsg->SetSourceBwp(GetBwpId());
                dciMsg->SetKDelay(kDelay);
                dciMsg->SetK1Delay(k1Delay);
                msg = dciMsg;
            }
            else
            {
                Ptr<NrUlDciMessage> dciMsg = Create<NrUlDciMessage>(dciElem);
 
                dciMsg->SetSourceBwp(GetBwpId());
                dciMsg->SetKDelay(kDelay);
                msg = dciMsg;
            }
 
            ctrlMsgs.push_back(msg);
        }
    }
    ctrlMsgs.sort();
    return ctrlMsgs;
}
 
std::list<Ptr<NrControlMessage>>
NrGnbPhy::RetrieveMsgsFromDCIs(const SfnSf& currentSlot)
{
    std::list<Ptr<NrControlMessage>> ctrlMsgs;
    uint64_t currentSlotN = currentSlot.Normalize() % m_tddPattern.size();
 
    uint32_t k1delay = m_dlHarqfbPosition[currentSlotN];
 
    // TODO: copy paste :(
    for (const auto& k0delay : m_toSendDl[currentSlotN])
    {
        SfnSf targetSlot = currentSlot;
 
        targetSlot.Add(k0delay);
 
        if (targetSlot == currentSlot)
        {
            NS_LOG_DEBUG(" in slot " << currentSlot << " send DL DCI for the same slot");
 
            ctrlMsgs.merge(RetrieveDciFromAllocation(m_currSlotAllocInfo,
                                                     DciInfoElementTdma::DL,
                                                     k0delay,
                                                     k1delay));
        }
        else if (SlotAllocInfoExists(targetSlot))
        {
            NS_LOG_DEBUG(" in slot " << currentSlot << " send DL DCI for " << targetSlot);
 
            ctrlMsgs.merge(RetrieveDciFromAllocation(PeekSlotAllocInfo(targetSlot),
                                                     DciInfoElementTdma::DL,
                                                     k0delay,
                                                     k1delay));
        }
        else
        {
            NS_LOG_DEBUG("No allocation found for slot " << targetSlot);
        }
    }
 
    for (const auto& k2delay : m_toSendUl[currentSlotN])
    {
        SfnSf targetSlot = currentSlot;
 
        targetSlot.Add(k2delay);
 
        if (targetSlot == currentSlot)
        {
            NS_LOG_DEBUG(" in slot " << currentSlot << " send UL DCI for the same slot");
 
            ctrlMsgs.merge(RetrieveDciFromAllocation(m_currSlotAllocInfo,
                                                     DciInfoElementTdma::UL,
                                                     k2delay,
                                                     k1delay));
        }
        else if (SlotAllocInfoExists(targetSlot))
        {
            NS_LOG_DEBUG(" in slot " << currentSlot << " send UL DCI for " << targetSlot);
 
            ctrlMsgs.merge(RetrieveDciFromAllocation(PeekSlotAllocInfo(targetSlot),
                                                     DciInfoElementTdma::UL,
                                                     k2delay,
                                                     k1delay));
        }
        else
        {
            NS_LOG_DEBUG("No allocation found for slot " << targetSlot);
        }
    }
    ctrlMsgs.sort();
    return ctrlMsgs;
}
 
Time
NrGnbPhy::DlCtrl(const std::shared_ptr<DciInfoElementTdma>& dci)
{
    NS_LOG_FUNCTION(this);
 
    NS_LOG_DEBUG("Starting DL CTRL TTI at symbol " << +m_currSymStart << " to "
                                                   << +m_currSymStart + dci->m_numSym);
 
    // TX control period
    Time varTtiPeriod = GetSymbolPeriod() * dci->m_numSym;
 
    bool transmitCsiRs = false;
    if (m_enableCsiRs)
    {
        // Check whether it is time to transmit CSI-RS
        uint16_t currentCsiRsOffset = m_currentSlot.Normalize() % m_csiRsPeriodicity;
        if (TimeToTransmitCsiRs(currentCsiRsOffset))
        {
            varTtiPeriod = ScheduleCsiRs(varTtiPeriod, currentCsiRsOffset);
            transmitCsiRs = true;
        }
    }
 
    // The function that is filling m_ctrlMsgs is NrPhy::encodeCtrlMsgs
    if (!m_ctrlMsgs.empty() || transmitCsiRs)
    {
        NS_LOG_DEBUG("gNB TXing DL CTRL with "
                     << m_ctrlMsgs.size() << " msgs, frame " << m_currentSlot << " symbols "
                     << static_cast<uint32_t>(dci->m_symStart) << "-"
                     << static_cast<uint32_t>(dci->m_symStart + dci->m_numSym - 1) << " start "
                     << Simulator::Now() << " end "
                     << Simulator::Now() + varTtiPeriod - NanoSeconds(1.0));
        for (auto& m_ctrlMsg : m_ctrlMsgs)
        {
            Ptr<NrControlMessage> msg = m_ctrlMsg;
            m_phyTxedCtrlMsgsTrace(m_currentSlot, GetCellId(), dci->m_rnti, GetBwpId(), msg);
        }
 
        SendCtrlChannels(varTtiPeriod -
                         NanoSeconds(1.0)); // -1 ns ensures control ends before data period
    }
    else
    {
        NS_LOG_DEBUG("No messages to send, skipping");
    }
 
    return varTtiPeriod;
}
 
bool
NrGnbPhy::TimeToTransmitCsiRs(uint16_t currentOffset) const
{
    if (!m_csiRsOffsetToUes.contains(currentOffset))
    {
        return false;
    }
    else
    {
        return !m_csiRsOffsetToUes.at(currentOffset).empty();
    }
}
 
void
NrGnbPhy::TransmitCsiRsPerUe(Ptr<NrUeNetDevice> ueDev)
{
    NS_LOG_FUNCTION(this);
    ChangeBeamformingVector(ueDev);
    uint64_t rnti = (DynamicCast<NrUePhy>(ueDev->GetPhy(GetBwpId())))->GetRnti();
 
    NS_LOG_DEBUG("Transmitting CSI-RS towards UE with IMSI : " << ueDev->GetImsi() << " at slot:"
                                                               << +m_currentSlot.Normalize());
    m_spectrumPhy->StartTxCsiRs(rnti, 0);
}
 
Time
NrGnbPhy::ScheduleCsiRs(Time ctrlVarTti, uint16_t currentOffset)
{
    NS_ASSERT_MSG(!m_spectrumPhy->IsTransmitting(),
                  "Should have finished transmission of CTRL already.");
 
    if (m_csiRsModel == CSI_RS_PER_UE)
    {
        // CSI-RS is the duration of 1 nanosecond plus we want a
        // 1 nanosecond pause between the independent CSI-RS transmissions
        ctrlVarTti -= m_deviceMap.size() * NanoSeconds(2);
 
        uint16_t ueCounter = 0;
        for (auto& i : m_csiRsOffsetToUes.at(currentOffset))
        {
            Ptr<NrUeNetDevice> ueDev = DynamicCast<NrUeNetDevice>(i);
            Simulator::Schedule(ctrlVarTti + NanoSeconds(2.0) * ueCounter,
                                &NrGnbPhy::TransmitCsiRsPerUe,
                                this,
                                ueDev);
 
            ueCounter++;
        }
    }
    return ctrlVarTti;
}
 
Time
NrGnbPhy::UlCtrl(const std::shared_ptr<DciInfoElementTdma>& dci)
{
    NS_LOG_FUNCTION(this);
 
    NS_LOG_DEBUG("Starting UL CTRL TTI at symbol " << +m_currSymStart << " to "
                                                   << +m_currSymStart + dci->m_numSym);
 
    Time varTtiPeriod = GetSymbolPeriod() * dci->m_numSym;
 
    NS_LOG_DEBUG("gNB RXng UL CTRL frame "
                 << m_currentSlot << " symbols " << static_cast<uint32_t>(dci->m_symStart) << "-"
                 << static_cast<uint32_t>(dci->m_symStart + dci->m_numSym - 1) << " start "
                 << Simulator::Now() << " end " << Simulator::Now() + varTtiPeriod);
    return varTtiPeriod;
}
 
Time
NrGnbPhy::DlData(const std::shared_ptr<DciInfoElementTdma>& dci)
{
    NS_LOG_FUNCTION(this);
    NS_LOG_DEBUG("Starting DL DATA TTI at symbol " << +m_currSymStart << " to "
                                                   << +m_currSymStart + dci->m_numSym << " for "
                                                   << +dci->m_rnti);
 
    Time varTtiPeriod = GetSymbolPeriod() * dci->m_numSym;
 
    Ptr<PacketBurst> pktBurst = GetPacketBurst(m_currentSlot, dci->m_symStart, dci->m_rnti);
 
    if (!pktBurst || pktBurst->GetNPackets() == 0)
    {
        // sometimes the UE will be scheduled when no data is queued.
        // In this case, don't send anything, don't put power... do nothing!
        return varTtiPeriod;
    }
 
    NS_LOG_INFO("gNB TXing DL DATA frame "
                << m_currentSlot << " symbols " << static_cast<uint32_t>(dci->m_symStart) << "-"
                << static_cast<uint32_t>(dci->m_symStart + dci->m_numSym - 1) << " start "
                << Simulator::Now() + NanoSeconds(1) << " end "
                << Simulator::Now() + varTtiPeriod - NanoSeconds(2.0));
 
    Simulator::Schedule(NanoSeconds(1.0),
                        &NrGnbPhy::SendDataChannels,
                        this,
                        pktBurst,
                        varTtiPeriod - NanoSeconds(2.0),
                        dci);
 
    return varTtiPeriod;
}
 
Time
NrGnbPhy::UlData(const std::shared_ptr<DciInfoElementTdma>& dci)
{
    NS_LOG_FUNCTION(this);
 
    NS_LOG_DEBUG("Starting UL DATA TTI at symbol " << +m_currSymStart << " to "
                                                   << +m_currSymStart + dci->m_numSym);
 
    Time varTtiPeriod = GetSymbolPeriod() * dci->m_numSym;
 
    m_spectrumPhy->AddExpectedTb({dci->m_ndi,
                                  dci->m_tbSize,
                                  dci->m_mcs,
                                  dci->m_rank,
                                  dci->m_rnti,
                                  FromRBGBitmaskToRBAssignment(dci->m_rbgBitmask),
                                  dci->m_harqProcess,
                                  dci->m_rv,
                                  false,
                                  dci->m_symStart,
                                  dci->m_numSym,
                                  m_currentSlot});
 
    bool found = false;
    for (auto& i : m_deviceMap)
    {
        Ptr<NrUeNetDevice> ueDev = DynamicCast<NrUeNetDevice>(i);
        uint64_t ueRnti = (DynamicCast<NrUePhy>(ueDev->GetPhy(GetBwpId())))->GetRnti();
        if (dci->m_rnti == ueRnti)
        {
            // Even if we change the beamforming vector, we hope that the scheduler
            // has scheduled UEs within the same beam (and, therefore, have the same
            // beamforming vector)
            // Beamforming vector should be available only when the node has a UPA antenna device
            if (DynamicCast<UniformPlanarArray>(m_spectrumPhy->GetAntenna()))
            {
                ChangeBeamformingVector(i); // assume the control signal is omni
            }
            found = true;
            break;
        }
    }
    // In case UE was not attached via NrHelper::AttachToGnb(),
    // assume quasi omni beamforming until we have the opportunity to scan for a beam
    if (!found)
    {
        ChangeBeamformingVector(nullptr);
    }
 
    NS_LOG_INFO("GNB RXing UL DATA frame "
                << m_currentSlot << " symbols " << static_cast<uint32_t>(dci->m_symStart) << "-"
                << static_cast<uint32_t>(dci->m_symStart + dci->m_numSym - 1) << " start "
                << Simulator::Now() << " end " << Simulator::Now() + varTtiPeriod);
    return varTtiPeriod;
}
 
void
NrGnbPhy::ChangeBeamformingVector(Ptr<NrNetDevice> dev)
{
    m_spectrumPhy->GetBeamManager()->ChangeBeamformingVector(dev);
}
 
void
NrGnbPhy::ChangeToQuasiOmniBeamformingVector()
{
    m_spectrumPhy->GetBeamManager()->ChangeToQuasiOmniBeamformingVector();
}
 
Time
NrGnbPhy::UlSrs(const std::shared_ptr<DciInfoElementTdma>& dci)
{
    NS_LOG_FUNCTION(this);
 
    NS_LOG_DEBUG("Starting UL SRS TTI at symbol " << +m_currSymStart << " to "
                                                  << +m_currSymStart + dci->m_numSym);
 
    Time varTtiPeriod = GetSymbolPeriod() * dci->m_numSym;
 
    m_spectrumPhy->AddExpectedSrsRnti(dci->m_rnti);
 
    bool found = false;
 
    // if yes, and the rnti for the current SRS is not found in the list,
    // the code will not abort
    for (auto& i : m_deviceMap)
    {
        Ptr<NrUeNetDevice> ueDev = DynamicCast<NrUeNetDevice>(i);
        uint64_t ueRnti = (DynamicCast<NrUePhy>(ueDev->GetPhy(0)))->GetRnti();
        if (dci->m_rnti == ueRnti)
        {
            // Even if we change the beamforming vector, we hope that the scheduler
            // has scheduled UEs within the same beam (and, therefore, have the same
            // beamforming vector)
            // Beamforming vector should be available only when the node has a UPA antenna device
            if (DynamicCast<UniformPlanarArray>(m_spectrumPhy->GetAntenna()))
            {
                ChangeBeamformingVector(i); // assume the control signal is omni
            }
            found = true;
            break;
        }
    }
 
    // In case UE was not attached via NrHelper::AttachToGnb(),
    // assume quasi omni beamforming until we have the opportunity to scan for a beam
    if (!found)
    {
        ChangeBeamformingVector(nullptr);
        NS_LOG_WARN("The UE for which is scheduled this SRS does not have yet initialized RNTI. "
                    "RAR message was not received yet.");
    }
 
    NS_LOG_INFO("GNB RXing UL SRS frame "
                << m_currentSlot << " symbols " << static_cast<uint32_t>(dci->m_symStart) << "-"
                << static_cast<uint32_t>(dci->m_symStart + dci->m_numSym - 1) << " start "
                << Simulator::Now() << " end " << Simulator::Now() + varTtiPeriod);
    return varTtiPeriod;
}
 
void
NrGnbPhy::StartVarTti(const std::shared_ptr<DciInfoElementTdma>& dci)
{
    NS_LOG_FUNCTION(this);
    if (DynamicCast<UniformPlanarArray>(m_spectrumPhy->GetAntenna()))
    {
        ChangeToQuasiOmniBeamformingVector(); // assume the control signal is omni
    }
    m_currSymStart = dci->m_symStart;
 
    Time varTtiPeriod;
 
    if (dci->m_type == DciInfoElementTdma::CTRL)
    {
        if (dci->m_format == DciInfoElementTdma::DL)
        {
            varTtiPeriod = DlCtrl(dci);
        }
        else if (dci->m_format == DciInfoElementTdma::UL)
        {
            varTtiPeriod = UlCtrl(dci);
        }
    }
    else if (dci->m_type == DciInfoElementTdma::DATA || dci->m_type == DciInfoElementTdma::MSG3)
    {
        if (dci->m_format == DciInfoElementTdma::DL)
        {
            varTtiPeriod = DlData(dci);
        }
        else if (dci->m_format == DciInfoElementTdma::UL)
        {
            varTtiPeriod = UlData(dci);
        }
    }
    else if (dci->m_type == DciInfoElementTdma::SRS)
    {
        NS_ASSERT(dci->m_format == DciInfoElementTdma::UL);
        varTtiPeriod = UlSrs(dci);
    }
 
    Simulator::Schedule(varTtiPeriod, &NrGnbPhy::EndVarTti, this, dci);
}
 
void
NrGnbPhy::EndVarTti(const std::shared_ptr<DciInfoElementTdma>& lastDci)
{
    NS_LOG_FUNCTION(this << Simulator::Now().GetSeconds());
 
    NS_LOG_DEBUG("DCI started at symbol "
                 << static_cast<uint32_t>(lastDci->m_symStart) << " which lasted for "
                 << static_cast<uint32_t>(lastDci->m_numSym) << " symbols finished");
}
 
void
NrGnbPhy::EndSlot()
{
    NS_LOG_FUNCTION(this);
 
    Time slotStart = m_lastSlotStart + GetSlotPeriod() - Simulator::Now();
 
    if (m_channelStatus == TO_LOSE)
    {
        NS_LOG_INFO("Release the channel because we did not have any data to maintain the grant");
        m_channelStatus = NONE;
        m_channelLostTimer.Cancel();
    }
 
    NS_LOG_DEBUG("Slot started at " << m_lastSlotStart << " ended");
 
    if (m_nrFhPhySapProvider)
    {
        NS_LOG_DEBUG("End slot notified from PHY"); // TODO: Add active UEs nad BWPs?
        m_nrFhPhySapProvider->NotifyEndSlot(GetBwpId(), m_currentSlot);
    }
 
    m_currentSlot.Add(1);
    Simulator::Schedule(slotStart, &NrGnbPhy::StartSlot, this, m_currentSlot);
}
 
void
NrGnbPhy::SendDataChannels(const Ptr<PacketBurst>& pb,
                           const Time& varTtiPeriod,
                           const std::shared_ptr<DciInfoElementTdma>& dci)
{
    NS_LOG_FUNCTION(this);
    // update beamforming vectors (currently supports 1 user only)
 
    // In each time instance, there can only be a single BF vector. Only update BF vectors once
    // unless time has changed
    if (Simulator::Now() > m_lastBfChange)
    {
        NS_ASSERT_MSG(!m_spectrumPhy->IsTransmitting(),
                      "Cannot change analog BF after TX has started");
        m_lastBfChange = Simulator::Now();
        bool found = false;
        for (auto& i : m_deviceMap)
        {
            Ptr<NrUeNetDevice> ueDev = DynamicCast<NrUeNetDevice>(i);
            uint64_t ueRnti = (DynamicCast<NrUePhy>(ueDev->GetPhy(GetBwpId())))->GetRnti();
            if (dci->m_rnti == ueRnti)
            {
                if (DynamicCast<UniformPlanarArray>(m_spectrumPhy->GetAntenna()))
                {
                    ChangeBeamformingVector(i);
                }
 
                found = true;
                break;
            }
        }
        // In case UE was not attached via NrHelper::AttachToGnb(),
        // assume quasi omni beamforming until we have the opportunity to scan for a beam
        if (!found)
        {
            ChangeBeamformingVector(nullptr);
        }
    }
 
    // in the map we stored the RBG allocated by the MAC for this symbol.
    // If the transmission last n symbol (n > 1 && n < 12) the SetSubChannels
    // doesn't need to be called again. In fact, SendDataChannels will be
    // invoked only when the symStart changes.
    NS_ASSERT(m_rbgAllocationPerSym.find(dci->m_symStart) != m_rbgAllocationPerSym.end());
    auto nTotalAllocRbs =
        FromRBGBitmaskToRBAssignment(m_rbgAllocationPerSym.at(dci->m_symStart)).size();
    SetSubChannels(FromRBGBitmaskToRBAssignment(dci->m_rbgBitmask), nTotalAllocRbs);
 
    std::list<Ptr<NrControlMessage>> ctrlMsgs;
    m_spectrumPhy->StartTxDataFrames(pb, ctrlMsgs, dci, varTtiPeriod);
}
 
void
NrGnbPhy::SendCtrlChannels(const Time& varTtiPeriod)
{
    NS_LOG_FUNCTION(this << "Send Ctrl");
 
    std::vector<int> fullBwRb(GetRbNum());
    // The first time set the right values for the phy
    for (uint32_t i = 0; i < fullBwRb.size(); ++i)
    {
        fullBwRb[i] = static_cast<int>(i);
    }
 
    // Transmit power for the current signal is distributed over the full bandwidth. This is the
    // only signal, so the bandwidth occupied by all concurrent transmissions is also the full
    // bandwidth.
    SetSubChannels(fullBwRb, fullBwRb.size());
 
    if (m_nrFhPhySapProvider &&
        m_nrFhPhySapProvider->GetFhControlMethod() == NrFhControl::FhControlMethod::Dropping)
    {
        std::vector<Ptr<NrControlMessage>> fhCtrlMsgs(m_ctrlMsgs.begin(), m_ctrlMsgs.end());
        auto rng = std::default_random_engine{};
        std::shuffle(std::begin(fhCtrlMsgs), std::end(fhCtrlMsgs), rng);
 
        for (auto ctrlIt = fhCtrlMsgs.begin(); ctrlIt != fhCtrlMsgs.end(); /* no incr */)
        {
            Ptr<NrControlMessage> msg = (*ctrlIt);
            if (msg->GetMessageType() == NrControlMessage::DL_DCI)
            {
                auto dciMsg = DynamicCast<NrDlDciMessage>(msg);
                auto dciInfoElem = dciMsg->GetDciInfoElement();
                long rbgAssigned = std::count(dciInfoElem->m_rbgBitmask.begin(),
                                              dciInfoElem->m_rbgBitmask.end(),
                                              1);
 
                if (DoesFhAllocationFit(GetBwpId(),
                                        dciInfoElem->m_mcs,
                                        rbgAssigned * dciInfoElem->m_numSym,
                                        dciInfoElem->m_rank) == 0)
                {
                    // drop DL DCI because data does not fit in available FH BW
                    ctrlIt = fhCtrlMsgs.erase(ctrlIt);
                    m_ctrlMsgs.remove(msg);
                }
                else
                {
                    ++ctrlIt;
                    m_nrFhPhySapProvider->UpdateTracesBasedOnDroppedData(GetBwpId(),
                                                                         dciInfoElem->m_mcs,
                                                                         rbgAssigned,
                                                                         dciInfoElem->m_numSym,
                                                                         dciInfoElem->m_rank);
                }
            }
            else
            {
                ++ctrlIt;
            }
        }
        if (!m_ctrlMsgs.empty())
        {
            m_spectrumPhy->StartTxDlControlFrames(m_ctrlMsgs, varTtiPeriod);
        }
        m_ctrlMsgs.clear();
    }
    else
    {
        m_spectrumPhy->StartTxDlControlFrames(m_ctrlMsgs, varTtiPeriod);
        m_ctrlMsgs.clear();
    }
}
 
void
NrGnbPhy::AssignCsiRsOffset(const Ptr<NrUeNetDevice>& ueDevice)
{
    NS_LOG_FUNCTION(this);
 
    if (m_csiRsOffsetToUes.empty())
    {
        NS_ABORT_MSG_UNLESS(m_csiRsPeriodicity % m_tddPattern.size() == 0,
                            "CSI-RS periodicity should be a multiply of TDD pattern size");
        // how many patterns falls into the CSI periodicity
        uint8_t repetitions = m_csiRsPeriodicity / m_tddPattern.size();
 
        for (uint8_t round = 0; round < repetitions; ++round)
        {
            // count available slots for the CSI-RS
            for (size_t index = 0; index < m_tddPattern.size(); index++)
            {
                if (m_tddPattern[index] != LteNrTddSlotType::UL)
                {
                    m_csiRsOffsetToUes[m_tddPattern.size() * round + index] =
                        std::set<Ptr<NrUeNetDevice>>();
                }
            }
        }
    }
 
    size_t lastAssignedOffset = m_csiRsOffsetToUes.begin()->second.size();
 
    // searching for the next available offset value
    for (auto& i : m_csiRsOffsetToUes)
    {
        if (i.second.size() < lastAssignedOffset)
        {
            i.second.emplace(ueDevice);
            NS_LOG_DEBUG("Assigning CSI-RS offset for UE with IMSI: " << ueDevice->GetImsi());
            return;
        }
        lastAssignedOffset = i.second.size();
    }
    // we are here because all the offset have the same number of users assigned so
    // the new user starts from the first offset value
    NS_LOG_DEBUG("Assigning CSI-RS offset for UE with IMSI: " << ueDevice->GetImsi());
    m_csiRsOffsetToUes.begin()->second.emplace(ueDevice);
}
 
bool
NrGnbPhy::RegisterUe(uint64_t imsi, const Ptr<NrUeNetDevice>& ueDevice)
{
    NS_LOG_FUNCTION(this << imsi);
    std::set<uint64_t>::iterator it;
    it = m_ueAttached.find(imsi);
 
    if (it == m_ueAttached.end())
    {
        m_ueAttached.insert(imsi);
        m_deviceMap.push_back(ueDevice);
 
        if (m_enableCsiRs && HasDlSlot(m_tddPattern))
        {
            AssignCsiRsOffset(ueDevice);
        }
        return (true);
    }
    else
    {
        NS_LOG_ERROR("Programming error...UE already attached");
        return (false);
    }
}
 
void
NrGnbPhy::PhyDataPacketReceived(const Ptr<Packet>& p)
{
    Simulator::ScheduleWithContext(m_netDevice->GetNode()->GetId(),
                                   GetTbDecodeLatency(),
                                   &NrGnbPhySapUser::ReceivePhyPdu,
                                   m_phySapUser,
                                   p);
}
 
void
NrGnbPhy::GenerateDataCqiReport(const SpectrumValue& sinr)
{
    NS_LOG_FUNCTION(this << sinr);
 
    Values::const_iterator it;
    NrMacSchedSapProvider::SchedUlCqiInfoReqParameters ulcqi;
    ulcqi.m_ulCqi.m_type = UlCqiInfo::PUSCH;
    for (it = sinr.ConstValuesBegin(); it != sinr.ConstValuesEnd(); it++)
    {
        //   double sinrdb = 10 * std::log10 ((*it));
        //       NS_LOG_INFO ("ULCQI RB " << i << " value " << sinrdb);
        // convert from double to fixed point notaltion Sxxxxxxxxxxx.xxx
        //   int16_t sinrFp = nr::FfConverter::double2fpS11dot3 (sinrdb);
        ulcqi.m_ulCqi.m_sinr.push_back(
            *it); // will be processed by NrMacSchedulerCQIManagement::UlSBCQIReported, it will
                  // look into a map of assignment
    }
 
    // here we use the start symbol index of the var tti in place of the var tti index because
    // the absolute UL var tti index is not known to the scheduler when m_allocationMap gets
    // populated
    ulcqi.m_sfnSf = m_currentSlot;
    ulcqi.m_symStart = m_currSymStart;
    SpectrumValue newSinr = sinr;
    m_ulSinrTrace(0, newSinr, newSinr);
    m_phySapUser->UlCqiReport(ulcqi);
}
 
void
NrGnbPhy::PhyCtrlMessagesReceived(const Ptr<NrControlMessage>& msg)
{
    NS_LOG_FUNCTION(this);
 
    if (msg->GetMessageType() == NrControlMessage::DL_CQI)
    {
        Ptr<NrDlCqiMessage> dlcqi = DynamicCast<NrDlCqiMessage>(msg);
        DlCqiInfo dlcqiLE = dlcqi->GetDlCqi();
        m_phyRxedCtrlMsgsTrace(m_currentSlot, GetCellId(), dlcqiLE.m_rnti, GetBwpId(), msg);
 
        NS_LOG_INFO("Received DL_CQI for RNTI: " << dlcqiLE.m_rnti << " in slot " << m_currentSlot);
 
        m_phySapUser->ReceiveControlMessage(msg);
    }
    else if (msg->GetMessageType() == NrControlMessage::RACH_PREAMBLE)
    {
        NS_LOG_INFO("received RACH_PREAMBLE");
 
        Ptr<NrRachPreambleMessage> rachPreamble = DynamicCast<NrRachPreambleMessage>(msg);
        m_phyRxedCtrlMsgsTrace(m_currentSlot, GetCellId(), 0, GetBwpId(), msg);
        NS_LOG_INFO("Received RACH Preamble in slot " << m_currentSlot);
        m_phySapUser->ReceiveRachPreamble(rachPreamble->GetRapId());
    }
    else if (msg->GetMessageType() == NrControlMessage::DL_HARQ)
    {
        Ptr<NrDlHarqFeedbackMessage> dlharqMsg = DynamicCast<NrDlHarqFeedbackMessage>(msg);
        DlHarqInfo dlharq = dlharqMsg->GetDlHarqFeedback();
        if (m_ueAttachedRnti.find(dlharq.m_rnti) != m_ueAttachedRnti.end())
        {
            m_phyRxedCtrlMsgsTrace(m_currentSlot, GetCellId(), dlharq.m_rnti, GetBwpId(), msg);
 
            NS_LOG_INFO("Received DL_HARQ for RNTI: " << dlharq.m_rnti << " in slot "
                                                      << m_currentSlot);
            m_phySapUser->ReceiveControlMessage(msg);
        }
    }
    else
    {
        m_phyRxedCtrlMsgsTrace(m_currentSlot, GetCellId(), 0, GetBwpId(), msg);
        m_phySapUser->ReceiveControlMessage(msg);
    }
}
 
 
void
NrGnbPhy::DoSetBandwidth(uint16_t ulBandwidth, uint16_t dlBandwidth)
{
    NS_LOG_FUNCTION(this << +ulBandwidth << +dlBandwidth);
    NS_ASSERT(ulBandwidth == dlBandwidth);
    SetChannelBandwidth(dlBandwidth);
}
 
void
NrGnbPhy::DoSetEarfcn(uint16_t ulEarfcn, uint16_t dlEarfcn)
{
    NS_LOG_FUNCTION(this << ulEarfcn << dlEarfcn);
}
 
void
NrGnbPhy::DoAddUe([[maybe_unused]] uint16_t rnti)
{
    NS_LOG_FUNCTION(this << rnti);
    std::set<uint16_t>::iterator it;
    it = m_ueAttachedRnti.find(rnti);
    if (it == m_ueAttachedRnti.end())
    {
        m_ueAttachedRnti.insert(rnti);
    }
}
 
void
NrGnbPhy::DoRemoveUe(uint16_t rnti)
{
    NS_LOG_FUNCTION(this << rnti);
 
    std::set<uint16_t>::iterator it = m_ueAttachedRnti.find(rnti);
    if (it != m_ueAttachedRnti.end())
    {
        m_ueAttachedRnti.erase(it);
    }
    else
    {
        NS_FATAL_ERROR("Impossible to remove UE, not attached!");
    }
}
 
void
NrGnbPhy::DoSetPa(uint16_t rnti, double pa)
{
    NS_LOG_FUNCTION(this << rnti << pa);
}
 
void
NrGnbPhy::DoSetTransmissionMode(uint16_t rnti, uint8_t txMode)
{
    NS_LOG_FUNCTION(this << rnti << +txMode);
    // UL supports only SISO MODE
}
 
void
NrGnbPhy::DoSetSrsConfigurationIndex(uint16_t rnti, uint16_t srcCi)
{
    NS_LOG_FUNCTION(this << rnti << srcCi);
}
 
void
NrGnbPhy::DoSetMasterInformationBlock([[maybe_unused]] NrRrcSap::MasterInformationBlock mib)
{
    NS_LOG_FUNCTION(this);
}
 
void
NrGnbPhy::DoSetSystemInformationBlockType1(NrRrcSap::SystemInformationBlockType1 sib1)
{
    NS_LOG_FUNCTION(this);
    m_sib1 = sib1;
}
 
int8_t
NrGnbPhy::DoGetReferenceSignalPower() const
{
    NS_LOG_FUNCTION(this);
    return static_cast<int8_t>(m_txPower);
}
 
void
NrGnbPhy::SetPhySapUser(NrGnbPhySapUser* ptr)
{
    m_phySapUser = ptr;
}
 
void
NrGnbPhy::ReportUlHarqFeedback(const UlHarqInfo& mes)
{
    NS_LOG_FUNCTION(this);
    // forward to scheduler
    if (m_ueAttachedRnti.find(mes.m_rnti) != m_ueAttachedRnti.end())
    {
        NS_LOG_INFO("Received UL HARQ feedback " << mes.IsReceivedOk()
                                                 << " and forwarding to the scheduler");
        m_phySapUser->UlHarqFeedback(mes);
    }
}
 
void
NrGnbPhy::SetPattern(const std::string& pattern)
{
    NS_LOG_FUNCTION(this);
 
    static std::unordered_map<std::string, LteNrTddSlotType> lookupTable = {
        {"DL", LteNrTddSlotType::DL},
        {"UL", LteNrTddSlotType::UL},
        {"S", LteNrTddSlotType::S},
        {"F", LteNrTddSlotType::F},
    };
 
    std::vector<LteNrTddSlotType> vector;
    std::stringstream ss(pattern);
    std::string token;
    std::vector<std::string> extracted;
 
    while (std::getline(ss, token, '|'))
    {
        extracted.push_back(token);
    }
 
    for (const auto& v : extracted)
    {
        if (lookupTable.find(v) == lookupTable.end())
        {
            NS_FATAL_ERROR("Pattern type " << v << " not valid. Valid values are: DL UL F S");
        }
        vector.push_back(lookupTable[v]);
    }
 
    SetTddPattern(vector);
}
 
std::string
NrGnbPhy::GetPattern() const
{
    return NrPhy::GetPattern(m_tddPattern);
}
 
void
NrGnbPhy::SetPrimary()
{
    NS_LOG_FUNCTION(this);
    m_isPrimary = true;
}
 
void
NrGnbPhy::SetCsiRsModel(enum NrGnbPhy::CsiRsModel csiRsModel)
{
    m_csiRsModel = csiRsModel;
}
 
enum NrGnbPhy::CsiRsModel
NrGnbPhy::GetCsiRsModel() const
{
    return m_csiRsModel;
}
 
void
NrGnbPhy::SetCsiRsPeriodicity(uint16_t csiRsPeriodicity)
{
    m_csiRsPeriodicity = csiRsPeriodicity;
}
 
uint16_t
NrGnbPhy::GetCsiRsPeriodicity() const
{
    return m_csiRsPeriodicity;
}
 
void
NrGnbPhy::ChannelAccessGranted(const Time& time)
{
    NS_LOG_FUNCTION(this);
 
    if (time < GetSlotPeriod())
    {
        NS_LOG_INFO("Channel granted for less than the slot time. Ignoring the grant.");
        m_channelStatus = NONE;
        return;
    }
 
    m_channelStatus = GRANTED;
 
    Time toNextSlot = m_lastSlotStart + GetSlotPeriod() - Simulator::Now();
    Time grant = time - toNextSlot;
    int64_t slotGranted = grant.GetNanoSeconds() / GetSlotPeriod().GetNanoSeconds();
 
    NS_LOG_INFO("Channel access granted for " << time << ", which corresponds to " << slotGranted
                                              << " slot in which each slot is " << GetSlotPeriod()
                                              << ". We lost " << toNextSlot);
    NS_ASSERT(!m_channelLostTimer.IsPending());
 
    if (slotGranted < 1)
    {
        slotGranted = 1;
    }
    m_channelLostTimer = Simulator::Schedule(GetSlotPeriod() * slotGranted - NanoSeconds(1),
                                             &NrGnbPhy::ChannelAccessLost,
                                             this);
}
 
void
NrGnbPhy::ChannelAccessLost()
{
    NS_LOG_FUNCTION(this);
    NS_LOG_INFO("Channel access lost");
    m_channelStatus = NONE;
}
 
} // namespace ns3