RootNTuple.h

Go to the documentation of this file.

/* -*- c++ -*- */

#ifndef RootNTuple_H
#define RootNTuple_H

#include "RootDataType.h"

#include "datasrcs/DataSource.h"

class TTree;
class TBranch;

namespace hippodraw {

class RootBranch;

class MDL_HIPPOPLOT_API RootNTuple : public DataSource
{

 private:

  typedef std::vector < RootBranch * > BranchList_t;

  BranchList_t m_data;

  typedef std::vector < std::vector < double > * > CacheList_t;
  CacheList_t m_data_cache;

  std::vector < bool > m_multiDimensional;
  
  TTree * m_tree;

  mutable unsigned int m_cur_entry;

  bool rotateWheels( std::vector< int > & wheels,
                     const std::vector < int > & sz,
                     unsigned int d );

  int addColumn ( const std::string & label, RootBranch * );

  TBranch * getTBranch(  const std::string & column );
  

protected:

  RootNTuple ();

  RootNTuple ( const RootNTuple & );

 public:

  RootNTuple ( TTree * tree );

  virtual ~RootNTuple();

  virtual void notifyObservers ( ) const;

  virtual unsigned int rows () const;

  virtual bool empty () const;

  virtual void clear ();

  virtual void reserve ( unsigned int );

  virtual double 
  operator [] (  std::vector < unsigned int > & indices ) const;

  virtual double valueAt ( unsigned int row, unsigned int column ) const;

  virtual double valueAtNoCache ( unsigned int row,
                                  unsigned int column ) const;

  hippodraw::RootData::Type getType ( unsigned int column ) const;

  virtual double * doubleArrayAt ( unsigned int row,
                                   unsigned int column ) const;

  virtual float * floatArrayAt ( unsigned int row,
                                 unsigned int column ) const;

  virtual int * intArrayAt ( unsigned int row,
                             unsigned int column ) const;

  virtual unsigned int * uintArrayAt ( unsigned int row,
                                       unsigned int column ) const;

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

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

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

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

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

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

  virtual void setShape ( std::vector < unsigned int > & shape );

  const std::vector < unsigned int > & getShape () const;

  virtual bool isMultiDimensional ( const std::string & column ) const;
    
  virtual bool isUseable ( const std::string & label ) const;

  virtual bool setReleventIndex( const std::string & column,
                                 const std::vector< unsigned int >& index );
  
  virtual void fillShape ( std::vector < int > & v,
                           unsigned int column ) const;

  void smartExpandRootNTuple( const std::string & column ) const;

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

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

private:

  void expandIfNeeded ( const std::string & label ) const;

  void initBranches ();

  void fillDataCache ( unsigned int column );
  
};

} // namespace hippodraw

#endif // RootNTuple_H