diff --git a/src/lr-wpan/model/lr-wpan-phy.cc b/src/lr-wpan/model/lr-wpan-phy.cc index 653155d64..09fe56849 100644 --- a/src/lr-wpan/model/lr-wpan-phy.cc +++ b/src/lr-wpan/model/lr-wpan-phy.cc @@ -209,6 +209,21 @@ LrWpanPhy::GetRxSpectrumModel () const } } +Ptr +LrWpanPhy::GetRxAntenna () +{ + return m_antenna; +} + +void +LrWpanPhy::SetAntenna (Ptr a) +{ + NS_LOG_FUNCTION (this << a); + m_antenna = a; +} + + + void LrWpanPhy::StartRx (Ptr spectrumRxParams) { @@ -330,6 +345,7 @@ LrWpanPhy::PdDataRequest (const uint32_t psduLength, Ptr p) txParams->duration = CalculateTxTime (p); txParams->txPhy = GetObject (); txParams->psd = m_txPsd; + txParams->txAntenna = m_antenna; Ptr pb = CreateObject (); pb->AddPacket (p); txParams->packetBurst = pb; diff --git a/src/lr-wpan/model/lr-wpan-phy.h b/src/lr-wpan/model/lr-wpan-phy.h index 08856c79a..05bd51974 100644 --- a/src/lr-wpan/model/lr-wpan-phy.h +++ b/src/lr-wpan/model/lr-wpan-phy.h @@ -30,6 +30,7 @@ #include #include #include +#include #include #include #include @@ -211,6 +212,8 @@ public: Ptr GetChannel (void); void SetDevice (Ptr d); Ptr GetDevice (); + void SetAntenna (Ptr a); + Ptr GetRxAntenna (); virtual Ptr GetRxSpectrumModel () const; /** @@ -386,6 +389,7 @@ private: Ptr m_mobility; Ptr m_device; Ptr m_channel; + Ptr m_antenna; Ptr m_txPsd; Ptr m_rxPsd; Ptr m_noise;