BMacLayer.h

//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// 
// 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 Lesser General Public License for more details.
// 
// You should have received a copy of the GNU Lesser General Public License
// along with this program.  If not, see http://www.gnu.org/licenses/.
// 

#ifndef BMACLAYER_H_
#define BMACLAYER_H_

#include <string>
#include <sstream>
#include <vector>
#include <list>
#include "BaseMacLayer.h"
#include <DroppedPacket.h>
#include <MacPkt_m.h>

class  BMacLayer : public BaseMacLayer
{
  public:
  virtual ~BMacLayer();

    virtual void initialize(int);

    virtual void finish();

    virtual void handleLowerMsg(cMessage*);

    virtual void handleUpperMsg(cMessage*);

    virtual void handleSelfMsg(cMessage*);

    virtual void handleLowerControl(cMessage *msg);

  protected:
    typedef std::list<MacPkt*> MacQueue;

    MacQueue macQueue;

  long nbTxDataPackets;
  long nbTxPreambles;
  long nbRxDataPackets;
  long nbRxPreambles;
  long nbMissedAcks;
  long nbRecvdAcks;
  long nbDroppedDataPackets;
  long nbTxAcks;
  enum States {
    INIT, //0
    SLEEP,  //1
    CCA,  //2
    SEND_PREAMBLE,  //3
    WAIT_DATA,    //4
    SEND_DATA,    //5
    WAIT_TX_DATA_OVER,  //6
    WAIT_ACK,   //7
    SEND_ACK,   //8
    WAIT_ACK_TX   //9
    };
  States macState;

  enum TYPES {
    // packet types
    BMAC_PREAMBLE = 191,
    BMAC_DATA,
    BMAC_ACK,
    // self message types
    BMAC_RESEND_DATA,
    BMAC_ACK_TIMEOUT,
    BMAC_START_BMAC,
    BMAC_WAKE_UP,
    BMAC_SEND_ACK,
    BMAC_CCA_TIMEOUT,
    BMAC_ACK_TX_OVER,
    BMAC_SEND_PREAMBLE,
    BMAC_STOP_PREAMBLES,
    BMAC_DATA_TX_OVER,
    BMAC_DATA_TIMEOUT
  };

  // messages used in the FSM
  cMessage *resend_data;
  cMessage *ack_timeout;
  cMessage *start_bmac;
  cMessage *wakeup;
  cMessage *send_ack;
  cMessage *cca_timeout;
  cMessage *ack_tx_over;
  cMessage *send_preamble;
  cMessage *stop_preambles;
  cMessage *data_tx_over;
  cMessage *data_timeout;

  int lastDataPktSrcAddr;
  int lastDataPktDestAddr;
  int txAttempts;
  DroppedPacket droppedPacket;

  int catDroppedPacket;

  int nicId;

  double queueLength;
  bool animation;
  double slotDuration;
  double bitrate;
  double headerLength;
  double checkInterval;
  double txPower;
  bool useMacAcks;
  int maxTxAttempts;
  bool stats;

  enum BMAC_COLORS {
    GREEN = 1,
    BLUE = 2,
    RED = 3,
    BLACK = 4,
    YELLOW = 5
  };

  void changeDisplayColor(BMAC_COLORS color);

  void sendDataPacket();

  void sendMacAck();

  void sendPreamble();

  void attachSignal(MacPkt *macPkt);

  bool addToQueue(cMessage * msg);
};

#endif /* BMACLAYER_H_ */