ATLAS Offline Software
Public Member Functions | Protected Member Functions | Protected Attributes | Private Types | Private Member Functions | Private Attributes | List of all members
Rec::MuSAVtxFitter Class Reference

#include <MuSAVtxFitter.h>

Inheritance diagram for Rec::MuSAVtxFitter:
Collaboration diagram for Rec::MuSAVtxFitter:

Public Member Functions

virtual ~MuSAVtxFitter ()
 
virtual StatusCode initialize () override
 
StatusCode fillCollections (std::vector< MuSAVtxFitterTool::WrkVrt > &workVerticesContainer, xAOD::VertexContainer *MuSAVtxContainer, xAOD::TrackParticleContainer *MuSAExtrapolatedTracksContainer, const xAOD::MuonContainer &muonContainer, const xAOD::TrackParticleContainer &MSTPContainer, const EventContext &ctx) const
 
virtual StatusCode execute (const EventContext &ctx) const override
 
virtual StatusCode sysInitialize () override
 Override sysInitialize. More...
 
virtual bool isClonable () const override
 Specify if the algorithm is clonable. More...
 
virtual unsigned int cardinality () const override
 Cardinality (Maximum number of clones that can exist) special value 0 means that algorithm is reentrant. More...
 
virtual StatusCode sysExecute (const EventContext &ctx) override
 Execute an algorithm. More...
 
virtual const DataObjIDColl & extraOutputDeps () const override
 Return the list of extra output dependencies. More...
 
virtual bool filterPassed (const EventContext &ctx) const
 
virtual void setFilterPassed (bool state, const EventContext &ctx) const
 
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, V, H > &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...
 

Protected Attributes

SG::ReadHandleKey< xAOD::MuonContainerm_muonContainer { this, "MuonContainerName", "Muons", "Muon container key" }
 
SG::ReadHandleKey< xAOD::EventInfom_eventInfo { this, "EventInfoName", "EventInfo", "Event info key" }
 
SG::ReadHandleKey< xAOD::TrackParticleContainerm_MSTPContainer {this, "MSTPContainerName", "MuonSpectrometerTrackParticles", "MSTP Key"}
 
SG::WriteHandleKey< xAOD::VertexContainerm_MuSAVertices { this, "MuSAVtxContainerName", "MuSAVertices", "MuSA vtx container" }
 
SG::WriteHandleKey< xAOD::TrackParticleContainerm_MuSAExtrapolatedTracks { this, "MuSAExtrapolatedTracksName", "MuSAExtrapolatedTrackParticles", "MuSA extrapolated tracks" }
 
ToolHandle< Rec::MuSAVtxFitterToolm_MuSAVtxFitterTool {this, "MuSAVtxToolName", "Rec::MuSAVtxFitterTool"}
 
ToolHandle< Reco::ITrackToVertexm_trackToVertexTool {this, "TrackToVertexTool", "Reco::TrackToVertex"}
 

Private Types

typedef ServiceHandle< StoreGateSvcStoreGateSvc_t
 

Private Member Functions

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

Private Attributes

DataObjIDColl m_extendedExtraObjects
 Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks. More...
 
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 27 of file MuSAVtxFitter.h.

Member Typedef Documentation

◆ StoreGateSvc_t

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

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ ~MuSAVtxFitter()

MuSAVtxFitter::~MuSAVtxFitter ( )
virtualdefault

Member Function Documentation

◆ cardinality()

unsigned int AthCommonReentrantAlgorithm< Gaudi::Algorithm >::cardinality
overridevirtualinherited

Cardinality (Maximum number of clones that can exist) special value 0 means that algorithm is reentrant.

Override this to return 0 for reentrant algorithms.

Override this to return 0 for reentrant algorithms.

Definition at line 75 of file AthCommonReentrantAlgorithm.cxx.

64 {
65  return 0;
66 }

◆ declareGaudiProperty() [1/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareGaudiProperty ( Gaudi::Property< T, V, H > &  hndl,
const SG::VarHandleKeyArrayType  
)
inlineprivateinherited

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

Definition at line 170 of file AthCommonDataStore.h.

172  {
173  return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
174  hndl.value(),
175  hndl.documentation());
176 
177  }

◆ declareGaudiProperty() [2/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareGaudiProperty ( Gaudi::Property< T, V, H > &  hndl,
const SG::VarHandleKeyType  
)
inlineprivateinherited

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

Definition at line 156 of file AthCommonDataStore.h.

158  {
159  return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
160  hndl.value(),
161  hndl.documentation());
162 
163  }

◆ declareGaudiProperty() [3/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareGaudiProperty ( Gaudi::Property< T, V, H > &  hndl,
const SG::VarHandleType  
)
inlineprivateinherited

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

Definition at line 184 of file AthCommonDataStore.h.

186  {
187  return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
188  hndl.value(),
189  hndl.documentation());
190  }

◆ declareGaudiProperty() [4/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareGaudiProperty ( Gaudi::Property< T, V, H > &  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.

200  {
201  return PBASE::declareProperty(t);
202  }

◆ declareProperty() [1/6]

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

Declare a new Gaudi property.

Parameters
nameName of the property.
hndlObject holding the property value.
docDocumentation 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.

249  {
250  this->declare(hndl.vhKey());
251  hndl.vhKey().setOwner(this);
252 
253  return PBASE::declareProperty(name,hndl,doc);
254  }

◆ declareProperty() [2/6]

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

Declare a new Gaudi property.

Parameters
nameName of the property.
hndlObject holding the property value.
docDocumentation 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.

225  {
226  this->declare(hndl);
227  hndl.setOwner(this);
228 
229  return PBASE::declareProperty(name,hndl,doc);
230  }

◆ declareProperty() [3/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Gaudi::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.

263  {
264 
265  // std::ostringstream ost;
266  // ost << Algorithm::name() << " VHKA declareProp: " << name
267  // << " size: " << hndArr.keys().size()
268  // << " mode: " << hndArr.mode()
269  // << " vhka size: " << m_vhka.size()
270  // << "\n";
271  // debug() << ost.str() << endmsg;
272 
273  hndArr.setOwner(this);
274  m_vhka.push_back(&hndArr);
275 
276  Gaudi::Details::PropertyBase* p = PBASE::declareProperty(name, hndArr, doc);
277  if (p != 0) {
278  p->declareUpdateHandler(&AthCommonDataStore<PBASE>::updateVHKA, this);
279  } else {
280  ATH_MSG_ERROR("unable to call declareProperty on VarHandleKeyArray "
281  << name);
282  }
283 
284  return p;
285 
286  }

◆ declareProperty() [4/6]

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

Declare a new Gaudi property.

Parameters
nameName of the property.
propertyObject holding the property value.
docDocumentation 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.

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

◆ declareProperty() [5/6]

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

Declare a new Gaudi property.

Parameters
nameName of the property.
propertyObject holding the property value.
docDocumentation 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.

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

◆ declareProperty() [6/6]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareProperty ( Gaudi::Property< T, V, H > &  t)
inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

145  {
146  typedef typename SG::HandleClassifier<T>::type htype;
148  }

◆ detStore()

const ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< Gaudi::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< Gaudi::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< Gaudi::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 MuSAVtxFitter::execute ( const EventContext &  ctx) const
overridevirtual

Definition at line 219 of file MuSAVtxFitter.cxx.

220 {
222  ATH_CHECK(muonContainer.isValid());
223 
224  SG::ReadHandle MSTPContainer(m_MSTPContainer, ctx);
225  ATH_CHECK(MSTPContainer.isValid());
226 
227  SG::ReadHandle eventInfo(m_eventInfo, ctx);
228  ATH_CHECK(eventInfo.isValid());
229 
230  SG::WriteHandle MuSAVtxContainer(m_MuSAVertices, ctx);
231  ATH_CHECK(MuSAVtxContainer.record(std::make_unique<xAOD::VertexContainer>(), std::make_unique<xAOD::VertexAuxContainer>()));
232 
233  SG::WriteHandle MuSAExtrapolatedTracksContainer(m_MuSAExtrapolatedTracks, ctx);
234  ATH_CHECK(MuSAExtrapolatedTracksContainer.record(std::make_unique<xAOD::TrackParticleContainer>(), std::make_unique<xAOD::TrackParticleAuxContainer>()));
235 
236  std::vector<MuSAVtxFitterTool::WrkVrt> workVerticesContainer;
237 
238  // Perform primary MuSA vertex fit
239  ATH_CHECK(m_MuSAVtxFitterTool->doMuSAVtxFit(workVerticesContainer, *muonContainer, *eventInfo, ctx));
240 
241  // If there's no MuSA vertices to begin with, stop here to avoid wasting CPU cycles
242  if (workVerticesContainer.empty()) {
243  ATH_MSG_DEBUG("No MuSA vertices found");
244  return StatusCode::SUCCESS;
245  }
246 
247  // Perform user-configurable post-fit selections
248  ATH_CHECK(m_MuSAVtxFitterTool->doPostFitSelections(workVerticesContainer));
249 
250  // If there's no MuSA vertices remaining, stop here to again avoid wasting CPU cycles
251  if (workVerticesContainer.empty()) {
252  ATH_MSG_DEBUG("No MuSA vertices found after post-fit selection!");
253  return StatusCode::SUCCESS;
254  }
255 
256  // Perform "ambiguity resolution" to prevent re-used SA muons in events with more than one MuSA vtx
257  ATH_CHECK(m_MuSAVtxFitterTool->selectBestVertices(workVerticesContainer));
258 
259  // Again avoiding wasting CPU cycles if no MuSA vertices remain
260  if (workVerticesContainer.empty()) {
261  ATH_MSG_DEBUG("No MuSA vertices found after best overlap selection!");
262  return StatusCode::SUCCESS;
263  } else {
264  ATH_MSG_DEBUG("Found " << workVerticesContainer.size() << " MuSA vertices");
265  }
266 
267  ATH_CHECK(fillCollections(workVerticesContainer, MuSAVtxContainer.ptr(), MuSAExtrapolatedTracksContainer.ptr(), *muonContainer, *MSTPContainer, ctx));
268  ATH_MSG_DEBUG("Saved MuSA collections!");
269 
270  return StatusCode::SUCCESS;
271 }

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< Gaudi::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 & AthCommonReentrantAlgorithm< Gaudi::Algorithm >::extraOutputDeps
overridevirtualinherited

Return the list of extra output dependencies.

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

Definition at line 94 of file AthCommonReentrantAlgorithm.cxx.

90 {
91  // If we didn't find any symlinks to add, just return the collection
92  // from the base class. Otherwise, return the extended collection.
93  if (!m_extendedExtraObjects.empty()) {
95  }
96  return BaseAlg::extraOutputDeps();
97 }

◆ fillCollections()

StatusCode MuSAVtxFitter::fillCollections ( std::vector< MuSAVtxFitterTool::WrkVrt > &  workVerticesContainer,
xAOD::VertexContainer MuSAVtxContainer,
xAOD::TrackParticleContainer MuSAExtrapolatedTracksContainer,
const xAOD::MuonContainer muonContainer,
const xAOD::TrackParticleContainer MSTPContainer,
const EventContext &  ctx 
) const

Definition at line 75 of file MuSAVtxFitter.cxx.

81 {
82  ATH_MSG_DEBUG("MuSAVtxFitter::fillCollections");
83  for (const auto& workVertex : workVerticesContainer) {
84 
85  xAOD::Vertex* MuSAVertex = new xAOD::Vertex();
86  MuSAVtxContainer->emplace_back(MuSAVertex);
87 
88  // Place the new extrapolated tracks in the new container (we need the index for bookkeeping)
89  for (size_t i = 0; i < workVertex.newExtrapolatedTracks.size(); i++) {
90 
91  const xAOD::TrackParticle* newTrack = workVertex.newExtrapolatedTracks[i].get();
92  // copy transfers ownership of the shared_ptr in wrkvrt to container
93  xAOD::TrackParticle* containerTrack = MuSAExtrapolatedTracksContainer->push_back(new xAOD::TrackParticle());
94  *containerTrack=*newTrack;
95 
96  // Link the new tracks to the vertex
97  ElementLink<xAOD::TrackParticleContainer> link_trk(*MuSAExtrapolatedTracksContainer, MuSAExtrapolatedTracksContainer->size() - 1);
98  MuSAVertex->addTrackAtVertex(link_trk, 1.0);
99 
100  //Also link the extrapolated tracks and vertices to their original muons -- extrapolated tracks and their muons should have the same index in workVertex
101 
102  bool muonLinkSuccess = false;
103  bool MSTPLinkSuccess = false;
104 
107  muonLinkSuccess = link_muon.toContainedElement(muonContainer, workVertex.muonCandidates[i]);
108 
109  const xAOD::Muon* muon = muonContainer[link_muon.index()];
110  const xAOD::TrackParticle* MSTP = muon->trackParticle(xAOD::Muon::MuonSpectrometerTrackParticle);
111  if (MSTP) {
112  MSTPLinkSuccess = link_MSTP.toContainedElement(MSTPContainer, MSTP);
113  }
114 
115  // Check that links are valid
116  if (!MSTPLinkSuccess) {
117  ATH_MSG_WARNING("Failed to link MSTP!");
118  } else {
119  acc_MSTPLink(*containerTrack) = link_MSTP;
120  auto& mstpLinks = acc_MSTPLinks(*MuSAVertex);
121  mstpLinks.push_back(link_MSTP);
122  }
123 
124  if (!muonLinkSuccess) {
125  ATH_MSG_WARNING("Failed to link muon!");
126  } else {
127  acc_MuonLink(*containerTrack) = link_muon;
128  auto& muonLinks = acc_MuonLinks(*MuSAVertex);
129  muonLinks.push_back(link_muon);
130  }
131  }
132 
133  MuSAVertex->setPosition(workVertex.pos);
134  MuSAVertex->setVertexType(xAOD::VxType::SecVtx);
135  MuSAVertex->setFitQuality(workVertex.chi2, 1);
136  std::vector<float> fCov(workVertex.cov.cbegin(), workVertex.cov.cend());
137  MuSAVertex->setCovariance(fCov);
138 
139  vtx_pxAcc(*MuSAVertex) = workVertex.mom.Px();
140  vtx_pyAcc(*MuSAVertex) = workVertex.mom.Py();
141  vtx_pzAcc(*MuSAVertex) = workVertex.mom.Pz();
142 
143  vtx_massAcc(*MuSAVertex) = workVertex.mom.M();
144  vtx_chargeAcc(*MuSAVertex) = workVertex.charge;
145 
146  minOpAngAcc(*MuSAVertex) = workVertex.minOpAng;
147 
148  chi2_coreAcc(*MuSAVertex) = workVertex.chi2Core;
149  ndof_coreAcc(*MuSAVertex) = 1;
150  chi2_assocAcc(*MuSAVertex) = workVertex.chi2;
151  ndof_assocAcc(*MuSAVertex) = workVertex.ndof();
152 
153  TLorentzVector sumP4_muon;
154  TLorentzVector sumP4_electron;
155  TLorentzVector sumP4_selected;
156 
157  constexpr double muonMass = 105.658; // Muon mass in MeV
158 
159  for (size_t i = 0; i < workVertex.newExtrapolatedTracks.size(); i++) {
160  const xAOD::TrackParticle* track = workVertex.newExtrapolatedTracks[i].get();
161  double pt_wrtSV = track->pt();
162  double eta_wrtSV = track->eta();
163  double phi_wrtSV = track->phi();
164 
165  TLorentzVector p4wrtSV_muon;
166  TLorentzVector p4wrtSV_electron;
167 
168  p4wrtSV_muon.SetPtEtaPhiM(pt_wrtSV, eta_wrtSV, phi_wrtSV, muonMass);
169  p4wrtSV_electron.SetPtEtaPhiM(pt_wrtSV, eta_wrtSV, phi_wrtSV, VKalVrtAthena::PhysConsts::mass_electron);
170 
171  sumP4_muon += p4wrtSV_muon;
172  sumP4_electron += p4wrtSV_electron;
173  sumP4_selected += p4wrtSV_muon;
174 
175  xAOD::TrackParticle* containerTrack = MuSAExtrapolatedTracksContainer->at(i);
176 
177  std::unique_ptr<Trk::Perigee> sv_perigee = m_trackToVertexTool->perigeeAtVertex(ctx, *containerTrack, workVertex.pos);
178  if (!sv_perigee) {
179  ATH_MSG_WARNING("Failed in obtaining the SV perigee for track!");
180  continue;
181  }
182 
183  double qOverP_wrtSV = sv_perigee->parameters()[Trk::qOverP];
184  double theta_wrtSV = sv_perigee->parameters()[Trk::theta];
185  double p_wrtSV = 1.0 / std::abs(qOverP_wrtSV);
186  pt_wrtSV = p_wrtSV * sin(theta_wrtSV);
187  eta_wrtSV = -log(tan(theta_wrtSV / 2.0));
188  phi_wrtSV = sv_perigee->parameters()[Trk::phi];
189  double d0_wrtSV = sv_perigee->parameters()[Trk::d0];
190  double z0_wrtSV = sv_perigee->parameters()[Trk::z0];
191  double sqrd0Err_wrtSV = (*sv_perigee->covariance())(Trk::d0, Trk::d0);
192  double sqrz0Err_wrtSV = (*sv_perigee->covariance())(Trk::z0, Trk::z0);
193  double sqrQoPErr_wrtSV = (*sv_perigee->covariance())(Trk::qOverP, Trk::qOverP);
194 
195  qOverP_wrtSVAcc(*containerTrack) = qOverP_wrtSV;
196  theta_wrtSVAcc(*containerTrack) = theta_wrtSV;
197  p_wrtSVAcc(*containerTrack) = p_wrtSV;
198  pt_wrtSVAcc(*containerTrack) = pt_wrtSV;
199  eta_wrtSVAcc(*containerTrack) = eta_wrtSV;
200  phi_wrtSVAcc(*containerTrack) = phi_wrtSV;
201  d0_wrtSVAcc(*containerTrack) = d0_wrtSV;
202  z0_wrtSVAcc(*containerTrack) = z0_wrtSV;
203  sqrd0Err_wrtSVAcc(*containerTrack) = sqrd0Err_wrtSV;
204  sqrz0Err_wrtSVAcc(*containerTrack) = sqrz0Err_wrtSV;
205  sqrQoPErr_wrtSVAcc(*containerTrack) = sqrQoPErr_wrtSV;
206  }
207 
208  massAcc(*MuSAVertex) = sumP4_muon.M();
209  mass_eAcc(*MuSAVertex) = sumP4_electron.M();
210  mass_selectedTracksAcc(*MuSAVertex) = sumP4_selected.M();
211  num_trksAcc(*MuSAVertex) = workVertex.nTracksTotal();
212  num_selectedTracksAcc(*MuSAVertex) = workVertex.selectedTrackIndices.size();
213  num_associatedTracksAcc(*MuSAVertex) = workVertex.associatedTrackIndices.size();
214  dCloseVrtAcc(*MuSAVertex) = workVertex.closestWrkVrtValue;
215  }
216  return StatusCode::SUCCESS;
217 }

◆ filterPassed()

virtual bool AthCommonReentrantAlgorithm< Gaudi::Algorithm >::filterPassed ( const EventContext &  ctx) const
inlinevirtualinherited

Definition at line 96 of file AthCommonReentrantAlgorithm.h.

96  {
97  return execState( ctx ).filterPassed();
98  }

◆ initialize()

StatusCode MuSAVtxFitter::initialize ( )
overridevirtual

Definition at line 55 of file MuSAVtxFitter.cxx.

56 {
60  ATH_CHECK( m_MuSAVertices.initialize() );
62 
63  ATH_CHECK( m_MuSAVtxFitterTool.retrieve() );
64  ATH_MSG_DEBUG("Retrieved tool " << m_MuSAVtxFitterTool);
65 
66  ATH_CHECK( m_trackToVertexTool.retrieve() );
67  ATH_MSG_DEBUG("Retrieved tool " << m_trackToVertexTool);
68 
69  ATH_MSG_DEBUG("Initialization successful");
70 
71  return StatusCode::SUCCESS;
72 
73 }

◆ inputHandles()

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

◆ isClonable()

bool AthCommonReentrantAlgorithm< Gaudi::Algorithm >::isClonable
overridevirtualinherited

Specify if the algorithm is clonable.

Reentrant algorithms are clonable.

Definition at line 68 of file AthCommonReentrantAlgorithm.cxx.

52 {
53  // Reentrant algorithms are clonable.
54  return true;
55 }

◆ msg() [1/2]

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

Definition at line 24 of file AthCommonMsg.h.

24  {
25  return this->msgStream();
26  }

◆ msg() [2/2]

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

Definition at line 27 of file AthCommonMsg.h.

27  {
28  return this->msgStream(lvl);
29  }

◆ msgLvl()

bool AthCommonMsg< Gaudi::Algorithm >::msgLvl ( const MSG::Level  lvl) const
inlineinherited

Definition at line 30 of file AthCommonMsg.h.

30  {
31  return this->msgLevel(lvl);
32  }

◆ outputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< Gaudi::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< Gaudi::Algorithm > >::renounce ( T &  h)
inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

381  {
382  h.renounce();
383  PBASE::renounce (h);
384  }

◆ renounceArray()

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

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

364  {
365  handlesArray.renounce();
366  }

◆ setFilterPassed()

virtual void AthCommonReentrantAlgorithm< Gaudi::Algorithm >::setFilterPassed ( bool  state,
const EventContext &  ctx 
) const
inlinevirtualinherited

Definition at line 100 of file AthCommonReentrantAlgorithm.h.

100  {
101  execState( ctx ).setFilterPassed( state );
102  }

◆ sysExecute()

StatusCode AthCommonReentrantAlgorithm< Gaudi::Algorithm >::sysExecute ( const EventContext &  ctx)
overridevirtualinherited

Execute an algorithm.

We override this in order to work around an issue with the Algorithm base class storing the event context in a member variable that can cause crashes in MT jobs.

Definition at line 85 of file AthCommonReentrantAlgorithm.cxx.

77 {
78  return BaseAlg::sysExecute (ctx);
79 }

◆ sysInitialize()

StatusCode AthCommonReentrantAlgorithm< Gaudi::Algorithm >::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< Gaudi::Algorithm > >.

Reimplemented in InputMakerBase, and HypoBase.

Definition at line 61 of file AthCommonReentrantAlgorithm.cxx.

107  {
109 
110  if (sc.isFailure()) {
111  return sc;
112  }
113 
114  ServiceHandle<ICondSvc> cs("CondSvc",name());
115  for (auto h : outputHandles()) {
116  if (h->isCondition() && h->mode() == Gaudi::DataHandle::Writer) {
117  // do this inside the loop so we don't create the CondSvc until needed
118  if ( cs.retrieve().isFailure() ) {
119  ATH_MSG_WARNING("no CondSvc found: won't autoreg WriteCondHandles");
120  return StatusCode::SUCCESS;
121  }
122  if (cs->regHandle(this,*h).isFailure()) {
123  sc = StatusCode::FAILURE;
124  ATH_MSG_ERROR("unable to register WriteCondHandle " << h->fullKey()
125  << " with CondSvc");
126  }
127  }
128  }
129  return sc;
130 }

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< Gaudi::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< Gaudi::Algorithm > >::updateVHKA ( Gaudi::Details::PropertyBase &  )
inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

308  {
309  // debug() << "updateVHKA for property " << p.name() << " " << p.toString()
310  // << " size: " << m_vhka.size() << endmsg;
311  for (auto &a : m_vhka) {
312  std::vector<SG::VarHandleKey*> keys = a->keys();
313  for (auto k : keys) {
314  k->setOwner(this);
315  }
316  }
317  }

Member Data Documentation

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_detStore
privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_eventInfo

SG::ReadHandleKey<xAOD::EventInfo> Rec::MuSAVtxFitter::m_eventInfo { this, "EventInfoName", "EventInfo", "Event info key" }
protected

Definition at line 43 of file MuSAVtxFitter.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_evtStore
privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthCommonReentrantAlgorithm< Gaudi::Algorithm >::m_extendedExtraObjects
privateinherited

Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks.

Empty if no symlinks were found.

Definition at line 114 of file AthCommonReentrantAlgorithm.h.

◆ m_MSTPContainer

SG::ReadHandleKey<xAOD::TrackParticleContainer> Rec::MuSAVtxFitter::m_MSTPContainer {this, "MSTPContainerName", "MuonSpectrometerTrackParticles", "MSTP Key"}
protected

Definition at line 44 of file MuSAVtxFitter.h.

◆ m_muonContainer

SG::ReadHandleKey<xAOD::MuonContainer> Rec::MuSAVtxFitter::m_muonContainer { this, "MuonContainerName", "Muons", "Muon container key" }
protected

Definition at line 42 of file MuSAVtxFitter.h.

◆ m_MuSAExtrapolatedTracks

SG::WriteHandleKey<xAOD::TrackParticleContainer> Rec::MuSAVtxFitter::m_MuSAExtrapolatedTracks { this, "MuSAExtrapolatedTracksName", "MuSAExtrapolatedTrackParticles", "MuSA extrapolated tracks" }
protected

Definition at line 50 of file MuSAVtxFitter.h.

◆ m_MuSAVertices

SG::WriteHandleKey<xAOD::VertexContainer> Rec::MuSAVtxFitter::m_MuSAVertices { this, "MuSAVtxContainerName", "MuSAVertices", "MuSA vtx container" }
protected

Definition at line 47 of file MuSAVtxFitter.h.

◆ m_MuSAVtxFitterTool

ToolHandle<Rec::MuSAVtxFitterTool> Rec::MuSAVtxFitter::m_MuSAVtxFitterTool {this, "MuSAVtxToolName", "Rec::MuSAVtxFitterTool"}
protected

Definition at line 53 of file MuSAVtxFitter.h.

◆ m_trackToVertexTool

ToolHandle<Reco::ITrackToVertex> Rec::MuSAVtxFitter::m_trackToVertexTool {this, "TrackToVertexTool", "Reco::TrackToVertex"}
protected

Definition at line 54 of file MuSAVtxFitter.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_varHandleArraysDeclared
privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

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

Definition at line 398 of file AthCommonDataStore.h.


The documentation for this class was generated from the following files:
muonContainer
xAOD::MuonContainer * muonContainer
Definition: TrigGlobEffCorrValidation.cxx:188
Rec::MuSAVtxFitter::m_muonContainer
SG::ReadHandleKey< xAOD::MuonContainer > m_muonContainer
Definition: MuSAVtxFitter.h:42
xAOD::muon
@ muon
Definition: TrackingPrimitives.h:196
xAOD::Vertex_v1::setPosition
void setPosition(const Amg::Vector3D &position)
Sets the 3-position.
Rec::MuSAVtxFitter::m_MuSAExtrapolatedTracks
SG::WriteHandleKey< xAOD::TrackParticleContainer > m_MuSAExtrapolatedTracks
Definition: MuSAVtxFitter.h:50
xAOD::Vertex_v1::setFitQuality
void setFitQuality(float chiSquared, float numberDoF)
Set the 'Fit Quality' information.
Definition: Vertex_v1.cxx:150
xAOD::Vertex
Vertex_v1 Vertex
Define the latest version of the vertex class.
Definition: Event/xAOD/xAODTracking/xAODTracking/Vertex.h:16
DataVector::emplace_back
value_type emplace_back(value_type pElem)
Add an element to the end of the collection.
SG::ReadHandle
Definition: StoreGate/StoreGate/ReadHandle.h:67
AthCommonDataStore::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 > renounce(T &h)
Definition: AthCommonDataStore.h:380
AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_evtStore
StoreGateSvc_t m_evtStore
Pointer to StoreGate (event store by default)
Definition: AthCommonDataStore.h:390
AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_vhka
std::vector< SG::VarHandleKeyArray * > m_vhka
Definition: AthCommonDataStore.h:398
Trk::z0
@ z0
Definition: ParamDefs.h:64
read_hist_ntuple.t
t
Definition: read_hist_ntuple.py:5
xAOD::Vertex_v1::setCovariance
void setCovariance(const std::vector< float > &value)
Sets the covariance matrix as a simple vector of values.
xAOD::Muon_v1
Class describing a Muon.
Definition: Muon_v1.h:38
AthenaPoolTestRead.sc
sc
Definition: AthenaPoolTestRead.py:27
SG::VarHandleKeyArray::setOwner
virtual void setOwner(IDataHandleHolder *o)=0
Rec::MuSAVtxFitter::m_MuSAVertices
SG::WriteHandleKey< xAOD::VertexContainer > m_MuSAVertices
Definition: MuSAVtxFitter.h:47
IDTPMcnv.htype
htype
Definition: IDTPMcnv.py:29
Rec::MuSAVtxFitter::m_trackToVertexTool
ToolHandle< Reco::ITrackToVertex > m_trackToVertexTool
Definition: MuSAVtxFitter.h:54
xAOD::Vertex_v1::setVertexType
void setVertexType(VxType::VertexType vType)
Set the type of the vertex.
AthCommonDataStore::declareGaudiProperty
Gaudi::Details::PropertyBase & declareGaudiProperty(Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
specialization for handling Gaudi::Property<SG::VarHandleKey>
Definition: AthCommonDataStore.h:156
AthCommonReentrantAlgorithm::extraOutputDeps
virtual const DataObjIDColl & extraOutputDeps() const override
Return the list of extra output dependencies.
Definition: AthCommonReentrantAlgorithm.cxx:89
xAOD::Vertex_v1::addTrackAtVertex
void addTrackAtVertex(const ElementLink< TrackParticleContainer > &tr, float weight=1.0)
Add a new track to the vertex.
Definition: Vertex_v1.cxx:314
python.utils.AtlRunQueryDQUtils.p
p
Definition: AtlRunQueryDQUtils.py:209
AthCommonDataStore
Definition: AthCommonDataStore.h:52
ATH_MSG_ERROR
#define ATH_MSG_ERROR(x)
Definition: AthMsgStreamMacros.h:33
Rec::MuSAVtxFitter::m_MSTPContainer
SG::ReadHandleKey< xAOD::TrackParticleContainer > m_MSTPContainer
Definition: MuSAVtxFitter.h:44
AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::outputHandles
virtual std::vector< Gaudi::DataHandle * > outputHandles() const override
Return this algorithm's output handles.
lumiFormat.i
int i
Definition: lumiFormat.py:85
Trk::theta
@ theta
Definition: ParamDefs.h:66
EL::StatusCode
::StatusCode StatusCode
StatusCode definition for legacy code.
Definition: PhysicsAnalysis/D3PDTools/EventLoop/EventLoop/StatusCode.h:22
xAOD::VxType::SecVtx
@ SecVtx
Secondary vertex.
Definition: TrackingPrimitives.h:573
ATH_MSG_DEBUG
#define ATH_MSG_DEBUG(x)
Definition: AthMsgStreamMacros.h:29
AthCommonDataStore::declareProperty
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
Definition: AthCommonDataStore.h:145
AthCommonReentrantAlgorithm::sysExecute
virtual StatusCode sysExecute(const EventContext &ctx) override
Execute an algorithm.
Definition: AthCommonReentrantAlgorithm.cxx:76
Rec::MuSAVtxFitter::m_eventInfo
SG::ReadHandleKey< xAOD::EventInfo > m_eventInfo
Definition: MuSAVtxFitter.h:43
ATH_CHECK
#define ATH_CHECK
Definition: AthCheckMacros.h:40
drawFromPickle.tan
tan
Definition: drawFromPickle.py:36
AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_detStore
StoreGateSvc_t m_detStore
Pointer to StoreGate (detector store by default)
Definition: AthCommonDataStore.h:393
SG::VarHandleKey::initialize
StatusCode initialize(bool used=true)
If this object is used as a property, then this should be called during the initialize phase.
Definition: AthToolSupport/AsgDataHandles/Root/VarHandleKey.cxx:103
SG::VarHandleKeyArray::renounce
virtual void renounce()=0
SG::HandleClassifier::type
std::conditional< std::is_base_of< SG::VarHandleKeyArray, T >::value, VarHandleKeyArrayType, type2 >::type type
Definition: HandleClassifier.h:54
merge_scale_histograms.doc
string doc
Definition: merge_scale_histograms.py:9
VKalVrtAthena::PhysConsts::mass_electron
constexpr double mass_electron
Definition: Reconstruction/VKalVrt/VrtSecInclusive/VrtSecInclusive/Constants.h:15
name
std::string name
Definition: Control/AthContainers/Root/debug.cxx:240
Trk::d0
@ d0
Definition: ParamDefs.h:63
DataVector::push_back
value_type push_back(value_type pElem)
Add an element to the end of the collection.
SG::WriteHandle
Definition: StoreGate/StoreGate/WriteHandle.h:73
a
TList * a
Definition: liststreamerinfos.cxx:10
h
xAOD::Vertex_v1
Class describing a Vertex.
Definition: Vertex_v1.h:42
ATH_MSG_WARNING
#define ATH_MSG_WARNING(x)
Definition: AthMsgStreamMacros.h:32
Trk::qOverP
@ qOverP
perigee
Definition: ParamDefs.h:67
python.CaloCondTools.log
log
Definition: CaloCondTools.py:20
SG::VarHandleBase::vhKey
SG::VarHandleKey & vhKey()
Return a non-const reference to the HandleKey.
Definition: StoreGate/src/VarHandleBase.cxx:629
AthCommonReentrantAlgorithm< Gaudi::Algorithm >::sysInitialize
virtual StatusCode sysInitialize() override
Override sysInitialize.
Definition: AthCommonReentrantAlgorithm.cxx:107
Rec::MuSAVtxFitter::m_MuSAVtxFitterTool
ToolHandle< Rec::MuSAVtxFitterTool > m_MuSAVtxFitterTool
Definition: MuSAVtxFitter.h:53
Trk::phi
@ phi
Definition: ParamDefs.h:75
python.Bindings.keys
keys
Definition: Control/AthenaPython/python/Bindings.py:801
xAOD::track
@ track
Definition: TrackingPrimitives.h:513
xAOD::TrackParticle_v1
Class describing a TrackParticle.
Definition: TrackParticle_v1.h:43
Rec::MuSAVtxFitter::fillCollections
StatusCode fillCollections(std::vector< MuSAVtxFitterTool::WrkVrt > &workVerticesContainer, xAOD::VertexContainer *MuSAVtxContainer, xAOD::TrackParticleContainer *MuSAExtrapolatedTracksContainer, const xAOD::MuonContainer &muonContainer, const xAOD::TrackParticleContainer &MSTPContainer, const EventContext &ctx) const
Definition: MuSAVtxFitter.cxx:75
LHEF::Writer
Pythia8::Writer Writer
Definition: Prophecy4fMerger.cxx:12
DataVector::at
const T * at(size_type n) const
Access an element, as an rvalue.
drawFromPickle.sin
sin
Definition: drawFromPickle.py:36
AthCommonReentrantAlgorithm< Gaudi::Algorithm >::m_extendedExtraObjects
DataObjIDColl m_extendedExtraObjects
Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks.
Definition: AthCommonReentrantAlgorithm.h:114
DataVector::size
size_type size() const noexcept
Returns the number of elements in the collection.
fitman.k
k
Definition: fitman.py:528
ServiceHandle< ICondSvc >