Helper class to easily to add xAOD::IParticles and associated decorator variables to the MuonTesterTree. More...

#include <IParticleFourMomBranch.h>

Inheritance diagram for MuonVal::IParticleFourMomBranch:

Collaboration diagram for MuonVal::IParticleFourMomBranch:

Public Types
using	DataDependency = SG::VarHandleKey *

Public Member Functions
	IParticleFourMomBranch (MuonTesterTree &tree, const std::string &particle)
	Construct an IParticleFourMomBranch. More...

bool	fill (const EventContext &ctx) override
	Fill the branch. More...

bool	init () override
	Initialize the branches and all children. More...

std::string	name () const override
	Name of the four momentum branch. More...

const TTree *	tree () const override
	Return the underyling TTree pointer. More...

TTree *	tree () override
	Returns the pointer to the underlying TTree object. More...

MuonTesterTree &	getTree ()
	Access to the MuonTrestTree parent. More...

bool	initialized () const
	Has the init method been called. More...

std::vector< DataDependency >	data_dependencies () override
	Returns a list of all Read(Coond)HandleKeys needed by the branch. More...

size_t	size () const
	How many particles have been pushed back already. More...

void	push_back (const xAOD::IParticle *p) override
	Similar to the IAuxElementDecoration branch but only accepting IParticles. More...

void	push_back (const xAOD::IParticle &p) override

void	operator+= (const xAOD::IParticle *p) override

void	operator+= (const xAOD::IParticle &p) override

size_t	find (const xAOD::IParticle &p) const
	Returns the position at which the particle has already be inserted in the chain – If the particle is not present then the size is returned – In cases of nullptr -1 is returned. More...

size_t	find (const xAOD::IParticle *p) const

size_t	find (std::function< bool(const xAOD::IParticle *)> func) const
	Find the first particle satisfying the selection criteria given by the client. More...

const std::vector< const xAOD::IParticle * > &	getCached () const
	Returns the list of currently cached particles. More...

template<typename T >
bool	addVariable (const std::string &variable, const std::string &accName="")
	Writes a variable of type <T> which can be directly accessed via an SG::AuxElement e.g. More...

template<typename T >
bool	addVariable (T defaultValue, const std::string &variable, const std::string &accName="")
	Write a variable of type <T> to the TTree. More...

template<typename T >
bool	addVariableGeV (const std::string &variable, const std::string &accName="")
	Write a variable of type <T> to the TTree, but divide it by 1k before dumping. More...

template<typename T >
bool	addVariableGeV (T defaultValue, const std::string &variable, const std::string &accName="")
	Write a variable of type <T> to the TTree, but divide it by 1k before dumping. More...

template<typename T >
std::shared_ptr< ParticleVariableBranch< T > >	getBranch (const std::string &var) const
	Returns the point to the underlying ParticleVariableBranch translating the IParticle Accessors into TTree branches. More...

bool	addVariable (std::shared_ptr< IParticleDecorationBranch > branch)

bool	msgLvl (const MSG::Level lvl) const
	Test the output level. More...

MsgStream &	msg () const
	The standard message stream. More...

MsgStream &	msg (const MSG::Level lvl) const
	The standard message stream. More...

void	setLevel (MSG::Level lvl)
	Change the current logging level. More...

Private Member Functions
void	initMessaging () const
	Initialize our message level and MessageSvc. More...

Private Attributes
MuonTesterTree &	m_parent

std::string	m_name {}

bool	m_init {false}

bool	m_updated {false}

VectorBranch< float >	m_pt {m_parent.tree(), m_name +"_pt"}

VectorBranch< float >	m_eta {m_parent.tree(), m_name + "_eta"}

VectorBranch< float >	m_phi {m_parent.tree(), m_name + "_phi"}

VectorBranch< float >	m_e {m_parent.tree(), m_name + "_e"}

VectorBranch< int >	m_q {m_parent.tree(), m_name+"_q"}

std::vector< std::shared_ptr< IParticleDecorationBranch > >	m_variables {}

std::vector< const xAOD::IParticle * >	m_cached_particles {}

std::string	m_nm
	Message source name. More...

boost::thread_specific_ptr< MsgStream >	m_msg_tls
	MsgStream instance (a std::cout like with print-out levels) More...

std::atomic< IMessageSvc * >	m_imsg { nullptr }
	MessageSvc pointer. More...

std::atomic< MSG::Level >	m_lvl { MSG::NIL }
	Current logging level. More...

std::atomic_flag m_initialized	ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
	Messaging initialized (initMessaging) More...

Detailed Description

Helper class to easily to add xAOD::IParticles and associated decorator variables to the MuonTesterTree.

The branch saves, the particle 4-momentum (E,Pt, Eta, Phi). Energy and Pt are saved in units of GeV. Extra information decorated to the particle can also easily appended to the tree output.

Potential setup:
auto part_br = std::make_shared<IParticleFourMomBranch>(m_testerTree, "TheCakeIsALie"); /// Adds a branch ptcone30 and fetches the infromation from ptcone30 part_br->addVariable<float>("ptcone30"); /// Adds a branch caloIso and fetches the information from etcone20 part_br->addVariable<float>("caloIso", "etcone20");

m_testerTree.addBranch(part_br); Then fill the branches: for (const xAOD::IParticle* part : *container) { part_br->push_back(part); }

Definition at line 31 of file IParticleFourMomBranch.h.

Member Typedef Documentation

◆ DataDependency

using MuonVal::IMuonTesterBranch::DataDependency = SG::VarHandleKey*

inherited

Definition at line 39 of file IMuonTesterBranch.h.

Constructor & Destructor Documentation

◆ IParticleFourMomBranch()

MuonVal::IParticleFourMomBranch::IParticleFourMomBranch	(	MuonTesterTree &	tree,
		const std::string &	particle
	)

Construct an IParticleFourMomBranch.

Parameters

tree	Instance of the parental MuonTesterTree to which the IParticleFourMom branch is added
particle	Name of the IParticleFourMomBranch inside the TTree

Definition at line 16 of file IParticleFourMomBranch.cxx.

                                                                          : 
         AthMessaging{"IParticle "+particle},
         m_parent{tree},
         m_name{particle}{}

Member Function Documentation

◆ addVariable() [1/3]

template<typename T >

bool MuonVal::IParticleFourMomBranch::addVariable	(	const std::string &	variable,
		const std::string &	accName = `""`
	)

Writes a variable of type <T> which can be directly accessed via an SG::AuxElement e.g.

SG::AuxElement::Accessor<float> acc_ptCone30{"pt_cone30"}; const float ptCone30 = acc_ptCone30(*partPtr); to the TTree. If the variable is not available the filling of the TTree fails

Parameters

variable	Name of the branch in the output tree
accName	Name of the accessor to obtain the information. If left empty, it's assumed that the accesor is the same as the variable name

◆ addVariable() [2/3]

bool MuonVal::IParticleFourMomBranch::addVariable ( std::shared_ptr< IParticleDecorationBranch > branch )

Definition at line 136 of file IParticleFourMomBranch.cxx.

                                                                                             {
         if (initialized()) {
             ATH_MSG_ERROR("The branch is already initialized. Cannot add more branches");
             return false;
         } else if (!branch) {
            ATH_MSG_ERROR("Nullptr is given.");
             return false;
         }
  
         auto itr = std::find_if(m_variables.begin(), m_variables.end(),
                                 [&branch](const std::shared_ptr<IParticleDecorationBranch>& known) {
                                     return known == branch || known->name() == branch->name();
                                 });
         if (itr != m_variables.end()) {
             if (typeid((*itr).get()) != typeid(branch.get())) {
                 ATH_MSG_ERROR("Different branches have been added here under "<<branch->name()<<".");
                 return false;
             }
             return true;
         }
         m_variables.push_back(branch);
         return m_parent.registerBranch(branch);
     }

◆ addVariable() [3/3]

template<typename T >

bool MuonVal::IParticleFourMomBranch::addVariable	(	T	defaultValue,
		const std::string &	variable,
		const std::string &	accName = `""`
	)

Write a variable of type <T> to the TTree.

If the information is not decorated to the particle a default value is used

Parameters

defValue	Default value to use as a backup option
variable	Name of the branch in the output tree
accName	Name of the accessor to obtain the information. If left empty, it's assumed that the accesor is the same as the variable name

◆ addVariableGeV() [1/2]

template<typename T >

bool MuonVal::IParticleFourMomBranch::addVariableGeV	(	const std::string &	variable,
		const std::string &	accName = `""`
	)

Write a variable of type <T> to the TTree, but divide it by 1k before dumping.

Useful for the conversion from MeV to GeV

Parameters

variable	Name of the branch in the output tree \| *
accName	Name of the accessor to obtain the information. If left empty, it's assumed that the accesor is the same as the variable name

◆ addVariableGeV() [2/2]

template<typename T >

bool MuonVal::IParticleFourMomBranch::addVariableGeV	(	T	defaultValue,
		const std::string &	variable,
		const std::string &	accName = `""`
	)

Write a variable of type <T> to the TTree, but divide it by 1k before dumping.

Useful for the conversion from MeV to GeV

Parameters

defValue	Default value to use as a backup option
variable	Name of the branch in the output tree
accName	Name of the accessor to obtain the information. If left empty, it's assumed that the accesor is the same as the variable name

◆ data_dependencies()

std::vector< IParticleFourMomBranch::DataDependency > MuonVal::IParticleFourMomBranch::data_dependencies ( )

overridevirtual

Returns a list of all Read(Coond)HandleKeys needed by the branch.

Implements MuonVal::IMuonTesterBranch.

Definition at line 159 of file IParticleFourMomBranch.cxx.

                                                                                               {
         std::vector<DataDependency> to_ret{};
         for (const std::shared_ptr<IParticleDecorationBranch>& br : m_variables) {
             std::vector<DataDependency> childDep{br->data_dependencies()};
             to_ret.insert(to_ret.end(),childDep.begin(), childDep.end());
         }
         return to_ret;
     }

◆ fill()

bool MuonVal::IParticleFourMomBranch::fill ( const EventContext & ctx )

overridevirtual

Fill the branch.

Implements MuonVal::IMuonTesterBranch.

Definition at line 92 of file IParticleFourMomBranch.cxx.

                                                              {
         if (!m_updated) {
             m_cached_particles.clear();
         }
         if (!m_pt.fill(ctx) || !m_eta.fill(ctx) || !m_phi.fill(ctx) || !m_e.fill(ctx) || !m_q.fill(ctx)){
             return false;
         }
         for (const auto& var : m_variables) {
             if (!var->fill(ctx)) {
                 return false;
             }
         }
         m_init = true;
         m_updated = false;
         return true;
     }

◆ find() [1/3]

size_t MuonVal::IParticleFourMomBranch::find ( const xAOD::IParticle & p ) const

Returns the position at which the particle has already be inserted in the chain – If the particle is not present then the size is returned – In cases of nullptr -1 is returned.

Definition at line 68 of file IParticleFourMomBranch.cxx.

                                                                     {
         return find(&p);
     }

◆ find() [2/3]

size_t MuonVal::IParticleFourMomBranch::find ( const xAOD::IParticle * p ) const

Definition at line 71 of file IParticleFourMomBranch.cxx.

                                                                     {
         if (!p) {
             return dummyIdx;
         }
         return find([p](const xAOD::IParticle* cached) { 
             return p == cached; 
         });
     }

◆ find() [3/3]

size_t MuonVal::IParticleFourMomBranch::find ( std::function< bool(const xAOD::IParticle *)> func ) const

Find the first particle satisfying the selection criteria given by the client.

If none of the particle is actually matching the current size of the branch is returned

Definition at line 79 of file IParticleFourMomBranch.cxx.

                                                                                           {
         if (!m_updated) {
             return dummyIdx;
         }
         size_t j{0};
         for (const xAOD::IParticle* p : m_cached_particles) {
             if (func(p)) {
                 return j;
             }
             ++j;
         }
         return dummyIdx;
     }

◆ getBranch()

template<typename T >

std::shared_ptr<ParticleVariableBranch<T> > MuonVal::IParticleFourMomBranch::getBranch ( const std::string & var ) const

Returns the point to the underlying ParticleVariableBranch translating the IParticle Accessors into TTree branches.

Parameters

variable Name of the branch as parsed in addVariable(GeV)

◆ getCached()

const std::vector< const xAOD::IParticle * > & MuonVal::IParticleFourMomBranch::getCached ( ) const

Returns the list of currently cached particles.

Definition at line 25 of file IParticleFourMomBranch.cxx.

                                                                                   {
         return m_cached_particles;
     }

◆ getTree()

MuonTesterTree & MuonVal::IParticleFourMomBranch::getTree ( )

Access to the MuonTrestTree parent.

Definition at line 22 of file IParticleFourMomBranch.cxx.

                                                     {
         return m_parent;
     }

◆ init()

bool MuonVal::IParticleFourMomBranch::init ( )

overridevirtual

Initialize the branches and all children.

Implements MuonVal::IMuonTesterBranch.

Definition at line 111 of file IParticleFourMomBranch.cxx.

                                       {
         if (initialized()) {
             return true;
         }
  
         if (!m_pt.init() || !m_eta.init() || !m_phi.init() || !m_e.init() || !m_q.init()) {
             return false;
         }
         std::vector<IMuonTesterBranch*> branchesToInit{};
         for (auto& v : m_variables) {
             branchesToInit.push_back(v.get());
         }
         std::sort(branchesToInit.begin(), branchesToInit.end(),
                 [](IMuonTesterBranch* a, IMuonTesterBranch* b) {
                     return a->name() < b->name();
                 });
         for (auto& br : branchesToInit) {
             if (!br->init()) {
                 ATH_MSG_ERROR("Failed to initialize "<<br->name());
                 return false;
             }
         }
         return true;
     }

◆ initialized()

bool MuonVal::IParticleFourMomBranch::initialized ( ) const

Has the init method been called.

Definition at line 108 of file IParticleFourMomBranch.cxx.

                                                    {
         return m_init;
     }

◆ initMessaging()

void AthMessaging::initMessaging ( ) const

privateinherited

Initialize our message level and MessageSvc.

This method should only be called once.

Definition at line 39 of file AthMessaging.cxx.

 {
   m_imsg = Athena::getMessageSvc();
   m_lvl = m_imsg ?
     static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
     MSG::INFO;
 }

◆ msg() [1/2]

MsgStream & AthMessaging::msg ( ) const

inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 164 of file AthMessaging.h.

 {
   MsgStream* ms = m_msg_tls.get();
   if (!ms) {
     if (!m_initialized.test_and_set()) initMessaging();
     ms = new MsgStream(m_imsg,m_nm);
     m_msg_tls.reset( ms );
   }
  
   ms->setLevel (m_lvl);
   return *ms;
 }

◆ msg() [2/2]

MsgStream & AthMessaging::msg ( const MSG::Level lvl ) const

inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 179 of file AthMessaging.h.

180 { return msg() << lvl; }

◆ msgLvl()

bool AthMessaging::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Test the output level.

Parameters

lvl	The message level to test against

Returns: boolean Indicating if messages at given level will be printed

Return values

true	Messages at level "lvl" will be printed

Definition at line 151 of file AthMessaging.h.

 {
   if (!m_initialized.test_and_set()) initMessaging();
   if (m_lvl <= lvl) {
     msg() << lvl;
     return true;
   } else {
     return false;
   }
 }

◆ name()

std::string MuonVal::IParticleFourMomBranch::name ( ) const

overridevirtual

Name of the four momentum branch.

Implements MuonVal::IMuonTesterBranch.

Definition at line 30 of file IParticleFourMomBranch.cxx.

30 { return m_name; }

◆ operator+=() [1/2]

void MuonVal::IParticleFourMomBranch::operator+= ( const xAOD::IParticle & p )

overridevirtual

Implements MuonVal::IParticleDecorationBranch.

Definition at line 34 of file IParticleFourMomBranch.cxx.

34 {push_back(p); }

◆ operator+=() [2/2]

void MuonVal::IParticleFourMomBranch::operator+= ( const xAOD::IParticle * p )

overridevirtual

Implements MuonVal::IParticleDecorationBranch.

Definition at line 33 of file IParticleFourMomBranch.cxx.

33 {push_back(p); }

◆ push_back() [1/2]

void MuonVal::IParticleFourMomBranch::push_back ( const xAOD::IParticle & p )

overridevirtual

Implements MuonVal::IParticleDecorationBranch.

Definition at line 35 of file IParticleFourMomBranch.cxx.

35 { push_back(&p); }

◆ push_back() [2/2]

void MuonVal::IParticleFourMomBranch::push_back ( const xAOD::IParticle * p )

overridevirtual

Similar to the IAuxElementDecoration branch but only accepting IParticles.

Avoid that the particle is added twice to the tree

Implements MuonVal::IParticleDecorationBranch.

Definition at line 36 of file IParticleFourMomBranch.cxx.

                                                                  {
         if (!m_updated) {
             m_cached_particles.clear();
             m_updated = true;
         }
         if (!p || find(p) < size()){
             if (p) ATH_MSG_VERBOSE("Rejected particle ("<<p<<") "<<p->pt()<<", "<<p->eta()<<", "<<p->phi()<<". Size: "<<size());
             return;
         }
  
         m_cached_particles.push_back(p);
         m_pt.push_back(p->pt() * MeVtoGeV);
         m_eta.push_back(p->eta());
         m_phi.push_back(p->phi());
         m_e.push_back(p->e() * MeVtoGeV);
         int q{0};
         if (acc_charge.isAvailable(*p)) {
             q = acc_charge(*p);
         } else if (p->type() == xAOD::Type::ObjectType::TrackParticle){
             q = static_cast<const xAOD::TrackParticle*>(p)->charge();
         } else if (p->type() == xAOD::Type::ObjectType::TruthParticle) {
             q = static_cast<const xAOD::TruthParticle*>(p)->charge();
         }
         ATH_MSG_VERBOSE("New particle ("<<p<<") "<<p->pt()<<", "<<p->eta()<<", "<<p->phi()<<", q: "<<q<<". Size: "<<size());
  
         m_q.push_back(q);
         for (const auto& var : m_variables) {
             var->push_back(p);
         }
     }

◆ setLevel()

void AthMessaging::setLevel ( MSG::Level lvl )

inherited

Change the current logging level.

Use this rather than msg().setLevel() for proper operation with MT.

Definition at line 28 of file AthMessaging.cxx.

 {
   m_lvl = lvl;
 }

◆ size()

size_t MuonVal::IParticleFourMomBranch::size ( ) const

How many particles have been pushed back already.

Definition at line 29 of file IParticleFourMomBranch.cxx.

29 { return m_updated ? m_cached_particles.size() : 0; }

◆ tree() [1/2]

const TTree * MuonVal::IParticleFourMomBranch::tree ( ) const

overridevirtual

Return the underyling TTree pointer.

Implements MuonVal::IMuonTesterBranch.

Definition at line 31 of file IParticleFourMomBranch.cxx.

31 { return m_pt.tree(); }

◆ tree() [2/2]

TTree * MuonVal::IParticleFourMomBranch::tree ( )

overridevirtual

Returns the pointer to the underlying TTree object.

Implements MuonVal::IMuonTesterBranch.

Definition at line 32 of file IParticleFourMomBranch.cxx.

32 { return m_pt.tree(); }

Member Data Documentation

◆ ATLAS_THREAD_SAFE

std::atomic_flag m_initialized AthMessaging::ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT

mutableprivateinherited

Messaging initialized (initMessaging)

Definition at line 141 of file AthMessaging.h.

◆ m_cached_particles

std::vector<const xAOD::IParticle*> MuonVal::IParticleFourMomBranch::m_cached_particles {}

private

Definition at line 137 of file IParticleFourMomBranch.h.

◆ m_e

VectorBranch<float> MuonVal::IParticleFourMomBranch::m_e {m_parent.tree(), m_name + "_e"}

private

Definition at line 133 of file IParticleFourMomBranch.h.

◆ m_eta

VectorBranch<float> MuonVal::IParticleFourMomBranch::m_eta {m_parent.tree(), m_name + "_eta"}

private

Definition at line 131 of file IParticleFourMomBranch.h.

◆ m_imsg

std::atomic<IMessageSvc*> AthMessaging::m_imsg { nullptr }

mutableprivateinherited

MessageSvc pointer.

Definition at line 135 of file AthMessaging.h.

◆ m_init

bool MuonVal::IParticleFourMomBranch::m_init {false}

private

Definition at line 128 of file IParticleFourMomBranch.h.

◆ m_lvl

std::atomic<MSG::Level> AthMessaging::m_lvl { MSG::NIL }

mutableprivateinherited

Current logging level.

Definition at line 138 of file AthMessaging.h.

◆ m_msg_tls

boost::thread_specific_ptr<MsgStream> AthMessaging::m_msg_tls

mutableprivateinherited

MsgStream instance (a std::cout like with print-out levels)

Definition at line 132 of file AthMessaging.h.

◆ m_name

std::string MuonVal::IParticleFourMomBranch::m_name {}

private

Definition at line 127 of file IParticleFourMomBranch.h.

◆ m_nm

std::string AthMessaging::m_nm

privateinherited

Message source name.

Definition at line 129 of file AthMessaging.h.

◆ m_parent

MuonTesterTree& MuonVal::IParticleFourMomBranch::m_parent

private

Definition at line 126 of file IParticleFourMomBranch.h.

◆ m_phi

VectorBranch<float> MuonVal::IParticleFourMomBranch::m_phi {m_parent.tree(), m_name + "_phi"}

private

Definition at line 132 of file IParticleFourMomBranch.h.

◆ m_pt

VectorBranch<float> MuonVal::IParticleFourMomBranch::m_pt {m_parent.tree(), m_name +"_pt"}

private

Definition at line 130 of file IParticleFourMomBranch.h.

◆ m_q

VectorBranch<int> MuonVal::IParticleFourMomBranch::m_q {m_parent.tree(), m_name+"_q"}

private

Definition at line 134 of file IParticleFourMomBranch.h.

◆ m_updated

bool MuonVal::IParticleFourMomBranch::m_updated {false}

private

Definition at line 129 of file IParticleFourMomBranch.h.

◆ m_variables

std::vector<std::shared_ptr<IParticleDecorationBranch> > MuonVal::IParticleFourMomBranch::m_variables {}

private

Definition at line 136 of file IParticleFourMomBranch.h.

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

Public Types

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ DataDependency

Constructor & Destructor Documentation

◆ IParticleFourMomBranch()

Member Function Documentation

◆ addVariable() [1/3]

◆ addVariable() [2/3]

◆ addVariable() [3/3]

◆ addVariableGeV() [1/2]

◆ addVariableGeV() [2/2]

◆ data_dependencies()

◆ fill()

◆ find() [1/3]

◆ find() [2/3]

◆ find() [3/3]

◆ getBranch()

◆ getCached()

◆ getTree()

◆ init()

◆ initialized()

◆ initMessaging()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ name()

◆ operator+=() [1/2]

◆ operator+=() [2/2]

◆ push_back() [1/2]

◆ push_back() [2/2]

◆ setLevel()

◆ size()

◆ tree() [1/2]

◆ tree() [2/2]

Member Data Documentation

◆ ATLAS_THREAD_SAFE

◆ m_cached_particles

◆ m_e

◆ m_eta

◆ m_imsg

◆ m_init

◆ m_lvl

◆ m_msg_tls

◆ m_name

◆ m_nm

◆ m_parent

◆ m_phi

◆ m_pt

◆ m_q

◆ m_updated

◆ m_variables