nr-bearer-stats-connector.h

// Copyright (c) 2011 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
// Copyright (c) 2015, NYU WIRELESS, Tandon School of Engineering, New York University
//
// SPDX-License-Identifier: GPL-2.0-only
 
#ifndef NR_BEARER_STATS_CONNECTOR_H
#define NR_BEARER_STATS_CONNECTOR_H
 
#include <ns3/ptr.h>
#include <ns3/simple-ref-count.h>
 
#include <map>
#include <set>
 
namespace ns3
{
 
class NrBearerStatsBase;
 
class NrBearerStatsConnector
{
  public:
    NrBearerStatsConnector();
 
    void EnableRlcStats(Ptr<NrBearerStatsBase> rlcStats);
 
    void EnablePdcpStats(Ptr<NrBearerStatsBase> pdcpStats);
 
    void EnsureConnected();
 
    // trace sinks, to be used with MakeBoundCallback
 
    static void NotifyRandomAccessSuccessfulUe(NrBearerStatsConnector* c,
                                               std::string context,
                                               uint64_t imsi,
                                               uint16_t cellid,
                                               uint16_t rnti);
 
    static void NotifyConnectionSetupUe(NrBearerStatsConnector* c,
                                        std::string context,
                                        uint64_t imsi,
                                        uint16_t cellid,
                                        uint16_t rnti);
 
    static void NotifyConnectionReconfigurationUe(NrBearerStatsConnector* c,
                                                  std::string context,
                                                  uint64_t imsi,
                                                  uint16_t cellid,
                                                  uint16_t rnti);
 
    static void NotifyHandoverStartUe(NrBearerStatsConnector* c,
                                      std::string context,
                                      uint64_t imsi,
                                      uint16_t cellid,
                                      uint16_t rnti,
                                      uint16_t targetCellId);
 
    static void NotifyHandoverEndOkUe(NrBearerStatsConnector* c,
                                      std::string context,
                                      uint64_t imsi,
                                      uint16_t cellid,
                                      uint16_t rnti);
 
    static void NotifyNewUeContextEnb(NrBearerStatsConnector* c,
                                      std::string context,
                                      uint16_t cellid,
                                      uint16_t rnti);
 
    static void NotifyConnectionReconfigurationEnb(NrBearerStatsConnector* c,
                                                   std::string context,
                                                   uint64_t imsi,
                                                   uint16_t cellid,
                                                   uint16_t rnti);
 
    static void NotifyHandoverStartEnb(NrBearerStatsConnector* c,
                                       std::string context,
                                       uint64_t imsi,
                                       uint16_t cellid,
                                       uint16_t rnti,
                                       uint16_t targetCellId);
 
    static void NotifyHandoverEndOkEnb(NrBearerStatsConnector* c,
                                       std::string context,
                                       uint64_t imsi,
                                       uint16_t cellid,
                                       uint16_t rnti);
 
    Ptr<NrBearerStatsBase> GetRlcStats();
    Ptr<NrBearerStatsBase> GetPdcpStats();
 
  private:
    void StoreUeManagerPath(std::string ueManagerPath, uint16_t cellId, uint16_t rnti);
 
    void ConnectSrb0Traces(std::string ueRrcPath, uint64_t imsi, uint16_t cellId, uint16_t rnti);
 
    void ConnectSrb1TracesUe(std::string ueRrcPath, uint64_t imsi, uint16_t cellId, uint16_t rnti);
 
    void ConnectTracesUeIfFirstTime(std::string context,
                                    uint64_t imsi,
                                    uint16_t cellid,
                                    uint16_t rnti);
 
    void ConnectTracesEnbIfFirstTime(std::string context,
                                     uint64_t imsi,
                                     uint16_t cellid,
                                     uint16_t rnti);
 
    void ConnectTracesUe(std::string context, uint64_t imsi, uint16_t cellid, uint16_t rnti);
 
    void DisconnectTracesUe(std::string context, uint64_t imsi, uint16_t cellid, uint16_t rnti);
 
    void ConnectTracesEnb(std::string context, uint64_t imsi, uint16_t cellid, uint16_t rnti);
 
    void DisconnectTracesEnb(std::string context, uint64_t imsi, uint16_t cellid, uint16_t rnti);
 
    Ptr<NrBearerStatsBase> m_rlcStats;  
    Ptr<NrBearerStatsBase> m_pdcpStats; 
 
    bool m_connected; 
    std::set<uint64_t>
        m_imsiSeenUe; 
    std::set<uint64_t>
        m_imsiSeenEnb; 
 
    struct CellIdRnti
    {
        uint16_t cellId; 
        uint16_t rnti;   
    };
 
    friend bool operator<(const CellIdRnti& a, const CellIdRnti& b);
 
    std::map<CellIdRnti, std::string> m_ueManagerPathByCellIdRnti;
};
 
} // namespace ns3
 
#endif // NR_BEARER_STATS_CONNECTOR_H