#include <TreeReader.h>

Classes
struct	Bool

class	Value

Public Member Functions
void	addBranch (const std::string &expression, AtomicId name=AtomicId(), bool opt=true)

void	addBranch (TBranch *branch, AtomicId name=AtomicId(), bool opt=true)

template<>
void	addMulti (AtomicId name, const std::vector< Bool_t > *value)

template<>
void	addMulti (AtomicId name, const std::vector< Double_t > *value)

template<>
void	addMulti (AtomicId name, const std::vector< Float_t > *value)

template<>
void	addMulti (AtomicId name, const std::vector< Int_t > *value)

template<typename T >
void	addMulti (AtomicId name, const std::vector< T > *value)

template<>
void	addSingle (AtomicId name, const Bool_t *value, bool opt)

template<>
void	addSingle (AtomicId name, const Bool_t *value, bool opt)

template<>
void	addSingle (AtomicId name, const Double_t *value, bool opt)

template<>
void	addSingle (AtomicId name, const Double_t *value, bool opt)

template<>
void	addSingle (AtomicId name, const Float_t *value, bool opt)

template<>
void	addSingle (AtomicId name, const Float_t *value, bool opt)

template<>
void	addSingle (AtomicId name, const Int_t *value, bool opt)

template<>
void	addSingle (AtomicId name, const Int_t *value, bool opt)

template<typename T >
void	addSingle (AtomicId name, const T *value, bool opt=false)

void	addTypeMulti (AtomicId name, const void *value, char type)

void	addTypeSingle (AtomicId name, const void *value, char type, bool opt)

void	automaticAdd (bool skipTarget=false, bool skipWeight=false)

Variable::ValueList	fill ()

double	fill (const MVAComputer *mva)

uint64_t	loop (const MVAComputer *mva)

TreeReader &	operator= (const TreeReader &orig)

void	reset ()

void	setOptional (AtomicId name, bool opt, double optVal=kOptVal)

void	setTree (TTree *tree)

	TreeReader ()

	TreeReader (const TreeReader &orig)

	TreeReader (TTree *tree, bool skipTarget=false, bool skipWeight=false)

void	update ()

std::vector< AtomicId >	variables () const

virtual	~TreeReader ()

Static Public Attributes
static const double	kOptVal = -999.0

Private Attributes
std::vector< std::pair< void *, std::vector< Bool_t > > >	multiBool

std::vector< std::pair< void *, std::vector< Double_t > > >	multiDouble

std::vector< std::pair< void *, std::vector< Float_t > > >	multiFloat

std::vector< std::pair< void *, std::vector< Int_t > > >	multiInt

std::vector< Bool >	singleBool

std::vector< Double_t >	singleDouble

std::vector< Float_t >	singleFloat

std::vector< Int_t >	singleInt

TTree *	tree

bool	upToDate

std::map< AtomicId, Value >	valueMap

Variable::ValueList	values

Friends
class	Value

Detailed Description

Definition at line 19 of file TreeReader.h.

Constructor & Destructor Documentation

◆ TreeReader() [1/3]

PhysicsTools::TreeReader::TreeReader ( )

Definition at line 25 of file TreeReader.cc.

25 : tree(nullptr), upToDate(false) {}

◆ TreeReader() [2/3]

PhysicsTools::TreeReader::TreeReader ( const TreeReader & orig )

Definition at line 27 of file TreeReader.cc.

27 { this->operator=(orig); }

References operator=().

◆ TreeReader() [3/3]

PhysicsTools::TreeReader::TreeReader	(	TTree *	tree,
		bool	skipTarget = `false`,
		bool	skipWeight = `false`
	)

Definition at line 29 of file TreeReader.cc.

                                                                       : tree(tree), upToDate(false) {
     automaticAdd(skipTarget, skipWeight);
   }

References automaticAdd().

◆ ~TreeReader()

PhysicsTools::TreeReader::~TreeReader ( )

virtual

Definition at line 33 of file TreeReader.cc.

33 {}

Member Function Documentation

◆ addBranch() [1/2]

void PhysicsTools::TreeReader::addBranch	(	const std::string &	expression,
		AtomicId	name = `AtomicId()`,
		bool	opt = `true`
	)

Definition at line 60 of file TreeReader.cc.

                                                                                  {
     if (!tree)
       throw cms::Exception("NoTreeAvailable") << "No TTree set in TreeReader::addBranch." << std::endl;
  
     TBranch *branch = tree->GetBranch(expression.c_str());
     if (!branch)
       throw cms::Exception("BranchMissing") << "Tree branch \"" << expression << "\" missing." << std::endl;
  
     addBranch(branch, name, opt);
   }

References MicroEventContent_cff::branch, Exception, Skims_PA_cff::name, and runTheMatrix::opt.

Referenced by automaticAdd().

◆ addBranch() [2/2]

void PhysicsTools::TreeReader::addBranch	(	TBranch *	branch,
		AtomicId	name = `AtomicId()`,
		bool	opt = `true`
	)

Definition at line 71 of file TreeReader.cc.

                                                                      {
     TString branchName = branch->GetName();
     if (!name)
       name = (const char *)branchName;
  
     TLeaf *leaf = dynamic_cast<TLeaf *>(branch->GetLeaf(branchName));
     if (!leaf)
       throw cms::Exception("InvalidBranch") << "Tree branch \"" << branchName << "\" has no leaf." << std::endl;
  
     TString typeName = leaf->GetTypeName();
     char typeId = 0;
     bool multi = false;
     if (typeName == "Double_t" || typeName == "double")
       typeId = 'D';
     else if (typeName == "Float_t" || typeName == "float")
       typeId = 'F';
     else if (typeName == "Int_t" || typeName == "int")
       typeId = 'I';
     else if (typeName == "Bool_t" || typeName == "bool")
       typeId = 'B';
     else {
       multi = true;
       if (typeName == "vector<double>" || typeName == "Vector<Double_t>")
         typeId = 'D';
       else if (typeName == "vector<float>" || typeName == "Vector<Float_t>")
         typeId = 'F';
       else if (typeName == "vector<int>" || typeName == "Vector<Int_t>")
         typeId = 'I';
       else if (typeName == "vector<bool>" || typeName == "Vector<Bool_t>")
         typeId = 'B';
     }
  
     if (!typeId)
       throw cms::Exception("InvalidBranch") << "Tree branch \"" << branchName
                                             << "\" is of "
                                                "unsupported type \""
                                             << typeName << "\"." << std::endl;
  
     if (multi)
       addTypeMulti(name, nullptr, typeId);
     else
       addTypeSingle(name, nullptr, typeId, opt);
  
     valueMap[name].setBranchName(branch->GetName());
   }

References addTypeMulti(), addTypeSingle(), MicroEventContent_cff::branch, electrons_cff::branchName, Exception, trackerHitRTTI::multi, Skims_PA_cff::name, runTheMatrix::opt, dqmiodumpindices::typeName, and valueMap.

◆ addMulti() [1/5]

void PhysicsTools::TreeReader::addMulti	(	AtomicId	name,
		const std::vector< Bool_t > *	value
	)

Definition at line 421 of file TreeReader.cc.

◆ addMulti() [2/5]

void PhysicsTools::TreeReader::addMulti	(	AtomicId	name,
		const std::vector< Double_t > *	value
	)

Definition at line 418 of file TreeReader.cc.

◆ addMulti() [3/5]

void PhysicsTools::TreeReader::addMulti	(	AtomicId	name,
		const std::vector< Float_t > *	value
	)

Definition at line 419 of file TreeReader.cc.

◆ addMulti() [4/5]

void PhysicsTools::TreeReader::addMulti	(	AtomicId	name,
		const std::vector< Int_t > *	value
	)

Definition at line 420 of file TreeReader.cc.

◆ addMulti() [5/5]

template<typename T >

void PhysicsTools::TreeReader::addMulti	(	AtomicId	name,
		const std::vector< T > *	value
	)

◆ addSingle() [1/9]

template<>

void PhysicsTools::TreeReader::addSingle	(	AtomicId	name,
		const Bool_t *	value,
		bool	opt
	)

◆ addSingle() [2/9]

template<>

void PhysicsTools::TreeReader::addSingle	(	AtomicId	name,
		const Bool_t *	value,
		bool	opt
	)

Definition at line 421 of file TreeReader.cc.

◆ addSingle() [3/9]

template<>

void PhysicsTools::TreeReader::addSingle	(	AtomicId	name,
		const Double_t *	value,
		bool	opt
	)

◆ addSingle() [4/9]

template<>

void PhysicsTools::TreeReader::addSingle	(	AtomicId	name,
		const Double_t *	value,
		bool	opt
	)

Definition at line 418 of file TreeReader.cc.

◆ addSingle() [5/9]

template<>

void PhysicsTools::TreeReader::addSingle	(	AtomicId	name,
		const Float_t *	value,
		bool	opt
	)

◆ addSingle() [6/9]

template<>

void PhysicsTools::TreeReader::addSingle	(	AtomicId	name,
		const Float_t *	value,
		bool	opt
	)

Definition at line 419 of file TreeReader.cc.

◆ addSingle() [7/9]

template<>

void PhysicsTools::TreeReader::addSingle	(	AtomicId	name,
		const Int_t *	value,
		bool	opt
	)

◆ addSingle() [8/9]

template<>

void PhysicsTools::TreeReader::addSingle	(	AtomicId	name,
		const Int_t *	value,
		bool	opt
	)

Definition at line 420 of file TreeReader.cc.

◆ addSingle() [9/9]

template<typename T >

void PhysicsTools::TreeReader::addSingle	(	AtomicId	name,
		const T *	value,
		bool	opt = `false`
	)

◆ addTypeMulti()

void PhysicsTools::TreeReader::addTypeMulti	(	AtomicId	name,
		const void *	value,
		char	type
	)

Definition at line 173 of file TreeReader.cc.

                                                                            {
     std::map<AtomicId, Value>::const_iterator pos = valueMap.find(name);
     if (pos != valueMap.end())
       throw cms::Exception("DuplicateVariable") << "Duplicate Variable \"" << name << "\"." << std::endl;
  
     if (type != 'D' && type != 'F' && type != 'I' && type != 'B')
       throw cms::Exception("InvalidType") << "Unsupported type '" << type
                                           << "' in call to"
                                              "TreeReader::addTypeMulti."
                                           << std::endl;
  
     int index = -1;
     if (!value) {
       switch (type) {
         case 'D':
           index = (int)multiDouble.size();
           multiDouble.push_back(makeMulti<Double_t>());
           break;
         case 'F':
           index = (int)multiFloat.size();
           multiFloat.push_back(makeMulti<Float_t>());
           break;
         case 'I':
           index = (int)multiInt.size();
           multiInt.push_back(makeMulti<Int_t>());
           break;
         case 'B':
           index = (int)multiBool.size();
           multiBool.push_back(makeMulti<Bool_t>());
           break;
       }
     }
  
     valueMap[name] = Value(index, true, false, type);
     if (value)
       valueMap[name].setPtr(value);
  
     upToDate = false;
   }

References Exception, createfilelist::int, multiBool, multiDouble, multiFloat, multiInt, Skims_PA_cff::name, upToDate, Value, and valueMap.

Referenced by addBranch().

◆ addTypeSingle()

void PhysicsTools::TreeReader::addTypeSingle	(	AtomicId	name,
		const void *	value,
		char	type,
		bool	opt
	)

Definition at line 128 of file TreeReader.cc.

                                                                                       {
     std::map<AtomicId, Value>::const_iterator pos = valueMap.find(name);
     if (pos != valueMap.end())
       throw cms::Exception("DuplicateVariable") << "Duplicate Variable \"" << name << "\"." << std::endl;
  
     if (type != 'D' && type != 'F' && type != 'I' && type != 'B')
       throw cms::Exception("InvalidType") << "Unsupported type '" << type
                                           << "' in call to"
                                              "TreeReader::addTypeSingle."
                                           << std::endl;
  
     int index = -1;
     if (!value) {
       switch (type) {
         case 'D':
           index = (int)singleDouble.size();
           singleDouble.push_back(Double_t());
           break;
         case 'F':
           index = (int)singleFloat.size();
           singleFloat.push_back(Float_t());
           break;
         case 'I':
           index = (int)singleInt.size();
           singleInt.push_back(Int_t());
           break;
         case 'B':
           index = (int)singleBool.size();
           singleBool.push_back(Bool());
           break;
       }
     }
  
     valueMap[name] = Value(index, false, opt, type);
     if (value)
       valueMap[name].setPtr(value);
  
     upToDate = false;
   }

References Exception, createfilelist::int, Skims_PA_cff::name, runTheMatrix::opt, singleBool, singleDouble, singleFloat, singleInt, upToDate, Value, and valueMap.

Referenced by addBranch().

◆ automaticAdd()

void PhysicsTools::TreeReader::automaticAdd	(	bool	skipTarget = `false`,
		bool	skipWeight = `false`
	)

Definition at line 213 of file TreeReader.cc.

                                                                 {
     if (!tree)
       throw cms::Exception("NoTreeAvailable") << "No TTree set in TreeReader::automaticAdd." << std::endl;
  
     TIter iter(tree->GetListOfBranches());
     TObject *obj;
     while ((obj = iter())) {
       TBranch *branch = dynamic_cast<TBranch *>(obj);
       if (!branch)
         continue;
  
       if (skipTarget && !std::strcmp(branch->GetName(), "__TARGET__"))
         continue;
  
       if (skipWeight && !std::strcmp(branch->GetName(), "__WEIGHT__"))
         continue;
  
       addBranch(branch);
     }
   }

References addBranch(), MicroEventContent_cff::branch, Exception, and getGTfromDQMFile::obj.

Referenced by TreeReader().

◆ fill() [1/2]

Variable::ValueList PhysicsTools::TreeReader::fill ( void )

Definition at line 286 of file TreeReader.cc.

                                      {
     for (std::map<AtomicId, Value>::const_iterator iter = valueMap.begin(); iter != valueMap.end(); iter++)
       iter->second.fill(iter->first, this);
  
     Variable::ValueList result = values;
     values.clear();
  
     return result;
   }

References PhysicsTools::Variable::ValueList::clear(), mps_fire::result, valueMap, and values.

Referenced by loop().

◆ fill() [2/2]

double PhysicsTools::TreeReader::fill ( const MVAComputer * mva )

Definition at line 276 of file TreeReader.cc.

                                                 {
     for (std::map<AtomicId, Value>::const_iterator iter = valueMap.begin(); iter != valueMap.end(); iter++)
       iter->second.fill(iter->first, this);
  
     double result = mva->eval(values);
     values.clear();
  
     return result;
   }

References PhysicsTools::Variable::ValueList::clear(), beam_dqm_sourceclient-live_cfg::mva, mps_fire::result, valueMap, and values.

◆ loop()

uint64_t PhysicsTools::TreeReader::loop ( const MVAComputer * mva )

Definition at line 260 of file TreeReader.cc.

                                                   {
     if (!tree)
       throw cms::Exception("NoTreeAvailable") << "No TTree set in TreeReader::automaticAdd." << std::endl;
  
     if (!upToDate)
       update();
  
     Long64_t entries = tree->GetEntries();
     for (Long64_t entry = 0; entry < entries; entry++) {
       tree->GetEntry(entry);
       fill(mva);
     }
  
     return entries;
   }

References mps_splice::entry, Exception, fill(), beam_dqm_sourceclient-live_cfg::mva, update(), and upToDate.

◆ operator=()

TreeReader & PhysicsTools::TreeReader::operator= ( const TreeReader & orig )

Definition at line 35 of file TreeReader.cc.

                                                           {
     reset();
  
     tree = orig.tree;
  
     multiDouble.resize(orig.multiDouble.size());
     multiFloat.resize(orig.multiFloat.size());
     multiInt.resize(orig.multiInt.size());
     multiBool.resize(orig.multiBool.size());
  
     singleDouble.resize(orig.singleDouble.size());
     singleFloat.resize(orig.singleFloat.size());
     singleInt.resize(orig.singleInt.size());
     singleBool.resize(orig.singleBool.size());
  
     valueMap = orig.valueMap;
  
     return *this;
   }

References multiBool, multiDouble, multiFloat, multiInt, reset(), singleBool, singleDouble, singleFloat, singleInt, tree, and valueMap.

Referenced by TreeReader().

◆ reset()

void PhysicsTools::TreeReader::reset ( void )

Definition at line 234 of file TreeReader.cc.

                          {
     multiDouble.clear();
     multiFloat.clear();
     multiInt.clear();
     multiBool.clear();
  
     singleDouble.clear();
     singleFloat.clear();
     singleInt.clear();
     singleBool.clear();
  
     valueMap.clear();
  
     upToDate = false;
   }

References multiBool, multiDouble, multiFloat, multiInt, singleBool, singleDouble, singleFloat, singleInt, upToDate, and valueMap.

Referenced by operator=().

◆ setOptional()

void PhysicsTools::TreeReader::setOptional	(	AtomicId	name,
		bool	opt,
		double	optVal = `kOptVal`
	)

Definition at line 117 of file TreeReader.cc.

                                                                      {
     std::map<AtomicId, Value>::iterator pos = valueMap.find(name);
     if (pos == valueMap.end())
       throw cms::Exception("UnknownVariable") << "Variable \"" << name
                                               << "\" is not known to the "
                                                  "TreeReader."
                                               << std::endl;
  
     pos->second.setOpt(opt, optVal);
   }

References Skims_PA_cff::name, runTheMatrix::opt, and valueMap.

◆ setTree()

void PhysicsTools::TreeReader::setTree ( TTree * tree )

Definition at line 55 of file TreeReader.cc.

                                       {
     this->tree = tree;
     upToDate = false;
   }

References tree, and upToDate.

◆ update()

void PhysicsTools::TreeReader::update ( )

Definition at line 250 of file TreeReader.cc.

                           {
     if (!tree)
       throw cms::Exception("NoTreeAvailable") << "No TTree set in TreeReader::automaticAdd." << std::endl;
  
     for (std::map<AtomicId, Value>::iterator iter = valueMap.begin(); iter != valueMap.end(); iter++)
       iter->second.update(this);
  
     upToDate = true;
   }

References Exception, upToDate, and valueMap.

Referenced by progressbar.ProgressBar::__next__(), MatrixUtil.Matrix::__setitem__(), MatrixUtil.Steps::__setitem__(), progressbar.ProgressBar::finish(), loop(), and MatrixUtil.Steps::overwrite().

◆ variables()

std::vector< AtomicId > PhysicsTools::TreeReader::variables ( ) const

Definition at line 296 of file TreeReader.cc.

                                                   {
     std::vector<AtomicId> result;
     for (std::map<AtomicId, Value>::const_iterator iter = valueMap.begin(); iter != valueMap.end(); iter++)
       result.push_back(iter->first);
  
     return result;
   }

References mps_fire::result, and valueMap.

Friends And Related Function Documentation

◆ Value

friend class Value

friend

Definition at line 91 of file TreeReader.h.

Referenced by addTypeMulti(), and addTypeSingle().

Member Data Documentation

◆ kOptVal

const double PhysicsTools::TreeReader::kOptVal = -999.0

static

Definition at line 54 of file TreeReader.h.

◆ multiBool

std::vector<std::pair<void *, std::vector<Bool_t> > > PhysicsTools::TreeReader::multiBool

private

Definition at line 99 of file TreeReader.h.

Referenced by addTypeMulti(), operator=(), and reset().

◆ multiDouble

std::vector<std::pair<void *, std::vector<Double_t> > > PhysicsTools::TreeReader::multiDouble

private

Definition at line 96 of file TreeReader.h.

Referenced by addTypeMulti(), operator=(), and reset().

◆ multiFloat

std::vector<std::pair<void *, std::vector<Float_t> > > PhysicsTools::TreeReader::multiFloat

private

Definition at line 97 of file TreeReader.h.

Referenced by addTypeMulti(), operator=(), and reset().

◆ multiInt

std::vector<std::pair<void *, std::vector<Int_t> > > PhysicsTools::TreeReader::multiInt

private

Definition at line 98 of file TreeReader.h.

Referenced by addTypeMulti(), operator=(), and reset().

◆ singleBool

std::vector<Bool> PhysicsTools::TreeReader::singleBool

private

Definition at line 104 of file TreeReader.h.

Referenced by addTypeSingle(), operator=(), and reset().

◆ singleDouble

std::vector<Double_t> PhysicsTools::TreeReader::singleDouble

private

Definition at line 101 of file TreeReader.h.

Referenced by addTypeSingle(), operator=(), and reset().

◆ singleFloat

std::vector<Float_t> PhysicsTools::TreeReader::singleFloat

private

Definition at line 102 of file TreeReader.h.

Referenced by addTypeSingle(), operator=(), and reset().

◆ singleInt

std::vector<Int_t> PhysicsTools::TreeReader::singleInt

private

Definition at line 103 of file TreeReader.h.

Referenced by addTypeSingle(), operator=(), and reset().

◆ tree

TTree* PhysicsTools::TreeReader::tree

private

Definition at line 94 of file TreeReader.h.

Referenced by svgfig.SVG::__str__(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer::beginLoop(), MetTreeProducer.MetTreeProducer::declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer::declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer::fillTree(), operator=(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer::process(), MetTreeProducer.MetTreeProducer::process(), python.cmstools.EventTree::SetAlias(), and setTree().

◆ upToDate

bool PhysicsTools::TreeReader::upToDate

private

Definition at line 108 of file TreeReader.h.

Referenced by addTypeMulti(), addTypeSingle(), loop(), reset(), setTree(), and update().

◆ valueMap

std::map<AtomicId, Value> PhysicsTools::TreeReader::valueMap

private

Definition at line 106 of file TreeReader.h.

Referenced by addBranch(), addTypeMulti(), addTypeSingle(), fill(), operator=(), reset(), setOptional(), update(), and variables().

◆ values

Variable::ValueList PhysicsTools::TreeReader::values

private

Definition at line 107 of file TreeReader.h.

Referenced by fill(), PhysicsTools::TreeReader::Value::fill(), and genericValidation.ValidationWithPlotsSummaryBase.SummaryItem::value().

Classes

Public Member Functions

Static Public Attributes

Private Attributes

Friends

Detailed Description

Constructor & Destructor Documentation

◆ TreeReader() [1/3]

◆ TreeReader() [2/3]

◆ TreeReader() [3/3]

◆ ~TreeReader()

Member Function Documentation

◆ addBranch() [1/2]

◆ addBranch() [2/2]

◆ addMulti() [1/5]

◆ addMulti() [2/5]

◆ addMulti() [3/5]

◆ addMulti() [4/5]

◆ addMulti() [5/5]

◆ addSingle() [1/9]

◆ addSingle() [2/9]

◆ addSingle() [3/9]

◆ addSingle() [4/9]

◆ addSingle() [5/9]

◆ addSingle() [6/9]

◆ addSingle() [7/9]

◆ addSingle() [8/9]

◆ addSingle() [9/9]

◆ addTypeMulti()

◆ addTypeSingle()

◆ automaticAdd()

◆ fill() [1/2]

◆ fill() [2/2]

◆ loop()

◆ operator=()

◆ reset()

◆ setOptional()

◆ setTree()

◆ update()

◆ variables()

Friends And Related Function Documentation

◆ Value

Member Data Documentation

◆ kOptVal

◆ multiBool

◆ multiDouble

◆ multiFloat

◆ multiInt

◆ singleBool

◆ singleDouble

◆ singleFloat

◆ singleInt

◆ tree

◆ upToDate

◆ valueMap

◆ values