NTFillIterator.h

Go to the documentation of this file.

#ifndef __NTFILLITERATOR_H
#define __NTFILLITERATOR_H

#include "jbltools/sfh/FillIterator.h"
 
#include <Rtypes.h>     // declares Float_t
#include <TTree.h>    
#include <TBranch.h>    
#include <cassert>   
#include <cstring>   
#include <stdio.h>   
#include <iostream>
 
// Class NTFillIterator

template<class Ntuple, class T = Int_t> 
class NTFillIterator: public FillIterator {
  public:
    NTFillIterator (const Ntuple& nt_,        
                    const char *branchname    
                   )
    : FillIterator (branchname ? branchname : "?"),
      nt(nt_), 
      index (0)           
    {
      TTree *t = nt.fChain;
      assert (t);
      TBranch *b = t->GetBranch (branchname);
      assert (b);
      branchAddress = reinterpret_cast<T *>(b->GetAddress());
      assert (branchAddress);
      
#if (DEBUG>1)
      std::cout << "NTFillIterator: created '" << getName() << "'\n";
#endif
    }
    
    NTFillIterator (const Ntuple& nt_,              
                    T Ntuple::* p_mem,             
                    const char *objectname = "?"   
                   )
    : FillIterator (objectname ? objectname : "?"),
      nt(nt_), 
      index (0)           
    {
      branchAddress = &(nt.*p_mem);
      assert (branchAddress);
#if (DEBUG>1)
      std::cout << "NTFillIterator: created '" << getName() << "'\n";
#endif
    }
  
    virtual int operator() () const {
      return (index < *branchAddress) ? index : -1;
    }

    virtual bool next() {
      return (++index < *branchAddress);
    }

    virtual bool reset() {
      index = 0;
      return (*branchAddress > 0);
    }
    
    virtual bool isValid() const {
      return (index >= 0) && (index < *branchAddress);
    }              
 
  protected:  
    virtual ~NTFillIterator() {}
    
    // Data members
    const Ntuple& nt;                 
    const T *branchAddress;           
    int     index;                    
};


template<class Ntuple, class T> 
NTFillIterator<Ntuple, T>& ntfilliterator (const Ntuple& nt,            
                                           T Ntuple::* p_mem,           
                                           const char *objectname = "?" 
                                  ) {
  return *new NTFillIterator<Ntuple, T> (nt, p_mem, objectname);
}                                  

#endif /* #ifndef __NTFILLITERATOR_H */

---