PyDataSource.h

Go to the documentation of this file.

/* -*- mode: c++ -*- */

#ifndef _PyDataSource_H
#define _PyDataSource_H

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <exception>
#include <string>
#include <vector>

#include <boost/python.hpp>

class DataSource;

class PyDataSource {

public:

   PyDataSource();

   PyDataSource(const std::string & dataSource);

   PyDataSource ( const std::string & name, DataSource * source );

   ~PyDataSource();

   const DataSource & dataSource() const { return *m_dataSource; }

   unsigned int columns() const;

   unsigned int rows() const;

   const std::string & getTitle () const;

   void setTitle(const std::string & title);

   void setName(const std::string & name);

   const std::vector<std::string> & getLabels() const;

   const std::vector<double> & getColumn(const std::string & name) const;

   const std::vector<double> & getColumn(unsigned int index) const;

   void replaceColumn(const std::string &,
                      const std::vector<double> & col);

   void replaceColumn(unsigned int index, 
                      const std::vector<double> & col);

#ifdef HAVE_NUMARRAY

   void replaceColumn ( const std::string &,
                        boost::python::numeric::array array );

   void replaceColumn ( unsigned int index, 
                        boost::python::numeric::array array);
#endif

   int addColumn( const std::string & label,
                  const std::vector<double> & col );

#ifdef HAVE_NUMARRAY

   int addColumn( const std::string & label, 
                  boost::python::numeric::array array );
#endif

   void clear();

   bool hasColumn(const std::string & colname) const;

   std::string registerNTuple();

   void registerNTuple( const std::string & name );

#ifdef HAVE_NUMARRAY

   boost::python::numeric::array
   columnAsNumArray( const std::string & label ) const;

   boost::python::numeric::array
   columnAsNumArray( unsigned int index ) const;

   void saveColumnFromNumArray( const std::string & label,
                                boost::python::numeric::array array );

   void saveColumnFromNumArray( unsigned int index,
                                boost::python::numeric::array array );
#endif // HAVE_NUMARRAY

   class StopIteration : public std::exception {
   public:
      StopIteration(const std::string & what) : m_what(what) {}
      ~StopIteration() throw() {}
      const char * what() const throw() {
         return m_what.c_str();
      }
   private:
      const std::string m_what;
   };

private:

   std::string m_type;

   DataSource * m_dataSource;

};

#endif // _PyDataSource_H

---