TrafficGenerator.h

// Copyright 2008 Michael Marsh, University of Maryland.
//
// This file is part of pydtn.
//
// pydtn is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// pydtn 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 pydtn.  If not, see <http://www.gnu.org/licenses/>.
//
// The views and conclusions contained in the software and documentation
// are those of the authors and should not be interpreted as representing
// official policies, either expressed or implied, of the University
// of Maryland.
//
// pydtn extends and embeds the Python interpreter, which is
// Copyright 2001-2006 Python Software Foundation, All Rights Reserved,
// and is released under the PSF License Agreement.
//
// RANLUX random number generation uses the Boost library,
// Copyright 1994-2006 by various authors (details in individual files),
// which is released under the Boost Software License, Version 1.0.

#ifndef __PYDTN_TRAFFIC_GENERATOR_H__
#define __PYDTN_TRAFFIC_GENERATOR_H__

#include "config.h"

#include "TimeDistribution.h"
#include "BundleMaker.h"
#include "simlpy/Entity.h"
#include "simlpy/Event.h"

class TrafficTrigger;


class TrafficGenerator : public Entity
{
   public :
      TrafficGenerator();
      virtual ~TrafficGenerator();

      Entity* create() const;

      void configure( const Entity::ArgList& args );
      void emit( const Entity::ArgList& args );

      std::string identifier() const { return "trafficgen"; }

      void setTimeDistribution( TimeDistribution* td );

      void setBundleMaker( BundleMaker* bm );

      virtual bool handler( TrafficTrigger& event );

   private :
      TimeDistribution*  m_timeDist;
      BundleMaker*       m_maker;
};

class TrafficTrigger : public Event
{
   public :
      TrafficTrigger( Entity* source, TrafficGenerator* destination,
                      const Time& t );

      virtual ~TrafficTrigger();

      bool handle();

   private :
      TrafficGenerator*  m_destination;
};


#endif // __PYDTN_TRAFFIC_GENERATOR_H__

---