uan-mac-rc.h

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2009 University of Washington
 *
 * 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: Leonard Tracy <lentracy@gmail.com>
 */

#ifndef UAN_MAC_RC_H
#define UAN_MAC_RC_H

#include "uan-mac.h"
#include "uan-address.h"

#include "ns3/nstime.h"
#include "ns3/trace-source-accessor.h"
#include "ns3/traced-callback.h"
#include "ns3/event-id.h"
#include "ns3/random-variable-stream.h"

#include <list>
#include <utility>
#include <vector>


namespace ns3 {

class Address;
class UanTxMode;
class UanHeaderRcRts;
class UanHeaderRcCts;
class UanHeaderRcCtsGlobal;
class UanPhy;


class Reservation
{
public:
  Reservation ();
  Reservation (std::list<std::pair <Ptr<Packet>, UanAddress > > &list, uint8_t frameNo, uint32_t maxPkts = 0);
  ~Reservation ();
  uint32_t GetNoFrames () const;
  uint32_t GetLength () const;
  const std::list<std::pair <Ptr<Packet>, UanAddress > > &GetPktList (void) const;
  uint8_t GetFrameNo () const;
  uint8_t GetRetryNo () const;
  Time GetTimestamp (uint8_t n) const;
  bool IsTransmitted () const;
  void SetFrameNo (uint8_t fn);
  void AddTimestamp (Time t);
  void IncrementRetry ();
  void SetTransmitted (bool t = true);
private:
  std::list<std::pair <Ptr<Packet>, UanAddress > > m_pktList;
  uint32_t m_length;
  uint8_t m_frameNo;
  std::vector<Time> m_timestamp;
  uint8_t m_retryNo;
  bool m_transmitted;

};


class UanMacRc : public UanMac
{
public:
  enum {
    TYPE_DATA, TYPE_GWPING, TYPE_RTS, TYPE_CTS, TYPE_ACK
  };
  UanMacRc ();
  virtual ~UanMacRc ();

  static TypeId GetTypeId (void);

  // Inherited virtual functions
  virtual Address GetAddress (void);
  virtual void SetAddress (UanAddress addr);
  virtual bool Enqueue (Ptr<Packet> pkt, const Address &dest, uint16_t protocolNumber);
  virtual void SetForwardUpCb (Callback<void, Ptr<Packet>, const UanAddress&> cb);
  virtual void AttachPhy (Ptr<UanPhy> phy);
  virtual Address GetBroadcast (void) const;
  virtual void Clear (void);

  int64_t AssignStreams (int64_t stream);

private:
  enum State {
    UNASSOCIATED, GWPSENT, IDLE, RTSSENT, DATATX
  };

  State m_state;
  bool m_rtsBlocked;

  EventId m_startAgain;
  UanAddress m_address;
  double m_retryRate;
  UanAddress m_assocAddr;
  Ptr<UanPhy> m_phy;
  uint32_t m_numRates;
  uint32_t m_currentRate;
  uint32_t m_maxFrames;
  uint32_t m_queueLimit;
  uint8_t m_frameNo;
  Time m_sifs;
  Time m_learnedProp;

  uint16_t m_numRetryRates;
  double m_minRetryRate;
  double m_retryStep;

  uint32_t m_ctsSizeN;
  uint32_t m_ctsSizeG;

  bool m_cleared;

  std::list<std::pair <Ptr<Packet>, UanAddress > > m_pktQueue;
  std::list<Reservation> m_resList;

  Callback<void, Ptr<Packet>, const UanAddress& > m_forwardUpCb;

  TracedCallback<Ptr<const Packet>, UanTxMode > m_rxLogger;
  TracedCallback<Ptr<const Packet>, uint16_t > m_enqueueLogger;
  TracedCallback<Ptr<const Packet>, uint16_t > m_dequeueLogger;

  EventId m_rtsEvent;
  void ReceiveOkFromPhy (Ptr<Packet>, double sinr, UanTxMode mode);

  void Associate (void);
  void AssociateTimeout (void);
  void SendRts (void);
  void RtsTimeout (void);
  UanHeaderRcRts CreateRtsHeader (const Reservation &res);
  void ScheduleData (const UanHeaderRcCts &ctsh, const UanHeaderRcCtsGlobal &ctsg, uint32_t ctsBytes);
  void ProcessAck (Ptr<Packet> ack);
  void SendPacket (Ptr<Packet> pkt, uint32_t rate);
  bool IsPhy1Ok (void);
  void BlockRtsing (void);

  static uint32_t m_cntrlSends;

  Ptr<ExponentialRandomVariable> m_ev;

protected:
  void DoDispose ();
};

}

#endif /* UAN_MAC_RC_H */