PythonTimeDistribution.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 "PythonTimeDistribution.h"
#include "simlpy/SimulationException.h"

#include <iostream>

PythonTimeDistribution::PythonTimeDistribution() :
   m_f( 0 )
{
}

PythonTimeDistribution::~PythonTimeDistribution()
{
   if ( 0 != m_f )
   {
      Py_DECREF( m_f );
   }
}

Time
PythonTimeDistribution::operator()( const Time& t )
{
   if ( 0 == m_f )
   {
      std::cerr << "No function configured" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   PyObject* l = PyList_New(0);
   if ( 0 == l )
   {
      std::cerr << "Could not allocate a new list" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   PyObject* li = PyLong_FromLong( t.tv.tv_sec );
   if ( 0 == li )
   {
      Py_DECREF(l);
      std::cerr << "Could not allocate a new long" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   if ( 0 > PyList_Append(l,li) )
   {
      Py_DECREF(l);
      Py_DECREF(li);
      std::cerr << "Error adding to the argument list" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   li = PyLong_FromLong( t.tv.tv_usec );
   if ( 0 == li )
   {
      Py_DECREF(l);
      std::cerr << "Could not allocate a new long" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   if ( 0 > PyList_Append(l,li) )
   {
      Py_DECREF(l);
      Py_DECREF(li);
      std::cerr << "Error adding to the argument list" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   PyObject* r = PyObject_CallFunctionObjArgs( m_f, l, 0 );
   Py_DECREF(l);
   if ( 0 == r )
   {
      std::cerr << "Function generated an error" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   if ( ! PyList_Check(r) )
   {
      Py_DECREF(r);
      std::cerr << "Function should have returned a list, but didn't"
                << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   if ( PyList_Size(r) < 2 )
   {
      Py_DECREF(r);
      std::cerr << "Function should have returned a list of length 2"
                << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   li = PyList_GetItem(r,0);
   if ( ! PyLong_Check( li ) )
   {
      Py_DECREF(r);
      std::cerr << "Returned list should have long int values" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   long s = PyLong_AsLong(li);
   li = PyList_GetItem(r,1);
   if ( ! PyLong_Check( li ) )
   {
      Py_DECREF(r);
      std::cerr << "Returned list should have long int values" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   long us = PyLong_AsLong(li);
   Time retval(s,us);
   Py_DECREF(r);
   return retval;
}

void
PythonTimeDistribution::setFunc( PyObject* f )
{
   if ( 0 == f )
   {
      std::cerr << "NULL object passed" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   if ( ! PyCallable_Check( f ) )
   {
      std::cerr << "Argument is not callable" << std::endl;
      throw SimulationException( __FILE__ , __LINE__ );
   }
   m_f = f;
   Py_INCREF(m_f); // take ownership of the reference
}

---