nr-bearer-stats-calculator.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_RADIO_BEARER_STATS_CALCULATOR_H_
#define NR_RADIO_BEARER_STATS_CALCULATOR_H_
 
#include "nr-bearer-stats-simple.h"
 
#include "ns3/basic-data-calculators.h"
#include "ns3/nr-common.h"
#include "ns3/object.h"
#include "ns3/uinteger.h"
 
#include <fstream>
#include <map>
#include <string>
 
namespace ns3
{
namespace nr
{
typedef std::map<nr::ImsiLcidPair_t, uint32_t> Uint32Map;
typedef std::map<nr::ImsiLcidPair_t, uint64_t> Uint64Map;
typedef std::map<nr::ImsiLcidPair_t, Ptr<MinMaxAvgTotalCalculator<uint32_t>>> Uint32StatsMap;
typedef std::map<nr::ImsiLcidPair_t, Ptr<MinMaxAvgTotalCalculator<uint64_t>>> Uint64StatsMap;
typedef std::map<nr::ImsiLcidPair_t, double> DoubleMap;
typedef std::map<nr::ImsiLcidPair_t, nr::FlowId_t> FlowIdMap;
} // namespace nr
 
class NrBearerStatsCalculator : public NrBearerStatsBase
{
  public:
    NrBearerStatsCalculator();
    NrBearerStatsCalculator(std::string protocolType);
    ~NrBearerStatsCalculator() override;
    // Inherited from ns3::Object
    static TypeId GetTypeId();
    void DoDispose() override;
    void SetStartTime(Time t);
    Time GetStartTime() const;
    void SetEpoch(Time e);
    Time GetEpoch() const;
    void UlTxPdu(uint16_t cellId,
                 uint64_t imsi,
                 uint16_t rnti,
                 uint8_t lcid,
                 uint32_t packetSize) override;
    void UlRxPdu(uint16_t cellId,
                 uint64_t imsi,
                 uint16_t rnti,
                 uint8_t lcid,
                 uint32_t packetSize,
                 uint64_t delay) override;
    void DlTxPdu(uint16_t cellId,
                 uint64_t imsi,
                 uint16_t rnti,
                 uint8_t lcid,
                 uint32_t packetSize) override;
    void DlRxPdu(uint16_t cellId,
                 uint64_t imsi,
                 uint16_t rnti,
                 uint8_t lcid,
                 uint32_t packetSize,
                 uint64_t delay) override;
    uint32_t GetUlTxPackets(uint64_t imsi, uint8_t lcid);
    uint32_t GetUlRxPackets(uint64_t imsi, uint8_t lcid);
    uint64_t GetUlTxData(uint64_t imsi, uint8_t lcid);
    uint64_t GetUlRxData(uint64_t imsi, uint8_t lcid);
    uint32_t GetUlCellId(uint64_t imsi, uint8_t lcid);
    double GetUlDelay(uint64_t imsi, uint8_t lcid);
    std::vector<double> GetUlDelayStats(uint64_t imsi, uint8_t lcid);
    std::vector<double> GetUlPduSizeStats(uint64_t imsi, uint8_t lcid);
    uint32_t GetDlTxPackets(uint64_t imsi, uint8_t lcid);
    uint32_t GetDlRxPackets(uint64_t imsi, uint8_t lcid);
    uint64_t GetDlTxData(uint64_t imsi, uint8_t lcid);
    uint64_t GetDlRxData(uint64_t imsi, uint8_t lcid);
    uint32_t GetDlCellId(uint64_t imsi, uint8_t lcid);
    double GetDlDelay(uint64_t imsi, uint8_t lcid);
    std::vector<double> GetDlDelayStats(uint64_t imsi, uint8_t lcid);
    std::vector<double> GetDlPduSizeStats(uint64_t imsi, uint8_t lcid);
    std::string GetUlOutputFilename();
    std::string GetDlOutputFilename();
 
  private:
    void ShowResults();
    void WriteUlResults(std::ofstream& outFile);
    void WriteDlResults(std::ofstream& outFile);
    void ResetResults();
    void RescheduleEndEpoch();
    void EndEpoch();
 
    EventId m_endEpochEvent;        
    nr::FlowIdMap m_flowId;         
    nr::Uint32Map m_dlCellId;       
    nr::Uint32Map m_dlTxPackets;    
    nr::Uint32Map m_dlRxPackets;    
    nr::Uint64Map m_dlTxData;       
    nr::Uint64Map m_dlRxData;       
    nr::Uint64StatsMap m_dlDelay;   
    nr::Uint32StatsMap m_dlPduSize; 
    nr::Uint32Map m_ulCellId;       
    nr::Uint32Map m_ulTxPackets;    
    nr::Uint32Map m_ulRxPackets;    
    nr::Uint64Map m_ulTxData;       
    nr::Uint64Map m_ulRxData;       
    nr::Uint64StatsMap m_ulDelay;   
    nr::Uint32StatsMap m_ulPduSize; 
    Time m_startTime;
    Time m_epochDuration;
    bool m_firstWrite;
    bool m_pendingOutput;
    std::string m_protocolType;
    std::string m_dlRlcOutputFilename;
    std::string m_ulRlcOutputFilename;
    std::string m_dlPdcpOutputFilename;
    std::string m_ulPdcpOutputFilename;
    std::ofstream m_dlOutFile;
    std::ofstream m_ulOutFile;
};
 
} // namespace ns3
 
#endif /* NR_RADIO_BEARER_STATS_CALCULATOR_H_ */