nr-mac-scheduler-ue-info-pf.h

// Copyright (c) 2019 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#pragma once
 
#include "nr-mac-scheduler-ue-info-rr.h"
 
namespace ns3
{
 
class NrMacSchedulerUeInfoPF : public NrMacSchedulerUeInfo
{
  public:
    NrMacSchedulerUeInfoPF(float alpha, uint16_t rnti, BeamId beamId, const GetRbPerRbgFn& fn)
        : NrMacSchedulerUeInfo(rnti, beamId, fn),
          m_alpha(alpha)
    {
    }
 
    void ResetDlSchedInfo() override
    {
        m_lastAvgTputDl = m_avgTputDl;
        m_avgTputDl = 0.0;
        m_currTputDl = 0.0;
        m_potentialTputDl = 0.0;
        NrMacSchedulerUeInfo::ResetDlSchedInfo();
    }
 
    void ResetUlSchedInfo() override
    {
        m_lastAvgTputUl = m_avgTputUl;
        m_avgTputUl = 0.0;
        m_currTputUl = 0.0;
        m_potentialTputUl = 0.0;
        NrMacSchedulerUeInfo::ResetUlSchedInfo();
    }
 
    void ResetDlMetric() override
    {
        NrMacSchedulerUeInfo::ResetDlMetric();
        m_avgTputDl = m_lastAvgTputDl;
    }
 
    void ResetUlMetric() override
    {
        NrMacSchedulerUeInfo::ResetUlMetric();
        m_avgTputUl = m_lastAvgTputUl;
    }
 
    void UpdateDlPFMetric(const NrMacSchedulerNs3::FTResources& totAssigned, double timeWindow);
 
    void UpdateUlPFMetric(const NrMacSchedulerNs3::FTResources& totAssigned, double timeWindow);
 
    void CalculatePotentialTPutDl(const NrMacSchedulerNs3::FTResources& assignableInIteration);
 
    void CalculatePotentialTPutUl(const NrMacSchedulerNs3::FTResources& assignableInIteration);
 
    static bool CompareUeWeightsDl(const NrMacSchedulerNs3::UePtrAndBufferReq& lue,
                                   const NrMacSchedulerNs3::UePtrAndBufferReq& rue)
    {
        auto luePtr = dynamic_cast<NrMacSchedulerUeInfoPF*>(lue.first.get());
        auto ruePtr = dynamic_cast<NrMacSchedulerUeInfoPF*>(rue.first.get());
 
        double lPfMetric = std::pow(luePtr->m_potentialTputDl, luePtr->m_alpha) /
                           std::max(1E-9, luePtr->m_avgTputDl);
        double rPfMetric = std::pow(ruePtr->m_potentialTputDl, ruePtr->m_alpha) /
                           std::max(1E-9, ruePtr->m_avgTputDl);
 
        return (lPfMetric > rPfMetric);
    }
 
    static bool CompareUeWeightsUl(const NrMacSchedulerNs3::UePtrAndBufferReq& lue,
                                   const NrMacSchedulerNs3::UePtrAndBufferReq& rue)
    {
        auto luePtr = dynamic_cast<NrMacSchedulerUeInfoPF*>(lue.first.get());
        auto ruePtr = dynamic_cast<NrMacSchedulerUeInfoPF*>(rue.first.get());
 
        double lPfMetric = std::pow(luePtr->m_potentialTputUl, luePtr->m_alpha) /
                           std::max(1E-9, luePtr->m_avgTputUl);
        double rPfMetric = std::pow(ruePtr->m_potentialTputUl, ruePtr->m_alpha) /
                           std::max(1E-9, ruePtr->m_avgTputUl);
 
        return (lPfMetric > rPfMetric);
    }
 
    double m_currTputDl{0.0};      
    double m_avgTputDl{0.0};       
    double m_lastAvgTputDl{0.0};   
    double m_potentialTputDl{0.0}; 
    float m_alpha{0.0};            
 
    double m_currTputUl{0.0};      
    double m_avgTputUl{0.0};       
    double m_lastAvgTputUl{0.0};   
    double m_potentialTputUl{0.0}; 
};
 
} // namespace ns3