TrafficGenerator.cc

// 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.

#include "simlpy/interpreter_defs.h"
#include "simlpy/interpreter_hooks.h"

#include <iostream>

#include "TrafficGenerator.h"
#include "simlpy/Clock.h"
#include "PythonTimeDistribution.h"
#include "PythonBundleMaker.h"
#include "simlpy/SimulationException.h"

TrafficGenerator::TrafficGenerator() :
   m_timeDist( 0 ),
   m_maker( 0 )
{
}

TrafficGenerator::~TrafficGenerator()
{
   if ( 0 != m_timeDist ) delete m_timeDist;
   if ( 0 != m_maker ) delete m_maker;
}

Entity*
TrafficGenerator::create() const
{
   return new TrafficGenerator;
}

void
TrafficGenerator::configure( const Entity::ArgList& args )
{
   unsigned int nargs = args.size();
   if ( nargs < 2 )
   {
      std::cerr << "config() takes at least 2 arguments ("
                << nargs << " given)"
                << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }

   std::string command = parse_string(args[0]);
   if ( "time_dist" == command )
   {
      if ( ! PyCallable_Check( args[1] ) )
      {
         std::cerr << "second argument isn't callable" << std::endl;
         throw SimulationException( __FILE__ , __LINE__ );
      }
      PythonTimeDistribution* ptd = new PythonTimeDistribution;
      ptd->setFunc(args[1]);
      setTimeDistribution( ptd );
   }
   else if ( "generator" == command )
   {
      if ( ! PyCallable_Check( args[1] ) )
      {
         std::cerr << "second argument isn't callable" << std::endl;
         throw SimulationException( __FILE__ , __LINE__ );
      }
      PythonBundleMaker* pbm = new PythonBundleMaker;
      pbm->setFunc(args[1]);
      setBundleMaker( pbm );
   }
   else
   {
      std::cerr << "Unrecognized configuration option: " << command
                << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
}

void
TrafficGenerator::emit( const Entity::ArgList& args )
{
   int nargs = args.size();
   if ( nargs < 1 )
   {
      std::cerr << "emit() takes at least 1 argument ("
                << nargs << " given)"
                << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   Time t = parse_time(args[0]);
   Clock::schedule( new TrafficTrigger(this,this,t) );
}

void
TrafficGenerator::setTimeDistribution( TimeDistribution* td )
{
   if ( 0 == td ) return;
   if ( 0 != m_timeDist ) delete m_timeDist;
   m_timeDist = td;
}

void
TrafficGenerator::setBundleMaker( BundleMaker* bm )
{
   if ( 0 == bm ) return;
   if ( 0 != m_maker ) delete m_maker;
   m_maker = bm;
}

bool
TrafficGenerator::handler( TrafficTrigger& event )
{
   if ( 0 == m_timeDist ) return false;
   if ( 0 == m_maker ) return false;
   Time now = Clock::time();
   m_maker->generate();
   Time t = (*m_timeDist)( now );
   Clock::tentative( new TrafficTrigger( this, this, t ) );
   return true;
}


TrafficTrigger::TrafficTrigger( Entity* source,
                                TrafficGenerator* destination,
                                const Time& t ) :
   Event( source, t ),
   m_destination( destination )
{
}

TrafficTrigger::~TrafficTrigger()
{
}

bool
TrafficTrigger::handle()
{
   return m_destination->handler( *this );
}

---