#include <ChargedHadronSubtractionTool.h>

Inheritance diagram for ChargedHadronSubtractionTool:

Collaboration diagram for ChargedHadronSubtractionTool:

Public Member Functions
	ChargedHadronSubtractionTool (const std::string &name)

StatusCode	initialize ()
	Dummy implementation of the initialisation function. More...

StatusCode	process (xAOD::IParticleContainer *cont) const override final

virtual void	print () const
	Print the state of the tool. More...

ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	evtStore () const
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard `StoreGateSvc/DetectorStore` Returns (kind of) a pointer to the `StoreGateSvc`. More...

virtual StatusCode	sysInitialize () override
	Perform system initialization for an algorithm. More...

virtual StatusCode	sysStart () override
	Handle START transition. More...

virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles. More...

virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles. More...

Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T > &t)

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKey &hndl, const std::string &doc, const SG::VarHandleKeyType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleBase &hndl, const std::string &doc, const SG::VarHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKeyArray &hndArr, const std::string &doc, const SG::VarHandleKeyArrayType &)

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc, const SG::NotHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc="none")
	Declare a new Gaudi property. More...

void	updateVHKA (Gaudi::Details::PropertyBase &)

MsgStream &	msg () const

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

bool	msgLvl (const MSG::Level lvl) const

Protected Member Functions
StatusCode	setEtaPhi (xAOD::IParticle *obj, float eta, float phi) const

StatusCode	setEnergyPt (xAOD::IParticle obj, float e, float pt, const SG::AuxElement::Accessor< float > weightAcc=nullptr) const

StatusCode	setP4 (xAOD::IParticle obj, const xAOD::JetFourMom_t &p4, const SG::AuxElement::Accessor< float > weightAcc=nullptr) const

void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution More...

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void >	renounce (T &h)

void	extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
	Add StoreName to extra input/output deps as needed. More...

Protected Attributes
unsigned int	m_inputType =xAOD::Type::Other

bool	m_applyToChargedPFO =true

bool	m_applyToNeutralPFO =true

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
StatusCode	process_impl (xAOD::IParticleContainer *cont) const

template<class T , class U >
StatusCode	matchToPrimaryVertex (T &) const

template<class T , class U >
StatusCode	matchByPrimaryVertex (T &) const

const xAOD::Vertex *	getPrimaryVertex () const

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyArrayType &)
	specialization for handling Gaudi::Property<SG::VarHandleKeyArray> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleType &)
	specialization for handling Gaudi::Property<SG::VarHandleBase> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &t, const SG::NotHandleType &)
	specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray> More...

Static Private Member Functions
static double	calcAbsZ0SinTheta (const xAOD::TrackParticle &trk, const xAOD::Vertex &vtx)

Private Attributes
Gaudi::Property< bool >	m_useTrackToVertexTool = {this, "UseTrackToVertexTool", false, "True if we will use the track to vertex tool"}

Gaudi::Property< bool >	m_ignoreVertex = {this, "IgnoreVertex", false, "Dummy option for cosmics - accept everything"}

Gaudi::Property< float >	m_z0sinThetaCutValue = {this, "Z0sinThetaCutValue", 2.0, "True if we will use the track to vertex tool"}

Gaudi::Property< bool >	m_byVertex = {this, "DoByVertex", false, "True if we should match to each primary vertex, not just PV0"}

SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexContainer_key = {this, "VertexContainerKey", "PrimaryVertices", "key for the primary vertex container"}

SG::ReadHandleKey< jet::TrackVertexAssociation >	m_trkVtxAssoc_key = {this, "TrackVertexAssociation", "JetTrackVtxAssoc", "key for the TrackVertexAssociation object"}

StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default) More...

StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default) More...

std::vector< SG::VarHandleKeyArray * >	m_vhka

bool	m_varHandleArraysDeclared

Detailed Description

Removes charged PFO not associated to the PV

Author: John Stupak, Jennifer Roloff, and Steven Schramm

Definition at line 30 of file ChargedHadronSubtractionTool.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< AlgTool > >::StoreGateSvc_t

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ ChargedHadronSubtractionTool()

ChargedHadronSubtractionTool::ChargedHadronSubtractionTool ( const std::string & name )

Definition at line 9 of file ChargedHadronSubtractionTool.cxx.

                                                                                 : JetConstituentModifierBase(name)
 {
 }

Member Function Documentation

◆ calcAbsZ0SinTheta()

double ChargedHadronSubtractionTool::calcAbsZ0SinTheta	(	const xAOD::TrackParticle &	trk,
		const xAOD::Vertex &	vtx
	)

staticprivate

Definition at line 95 of file ChargedHadronSubtractionTool.cxx.

 {
   // vtz.z() provides z of that vertex w.r.t the center of the beamspot (z = 0).
   // Thus we correct the track z0 to be w.r.t z = 0
   const float z0 = trk.z0() + trk.vz() - vtx.z();
   const float theta = trk.theta();
   return std::abs(z0 * std::sin(theta));
 }

◆ declareGaudiProperty() [1/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleKeyArrayType &
	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKeyArray>

Definition at line 170 of file AthCommonDataStore.h.

   {
     return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                        hndl.value(), 
                                                        hndl.documentation());
  
   }

◆ declareGaudiProperty() [2/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleKeyType &
	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

   {
     return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                        hndl.value(), 
                                                        hndl.documentation());
  
   }

◆ declareGaudiProperty() [3/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleType &
	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleBase>

Definition at line 184 of file AthCommonDataStore.h.

   {
     return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                        hndl.value(), 
                                                        hndl.documentation());
   }

◆ declareGaudiProperty() [4/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	t,
		const SG::NotHandleType &
	)

inlineprivateinherited

specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray>

Definition at line 199 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(t);
   }

◆ declareProperty() [1/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		SG::VarHandleBase &	hndl,
		const std::string &	doc,
		const SG::VarHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleBase. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 245 of file AthCommonDataStore.h.

   {
     this->declare(hndl.vhKey());
     hndl.vhKey().setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [2/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKey &	hndl,
		const std::string &	doc,
		const SG::VarHandleKeyType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleKey. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 221 of file AthCommonDataStore.h.

   {
     this->declare(hndl);
     hndl.setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [3/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKeyArray &	hndArr,
		const std::string &	doc,
		const SG::VarHandleKeyArrayType &
	)

inlineinherited

Definition at line 259 of file AthCommonDataStore.h.

   {
  
     // std::ostringstream ost;
     // ost << Algorithm::name() << " VHKA declareProp: " << name 
     //     << " size: " << hndArr.keys().size() 
     //     << " mode: " << hndArr.mode() 
     //     << "  vhka size: " << m_vhka.size()
     //     << "\n";
     // debug() << ost.str() << endmsg;
  
     hndArr.setOwner(this);
     m_vhka.push_back(&hndArr);
  
     Gaudi::Details::PropertyBase* p =  PBASE::declareProperty(name, hndArr, doc);
     if (p != 0) {
       p->declareUpdateHandler(&AthCommonDataStore<PBASE>::updateVHKA, this);
     } else {
       ATH_MSG_ERROR("unable to call declareProperty on VarHandleKeyArray " 
                     << name);
     }
  
     return p;
  
   }

◆ declareProperty() [4/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc,
		const SG::NotHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This is the generic version, for types that do not derive from SG::VarHandleKey. It just forwards to the base class version of declareProperty.

Definition at line 333 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(name, property, doc);
   }

◆ declareProperty() [5/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc = `"none"`
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This dispatches to either the generic declareProperty or the one for VarHandle/Key/KeyArray.

Definition at line 352 of file AthCommonDataStore.h.

   {
     typedef typename SG::HandleClassifier<T>::type htype;
     return declareProperty (name, property, doc, htype());
   }

◆ declareProperty() [6/6]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty ( Gaudi::Property< T > & t )

inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

                                                                  {
     typedef typename SG::HandleClassifier<T>::type htype;
     return AthCommonDataStore<PBASE>::declareGaudiProperty(t, htype());
   }

◆ detStore()

const ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< AlgTool > >::detStore ( ) const

inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

95 { return m_detStore; }

◆ evtStore() [1/2]

ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( )

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

85 { return m_evtStore; }

◆ evtStore() [2/2]

const ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( ) const

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 90 of file AthCommonDataStore.h.

90 { return m_evtStore; }

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps )

protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

◆ getKey()

SG::sgkey_t asg::AsgTool::getKey ( const void * ptr ) const

inherited

Get the (hashed) key of an object that is in the event store.

This is a bit of a special one. StoreGateSvc and xAOD::TEvent both provide ways for getting the SG::sgkey_t key for an object that is in the store, based on a bare pointer. But they provide different interfaces for doing so.

In order to allow tools to efficiently perform this operation, they can use this helper function.

See also: asg::AsgTool::getName

Parameters

ptr	The bare pointer to the object that the event store should know about

Returns: The hashed key of the object in the store. If not found, an invalid (zero) key.

Definition at line 119 of file AsgTool.cxx.

                                                     {
  
 #ifdef XAOD_STANDALONE
       // In case we use @c xAOD::TEvent, we have a direct function call
       // for this.
       return evtStore()->event()->getKey( ptr );
 #else
       const SG::DataProxy* proxy = evtStore()->proxy( ptr );
       return ( proxy == nullptr ? 0 : proxy->sgkey() );
 #endif // XAOD_STANDALONE
    }

◆ getName()

const std::string & asg::AsgTool::getName ( const void * ptr ) const

inherited

Get the name of an object that is / should be in the event store.

This is a bit of a special one. StoreGateSvc and xAOD::TEvent both provide ways for getting the std::string name for an object that is in the store, based on a bare pointer. But they provide different interfaces for doing so.

In order to allow tools to efficiently perform this operation, they can use this helper function.

See also: asg::AsgTool::getKey

Parameters

ptr	The bare pointer to the object that the event store should know about

Returns: The string name of the object in the store. If not found, an empty string.

Definition at line 106 of file AsgTool.cxx.

                                                             {
  
 #ifdef XAOD_STANDALONE
       // In case we use @c xAOD::TEvent, we have a direct function call
       // for this.
       return evtStore()->event()->getName( ptr );
 #else
       const SG::DataProxy* proxy = evtStore()->proxy( ptr );
       static const std::string dummy = "";
       return ( proxy == nullptr ? dummy : proxy->name() );
 #endif // XAOD_STANDALONE
    }

◆ getPrimaryVertex()

const xAOD::Vertex * ChargedHadronSubtractionTool::getPrimaryVertex ( ) const

private

Definition at line 53 of file ChargedHadronSubtractionTool.cxx.

 {
   // Retrieve Primary Vertices
   auto handle = SG::makeHandle(m_vertexContainer_key);
   if (!handle.isValid())
   {
     ATH_MSG_WARNING(" This event has no primary vertex container");
     return nullptr;
   }
  
   const xAOD::VertexContainer *pvtxs = handle.cptr();
   if (pvtxs->empty())
   {
     ATH_MSG_WARNING(" Failed to retrieve valid primary vertex container");
     return nullptr;
   }
  
   // Usually the 0th vertex is the primary one, but this is not always
   //  the case. So we will choose the first vertex of type PriVtx
   for (const auto *theVertex : *pvtxs)
   {
     if (theVertex->vertexType() == xAOD::VxType::PriVtx)
     {
       return theVertex;
     } // If we have a vertex of type primary vertex
   }   // iterate over the vertices and check their type
  
   // If we failed to find an appropriate vertex, return the dummy vertex
   ATH_MSG_DEBUG("Could not find a primary vertex in this event");
   for (const auto *theVertex : *pvtxs)
   {
     if (theVertex->vertexType() == xAOD::VxType::NoVtx)
     {
       return theVertex;
     }
   }
  
   // If there is no primary vertex, then we cannot do PV matching.
   ATH_MSG_WARNING("Primary vertex container is empty");
   return nullptr;
 }

◆ getProperty()

template<class T >

const T* asg::AsgTool::getProperty ( const std::string & name ) const

inherited

Get one of the tool's properties.

◆ initialize()

StatusCode ChargedHadronSubtractionTool::initialize ( )

virtual

Dummy implementation of the initialisation function.

It's here to allow the dual-use tools to skip defining an initialisation function. Since many are doing so...

Reimplemented from asg::AsgTool.

Definition at line 13 of file ChargedHadronSubtractionTool.cxx.

 {
   if (m_inputType != xAOD::Type::ParticleFlow && m_inputType != xAOD::Type::FlowElement)
   {
     ATH_MSG_ERROR("ChargedHadronSubtractionTool requires PFO inputs. It cannot operate on objects of type "
                   << m_inputType);
     return StatusCode::FAILURE;
   }
   ATH_MSG_INFO("Running ChargedHadronSubtractionTool by-vertex: " << m_byVertex.value());
   ATH_MSG_INFO("Running ChargedHadronSubtractionTool using TTVA: " << m_useTrackToVertexTool.value());
  
   ATH_CHECK(m_trkVtxAssoc_key.initialize(m_useTrackToVertexTool && !m_ignoreVertex));
   ATH_CHECK(m_vertexContainer_key.initialize(!m_ignoreVertex));
  
   return StatusCode::SUCCESS;
 }

◆ inputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< AlgTool > >::inputHandles ( ) const

overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ matchByPrimaryVertex()

template<class T , class U >

StatusCode ChargedHadronSubtractionTool::matchByPrimaryVertex ( T & cont ) const

private

Definition at line 184 of file ChargedHadronSubtractionTool.h.

 {
   const static SG::AuxElement::Accessor<std::vector<unsigned>> matchingPVs("MatchingPVs");
   const static SG::AuxElement::Accessor<std::vector<unsigned>> matchingPUSBs("MatchingPUsidebands");
  
   // Retrieve Primary Vertices
   auto handle = SG::makeHandle(m_vertexContainer_key);
   if (!handle.isValid())
   {
     ATH_MSG_WARNING(" This event has no primary vertex container");
     return StatusCode::FAILURE;
   }
  
   const xAOD::VertexContainer *pvtxs = handle.cptr();
   if (pvtxs->empty())
   {
     ATH_MSG_WARNING(" Failed to retrieve valid primary vertex container");
     return StatusCode::FAILURE;
   }
  
   // Use only one of TVA or PV
   const jet::TrackVertexAssociation *trkVtxAssoc = nullptr;
   if (m_useTrackToVertexTool)
   {
     auto handle = SG::makeHandle(m_trkVtxAssoc_key);
     if (!handle.isValid())
     {
       ATH_MSG_ERROR("Can't retrieve TrackVertexAssociation : " << m_trkVtxAssoc_key.key());
       return StatusCode::FAILURE;
     }
     trkVtxAssoc = handle.cptr();
   }
  
   for (U *ppfo : cont)
   {
     // Ignore neutral PFOs
     if constexpr (std::is_same_v<U, xAOD::PFO>) {
       if (std::abs(ppfo->charge()) < FLT_MIN)
       continue;
     } else if constexpr (std::is_same_v<U, xAOD::FlowElement>) {
       if (!ppfo->isCharged())
       continue;
     }
  
     // Get the track for this charged PFO
     const xAOD::TrackParticle* ptrk = nullptr;
  
     // Use different methods to get TrackParticle based on U
     if constexpr (std::is_same_v<U, xAOD::PFO>) {
       ptrk = ppfo->track(0);
     } else if constexpr (std::is_same_v<U, xAOD::FlowElement>) {
       ptrk = dynamic_cast<const xAOD::TrackParticle*>(ppfo->chargedObject(0));
     }
     if (ptrk == nullptr)
     {
       ATH_MSG_WARNING("Charged PFO with index " << ppfo->index() << " has no ID track!");
       continue;
     }
  
     std::vector<unsigned> matchingVertexList;
     std::vector<unsigned> matchingPUSBList;
  
     if (trkVtxAssoc)
     { 
       // Use TrackVertexAssociation
       const xAOD::Vertex *thisTracksVertex = trkVtxAssoc->associatedVertex(ptrk);
       if (thisTracksVertex == nullptr)
       {
         ATH_MSG_DEBUG("No vertex associated to track " << ptrk->index() << "! So it cannot be associated to the primary vertex");
       }
       else
       {
         matchingVertexList.push_back(thisTracksVertex->index());
         for (const xAOD::Vertex *vtx : *pvtxs){
           if (vtx != nullptr && vtx->vertexType() != xAOD::VxType::NoVtx)
           {
             float z0 = ptrk->z0() + ptrk->vz() - vtx->z();
             float theta = ptrk->theta();
           
             if (std::abs(z0 * sin(theta)) < 2.0 * m_z0sinThetaCutValue && std::abs(z0 * sin(theta)) >= m_z0sinThetaCutValue)
               matchingPUSBList.push_back(vtx->index());
           }
         }
       }
     }
     else{
       // Use z0sinThetaCutValue
       // Loop over the primary vertices to determine which ones potentially match
       for (const xAOD::Vertex *vtx : *pvtxs)
       {
         bool matchedToVertex = false;
         bool matchedToPileupSideband = false;
  
         if (vtx == nullptr)
         {
           ATH_MSG_WARNING("Encountered a nullptr vertex when trying to match charged PFOs to vertices");
           continue;
         }
         else if (vtx->vertexType() == xAOD::VxType::NoVtx)
         {                         // No reconstructed vertices
           matchedToVertex = true; // simply match all cPFOs in this case
         }
         else
         { // Had a good reconstructed vertex
           const double absZ0sinTheta = calcAbsZ0SinTheta(*ptrk, *vtx);
           matchedToVertex = (absZ0sinTheta < m_z0sinThetaCutValue);
           if (absZ0sinTheta < 2.0 * m_z0sinThetaCutValue && absZ0sinTheta >= m_z0sinThetaCutValue)
             matchedToPileupSideband = true;
         }
  
         if (matchedToVertex)
           matchingVertexList.push_back(vtx->index());
         if (matchedToPileupSideband)
           matchingPUSBList.push_back(vtx->index());
       }
     }
  
     matchingPVs(*ppfo) = matchingVertexList;
     matchingPUSBs(*ppfo) = matchingPUSBList;
   }
  
   return StatusCode::SUCCESS;
 }

◆ matchToPrimaryVertex()

template<class T , class U >

StatusCode ChargedHadronSubtractionTool::matchToPrimaryVertex ( T & cont ) const

private

Definition at line 65 of file ChargedHadronSubtractionTool.h.

 {
   const static SG::AuxElement::Accessor<char> PVMatchedAcc("matchedToPV");
   const static SG::AuxElement::Accessor<char> PUsidebandMatchedAcc("matchedToPUsideband");
  
   // Use only one of TVA or PV
   const jet::TrackVertexAssociation *trkVtxAssoc = nullptr;
   const xAOD::Vertex *vtx = nullptr;
   if (!m_ignoreVertex)
   {
     // In cosmics, there's no PV container so we need to avoid attempting
     // to retrieve anything related to it
     if (m_useTrackToVertexTool)
     {
       auto handle = SG::makeHandle(m_trkVtxAssoc_key);
       if (!handle.isValid())
       {
         ATH_MSG_ERROR("Can't retrieve TrackVertexAssociation : " << m_trkVtxAssoc_key.key());
         return StatusCode::FAILURE;
       }
       trkVtxAssoc = handle.cptr();
     }
     else
     {
       vtx = getPrimaryVertex();
       if (vtx == nullptr)
       {
         ATH_MSG_ERROR("Primary vertex container was empty or no valid vertex found!");
         return StatusCode::FAILURE;
       }
       else if (vtx->vertexType() == xAOD::VxType::NoVtx)
       {
         ATH_MSG_VERBOSE("No genuine primary vertex found. Will consider all PFOs matched.");
       }
     }
   }
  
   for (U *ppfo : cont)
   {
     // Ignore neutral PFOs
     if constexpr (std::is_same_v<U, xAOD::PFO>) {
       if (std::abs(ppfo->charge()) < FLT_MIN)
       continue;
     } else if constexpr (std::is_same_v<U, xAOD::FlowElement>) {
       if (!ppfo->isCharged())
       continue;
     }
  
     bool matchedToPrimaryVertex = false;
     bool matchedToPileupSideband = false;
     if (m_ignoreVertex)
     {
       // If we don't use vertex information, don't bother computing the decision
       // Just pass every cPFO -- there shouldn't be many in cosmics!
       matchedToPrimaryVertex = true;
     }
     else
     {
       const xAOD::TrackParticle* ptrk = nullptr;
  
       // Use different methods to get TrackParticle based on U
       if constexpr (std::is_same_v<U, xAOD::PFO>) {
         ptrk = ppfo->track(0);
       } else if constexpr (std::is_same_v<U, xAOD::FlowElement>) {
         ptrk = dynamic_cast<const xAOD::TrackParticle*>(ppfo->chargedObject(0));
       }
       if (ptrk == nullptr)
       {
         ATH_MSG_WARNING("Charged PFO with index " << ppfo->index() << " has no ID track!");
         continue;
       }
       if (trkVtxAssoc)
       { // Use TrackVertexAssociation
         const xAOD::Vertex *thisTracksVertex = trkVtxAssoc->associatedVertex(ptrk);
         if (thisTracksVertex == nullptr)
         {
           ATH_MSG_DEBUG("No vertex associated to track " << ptrk->index() << "! So it cannot be associated to the primary vertex");
           matchedToPrimaryVertex = false;
         }
         else
         {
           matchedToPrimaryVertex = (xAOD::VxType::PriVtx == thisTracksVertex->vertexType());
  
           // r21 PU sideband definition (see below)
           // needed for comparisons with new r22 definition (neutral only)
           vtx = getPrimaryVertex();
           if (vtx != nullptr && vtx->vertexType() != xAOD::VxType::NoVtx)
           {
             const double absZ0sinTheta = calcAbsZ0SinTheta(*ptrk, *vtx);
             if (absZ0sinTheta < 2.0 * m_z0sinThetaCutValue && absZ0sinTheta >= m_z0sinThetaCutValue)
               matchedToPileupSideband = true;
           }
         }
       }
       else
       { // Use Primary Vertex
         if (vtx->vertexType() == xAOD::VxType::NoVtx)
         {                                // No reconstructed vertices
           matchedToPrimaryVertex = true; // simply match all cPFOs in this case
         }
         else
         { // Had a good reconstructed vertex.
           // vtz.z() provides z of that vertex w.r.t the center of the beamspot (z = 0).
           // Thus we correct the track z0 to be w.r.t z = 0
           float z0 = ptrk->z0() + ptrk->vz() - vtx->z();
           float theta = ptrk->theta();
           matchedToPrimaryVertex = (std::abs(z0 * sin(theta)) < m_z0sinThetaCutValue);
           if (std::abs(z0 * sin(theta)) < 2.0 * m_z0sinThetaCutValue && std::abs(z0 * sin(theta)) >= m_z0sinThetaCutValue)
             matchedToPileupSideband = true;
         }
       } // TVA vs PV decision
     }
     PVMatchedAcc(*ppfo) = matchedToPrimaryVertex;
     PUsidebandMatchedAcc(*ppfo) = matchedToPileupSideband;
   }
  
   return StatusCode::SUCCESS;
 }

◆ msg() [1/2]

MsgStream& AthCommonMsg< AlgTool >::msg ( ) const

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                 {
     return this->msgStream();
   }

◆ msg() [2/2]

MsgStream& AthCommonMsg< AlgTool >::msg ( const MSG::Level lvl ) const

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                   {
     return this->msgStream(lvl);
   }

◆ msg_level_name()

const std::string & asg::AsgTool::msg_level_name ( ) const

inherited

A deprecated function for getting the message level's name.

Instead of using this, weirdly named function, user code should get the string name of the current minimum message level (in case they really need it...), with:

MSG::name( msg().level() )

This function's name doesn't follow the ATLAS coding rules, and as such will be removed in the not too distant future.

Returns: The string name of the current minimum message level that's printed

Definition at line 101 of file AsgTool.cxx.

                                                   {
  
       return MSG::name( msg().level() );
    }

◆ msgLvl()

bool AthCommonMsg< AlgTool >::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                                {
     return this->msgLevel(lvl);
   }

◆ outputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< AlgTool > >::outputHandles ( ) const

overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ print()

void asg::AsgTool::print ( ) const

virtualinherited

Print the state of the tool.

Implements asg::IAsgTool.

Reimplemented in JetRecTool, JetFinder, JetModifiedMassDrop, JetFromPseudojet, JetReclusterer, JetReclusteringTool, JetTruthLabelingTool, JetPileupLabelingTool, HI::HIPileupTool, JetDumper, JetBottomUpSoftDrop, JetRecursiveSoftDrop, JetSoftDrop, JetConstituentsRetriever, JetSubStructureMomentToolsBase, JetSplitter, JetToolRunner, JetPruner, JetPseudojetRetriever, JetTrimmer, AsgHelloTool, and KtDeltaRTool.

Definition at line 131 of file AsgTool.cxx.

                              {
  
       ATH_MSG_INFO( "AsgTool " << name() << " @ " << this );
       return;
    }

◆ process()

StatusCode JetConstituentModifierBase::process ( xAOD::IParticleContainer * cont ) const

finaloverridevirtualinherited

Implements IJetConstituentModifier.

Definition at line 23 of file JetConstituentModifierBase.cxx.

 {
   // Test that we are operating on the type of object that
   // we will be writing out.
   // By implication, any supporting containers should not
   // be the ones passed to this method...
   if(!cont->empty() && cont->front()->type() != m_inputType) {
     ATH_MSG_ERROR("Object type mismatch! This tool expects " << m_inputType
                << ", but received " << cont->front()->type());
     return StatusCode::FAILURE;
   }
  
   ATH_CHECK(process_impl(cont));
  
   return StatusCode::SUCCESS;
 }

◆ process_impl()

StatusCode ChargedHadronSubtractionTool::process_impl ( xAOD::IParticleContainer * cont ) const

privatevirtual

Reimplemented from JetConstituentModifierBase.

Definition at line 30 of file ChargedHadronSubtractionTool.cxx.

 {
  
   // Type-checking happens in the JetConstituentModifierBase class
   // so it is safe just to static_cast
   if (m_inputType == xAOD::Type::FlowElement)
   {
     xAOD::FlowElementContainer *feCont = static_cast<xAOD::FlowElementContainer *>(cont);
  
     // Check that these FlowElements are actually PFOs; nothing else is supported
     if (!feCont->empty() && !(feCont->front()->signalType() & xAOD::FlowElement::PFlow))
     {
       ATH_MSG_ERROR("This tool correctly recieved a FlowElement, but it wasn't a PFO!"
                     << " Signal type was " << feCont->front()->signalType());
       return StatusCode::FAILURE;
     }
  
     return !m_byVertex ? matchToPrimaryVertex<xAOD::FlowElementContainer, xAOD::FlowElement>(*feCont) : matchByPrimaryVertex<xAOD::FlowElementContainer, xAOD::FlowElement>(*feCont);
   }
   xAOD::PFOContainer *pfoCont = static_cast<xAOD::PFOContainer *>(cont);
   return !m_byVertex ? matchToPrimaryVertex<xAOD::PFOContainer, xAOD::PFO>(*pfoCont) : matchByPrimaryVertex<xAOD::PFOContainer, xAOD::PFO>(*pfoCont);
 }

◆ renounce()

std::enable_if_t<std::is_void_v<std::result_of_t<decltype(&T::renounce)(T)> > && !std::is_base_of_v<SG::VarHandleKeyArray, T> && std::is_base_of_v<Gaudi::DataHandle, T>, void> AthCommonDataStore< AthCommonMsg< AlgTool > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

   {
     h.renounce();
     PBASE::renounce (h);
   }

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::renounceArray ( SG::VarHandleKeyArray & handlesArray )

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                           {
     handlesArray.renounce();
   }

◆ setEnergyPt()

StatusCode JetConstituentModifierBase::setEnergyPt	(	xAOD::IParticle *	obj,
		float	e,
		float	pt,
		const SG::AuxElement::Accessor< float > *	weightAcc = `nullptr`
	)		const

protectedinherited

Definition at line 61 of file JetConstituentModifierBase.cxx.

 {
   switch(m_inputType) {
   case xAOD::Type::CaloCluster:
     {
       xAOD::CaloCluster* clus = static_cast<xAOD::CaloCluster*>(obj);
       // Clusters get pt via the energy
       // This currently leaves the mass unaltered.
       if(weightAcc) (*weightAcc)(*clus) = clus->calE() > 0. ? e / clus->calE() : 0.;
       clus->setCalE(e);
     }
     break;
   case xAOD::Type::ParticleFlow:
     {
       xAOD::PFO* pfo = static_cast<xAOD::PFO*>(obj);
       if( (m_applyToChargedPFO && pfo->isCharged()) || 
           (m_applyToNeutralPFO && !pfo->isCharged()) ) {
         if(weightAcc) (*weightAcc)(*pfo) = pfo->pt() > 0. ? pt / pfo->pt() : 0.;
         // KTJ: Temporary fix
         // Defeats the purpose, but we need to use this to reset the 4-vec cache
         pfo->setP4(pt, pfo->eta(), pfo->phi());
       }
     }
     break;
   case xAOD::Type::FlowElement:
     {
       xAOD::FlowElement* pfo = static_cast<xAOD::FlowElement*>(obj);
       if( (m_applyToChargedPFO && pfo->isCharged()) || 
           (m_applyToNeutralPFO && !pfo->isCharged()) ) {
         if(weightAcc) (*weightAcc)(*pfo) = pfo->pt() > 0. ? pt / pfo->pt() : 0.;
         pfo->setP4(pt, pfo->eta(), pfo->phi(), 0.);
       }
     }
     break;
   case xAOD::Type::TrackCaloCluster:
     {
       xAOD::TrackCaloCluster* tcc = static_cast<xAOD::TrackCaloCluster*>(obj);
       if( tcc->taste() != 0) {
         if(weightAcc) (*weightAcc)(*tcc) = tcc->pt() > 0. ? pt / tcc->pt() : 0.;
         tcc->setParameters(pt, tcc->eta(), tcc->phi(), tcc->m(), xAOD::TrackCaloCluster::Taste(tcc->taste()), tcc->trackParticleLink(), tcc->caloClusterLinks());
       }
     }
     break;
   default:
     // Should not get here, because type-checking should happen in process()
     ATH_MSG_ERROR("No specialisation for object type " << m_inputType);
     return StatusCode::FAILURE;
   }
   return StatusCode::SUCCESS;
 }

◆ setEtaPhi()

StatusCode JetConstituentModifierBase::setEtaPhi	(	xAOD::IParticle *	obj,
		float	eta,
		float	phi
	)		const

protectedinherited

Definition at line 41 of file JetConstituentModifierBase.cxx.

 {
   switch(m_inputType) {
     // The main (only?) application is origin-correcting LC topoclusters
     // By convention we leave the raw p4 unmodified
   case xAOD::Type::CaloCluster:
     {
       xAOD::CaloCluster* clus = static_cast<xAOD::CaloCluster*>(obj);
       clus->setCalEta(eta);
       clus->setCalPhi(phi);
     }
     break;
   default:
     // Should not get here, because type-checking should happen in process()
     ATH_MSG_ERROR("No specialisation for object type " << m_inputType);
     return StatusCode::FAILURE;
   }
   return StatusCode::SUCCESS;
 }

◆ setP4()

StatusCode JetConstituentModifierBase::setP4	(	xAOD::IParticle *	obj,
		const xAOD::JetFourMom_t &	p4,
		const SG::AuxElement::Accessor< float > *	weightAcc = `nullptr`
	)		const

protectedinherited

Definition at line 113 of file JetConstituentModifierBase.cxx.

                                                                                                  {
   switch(m_inputType) {
   case xAOD::Type::CaloCluster:
     {
       xAOD::CaloCluster* clus = static_cast<xAOD::CaloCluster*>(obj);
       // This currently leaves the mass unaltered
       if(weightAcc) (*weightAcc)(*clus) = clus->calE() > 0. ? p4.e() / clus->calE() : 0.;
       clus->setCalE(p4.e());
       clus->setCalEta(p4.eta());
       clus->setCalPhi(p4.phi());
     }
     break;
   case xAOD::Type::ParticleFlow:
     {
       xAOD::PFO* pfo = static_cast<xAOD::PFO*>(obj);
       // The PFO setter defaults to m=0
       if( (m_applyToChargedPFO && pfo->isCharged()) || 
           (m_applyToNeutralPFO && !pfo->isCharged()) ) {
         if(weightAcc) (*weightAcc)(*pfo) = pfo->pt() > 0. ? p4.pt() / pfo->pt() : 0.;
         pfo->setP4(p4.pt(),p4.eta(),p4.phi(),p4.mass());
       }
     }
     break;
   case xAOD::Type::FlowElement:
     {
       xAOD::FlowElement* pfo = static_cast<xAOD::FlowElement*>(obj);
       if( (m_applyToChargedPFO && pfo->isCharged()) || 
           (m_applyToNeutralPFO && !pfo->isCharged()) ) {
         if(weightAcc) (*weightAcc)(*pfo) = pfo->pt() > 0. ? p4.pt() / pfo->pt() : 0.;
         pfo->setP4(p4.pt(),p4.eta(),p4.phi(),p4.mass());
       }
     }
     break;
   case xAOD::Type::TrackCaloCluster:
     {
       xAOD::TrackCaloCluster* tcc = static_cast<xAOD::TrackCaloCluster*>(obj);
       if( tcc->taste() != 0) {
               if(weightAcc) (*weightAcc)(*tcc) = tcc->pt() > 0. ? p4.pt() / tcc->pt() : 0.;
         tcc->setParameters(p4.pt(), p4.eta(), p4.phi(), p4.mass(), xAOD::TrackCaloCluster::Taste(tcc->taste()), tcc->trackParticleLink(), tcc->caloClusterLinks());
       }
       break;
     }
  
   default:
     // Should not get here, because type-checking should happen in process()
     ATH_MSG_ERROR("No specialisation for object type " << m_inputType);
     return StatusCode::FAILURE;
   }
   return StatusCode::SUCCESS;
 }

◆ sysInitialize()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysInitialize ( )

overridevirtualinherited

Perform system initialization for an algorithm.

We override this to declare all the elements of handle key arrays at the end of initialization. See comments on updateVHKA.

Reimplemented in DerivationFramework::CfAthAlgTool, AthCheckedComponent< AthAlgTool >, AthCheckedComponent<::AthAlgTool >, and asg::AsgMetadataTool.

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysStart ( )

overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::updateVHKA ( Gaudi::Details::PropertyBase & )

inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

                                                {
     // debug() << "updateVHKA for property " << p.name() << " " << p.toString() 
     //         << "  size: " << m_vhka.size() << endmsg;
     for (auto &a : m_vhka) {
       std::vector<SG::VarHandleKey*> keys = a->keys();
       for (auto k : keys) {
         k->setOwner(this);
       }
     }
   }

Member Data Documentation

◆ m_applyToChargedPFO

bool JetConstituentModifierBase::m_applyToChargedPFO =true

protectedinherited

Definition at line 62 of file JetConstituentModifierBase.h.

◆ m_applyToNeutralPFO

bool JetConstituentModifierBase::m_applyToNeutralPFO =true

protectedinherited

Definition at line 63 of file JetConstituentModifierBase.h.

◆ m_byVertex

Gaudi::Property<bool> ChargedHadronSubtractionTool::m_byVertex = {this, "DoByVertex", false, "True if we should match to each primary vertex, not just PV0"}

private

Definition at line 53 of file ChargedHadronSubtractionTool.h.

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_ignoreVertex

Gaudi::Property<bool> ChargedHadronSubtractionTool::m_ignoreVertex = {this, "IgnoreVertex", false, "Dummy option for cosmics - accept everything"}

private

Definition at line 51 of file ChargedHadronSubtractionTool.h.

◆ m_inputType

unsigned int JetConstituentModifierBase::m_inputType =xAOD::Type::Other

protectedinherited

Definition at line 60 of file JetConstituentModifierBase.h.

◆ m_trkVtxAssoc_key

SG::ReadHandleKey<jet::TrackVertexAssociation> ChargedHadronSubtractionTool::m_trkVtxAssoc_key = {this, "TrackVertexAssociation", "JetTrackVtxAssoc", "key for the TrackVertexAssociation object"}

private

Definition at line 58 of file ChargedHadronSubtractionTool.h.

◆ m_useTrackToVertexTool

Gaudi::Property<bool> ChargedHadronSubtractionTool::m_useTrackToVertexTool = {this, "UseTrackToVertexTool", false, "True if we will use the track to vertex tool"}

private

Definition at line 50 of file ChargedHadronSubtractionTool.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< AlgTool > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vertexContainer_key

SG::ReadHandleKey<xAOD::VertexContainer> ChargedHadronSubtractionTool::m_vertexContainer_key = {this, "VertexContainerKey", "PrimaryVertices", "key for the primary vertex container"}

private

Definition at line 57 of file ChargedHadronSubtractionTool.h.

◆ m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< AlgTool > >::m_vhka

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

◆ m_z0sinThetaCutValue

Gaudi::Property<float> ChargedHadronSubtractionTool::m_z0sinThetaCutValue = {this, "Z0sinThetaCutValue", 2.0, "True if we will use the track to vertex tool"}

private

Definition at line 52 of file ChargedHadronSubtractionTool.h.

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

Public Member Functions

Protected Member Functions

Protected Attributes

Private Types

Private Member Functions

Static Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ StoreGateSvc_t

Constructor & Destructor Documentation

◆ ChargedHadronSubtractionTool()

Member Function Documentation

◆ calcAbsZ0SinTheta()

◆ declareGaudiProperty() [1/4]

◆ declareGaudiProperty() [2/4]

◆ declareGaudiProperty() [3/4]

◆ declareGaudiProperty() [4/4]

◆ declareProperty() [1/6]

◆ declareProperty() [2/6]

◆ declareProperty() [3/6]

◆ declareProperty() [4/6]

◆ declareProperty() [5/6]

◆ declareProperty() [6/6]

◆ detStore()

◆ evtStore() [1/2]

◆ evtStore() [2/2]

◆ extraDeps_update_handler()

◆ getKey()

◆ getName()

◆ getPrimaryVertex()

◆ getProperty()

◆ initialize()

◆ inputHandles()

◆ matchByPrimaryVertex()

◆ matchToPrimaryVertex()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msg_level_name()

◆ msgLvl()

◆ outputHandles()

◆ print()

◆ process()

◆ process_impl()

◆ renounce()

◆ renounceArray()

◆ setEnergyPt()

◆ setEtaPhi()

◆ setP4()

◆ sysInitialize()

◆ sysStart()

◆ updateVHKA()

Member Data Documentation

◆ m_applyToChargedPFO

◆ m_applyToNeutralPFO

◆ m_byVertex

◆ m_detStore

◆ m_evtStore

◆ m_ignoreVertex

◆ m_inputType

◆ m_trkVtxAssoc_key

◆ m_useTrackToVertexTool

◆ m_varHandleArraysDeclared

◆ m_vertexContainer_key

◆ m_vhka

◆ m_z0sinThetaCutValue