PriorityTrafficGenerator.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 __BACKBONE_PRIORITYTRAFFICGENERATOR_H_
#define __BACKBONE_PRIORITYTRAFFICGENERATOR_H_

#include <omnetpp.h>
#include <MACAddress.h>
#include "cModule.h"
#include <vector>
#include <string>

/**
 * TODO - Generated class
 */
class PriorityTrafficGenerator : public cSimpleModule
{
protected:
  long seqNum;

  // send parameters
  cPar *sendInterval;
  cPar *numPacketsPerBurst;
  cPar *packetLength;
  int etherType;
  MACAddress destMACAddress;

  // self messages
  cMessage *timerMsg;
  simtime_t stopTime;

  // receive statistics
  long packetsSent;
  long packetsReceived;
  static simsignal_t sentPkSignal;
  static simsignal_t rcvdPkSignal;

  bool dynamicAPConnections;
  cModule *currentlyAPConnected;
  std::vector<std::string> accessPointPaths;

  void reconnect();

public:
  PriorityTrafficGenerator();
  ~PriorityTrafficGenerator();

protected:
  virtual void initialize(int stage);
  virtual int numInitStages() const {return 2;}
  virtual void handleMessage(cMessage *msg);
  virtual void finish();

  virtual MACAddress resolveDestMACAddress();

  virtual void sendBurstPackets();
  virtual void receivePacket(cPacket *msg);
};

#endif