TreeReader.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TreeReader_h
#define TreeReader_h
 
//
//      Class TreeReader
//      TreeReader.h
//
// Class for Tree reading through TFomula
 
#include "TTree.h"
#include "TTreeFormula.h"
#include "TTreeFormulaManager.h"
#include "TString.h"
#include <map>
#include <vector>
/*
class std::vector<double>;
class std::vector<int>;
class std::vector<std::vector<double> >;
class std::vector<std::vector<int> >;*/
 
class TreeReader //: public TTreeFormulaManager
{
 private:
 
  TTree* m_tree;
  int    m_currentEntry = 0;            // current ntuple entry stored in buffer
  int    m_entries;                 // total number of entries
  bool   m_isChain = false;
  int    m_currentTree = 0;
  std::map<std::string, TTreeFormula*>     m_formulae;  // known formulae
 
 public:
 
  TreeReader();               // Default ctor
  virtual ~TreeReader();      // dtor
 
  TreeReader(TTree* n);       // ctor with ntuple
 
  void   SetTree(TTree* n);       //
  double GetVariable(const char* c, int entry=-2); // return variable s for a given entry (<0 -> current entry)
  int    GetEntry(int entry=-1);     // Read a given entry in the buffer (-1 -> next entry);
  int    GetEntries()             { return m_entries ; }
  TTree* GetTree()                { return m_tree    ; }
  void   Restart()                    {m_currentEntry = -1;}
 
  ClassDef(TreeReader,1)  // Integrate this class into ROOT (must be the last member)
  
};
 
#endif