nr-mac-scheduler-tdma-ai.h

// Copyright (c) 2024 Seoul National University (SNU)
// Copyright (c) 2024 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#pragma once
 
#include "nr-mac-scheduler-tdma-qos.h"
#include "nr-mac-scheduler-ue-info-ai.h"
 
namespace ns3
{
 
class NrMacSchedulerTdmaAi : public NrMacSchedulerTdmaQos
{
    friend class NrTestSchedulerAiCase;
 
  public:
    static TypeId GetTypeId();
 
    NrMacSchedulerTdmaAi();
 
  protected:
    std::shared_ptr<NrMacSchedulerUeInfo> CreateUeRepresentation(
        const NrMacCschedSapProvider::CschedUeConfigReqParameters& params) const override;
 
    std::function<bool(const NrMacSchedulerNs3::UePtrAndBufferReq& lhs,
                       const NrMacSchedulerNs3::UePtrAndBufferReq& rhs)>
    GetUeCompareDlFn() const override;
 
    std::function<bool(const NrMacSchedulerNs3::UePtrAndBufferReq& lhs,
                       const NrMacSchedulerNs3::UePtrAndBufferReq& rhs)>
    GetUeCompareUlFn() const override;
 
    void SetNotifyCbDl(NrMacSchedulerUeInfoAi::NotifyCb notifyCb);
 
    void SetNotifyCbUl(NrMacSchedulerUeInfoAi::NotifyCb notifyCb);
 
    std::vector<NrMacSchedulerUeInfoAi::LcObservation> GetUeObservationsDl(
        const std::vector<NrMacSchedulerNs3::UePtrAndBufferReq>& ueVector) const;
 
    std::vector<NrMacSchedulerUeInfoAi::LcObservation> GetUeObservationsUl(
        const std::vector<NrMacSchedulerNs3::UePtrAndBufferReq>& ueVector) const;
 
    bool GetIsGameOverDl() const;
 
    bool GetIsGameOverUl() const;
 
    float GetUeRewardsDl(const std::vector<NrMacSchedulerNs3::UePtrAndBufferReq>& ueVector) const;
 
    float GetUeRewardsUl(const std::vector<NrMacSchedulerNs3::UePtrAndBufferReq>& ueVector) const;
 
    void CallNotifyDlFn(
        const std::vector<NrMacSchedulerNs3::UePtrAndBufferReq>& ueVector) const override;
 
    void CallNotifyUlFn(
        const std::vector<NrMacSchedulerNs3::UePtrAndBufferReq>& ueVector) const override;
 
    void UpdateAllUeWeightsDl(
        const NrMacSchedulerUeInfoAi::UeWeightsMap& ueWeights,
        const std::vector<NrMacSchedulerNs3::UePtrAndBufferReq>& ueVector) const;
 
    void UpdateAllUeWeightsUl(
        const NrMacSchedulerUeInfoAi::UeWeightsMap& ueWeights,
        const std::vector<NrMacSchedulerNs3::UePtrAndBufferReq>& ueVector) const;
 
  private:
    double m_alpha{0.0};                           
    NrMacSchedulerUeInfoAi::NotifyCb m_notifyCbDl; 
    NrMacSchedulerUeInfoAi::NotifyCb m_notifyCbUl; 
};
 
} // namespace ns3