python_init.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 "Python.h"
#include "structmember.h"

#include "EpidemicStore.h"
#include "EpidemicApp.h"
#include "EpidemicForwarding.h"
#include "EpidemicCustody.h"

#include "mobility/MobileNode.h"
#include "mobility/WirelessLink.h"
#include "pydtn/parsers.h"
#include "mobility/parsers.h"
#include "simlpy/interpreter_hooks.h"
#include "simlpy/Clock.h"

#ifndef PyMODINIT_FUNC
#define PyMODINIT_FUNC extern "C" void
#endif

static uint32_t epidemic_max_hops = 0;

static char epidemic_set_max_hops_doc[] =
"Set the maximum number of allowed hops.\n\
\n\
This takes one integer argument, which will set the number\n\
of hops that a bundle may take via epidemic forwarding\n\
before further propagation is limited to the destination\n\
node.  The default value is 0, which is unlimited forwarding.\n";

static PyObject*
epidemic_set_max_hops( PyObject* self, PyObject* args )
{
   if ( ! PyTuple_Check( args ) )
   {
      PyErr_SetString(PyExc_TypeError,
                      "set_max_hops expects a list of arguments");
      return 0;
   }
   if ( 0 == PyTuple_Size( args ) )
   {
      PyErr_SetString(PyExc_TypeError,
                      "set_max_hops expects an argument");
      return 0;
   }

   try
   {
      long int val = parse_long( PyTuple_GetItem(args,0) );
      epidemic_max_hops = val;
   }
   catch ( ... )
   {
      PyErr_SetString(PyExc_TypeError,"integer expected");
      return 0;
   }

   Py_INCREF(Py_None);
   return Py_None;
}

static char epidemic_attach_doc[] = "Add an application instance to a node.\n\
   \n\
   Calling syntax:\n\
   \n\
   pydtn.mobility.epidemic.attach(<n>,<l>)\n\
   \n\
      <n> is a mobile node to which we want to add an instance.\n\
      <l> is the wireless link for the node.\n\
   \n\
   In general, this will be called automatically at node creation by\n\
   pydtn.mobility.node()\n";

static PyObject*
epidemic_attach( PyObject* self, PyObject* args )
{
   if ( ! PyTuple_Check( args ) )
   {
      PyErr_SetString(PyExc_TypeError,"attach expects a list of arguments");
      return 0;
   }
   if ( PyTuple_Size( args ) < 2 )
   {
      PyErr_SetString(PyExc_TypeError,
                      "attach expects two arguments");
      return 0;
   }

   Mobility::MobileNode* mn = Mobility::parse_mn( PyTuple_GetItem(args,0) );
   if ( 0 == mn ) return 0;

   Mobility::WirelessLink* wl = Mobility::parse_wl( PyTuple_GetItem(args,1) );
   if ( 0 == wl ) return 0;

   Epidemic::EpidemicStore* ps =
      new Epidemic::EpidemicStore( &(mn->node()), epidemic_max_hops );
   Epidemic::EpidemicApp* ma =
      new Epidemic::EpidemicApp( ps, &(wl->link()), mn->bundleLifetime().tv );
   Epidemic::EpidemicForwarding* fp =
      new Epidemic::EpidemicForwarding( &(mn->node()), &(wl->link()), ma );
   
   mn->setPersistentStore(ps);
   mn->addMobileApp(ma);
   mn->setForwardingPolicy(fp);
   mn->node().setCustodyPolicy( new Epidemic::EpidemicCustody(&(mn->node())) );


   Py_INCREF(Py_None);
   return Py_None;
}

static PyMethodDef ExampleMethods[] =
{
   {"set_max_hops",epidemic_set_max_hops,METH_VARARGS,epidemic_set_max_hops_doc},
   {"attach",epidemic_attach,METH_VARARGS,epidemic_attach_doc},
   {0,0,0,0} // sentinel
};

PyMODINIT_FUNC
init_epidemic(void)
{
   (void) Py_InitModule("_epidemic",ExampleMethods);
}

---