nr-mac-scheduler-ofdma-qos.cc

// Copyright (c) 2022 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#include "nr-mac-scheduler-ofdma-qos.h"
 
#include "nr-mac-scheduler-ue-info-qos.h"
 
#include "ns3/double.h"
#include "ns3/log.h"
 
#include <algorithm>
 
namespace ns3
{
 
NS_LOG_COMPONENT_DEFINE("NrMacSchedulerOfdmaQos");
NS_OBJECT_ENSURE_REGISTERED(NrMacSchedulerOfdmaQos);
 
TypeId
NrMacSchedulerOfdmaQos::GetTypeId()
{
    static TypeId tid =
        TypeId("ns3::NrMacSchedulerOfdmaQos")
            .SetParent<NrMacSchedulerOfdmaRR>()
            .AddConstructor<NrMacSchedulerOfdmaQos>()
            .AddAttribute("FairnessIndex",
                          "Value (between 0 and 1) that defines the PF metric (1 is the "
                          "traditional 3GPP PF, 0 is RR in throughput",
                          DoubleValue(1),
                          MakeDoubleAccessor(&NrMacSchedulerOfdmaQos::SetFairnessIndex,
                                             &NrMacSchedulerOfdmaQos::GetFairnessIndex),
                          MakeDoubleChecker<float>(0, 1))
            .AddAttribute(
                "LastAvgTPutWeight",
                "Weight of the last average throughput in the average throughput calculation",
                DoubleValue(99),
                MakeDoubleAccessor(&NrMacSchedulerOfdmaQos::SetTimeWindow,
                                   &NrMacSchedulerOfdmaQos::GetTimeWindow),
                MakeDoubleChecker<float>(0));
    return tid;
}
 
NrMacSchedulerOfdmaQos::NrMacSchedulerOfdmaQos()
    : NrMacSchedulerOfdmaRR()
{
}
 
void
NrMacSchedulerOfdmaQos::SetFairnessIndex(double v)
{
    NS_LOG_FUNCTION(this);
    m_alpha = v;
}
 
double
NrMacSchedulerOfdmaQos::GetFairnessIndex() const
{
    NS_LOG_FUNCTION(this);
    return m_alpha;
}
 
void
NrMacSchedulerOfdmaQos::SetTimeWindow(double v)
{
    NS_LOG_FUNCTION(this);
    m_timeWindow = v;
}
 
double
NrMacSchedulerOfdmaQos::GetTimeWindow() const
{
    NS_LOG_FUNCTION(this);
    return m_timeWindow;
}
 
std::shared_ptr<NrMacSchedulerUeInfo>
NrMacSchedulerOfdmaQos::CreateUeRepresentation(
    const NrMacCschedSapProvider::CschedUeConfigReqParameters& params) const
{
    NS_LOG_FUNCTION(this);
    return std::make_shared<NrMacSchedulerUeInfoQos>(
        m_alpha,
        params.m_rnti,
        params.m_beamId,
        std::bind(&NrMacSchedulerOfdmaQos::GetNumRbPerRbg, this));
}
 
std::function<bool(const NrMacSchedulerNs3::UePtrAndBufferReq& lhs,
                   const NrMacSchedulerNs3::UePtrAndBufferReq& rhs)>
NrMacSchedulerOfdmaQos::GetUeCompareDlFn() const
{
    return NrMacSchedulerUeInfoQos::CompareUeWeightsDl;
}
 
std::function<bool(const NrMacSchedulerNs3::UePtrAndBufferReq& lhs,
                   const NrMacSchedulerNs3::UePtrAndBufferReq& rhs)>
NrMacSchedulerOfdmaQos::GetUeCompareUlFn() const
{
    return NrMacSchedulerUeInfoQos::CompareUeWeightsUl;
}
 
void
NrMacSchedulerOfdmaQos::AssignedDlResources(const UePtrAndBufferReq& ue,
                                            [[maybe_unused]] const FTResources& assigned,
                                            const FTResources& totAssigned) const
{
    NS_LOG_FUNCTION(this);
    auto uePtr = std::dynamic_pointer_cast<NrMacSchedulerUeInfoQos>(ue.first);
    uePtr->UpdateDlQosMetric(totAssigned, m_timeWindow);
}
 
void
NrMacSchedulerOfdmaQos::NotAssignedDlResources(
    const NrMacSchedulerNs3::UePtrAndBufferReq& ue,
    [[maybe_unused]] const NrMacSchedulerNs3::FTResources& assigned,
    const NrMacSchedulerNs3::FTResources& totAssigned) const
{
    NS_LOG_FUNCTION(this);
    auto uePtr = std::dynamic_pointer_cast<NrMacSchedulerUeInfoQos>(ue.first);
    uePtr->UpdateDlQosMetric(totAssigned, m_timeWindow);
}
 
void
NrMacSchedulerOfdmaQos::AssignedUlResources(const UePtrAndBufferReq& ue,
                                            [[maybe_unused]] const FTResources& assigned,
                                            const FTResources& totAssigned) const
{
    NS_LOG_FUNCTION(this);
    auto uePtr = std::dynamic_pointer_cast<NrMacSchedulerUeInfoQos>(ue.first);
    uePtr->UpdateUlQosMetric(totAssigned, m_timeWindow);
}
 
void
NrMacSchedulerOfdmaQos::NotAssignedUlResources(
    const NrMacSchedulerNs3::UePtrAndBufferReq& ue,
    [[maybe_unused]] const NrMacSchedulerNs3::FTResources& assigned,
    const NrMacSchedulerNs3::FTResources& totAssigned) const
{
    NS_LOG_FUNCTION(this);
    auto uePtr = std::dynamic_pointer_cast<NrMacSchedulerUeInfoQos>(ue.first);
    uePtr->UpdateUlQosMetric(totAssigned, m_timeWindow);
}
 
void
NrMacSchedulerOfdmaQos::BeforeDlSched(const UePtrAndBufferReq& ue,
                                      const FTResources& assignableInIteration) const
{
    NS_LOG_FUNCTION(this);
    auto uePtr = std::dynamic_pointer_cast<NrMacSchedulerUeInfoQos>(ue.first);
    uePtr->CalculatePotentialTPutDl(assignableInIteration);
}
 
void
NrMacSchedulerOfdmaQos::BeforeUlSched(const UePtrAndBufferReq& ue,
                                      const FTResources& assignableInIteration) const
{
    NS_LOG_FUNCTION(this);
    auto uePtr = std::dynamic_pointer_cast<NrMacSchedulerUeInfoQos>(ue.first);
    uePtr->CalculatePotentialTPutUl(assignableInIteration);
}
 
} // namespace ns3