PyDataSource.h

Go to the documentation of this file.

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

#ifndef _PyDataSource_H
#define _PyDataSource_H

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

#ifndef _MSC_VER
#include <inttypes.h>
#endif

namespace boost {
  namespace python {
    namespace numeric {
      class array;
    }
  }
}

namespace hippodraw {

  class DataSource;

class PyDataSource
{

private:

  void checkRank ( boost::python::numeric::array array );

public:

   PyDataSource();

   PyDataSource(const std::string & dataSource);

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

   virtual ~PyDataSource();

   void static extractVector ( boost::python::numeric::array array,
                               std::vector<double> & col );

  template < typename T >
  void static copy_direct ( boost::python::numeric::array array,
                            std::vector<double > & col );

  static PyDataSource * getCurrentDataSource ();

   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);

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

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

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

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

   void clear();

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

   std::string registerNTuple();

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

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

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

  void saveColumn ( const std::string & label,
                    const std::vector < double > & v,
                    const std::vector < intptr_t > & shape );

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

   void saveColumnFromNumArray( unsigned int index,
                                boost::python::numeric::array array );


  void saveColumnFrom ( const std::string & label,
                        const std::vector < double > & array );

  void addRow ( const std::vector < double > & array );

  void append ( const DataSource * source );

  void append ( const PyDataSource * source );

   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;

};

} // namespace hippodraw

#endif // _PyDataSource_H