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 "FlowTracer.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 FlowTracer* flow_tracer = 0;

static char flowtrace_setup_doc[] = "Flow tracing configuration.\n\
   \n\
   This method should be called before creating any nodes if\n\
   flow tracing is desired.\n\
   \n\
   Calling syntax:\n\
   \n\
   pydtn.flowtrace.setup(<fname>)\n\
   \n\
      <fname> is either a filename or one of 'stdout' or 'stderr'.\n\
      In the former case, the flow 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*
flowtrace_setup( PyObject* self, PyObject* args )
{
   if ( 0 != flow_tracer )
   {
      std::cerr << "flow 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 )
   {
      flow_tracer = new FlowTracer( std::cout );
   }
   else if ( "stderr" == fnstr )
   {
      flow_tracer = new FlowTracer( std::cerr );
   }
   else
   {
      flow_tracer = new FlowTracer( *(new std::ofstream(fname)) );
   }
   gt->addTracer(flow_tracer);

   Py_INCREF(Py_None);
   return Py_None;
}

static PyMethodDef ExampleMethods[] =
{
   {"setup",flowtrace_setup,METH_VARARGS,flowtrace_setup_doc},
   {0,0,0,0} // sentinel
};

PyMODINIT_FUNC
init_flowtrace(void)
{
   (void) Py_InitModule("_flowtrace",ExampleMethods);
}

---