nr-stats-calculator.h

// Copyright (c) 2021 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#ifndef NR_STATS_CALCULATOR_H_
#define NR_STATS_CALCULATOR_H_
 
#include "ns3/object.h"
#include "ns3/string.h"
 
#include <map>
 
namespace ns3
{
 
class NrStatsCalculator : public Object
{
  public:
    NrStatsCalculator();
 
    ~NrStatsCalculator() override;
 
    static TypeId GetTypeId();
 
    void SetUlOutputFilename(std::string outputFilename);
 
    std::string GetUlOutputFilename();
 
    void SetDlOutputFilename(std::string outputFilename);
 
    std::string GetDlOutputFilename();
 
    bool ExistsImsiPath(std::string path);
 
    void SetImsiPath(std::string path, uint64_t imsi);
 
    uint64_t GetImsiPath(std::string path);
 
    bool ExistsCellIdPath(std::string path);
 
    void SetCellIdPath(std::string path, uint16_t cellId);
 
    uint16_t GetCellIdPath(std::string path);
 
  protected:
    static uint64_t FindImsiFromGnbRlcPath(std::string path);
 
    static uint64_t FindImsiFromNrUeNetDevice(std::string path);
 
    static uint16_t FindCellIdFromGnbRlcPath(std::string path);
 
    static uint64_t FindImsiFromGnbMac(std::string path, uint16_t rnti);
 
    static uint16_t FindCellIdFromGnbMac(std::string path, uint16_t rnti);
 
  private:
    std::map<std::string, uint64_t> m_pathImsiMap;
 
    std::map<std::string, uint16_t> m_pathCellIdMap;
 
    std::string m_dlOutputFilename;
 
    std::string m_ulOutputFilename;
};
 
} // namespace ns3
 
#endif /* NR_STATS_CALCULATOR_H_ */