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 "NamTracer.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 NamTracer* nam_tracer = 0;

static char namtrace_setup_doc[] = "NAM tracing configuration.\n\
   \n\
   This method should be called before creating any nodes if\n\
   nam tracing is desired.\n\
   \n\
   Calling syntax:\n\
   \n\
   pydtn.namtrace.setup(<fname>)\n\
   \n\
      <fname> is either a filename or one of 'stdout' or 'stderr'.\n\
      In the former case, the nam tracing information is written\n\
      to the file named.  In the latter cases, the information is\n\
      written to the appropriate stream.\n";

static PyObject*
namtrace_setup( PyObject* self, PyObject* args )
{
   if ( 0 != nam_tracer )
   {
      std::cerr << "nam tracing already enabled" << std::endl;
      return 0;
   }

   const char* fname;
   if ( ! PyArg_ParseTuple(args, "s", &fname) )
   {
      std::cerr << "Incorrect argument list" << std::endl;
      return 0;
   }
   std::string fnstr( fname ); // This lets us use operator==.
   GlobalTracer::enable();
   GlobalTracer* gt = GlobalTracer::instance();
   if ( 0 == gt )
   {
      std::cerr << "Could not instantiate the global tracer" << std::endl;
      return 0;
   }

   if ( "stdout" == fnstr )
   {
      nam_tracer = new NamTracer( std::cout );
   }
   else if ( "stderr" == fnstr )
   {
      nam_tracer = new NamTracer( std::cerr );
   }
   else
   {
      nam_tracer = new NamTracer( *(new std::ofstream(fname)) );
   }
   gt->addTracer(nam_tracer);

   Py_INCREF(Py_None);
   return Py_None;
}

static char namtrace_queue_doc[] = "Configure the queue tracing behavior.\n\
   \n\
   Calling syntax:\n\
   \n\
   pydtn.namtrace.queue(<state>)\n\
   \n\
      <state> is either True or False, and determines whether or\n\
      not queue information should be included in the trace.\n";

static PyObject*
namtrace_queue( PyObject* self, PyObject* args )
{
   if ( 0 == nam_tracer )
   {
      std::cerr << "nam tracing must first be enabled" << std::endl;
      return 0;
   }

   bool q;
   if ( ! PyArg_ParseTuple(args, "b", &q) )
   {
      std::cerr << "Incorrect argument list" << std::endl;
      return 0;
   }
   nam_tracer->setQueueState(q);

   Py_INCREF(Py_None);
   return Py_None;
}

static PyMethodDef ExampleMethods[] =
{
   {"setup",namtrace_setup,METH_VARARGS,namtrace_setup_doc},
   {"queue",namtrace_queue,METH_VARARGS,namtrace_queue_doc},
   {0,0,0,0} // sentinel
};

PyMODINIT_FUNC
init_namtrace(void)
{
   (void) Py_InitModule("_namtrace",ExampleMethods);
}

---