QtRootNTuple.h

Go to the documentation of this file.

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

#ifndef _QtRootNTuple_H_
#define _QtRootNTuple_H_

#include "RootNTuple.h"

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

namespace hippodraw {

class QtRootNTuple : public RootNTuple
{
private:

  QtRootNTuple ( );

  static void sliceRowDimension ( std::vector < int > & shape );

public:

  QtRootNTuple ( TTree * tree );

  unsigned int columns () const;

  unsigned int rows () const;

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

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

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

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

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

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

  bool isMultiDimensional ( const std::string & column ) const;

  const std::vector < int > getColumnShape ( const std::string & column );


  boost::python::numeric::array
  valueAt ( unsigned int row, const std::string & variable );

  boost::python::numeric::array
  getColumnAsArray ( const std::string & variable );


  void expandIfNeeded ( const std::vector < std::string > & labels ) const;

  std::string 
  createBinding ( const std::string & name, 
                  const std::vector < int > & indices ) const;
};

} // namespace hippodraw

#endif // _QtRootNTuple_H_