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 "SampleApp.h"
#include "Manager.h"
#include "pydtn/WrapNode.h"
#include "pydtn/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 SampleAppNS::Manager manager;

static char sampleapp_attach_doc[] = "Add an application instance to a node.\n\
   \n\
   Calling syntax:\n\
   \n\
   pydtn.sampleapp.attach(<n>)\n\
   \n\
      <n> is a node to which we want to add an instance.\n";

static PyObject*
sampleapp_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;

   SampleAppNS::SampleApp* app = new SampleAppNS::SampleApp(&manager);
   wn->addApplication( app );
   manager.add(app,wn);
   Py_INCREF(Py_None);
   return Py_None;
}

static char sampleapp_send_doc[] = "Send application data.\n\
   \n\
   Calling syntax:\n\
   \n\
   pydtn.sampleapp.send(<from>,<to>,<data>,<when>)\n\
   \n\
     <from> is the node that will send the data.  It must have\n\
     an instance of the application.\n\
     \n\
     <to> is the node that will receive the data.  It need not\n\
     have an instance of the application, but if it does it will\n\
     print a message for each application bundle received.\n\
     \n\
     <data> is the actual data to be sent.  See emit(<t>,'data',...)\n\
     from pydtn's sim.Entity('node') for details.\n\
     \n\
     <when> is the time at which the application data bundle will\n\
     be scheduled, given as a list:\n\
        [ seconds , microseconds ]\n";

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

   ItemWrapper item;

   WrapNode* from = parse_node( PyTuple_GetItem( args, 0 ) );
   WrapNode* to = parse_node( PyTuple_GetItem( args, 1 ) );

   std::string dataString = parse_string( PyTuple_GetItem( args, 2 ) );
   DTN::ByteString data;
   for ( unsigned int i = 0 ; i < dataString.length() ; ++i )
   {
      data += dataString[i];
   }

   Time when( parse_time( PyTuple_GetItem( args, 3 ) ) );

   SampleAppNS::SampleApp* app = manager.lookup(from);
   if ( 0 != app )
   {
      app->send(to->addr(),data,when);
   }

   Py_INCREF(Py_None);
   return Py_None;
}

static PyMethodDef ExampleMethods[] =
{
   {"attach",sampleapp_attach,METH_VARARGS,sampleapp_attach_doc},
   {"send",sampleapp_send,METH_VARARGS,sampleapp_send_doc},
   {0,0,0,0} // sentinel
};

PyMODINIT_FUNC
init_sampleapp(void)
{
   (void) Py_InitModule("_sampleapp",ExampleMethods);
}

---