lte-test-entities.h

Go to the documentation of this file.

/* -*-  Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2011 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation;
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Author: Manuel Requena <manuel.requena@cttc.es>
 */
 
#ifndef LTE_TEST_ENTITIES_H
#define LTE_TEST_ENTITIES_H
 
#include "ns3/simulator.h"
#include "ns3/test.h"
 
#include "ns3/lte-mac-sap.h"
#include "ns3/lte-rlc-sap.h"
#include "ns3/lte-pdcp-sap.h"
 
#include "ns3/net-device.h"
#include <ns3/epc-enb-s1-sap.h>
 
namespace ns3 {
 
class LteTestRrc : public Object
{
    friend class LtePdcpSpecificLtePdcpSapUser<LteTestRrc>;
//   friend class EnbMacMemberLteEnbCmacSapProvider;
//   friend class EnbMacMemberLteMacSapProvider<LteTestMac>;
//   friend class EnbMacMemberFfMacSchedSapUser;
//   friend class EnbMacMemberFfMacCschedSapUser;
//   friend class EnbMacMemberLteEnbPhySapUser;
 
  public:
    static TypeId GetTypeId (void);
 
    LteTestRrc (void);
    virtual ~LteTestRrc (void);
    virtual void DoDispose (void);
 
 
    void SetLtePdcpSapProvider (LtePdcpSapProvider* s);
    LtePdcpSapUser* GetLtePdcpSapUser (void);
 
    void Start ();
    void Stop ();
 
    void SendData (Time at, std::string dataToSend);
    std::string GetDataReceived (void);
 
    // Stats
    uint32_t GetTxPdus (void);
    uint32_t GetTxBytes (void);
    uint32_t GetRxPdus (void);
    uint32_t GetRxBytes (void);
 
    Time GetTxLastTime (void);
    Time GetRxLastTime (void);
 
    void SetArrivalTime (Time arrivalTime);
    void SetPduSize (uint32_t pduSize);
 
    void SetDevice (Ptr<NetDevice> device);
 
  private:
    virtual void DoReceivePdcpSdu (LtePdcpSapUser::ReceivePdcpSduParameters params);
 
    LtePdcpSapUser* m_pdcpSapUser; 
    LtePdcpSapProvider* m_pdcpSapProvider; 
 
    std::string m_receivedData; 
 
    uint32_t m_txPdus; 
    uint32_t m_txBytes; 
    uint32_t m_rxPdus; 
    uint32_t m_rxBytes; 
    Time     m_txLastTime; 
    Time     m_rxLastTime; 
 
    EventId m_nextPdu; 
    Time m_arrivalTime; 
    uint32_t m_pduSize; 
 
    Ptr<NetDevice> m_device; 
};
 
 
class LteTestPdcp : public Object
{
  friend class LteRlcSpecificLteRlcSapUser<LteTestPdcp>;
  
  public:
    static TypeId GetTypeId (void);
 
    LteTestPdcp (void);
    virtual ~LteTestPdcp (void);
    virtual void DoDispose (void);
 
 
    void SetLteRlcSapProvider (LteRlcSapProvider* s);
    LteRlcSapUser* GetLteRlcSapUser (void);
 
    void Start ();
 
    void SendData (Time time, std::string dataToSend);
    std::string GetDataReceived (void);
 
  private:
    virtual void DoReceivePdcpPdu (Ptr<Packet> p);
 
    LteRlcSapUser* m_rlcSapUser; 
    LteRlcSapProvider* m_rlcSapProvider; 
 
    std::string m_receivedData; 
};
 
 
class LteTestMac : public Object
{
//   friend class EnbMacMemberLteEnbCmacSapProvider;
    friend class EnbMacMemberLteMacSapProvider<LteTestMac>;
//   friend class EnbMacMemberFfMacSchedSapUser;
//   friend class EnbMacMemberFfMacCschedSapUser;
//   friend class EnbMacMemberLteEnbPhySapUser;
 
  public:
    static TypeId GetTypeId (void);
 
    LteTestMac (void);
    virtual ~LteTestMac (void);
    virtual void DoDispose (void);
 
    void SetDevice (Ptr<NetDevice> device);
 
    void SendTxOpportunity (Time time, uint32_t bytes);
    std::string GetDataReceived (void);
 
    bool Receive (Ptr<NetDevice> nd, Ptr<const Packet> p, uint16_t protocol, const Address& addr);
 
    void SetLteMacSapUser (LteMacSapUser* s);
    LteMacSapProvider* GetLteMacSapProvider (void);
 
    void SetLteMacLoopback (Ptr<LteTestMac> s);
 
    void SetPdcpHeaderPresent (bool present);
 
    void SetRlcHeaderType (uint8_t rlcHeaderType);
 
    typedef enum {
      UM_RLC_HEADER = 0,
      AM_RLC_HEADER = 1,
    } RlcHeaderType_t; 
 
    void SetTxOpportunityMode (uint8_t mode);
 
    typedef enum {
      MANUAL_MODE     = 0,
      AUTOMATIC_MODE  = 1,
      RANDOM_MODE     = 2
    } TxOpportunityMode_t; 
 
    void SetTxOppTime (Time txOppTime);
    void SetTxOppSize (uint32_t txOppSize);
 
    // Stats
    uint32_t GetTxPdus (void);
    uint32_t GetTxBytes (void);
    uint32_t GetRxPdus (void);
    uint32_t GetRxBytes (void);
 
  private:
    // forwarded from LteMacSapProvider
    void DoTransmitPdu (LteMacSapProvider::TransmitPduParameters params);
    void DoReportBufferStatus (LteMacSapProvider::ReportBufferStatusParameters params);
 
    LteMacSapProvider* m_macSapProvider; 
    LteMacSapUser* m_macSapUser; 
    Ptr<LteTestMac> m_macLoopback; 
 
    std::string m_receivedData; 
 
    uint8_t m_rlcHeaderType; 
    bool m_pdcpHeaderPresent; 
    uint8_t m_txOpportunityMode; 
 
    Ptr<NetDevice> m_device; 
 
    // TxOpportunity configuration
    EventId m_nextTxOpp; 
    Time m_txOppTime; 
    uint32_t m_txOppSize; 
    std::list<EventId> m_nextTxOppList; 
 
    // Stats
    uint32_t m_txPdus; 
    uint32_t m_txBytes; 
    uint32_t m_rxPdus; 
    uint32_t m_rxBytes; 
 
};
 
 
 
class EpcTestRrc : public Object
{
  friend class MemberEpcEnbS1SapUser<EpcTestRrc>;
 
public:
  EpcTestRrc ();
  virtual ~EpcTestRrc ();
 
  // inherited from Object
  virtual void DoDispose (void);
  static TypeId GetTypeId (void);
 
  void SetS1SapProvider (EpcEnbS1SapProvider* s);
 
  EpcEnbS1SapUser* GetS1SapUser ();
 
private:
 
  // S1 SAP methods
  void DoInitialContextSetupRequest (EpcEnbS1SapUser::InitialContextSetupRequestParameters params);
  void DoDataRadioBearerSetupRequest (EpcEnbS1SapUser::DataRadioBearerSetupRequestParameters params);
  void DoPathSwitchRequestAcknowledge (EpcEnbS1SapUser::PathSwitchRequestAcknowledgeParameters params);  
  
  EpcEnbS1SapProvider* m_s1SapProvider; 
  EpcEnbS1SapUser* m_s1SapUser; 
  
 
};
 
 
} // namespace ns3
 
#endif /* LTE_TEST_MAC_H */