#include <PrimaryVertexReFitter.h>

Inheritance diagram for Prompt::PrimaryVertexReFitter:

Collaboration diagram for Prompt::PrimaryVertexReFitter:

Public Member Functions
	PrimaryVertexReFitter (const std::string &name, ISvcLocator *pSvcLocator)

virtual StatusCode	initialize () override

virtual StatusCode	execute () override

virtual StatusCode	finalize () override

virtual StatusCode	sysInitialize () override
	Override sysInitialize. More...

virtual const DataObjIDColl &	extraOutputDeps () const override
	Return the list of extra output dependencies. 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	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
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...

Private Types
typedef SG::AuxElement::Decorator< float >	decoratorFloat_t

typedef SG::AuxElement::Decorator< ElementLink< xAOD::VertexContainer > >	decoratorElemVtx_t

typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
bool	decorateLepWithReFitPrimaryVertex (const FittingInput &input, const xAOD::TrackParticle tracklep, const xAOD::IParticle lep, const std::vector< const xAOD::TrackParticle * > &tracks, xAOD::VertexContainer &refitVtxContainer)

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

Private Attributes
ToolHandle< Prompt::VertexFittingTool >	m_vertexFitterTool

Gaudi::Property< bool >	m_printTime {this, "PrintTime", false}

Gaudi::Property< std::string >	m_distToRefittedPriVtxName

Gaudi::Property< std::string >	m_normDistToRefittedPriVtxName

Gaudi::Property< std::string >	m_lepVtxLinkName

Gaudi::Property< std::string >	m_lepRefittedVtxWithoutLeptonLinkName

TStopwatch	m_timerAll

TStopwatch	m_timerExec

SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_inDetTracksKey

SG::ReadHandleKey< xAOD::IParticleContainer >	m_leptonContainerKey

SG::ReadHandleKey< xAOD::VertexContainer >	m_primaryVertexContainerKey

SG::WriteHandleKey< xAOD::VertexContainer >	m_reFitPrimaryVertexKey

std::unique_ptr< decoratorFloat_t >	m_distToRefittedPriVtx

std::unique_ptr< decoratorFloat_t >	m_normdistToRefittedPriVtx

std::unique_ptr< decoratorElemVtx_t >	m_lepRefittedRMVtxLinkDec

DataObjIDColl	m_extendedExtraObjects

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

Definition at line 51 of file PrimaryVertexReFitter.h.

Member Typedef Documentation

◆ decoratorElemVtx_t

typedef SG::AuxElement::Decorator<ElementLink<xAOD::VertexContainer> > Prompt::PrimaryVertexReFitter::decoratorElemVtx_t

private

Definition at line 84 of file PrimaryVertexReFitter.h.

◆ decoratorFloat_t

typedef SG::AuxElement::Decorator<float> Prompt::PrimaryVertexReFitter::decoratorFloat_t

private

Definition at line 83 of file PrimaryVertexReFitter.h.

◆ StoreGateSvc_t

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

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ PrimaryVertexReFitter()

Prompt::PrimaryVertexReFitter::PrimaryVertexReFitter	(	const std::string &	name,
		ISvcLocator *	pSvcLocator
	)

Definition at line 18 of file PrimaryVertexReFitter.cxx.

                                                                                            :
     AthAlgorithm (name, pSvcLocator)
 {}

Member Function Documentation

◆ declareGaudiProperty() [1/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::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< Algorithm > >::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< Algorithm > >::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< Algorithm > >::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< Algorithm > >::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< Algorithm > >::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< Algorithm > >::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< Algorithm > >::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< Algorithm > >::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< Algorithm > >::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());
   }

◆ decorateLepWithReFitPrimaryVertex()

bool Prompt::PrimaryVertexReFitter::decorateLepWithReFitPrimaryVertex	(	const FittingInput &	input,
		const xAOD::TrackParticle *	tracklep,
		const xAOD::IParticle *	lep,
		const std::vector< const xAOD::TrackParticle * > &	tracks,
		xAOD::VertexContainer &	refitVtxContainer
	)

private

Definition at line 243 of file PrimaryVertexReFitter.cxx.

 {
     //
     // Check if the lepton track has been used for primary vertex reconstruction.
     // if true, then remove the lepton track from the input track list, re-fit primary vertex again.
     // Save the ElementLink of the re-fit primary vertex to the lepton
     //
     if(!input.priVtx) {
         ATH_MSG_WARNING("decorateLepWithReFitPrimaryVertex - invalid input primary vertex pointer");
         return false;
     }
  
     //--------------------------------------------------------
     // Remove the lepton track from the track list
     // get re-fitted non-prompt primary vertex
     //
     std::vector<const xAOD::TrackParticle*> priVtx_tracks_pass;
     bool isRefit = false;
  
     for(const xAOD::TrackParticle *track: tracks) {
         if(track == tracklep) {
             isRefit = true;
             ATH_MSG_DEBUG("decorateLepWithReFitPrimaryVertex -- lepton has been used, lepton pT =" << tracklep->pt() << ", track pT =" << track->pt());
             continue;
         }
  
         priVtx_tracks_pass.push_back(track);
     }
  
     ElementLink<xAOD::VertexContainer> refittedRM_pv_link;
  
     if(!isRefit) {
         ATH_MSG_DEBUG("decorateLepWithReFitPrimaryVertex -- Skip the primary vertex without lepton track");
  
         (*m_lepRefittedRMVtxLinkDec)(*lep) = refittedRM_pv_link;
         return false;
     }
  
     if(priVtx_tracks_pass.size() < 2) {
         ATH_MSG_DEBUG("decorateLepWithReFitPrimaryVertex -- Skip the primary vertex refitting: N tracks =" << priVtx_tracks_pass.size());
  
         (*m_lepRefittedRMVtxLinkDec)(*lep) = refittedRM_pv_link;
         return false;
     }
  
     // TODO: probably need to fix memory management here
     // but I'm not sure what happens with the ElementLink later
     // so didn't want the vertex to be deleted by accident
     xAOD::Vertex* refittedVtxRMLep = m_vertexFitterTool->fitVertexWithSeed(
         input, priVtx_tracks_pass, input.priVtx->position(),
         Prompt::kRefittedPriVtxWithoutLep
     ).release();
  
     if(refittedVtxRMLep) {
         //
         // Record vertex with output container
         //
         refitVtxContainer.push_back(std::move(refittedVtxRMLep));
  
         // TODO: I don't know if this is the correct use of an ElementLink
         //
         // Add refitted non-prompt vertex ElementLink to the lepton
         //
         refittedRM_pv_link.toContainedElement(refitVtxContainer, refittedVtxRMLep);
  
         ATH_MSG_DEBUG("decorateLepWithReFitPrimaryVertex -- save refitted non-prompt primary vertex with NTrack = " << refittedVtxRMLep->nTrackParticles());
  
         if(input.refittedPriVtx) {
             (*m_distToRefittedPriVtx)    (*refittedVtxRMLep) = Prompt::getDistance(input.refittedPriVtx->position(), refittedVtxRMLep->position());
             (*m_normdistToRefittedPriVtx)(*refittedVtxRMLep) = Prompt::getNormDist(
                 input.refittedPriVtx->position(),
                 refittedVtxRMLep->position(),
                 refittedVtxRMLep->covariance(),
                 msg(MSG::WARNING)
             );
         }
     }
  
     (*m_lepRefittedRMVtxLinkDec)(*lep) = refittedRM_pv_link;
  
     return true;
 }

◆ detStore()

const ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< Algorithm > >::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< Algorithm > >::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< Algorithm > >::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; }

◆ execute()

StatusCode Prompt::PrimaryVertexReFitter::execute ( )

overridevirtual

Definition at line 90 of file PrimaryVertexReFitter.cxx.

 {
     //
     // Start execute timer
     //
     TimerScopeHelper timer(m_timerExec);
  
     //
     // Find Inner Detector tracks
     //
     SG::ReadHandle<xAOD::TrackParticleContainer> h_inDetTracks(m_inDetTracksKey);
     if (!h_inDetTracks.isValid()){
         ATH_MSG_FATAL("execute - failed to find the InDetTrackParticles");
         return StatusCode::FAILURE;
     }
  
     const xAOD::TrackParticleContainer inDetTracks = *h_inDetTracks;
  
     //
     // Create output vertex collections and record them immediately
     // in StoreGate for memory management
     //
     std::unique_ptr<xAOD::VertexContainer> refitVtxContainer = std::make_unique< xAOD::VertexContainer>();
     std::unique_ptr<xAOD::VertexAuxContainer> refitVtxContainerAux = std::make_unique< xAOD::VertexAuxContainer>();
  
     refitVtxContainer->setStore(refitVtxContainerAux.get());
  
     // Take reference BEFORE pointers moved to SG
     xAOD::VertexContainer &refitVtxContainerRef = *refitVtxContainer;
  
     SG::WriteHandle<xAOD::VertexContainer> h_refitVtxContainer (m_reFitPrimaryVertexKey);
     ATH_CHECK(h_refitVtxContainer.record(
         std::move(refitVtxContainer), std::move(refitVtxContainerAux)
     ));
  
     //
     // Retrieve containers from evtStore
     //
     SG::ReadHandle<xAOD::IParticleContainer> h_leptonContainer(m_leptonContainerKey);
     if (!h_leptonContainer.isValid()){
         ATH_MSG_FATAL("execute - failed to find the lepton container");
         return StatusCode::FAILURE;
     }
  
     SG::ReadHandle<xAOD::VertexContainer> h_vertices(m_primaryVertexContainerKey);
     if (!h_vertices.isValid()){
         ATH_MSG_FATAL("execute - failed to find the vertices");
         return StatusCode::FAILURE;
     }
  
     const xAOD::IParticleContainer leptonContainer = *h_leptonContainer;
     const xAOD::VertexContainer    vertices        = *h_vertices;
  
     Prompt::FittingInput fittingInput(&inDetTracks, 0, 0);
  
     for(const xAOD::Vertex *vertex: vertices) {
         if(vertex->vertexType() == 1) {
             fittingInput.priVtx = dynamic_cast<const xAOD::Vertex*>(vertex);
             break;
         }
     }
  
     if(!fittingInput.priVtx) {
         ATH_MSG_INFO("Failed to find primary vertices - save empty containers");
         return StatusCode::SUCCESS;
     }
  
     //
     // Collect tracks used for primary vertex fit
     //
     std::vector<const xAOD::TrackParticle *> priVtx_tracks;
     priVtx_tracks.reserve(fittingInput.priVtx->nTrackParticles());
  
     for(unsigned k = 0; k < fittingInput.priVtx->nTrackParticles(); ++k) {
         const xAOD::TrackParticle *track  = fittingInput.priVtx->trackParticle(k);
  
         if(track) {
             priVtx_tracks.push_back(track);
         }
     }
  
     // Refit primary vertex
     std::unique_ptr<xAOD::Vertex> refittedPriVtx = m_vertexFitterTool->fitVertexWithSeed(
         fittingInput, priVtx_tracks,
         fittingInput.priVtx->position(),
         Prompt::kRefittedPriVtx
     );
  
     if(!refittedPriVtx) {
         ATH_MSG_WARNING("Failed to refit primary vertex - save empty containers");
         return StatusCode::SUCCESS;
     }
  
     //
     // Save refitted primary vertex for fitting service
     //
     fittingInput.refittedPriVtx = refittedPriVtx.get();
  
     ATH_MSG_DEBUG("execute --          primary vertex NTrack = " << fittingInput.priVtx         ->nTrackParticles());
     ATH_MSG_DEBUG("execute -- refitted primary vertex NTrack = " << fittingInput.refittedPriVtx->nTrackParticles());
  
     //
     // Dynamic cast IParticle container to electron or muon container
     //
     ATH_MSG_DEBUG("======================================="
             << "\n\t\t\t  Size of lepton container:    " << leptonContainer.size()
             << "\n-----------------------------------------------------------------");
  
     for(const xAOD::IParticle *lepton: leptonContainer) {
         const xAOD::TrackParticle *tracklep = 0;
         const xAOD::Electron *elec = dynamic_cast<const xAOD::Electron*>(lepton);
         const xAOD::Muon *muon = dynamic_cast<const xAOD::Muon*>(lepton);
  
         if(elec) {
             //
             // get GSF track
             //
             const xAOD::TrackParticle *bestmatchedGSFElTrack=elec->trackParticle(0);
  
             //
             // get origin ID track for later study
             //
             tracklep = xAOD::EgammaHelpers::getOriginalTrackParticleFromGSF(bestmatchedGSFElTrack);
         }
         else if(muon) {
             if(muon->inDetTrackParticleLink().isValid()) {
                 tracklep = *(muon->inDetTrackParticleLink());
             }
             else {
                 ATH_MSG_DEBUG("PrimaryVertexReFitter::execute - skip muon without valid inDetTrackParticleLink()");
                 continue;
             }
         }
  
         if(!tracklep) {
             ATH_MSG_WARNING("PrimaryVertexReFitter::execute - cannot find muon->inDetTrackParticleLink() nor electron->trackParticle()");
             continue;
         }
  
         decorateLepWithReFitPrimaryVertex(fittingInput, tracklep, lepton, priVtx_tracks, refitVtxContainerRef);
     }
  
     h_refitVtxContainer->push_back(std::move(refittedPriVtx));
  
     ATH_MSG_DEBUG("SV Vertex container " << m_reFitPrimaryVertexKey << " recorded in store");
  
     ATH_MSG_DEBUG("execute - all done");
     ATH_MSG_DEBUG("=======================================");
  
     return StatusCode::SUCCESS;
 }

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::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

◆ extraOutputDeps()

const DataObjIDColl & AthAlgorithm::extraOutputDeps ( ) const

overridevirtualinherited

Return the list of extra output dependencies.

This list is extended to include symlinks implied by inheritance relations.

Definition at line 50 of file AthAlgorithm.cxx.

 {
   // If we didn't find any symlinks to add, just return the collection
   // from the base class.  Otherwise, return the extended collection.
   if (!m_extendedExtraObjects.empty()) {
     return m_extendedExtraObjects;
   }
   return Algorithm::extraOutputDeps();
 }

◆ finalize()

StatusCode Prompt::PrimaryVertexReFitter::finalize ( )

overridevirtual

Definition at line 74 of file PrimaryVertexReFitter.cxx.

 {
     if(m_printTime) {
         //
         // Print full time stopwatch
         //
         m_timerAll.Stop();
  
         ATH_MSG_INFO("Real time: " << m_timerAll.RealTime() << "\t CPU time: " << m_timerAll.CpuTime());
         ATH_MSG_INFO("Execute time: " << PrintResetStopWatch(m_timerExec));
     }
  
     return StatusCode::SUCCESS;
 }

◆ initialize()

StatusCode Prompt::PrimaryVertexReFitter::initialize ( )

overridevirtual

Definition at line 23 of file PrimaryVertexReFitter.cxx.

 {
     ANA_MSG_DEBUG("ReFitPriVtxName = " << m_reFitPrimaryVertexKey);
     ANA_MSG_DEBUG("LeptonContainerName = " << m_leptonContainerKey);
     ANA_MSG_DEBUG("PriVertexContainerName = " << m_primaryVertexContainerKey);
  
     ANA_MSG_DEBUG("PrintTime = " << m_printTime);
  
     ANA_MSG_DEBUG("DistToRefittedPriVtxName = " << m_distToRefittedPriVtxName);
     ANA_MSG_DEBUG("NormDistToRefittedPriVtxName = " << m_normDistToRefittedPriVtxName);
     ANA_MSG_DEBUG("RefittedVtxLinkName = " << m_lepVtxLinkName);
     ANA_MSG_DEBUG("RefittedVtxWithoutLeptonLinkName = " << m_lepRefittedVtxWithoutLeptonLinkName);
  
     ATH_CHECK(m_inDetTracksKey.initialize());
  
     ATH_CHECK(m_leptonContainerKey.initialize());
     ATH_CHECK(m_primaryVertexContainerKey.initialize());
  
     ATH_CHECK(m_reFitPrimaryVertexKey.initialize());
  
     //
     // Must have non-empty container name for refitted primary vertex with/without lepton
     //
     if(m_reFitPrimaryVertexKey.empty()) {
         ATH_MSG_FATAL("initialize - SecVtx container invalid name: \"" << m_reFitPrimaryVertexKey << "\"");
         return StatusCode::FAILURE;
     }
  
     m_distToRefittedPriVtx = std::make_unique<decoratorFloat_t>  (m_distToRefittedPriVtxName);
     m_normdistToRefittedPriVtx = std::make_unique<decoratorFloat_t>  (m_normDistToRefittedPriVtxName);
     m_lepRefittedRMVtxLinkDec = std::make_unique<decoratorElemVtx_t>(m_lepRefittedVtxWithoutLeptonLinkName);
  
     ATH_CHECK(m_vertexFitterTool.retrieve());
  
     if(m_printTime) {
         //
         // Reset timers
         //
         m_timerAll .Reset();
         m_timerExec.Reset();
  
         //
         // Start full timer
         //
         m_timerAll.Start();
     }
  
     return StatusCode::SUCCESS;
 }

◆ inputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< Algorithm > >::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.

◆ msg() [1/2]

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

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                 {
     return this->msgStream();
   }

◆ msg() [2/2]

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

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                   {
     return this->msgStream(lvl);
   }

◆ msgLvl()

bool AthCommonMsg< Algorithm >::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< Algorithm > >::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.

◆ 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< Algorithm > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

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

◆ renounceArray()

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

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                           {
     handlesArray.renounce();
   }

◆ sysInitialize()

StatusCode AthAlgorithm::sysInitialize ( )

overridevirtualinherited

Override sysInitialize.

Override sysInitialize from the base class.

Loop through all output handles, and if they're WriteCondHandles, automatically register them and this Algorithm with the CondSvc

Scan through all outputHandles, and if they're WriteCondHandles, register them with the CondSvc

Reimplemented from AthCommonDataStore< AthCommonMsg< Algorithm > >.

Reimplemented in AthAnalysisAlgorithm, AthFilterAlgorithm, PyAthena::Alg, and AthHistogramAlgorithm.

Definition at line 66 of file AthAlgorithm.cxx.

                                        {
   StatusCode sc=AthCommonDataStore<AthCommonMsg<Algorithm>>::sysInitialize();
  
   if (sc.isFailure()) {
     return sc;
   }
   ServiceHandle<ICondSvc> cs("CondSvc",name());
   for (auto h : outputHandles()) {
     if (h->isCondition() && h->mode() == Gaudi::DataHandle::Writer) {
       // do this inside the loop so we don't create the CondSvc until needed
       if ( cs.retrieve().isFailure() ) {
         ATH_MSG_WARNING("no CondSvc found: won't autoreg WriteCondHandles");
         return StatusCode::SUCCESS;
       }
       if (cs->regHandle(this,*h).isFailure()) {
         sc = StatusCode::FAILURE;
         ATH_MSG_ERROR("unable to register WriteCondHandle " << h->fullKey()
                       << " with CondSvc");
       }
     }
   }
   return sc;
 }

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< Algorithm > >::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< Algorithm > >::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_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_distToRefittedPriVtx

std::unique_ptr<decoratorFloat_t> Prompt::PrimaryVertexReFitter::m_distToRefittedPriVtx

private

Definition at line 137 of file PrimaryVertexReFitter.h.

◆ m_distToRefittedPriVtxName

Gaudi::Property<std::string> Prompt::PrimaryVertexReFitter::m_distToRefittedPriVtxName

private

Initial value:

{
            this, "DistToRefittedPriVtxName", "default"
        }

Definition at line 100 of file PrimaryVertexReFitter.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthAlgorithm::m_extendedExtraObjects

privateinherited

Definition at line 79 of file AthAlgorithm.h.

◆ m_inDetTracksKey

SG::ReadHandleKey<xAOD::TrackParticleContainer> Prompt::PrimaryVertexReFitter::m_inDetTracksKey

private

Initial value:

{
            this, "InDetTrackParticlesKey", "InDetTrackParticles"
        }

Definition at line 117 of file PrimaryVertexReFitter.h.

◆ m_lepRefittedRMVtxLinkDec

std::unique_ptr<decoratorElemVtx_t> Prompt::PrimaryVertexReFitter::m_lepRefittedRMVtxLinkDec

private

Definition at line 139 of file PrimaryVertexReFitter.h.

◆ m_lepRefittedVtxWithoutLeptonLinkName

Gaudi::Property<std::string> Prompt::PrimaryVertexReFitter::m_lepRefittedVtxWithoutLeptonLinkName

private

Initial value:

{
            this, "RefittedVtxWithoutLeptonLinkName", "default"
        }

Definition at line 109 of file PrimaryVertexReFitter.h.

◆ m_leptonContainerKey

SG::ReadHandleKey<xAOD::IParticleContainer> Prompt::PrimaryVertexReFitter::m_leptonContainerKey

private

Initial value:

{
            this,
            "LeptonContainerName",
            "lepContainerNameDefault", "Name of lepton container"
        }

Definition at line 121 of file PrimaryVertexReFitter.h.

◆ m_lepVtxLinkName

Gaudi::Property<std::string> Prompt::PrimaryVertexReFitter::m_lepVtxLinkName

private

Initial value:

{
            this, "RefittedVtxLinkName", "default"
        }

Definition at line 106 of file PrimaryVertexReFitter.h.

◆ m_normdistToRefittedPriVtx

std::unique_ptr<decoratorFloat_t> Prompt::PrimaryVertexReFitter::m_normdistToRefittedPriVtx

private

Definition at line 138 of file PrimaryVertexReFitter.h.

◆ m_normDistToRefittedPriVtxName

Gaudi::Property<std::string> Prompt::PrimaryVertexReFitter::m_normDistToRefittedPriVtxName

private

Initial value:

{
            this, "NormDistToRefittedPriVtxName", "default"
        }

Definition at line 103 of file PrimaryVertexReFitter.h.

◆ m_primaryVertexContainerKey

SG::ReadHandleKey<xAOD::VertexContainer> Prompt::PrimaryVertexReFitter::m_primaryVertexContainerKey

private

Initial value:

{
            this, "PriVertexContainerName", "PrimaryVertices",
            "Name of primary vertex container"
        }

Definition at line 126 of file PrimaryVertexReFitter.h.

◆ m_printTime

Gaudi::Property<bool> Prompt::PrimaryVertexReFitter::m_printTime {this, "PrintTime", false}

private

Definition at line 98 of file PrimaryVertexReFitter.h.

◆ m_reFitPrimaryVertexKey

SG::WriteHandleKey<xAOD::VertexContainer> Prompt::PrimaryVertexReFitter::m_reFitPrimaryVertexKey

private

Initial value:

{
            this, "ReFitPriVtxName", "default"
        }

Definition at line 130 of file PrimaryVertexReFitter.h.

◆ m_timerAll

TStopwatch Prompt::PrimaryVertexReFitter::m_timerAll

private

Definition at line 113 of file PrimaryVertexReFitter.h.

◆ m_timerExec

TStopwatch Prompt::PrimaryVertexReFitter::m_timerExec

private

Definition at line 114 of file PrimaryVertexReFitter.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Algorithm > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vertexFitterTool

ToolHandle<Prompt::VertexFittingTool> Prompt::PrimaryVertexReFitter::m_vertexFitterTool

private

Initial value:

{
            this, "VertexFittingTool", "Prompt::VertexFittingTool/VertexFittingTool"
        }

Definition at line 91 of file PrimaryVertexReFitter.h.

◆ m_vhka

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

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

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

Public Member Functions

Protected Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ decoratorElemVtx_t

◆ decoratorFloat_t

◆ StoreGateSvc_t

Constructor & Destructor Documentation

◆ PrimaryVertexReFitter()

Member Function Documentation

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

◆ decorateLepWithReFitPrimaryVertex()

◆ detStore()

◆ evtStore() [1/2]

◆ evtStore() [2/2]

◆ execute()

◆ extraDeps_update_handler()

◆ extraOutputDeps()

◆ finalize()

◆ initialize()

◆ inputHandles()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ outputHandles()

◆ renounce()

◆ renounceArray()

◆ sysInitialize()

◆ sysStart()

◆ updateVHKA()

Member Data Documentation

◆ m_detStore

◆ m_distToRefittedPriVtx

◆ m_distToRefittedPriVtxName

◆ m_evtStore

◆ m_extendedExtraObjects

◆ m_inDetTracksKey

◆ m_lepRefittedRMVtxLinkDec

◆ m_lepRefittedVtxWithoutLeptonLinkName

◆ m_leptonContainerKey

◆ m_lepVtxLinkName

◆ m_normdistToRefittedPriVtx

◆ m_normDistToRefittedPriVtxName

◆ m_primaryVertexContainerKey

◆ m_printTime

◆ m_reFitPrimaryVertexKey

◆ m_timerAll

◆ m_timerExec

◆ m_varHandleArraysDeclared

◆ m_vertexFitterTool

◆ m_vhka