PhysicsTools::TreeReader Class Reference

#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<typename T >
void	addMulti (AtomicId name, const std::vector< 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<>
void	addMulti (AtomicId name, const std::vector< Bool_t > *value)
template<typename T >
void	addSingle (AtomicId name, const T *value, bool opt=false)
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 Int_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 Float_t *value, bool opt)
template<>
void	addSingle (AtomicId name, const Int_t *value, bool opt)
template<>
void	addSingle (AtomicId name, const Bool_t *value, bool opt)
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)
double	fill (const MVAComputer *mva)
Variable::ValueList	fill ()
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.

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

TreeReader() [2/3]

PhysicsTools::TreeReader::TreeReader

(

const TreeReader &

orig

)

Definition at line 27 of file TreeReader.cc.

References operator=().

{ this->operator=(orig); }

TreeReader() [3/3]

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

Definition at line 29 of file TreeReader.cc.

References automaticAdd().

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

~TreeReader()

PhysicsTools::TreeReader::~TreeReader ( )

virtual

Definition at line 33 of file TreeReader.cc.

{}

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.

References MicroEventContent_cff::branch, Exception, l1tGTMenu_BTagSeeds_cff::expression, Skims_PA_cff::name, and das-up-to-nevents::opt.

Referenced by automaticAdd().

                                                                                 {
    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);
  }

addBranch() [2/2]

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

Definition at line 71 of file TreeReader.cc.

References addTypeMulti(), addTypeSingle(), MicroEventContent_cff::branch, electrons_cff::branchName, Exception, trackerHitRTTI::multi, Skims_PA_cff::name, das-up-to-nevents::opt, dqmiodumpindices::typeName, and valueMap.

                                                                     {
    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());
  }

addMulti() [1/5]

template<typename T >

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

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]

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

Definition at line 421 of file TreeReader.cc.

addSingle() [1/9]

template<typename T >

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

addSingle() [2/9]

template<>

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

addSingle() [3/9]

template<>

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

addSingle() [4/9]

template<>

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

addSingle() [5/9]

template<>

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

addSingle() [6/9]

template<>

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

Definition at line 418 of file TreeReader.cc.

addSingle() [7/9]

template<>

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

Definition at line 419 of file TreeReader.cc.

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<>

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

Definition at line 421 of file TreeReader.cc.

addTypeMulti()

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

Definition at line 173 of file TreeReader.cc.

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

Referenced by addBranch().

                                                                           {
    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;
  }

addTypeSingle()

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

Definition at line 128 of file TreeReader.cc.

References Exception, createfilelist::int, Skims_PA_cff::name, das-up-to-nevents::opt, singleBool, singleDouble, singleFloat, singleInt, upToDate, Value, and valueMap.

Referenced by addBranch().

                                                                                      {
    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;
  }

automaticAdd()

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

Definition at line 213 of file TreeReader.cc.

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

Referenced by TreeReader().

                                                                {
    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);
    }
  }

fill() [1/2]

double PhysicsTools::TreeReader::fill

(

const MVAComputer *

mva

)

Definition at line 276 of file TreeReader.cc.

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

                                                {
    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;
  }

fill() [2/2]

Variable::ValueList PhysicsTools::TreeReader::fill

(

void

)

Definition at line 286 of file TreeReader.cc.

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

Referenced by loop().

                                     {
    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;
  }

loop()

uint64_t PhysicsTools::TreeReader::loop

(

const MVAComputer *

mva

)

Definition at line 260 of file TreeReader.cc.

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

                                                  {
    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;
  }

operator=()

TreeReader & PhysicsTools::TreeReader::operator=

(

const TreeReader &

orig

)

Definition at line 35 of file TreeReader.cc.

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

Referenced by TreeReader().

                                                          {
    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;
  }

reset()

void PhysicsTools::TreeReader::reset

(

void

)

Definition at line 234 of file TreeReader.cc.

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

Referenced by operator=().

                         {
    multiDouble.clear();
    multiFloat.clear();
    multiInt.clear();
    multiBool.clear();

    singleDouble.clear();
    singleFloat.clear();
    singleInt.clear();
    singleBool.clear();

    valueMap.clear();

    upToDate = false;
  }

setOptional()

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

Definition at line 117 of file TreeReader.cc.

References Skims_PA_cff::name, das-up-to-nevents::opt, and valueMap.

                                                                     {
    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);
  }

setTree()

void PhysicsTools::TreeReader::setTree

(

TTree *

tree

)

Definition at line 55 of file TreeReader.cc.

References tree, and upToDate.

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

update()

void PhysicsTools::TreeReader::update

(

)

Definition at line 250 of file TreeReader.cc.

References Exception, upToDate, and valueMap.

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

                          {
    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;
  }

variables()

std::vector< AtomicId > PhysicsTools::TreeReader::variables

(

)

const

Definition at line 296 of file TreeReader.cc.

References mps_fire::result, and valueMap.

                                                  {
    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;
  }

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(), and PhysicsTools::TreeReader::Value::fill().