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 "dtnaodv/AODVForwardingPolicy.h"
#include "dtnaodv/AODVApp.h"
#include "pydtn/WrapNode.h"
#include "mobility/WirelessLink.h"
#include "pydtn/parsers.h"
#include "mobility/parsers.h"
#include "simlpy/interpreter_hooks.h"
#include "simlpy/Clock.h"

#include <iostream>
#include <fstream>

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

static char aodv_attach_doc[] = "Add AODV routing to a node.\n\
   \n\
   Calling syntax:\n\
   \n\
   pydtn.aodv.attach(<n>,<l>)\n\
   \n\
      <n> is a node to which we want to add AODV routing.\n\
      <l> is a wireless link that will employ AODV.\n";

static PyObject*
aodv_attach( PyObject* self, PyObject* args )
{
   if ( ! PyTuple_Check( args ) )
   {
      PyErr_SetString(PyExc_TypeError,"attach expects a list of arguments");
      return 0;
   }
   if ( 0 == PyTuple_Size( args ) )
   {
      PyErr_SetString(PyExc_TypeError,
                      "attach expects a non-empty argument list");
      return 0;
   }

   WrapNode* wn = parse_node( PyTuple_GetItem( args, 0 ) );
   if ( 0 == wn ) return 0;

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

   uint16_t prefix = 0;

   AODVForwardingPolicy* pol =
      new AODVForwardingPolicy( &(wn->node()), prefix );
   if ( 0 == pol ) return 0;
   pol->addLink( &(wl->link()) );
   wn->setForwardingPolicy(pol);

   AODVApp* app = new AODVApp( pol );
   if ( 0 == app ) return 0;
   wn->addApplication(app);

   Py_INCREF(Py_None);
   return Py_None;
}

static PyMethodDef ExampleMethods[] =
{
   {"attach",aodv_attach,METH_VARARGS,aodv_attach_doc},
   {0,0,0,0} // sentinel
};

PyMODINIT_FUNC
init_aodv(void)
{
   (void) Py_InitModule("_aodv",ExampleMethods);
}

---