IntFunIterator.h

Go to the documentation of this file.

#ifndef __INTFUNITERATOR_H
#define __INTFUNITERATOR_H

#include "jbltools/sfh/IntFun.h"
 
#include <Rtypes.h>     // declares Float_t
#include <TTree.h>    
#include <TBranch.h>    
#include <cassert>   
 
// Class IntFunIterator

class IntFunIterator: public FillIterator {
  public:
    IntFunIterator (const IntFun& intFun_    
                   )
    : FillIterator (std::string(intFun_.getName())+"_iterator"),
      intFun (intFun_), 
      index (0)             
    {
      assert (&intFun);
      assert (!intFun.getIterator());
    }
    IntFunIterator (const IntFun& intFun_,   
                    const std::string& name_ 
                    )
    : FillIterator (name_),
      intFun (intFun_), 
      index (0)             
    {
      assert (&intFun);
      assert (!intFun.getIterator());
    }
  
    virtual int operator() () const {
      return (index < intFun()) ? index : -1;
    }

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

    virtual bool reset() {
      index = 0;
      return (intFun() > 0);
    }
    
    virtual bool isValid() const {
      return (index >= 0) && (index < intFun());
    }              
 
  protected:  
    virtual ~IntFunIterator() {}
    
    // Data members
    const IntFun& intFun;                 
    int index;                            
};


#endif /* #ifndef __INTFUNITERATOR_H */

---