TreeReader Class Reference

#include <TreeReader.h>

Collaboration diagram for TreeReader:

Public Member Functions
	TreeReader ()
virtual	~TreeReader ()
	TreeReader (TTree *n)
void	SetTree (TTree *n)
double	GetVariable (const char *c, int entry=-2)
int	GetEntry (int entry=-1)
int	GetEntries ()
TTree *	GetTree ()
void	Restart ()

Private Attributes
TTree *	m_tree
int	m_currentEntry = 0
int	m_entries
bool	m_isChain = false
int	m_currentTree = 0
std::map< std::string, TTreeFormula * >	m_formulae

Detailed Description

Definition at line 29 of file TreeReader.h.

Constructor & Destructor Documentation

TreeReader() [1/2]

TreeReader::TreeReader

(

)

Definition at line 26 of file TreeReader.cxx.

{
  // Default constructor.
  m_isChain = false;
  m_currentTree = -1;
  m_tree = nullptr;
  m_currentEntry = -1;
  m_entries = -1;
}

~TreeReader()

TreeReader::~TreeReader ( )

virtual

Definition at line 36 of file TreeReader.cxx.

{
  m_formulae.clear();
}

TreeReader() [2/2]

TreeReader::TreeReader

(

TTree *

n

)

Definition at line 43 of file TreeReader.cxx.

{
  // Constructor.
  m_tree = nullptr;
  m_entries = -1;
  SetTree(n);
}

Member Function Documentation

GetEntries()

int TreeReader::GetEntries ( )

inline

Definition at line 50 of file TreeReader.h.

{ return m_entries ; }

GetEntry()

int TreeReader::GetEntry

(

int

entry = -1

)

Definition at line 102 of file TreeReader.cxx.

{
  // Read a given entry in the buffer (-1 -> next entry).
  // Return kFALSE if not found.
  //   entry += 1;
  if(m_entries==0) return 0;
  if(entry==-1) entry = m_currentEntry+1;
  if(entry<m_entries)
  {
    int entryNumber = m_tree->GetEntryNumber(entry);
    if (entryNumber < 0) return 0;
    Long64_t localEntry = m_tree->LoadTree(entryNumber);
    if (localEntry < 0) return 0;
    m_currentEntry = entry;
    if(m_isChain) // check file change in chain
    {
      int I = static_cast<TChain*>(m_tree)->GetTreeNumber();
      if(I!=m_currentTree)
      {
        m_currentTree = I;
        //fManager->Clear();
        std::map<std::string, TTreeFormula*>::iterator itr = m_formulae.begin();
        std::map<std::string, TTreeFormula*>::iterator itrE= m_formulae.end();
        TTreeFormula* dummy = m_formulae["__DUMMY__"];
        for(;itr!=itrE;++itr)
        {
          if(itr->second!=dummy) itr->second->Notify(); //itr->second->UpdateFormulaLeaves();
        }
      }
    }
    return 1;
  }
  return 0;
}

GetTree()

TTree * TreeReader::GetTree ( )

inline

Definition at line 51 of file TreeReader.h.

{ return m_tree    ; }

GetVariable()

double TreeReader::GetVariable	(	const char *	c,
		int	entry = -2 )

Definition at line 69 of file TreeReader.cxx.

{
  // Get vaviable.
  // Return variable for a given entry (<0 -> current entry).
  if(entry>=0 && entry!=m_currentEntry) this->GetEntry(entry);
  std::string s = c;
  TTreeFormula *f = m_formulae[s];
  if(!f)
  {
    f = new TTreeFormula(c,c,m_tree);
    f->SetQuickLoad(kTRUE);
    //   fManager->Add(f);
    //   fManager->Sync();
    if(f->GetNdim()!=1)  //invalid fomula
    {
      delete f;
      f = m_formulae["__DUMMY__"];
      std::cout << "in [TreeReader] : " << s << " is not valid -> return 0" << std::endl;
    }
    //  else {f->Notify();}
    m_formulae[s] = f;
  }
  if(f == m_formulae["__DUMMY__"]) return 0;
  int valid = f->GetNdata()  ;
  if(!valid) return 0;
  // std::cout << "Evaluating formula : " << s << std::flush;
  // std::cout << "  " << f->EvalInstance(0) << std::endl;
  return f->EvalInstance(0);
}

Restart()

void TreeReader::Restart ( )

inline

Definition at line 52 of file TreeReader.h.

{m_currentEntry = -1;}

SetTree()

void TreeReader::SetTree

(

TTree *

n

)

Definition at line 53 of file TreeReader.cxx.

{
  // check for null pointer BEFORE trying to use it
  if(!n) return;
  // Set tree.
  m_tree = n;
  m_currentEntry = -1;
  m_formulae.clear();
  m_formulae["__DUMMY__"] = new TTreeFormula("__DUMMY__","0",m_tree);
  m_isChain = (n->IsA() == TClass::GetClass("TChain"));
  m_currentTree = 0;
  m_entries = (int) m_tree->GetEntries();
}

Member Data Documentation

m_currentEntry

int TreeReader::m_currentEntry = 0

private

Definition at line 34 of file TreeReader.h.

m_currentTree

int TreeReader::m_currentTree = 0

private

Definition at line 37 of file TreeReader.h.

m_entries

int TreeReader::m_entries

private

Definition at line 35 of file TreeReader.h.

m_formulae

std::map<std::string, TTreeFormula*> TreeReader::m_formulae

private

Definition at line 38 of file TreeReader.h.

m_isChain

bool TreeReader::m_isChain = false

private

Definition at line 36 of file TreeReader.h.

m_tree

TTree* TreeReader::m_tree

private

Definition at line 33 of file TreeReader.h.

---

The documentation for this class was generated from the following files:

• TreeReader.h
• TreeReader.cxx