nr-bearer-stats-simple.h

// Copyright (c) 2022 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#ifndef NR_RADIO_BEARER_STATS_SIMPLE_H_
#define NR_RADIO_BEARER_STATS_SIMPLE_H_
 
#include "nr-stats-calculator.h"
 
#include "ns3/basic-data-calculators.h"
#include "ns3/nr-common.h"
#include "ns3/object.h"
#include "ns3/uinteger.h"
 
#include <fstream>
#include <string>
 
namespace ns3
{
 
/*
 * \ingroup helpers
 *
 * Defines the minimum set of functions that RLC or PDC stats class should implement.
 * See also NrBearerStatsSimple and NrBearerStatsSimple.
 */
class NrBearerStatsBase : public Object
{
  public:
    // Inherited from ns3::Object
    static TypeId GetTypeId();
    void DoDispose() override;
    virtual void UlTxPdu(uint16_t cellId,
                         uint64_t imsi,
                         uint16_t rnti,
                         uint8_t lcid,
                         uint32_t packetSize) = 0;
    virtual void UlRxPdu(uint16_t cellId,
                         uint64_t imsi,
                         uint16_t rnti,
                         uint8_t lcid,
                         uint32_t packetSize,
                         uint64_t delay) = 0;
    virtual void DlTxPdu(uint16_t cellId,
                         uint64_t imsi,
                         uint16_t rnti,
                         uint8_t lcid,
                         uint32_t packetSize) = 0;
    virtual void DlRxPdu(uint16_t cellId,
                         uint64_t imsi,
                         uint16_t rnti,
                         uint8_t lcid,
                         uint32_t packetSize,
                         uint64_t delay) = 0;
};
 
class NrBearerStatsSimple : public NrBearerStatsBase
{
  public:
    NrBearerStatsSimple();
 
    NrBearerStatsSimple(std::string protocolType);
 
    ~NrBearerStatsSimple() override;
 
    // Inherited from ns3::Object
    static TypeId GetTypeId();
    void DoDispose() override;
 
    std::string GetUlTxOutputFilename();
 
    std::string GetUlRxOutputFilename();
 
    std::string GetDlTxOutputFilename();
 
    std::string GetDlRxOutputFilename();
 
    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;
 
  private:
    std::string m_protocolType;
    std::string m_dlRlcTxOutputFilename;  
    std::string m_dlRlcRxOutputFilename;  
    std::string m_ulRlcTxOutputFilename;  
    std::string m_ulRlcRxOutputFilename;  
    std::string m_dlPdcpTxOutputFilename; 
    std::string m_dlPdcpRxOutputFilename; 
    std::string m_ulPdcpTxOutputFilename; 
    std::string m_ulPdcpRxOutputFilename; 
    std::ofstream m_dlTxOutFile; 
    std::ofstream m_dlRxOutFile; 
    std::ofstream m_ulTxOutFile; 
    std::ofstream m_ulRxOutFile; 
};
 
} // namespace ns3
 
#endif /* NR_RADIO_BEARER_STATS_SIMPLE_H_ */