72 {
73 std::vector<Trk::VxCascadeInfo*> cascadeinfoContainer;
74 constexpr int topoN = 2;
75 std::array<SG::WriteHandle<xAOD::VertexContainer>, topoN> Vtxwritehandles;
77
78 for (
int i =0;
i<topoN;
i++) {
80 ATH_CHECK( Vtxwritehandles[i].record (std::make_unique<xAOD::VertexContainer>(),
81 std::make_unique<xAOD::VertexAuxContainer>()) );
82 }
83
84
85
86
90 return StatusCode::FAILURE;
91 }
93
94 if (pvContainer->size()==0){
95 ATH_MSG_WARNING(
"You have no primary vertices: " << pvContainer->size());
96 return StatusCode::RECOVERABLE;
97 }
99
100
101
102
103 SG::WriteHandle<xAOD::VertexContainer> refPvContainer;
105
107 ATH_CHECK(refPvContainer.
record(std::make_unique<xAOD::VertexContainer>(), std::make_unique<xAOD::VertexAuxContainer>()));
108 }
109
111
116
117
118 SG::AuxElement::Decorator<VertexLinkVector> CascadeLinksDecor("CascadeVertexLinks");
119 SG::AuxElement::Decorator<VertexLinkVector> JpsiLinksDecor("JpsiVertexLinks");
120 SG::AuxElement::Decorator<VertexLinkVector> V0LinksDecor("V0VertexLinks");
121 SG::AuxElement::Decorator<float> chi2_decor("ChiSquared");
122 SG::AuxElement::Decorator<float> ndof_decor("NumberDoF");
123 SG::AuxElement::Decorator<float> Pt_decor("Pt");
124 SG::AuxElement::Decorator<float> PtErr_decor("PtErr");
125 SG::AuxElement::Decorator<float> Mass_svdecor("V0_mass");
126 SG::AuxElement::Decorator<float> MassErr_svdecor("V0_massErr");
127 SG::AuxElement::Decorator<float> Pt_svdecor("V0_Pt");
128 SG::AuxElement::Decorator<float> PtErr_svdecor("V0_PtErr");
129 SG::AuxElement::Decorator<float> Lxy_svdecor("V0_Lxy");
130 SG::AuxElement::Decorator<float> LxyErr_svdecor("V0_LxyErr");
131 SG::AuxElement::Decorator<float> Tau_svdecor("V0_Tau");
132 SG::AuxElement::Decorator<float> TauErr_svdecor("V0_TauErr");
133
134 ATH_MSG_DEBUG(
"cascadeinfoContainer size " << cascadeinfoContainer.size());
135
136
138 if (!jpsiContainer.
isValid()) {
140 return StatusCode::FAILURE;
141 }
145 return StatusCode::FAILURE;
146 }
147
148 for (Trk::VxCascadeInfo*
x : cascadeinfoContainer) {
151
152 return StatusCode::FAILURE;
153 }
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173 const std::vector<xAOD::Vertex*> &cascadeVertices =
x->vertices();
174 if(cascadeVertices.size()!=topoN)
176 if(cascadeVertices[0] ==
nullptr || cascadeVertices[1] ==
nullptr)
ATH_MSG_ERROR(
"Error null vertex");
177
178 for(
int i =0;
i<topoN;
i++) Vtxwritehandles[i]->
push_back(cascadeVertices[i]);
179
180 x->setSVOwnership(
false);
181 const auto mainVertex = cascadeVertices[1];
182
183 const std::vector< std::vector<TLorentzVector> > &moms =
x->getParticleMoms();
184
185
186 std::vector<const xAOD::Vertex*> verticestoLink;
187 verticestoLink.push_back(cascadeVertices[0]);
191
192
194 ATH_MSG_DEBUG(
"1 pt Jpsi tracks " << cascadeVertices[1]->trackParticle(0)->
pt() <<
", " << cascadeVertices[1]->trackParticle(1)->
pt());
196
197
199 ATH_MSG_DEBUG(
"1 pt V0 tracks " << cascadeVertices[0]->trackParticle(0)->
pt() <<
", " << cascadeVertices[0]->trackParticle(1)->
pt());
201
202
203 std::vector<const xAOD::Vertex*> jpsiVerticestoLink;
204 if (jpsiVertex) jpsiVerticestoLink.push_back(jpsiVertex);
208
209 std::vector<const xAOD::Vertex*> v0VerticestoLink;
210 if (v0Vertex) v0VerticestoLink.push_back(v0Vertex);
214
218 std::vector<double> massesJpsi(2, mass_track);
219 std::vector<double> massesV0;
220 std::vector<double> Masses(2, mass_track);
237 }
238
239
240
241
242
243 xAOD::BPhysHypoHelper vtx(
m_hypoName, mainVertex);
244
246
247
248
249
250
253
255 PtErr_decor(*mainVertex) =
m_CascadeTools->pTError(moms[1],
x->getCovariance()[1]);
256
257 chi2_decor(*mainVertex) =
x->fitChi2();
258 ndof_decor(*mainVertex) =
x->nDoF();
259
261 if (
m_refitPV) { refPVContainer = refPvContainer.
ptr(); }
264
265
266
267
268 Mass_svdecor(*mainVertex) =
m_CascadeTools->invariantMass(moms[0]);
269 MassErr_svdecor(*mainVertex) =
m_CascadeTools->invariantMassError(moms[0],
x->getCovariance()[0]);
271 PtErr_svdecor(*mainVertex) =
m_CascadeTools->pTError(moms[0],
x->getCovariance()[0]);
272 Lxy_svdecor(*mainVertex) =
m_CascadeTools->lxy(moms[0],cascadeVertices[0],cascadeVertices[1]);
273 LxyErr_svdecor(*mainVertex) =
m_CascadeTools->lxyError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1]);
274 Tau_svdecor(*mainVertex) =
m_CascadeTools->tau(moms[0],cascadeVertices[0],cascadeVertices[1]);
275 TauErr_svdecor(*mainVertex) =
m_CascadeTools->tauError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1]);
276
277
279 <<
" chi2_1 " <<
m_V0Tools->chisq(cascadeVertices[0])
280 <<
" chi2_2 " <<
m_V0Tools->chisq(cascadeVertices[1])
284 <<
" error " <<
m_V0Tools->invariantMassError(cascadeVertices[0],massesV0)
285 <<
" mass_J " <<
m_V0Tools->invariantMass(cascadeVertices[1],massesJpsi)
286 <<
" error " <<
m_V0Tools->invariantMassError(cascadeVertices[1],massesJpsi));
287
290 double Mass_B_err =
m_CascadeTools->invariantMassError(moms[1],
x->getCovariance()[1]);
291 double Mass_V0_err =
m_CascadeTools->invariantMassError(moms[0],
x->getCovariance()[0]);
292 ATH_MSG_DEBUG(
"Mass_B " << Mass_B <<
" Mass_V0 " << Mass_V0);
293 ATH_MSG_DEBUG(
"Mass_B_err " << Mass_B_err <<
" Mass_V0_err " << Mass_V0_err);
294 double mprob_B =
m_CascadeTools->massProbability(mass_b,Mass_B,Mass_B_err);
295 double mprob_V0 =
m_CascadeTools->massProbability(mass_v0,Mass_V0,Mass_V0_err);
296 ATH_MSG_DEBUG(
"mprob_B " << mprob_B <<
" mprob_V0 " << mprob_V0);
297
299 <<
" Mass_v0 " <<
m_CascadeTools->invariantMass(moms[0],massesV0));
301 <<
" Mass_v0_err " <<
m_CascadeTools->invariantMassError(moms[0],
x->getCovariance()[0],massesV0));
304 <<
" pt_v0 " <<
m_V0Tools->pT(cascadeVertices[0]));
306 <<
" ptErr_v " <<
m_CascadeTools->pTError(moms[0],
x->getCovariance()[0])
307 <<
" ptErr_v0 " <<
m_V0Tools->pTError(cascadeVertices[0]));
311 <<
" lxyErr_v " <<
m_CascadeTools->lxyError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
312 <<
" lxyErr_v0 " <<
m_V0Tools->lxyError(cascadeVertices[0],cascadeVertices[1]));
314 <<
" tau_v0 " <<
m_V0Tools->tau(cascadeVertices[0],cascadeVertices[1],massesV0));
316 <<
" tau_v " <<
m_CascadeTools->tau(moms[0],cascadeVertices[0],cascadeVertices[1])
317 <<
" tau_V " <<
m_CascadeTools->tau(moms[0],cascadeVertices[0],cascadeVertices[1],mass_v0));
319 <<
" tauErr_v " <<
m_CascadeTools->tauError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
320 <<
" tauErr_v0 " <<
m_V0Tools->tauError(cascadeVertices[0],cascadeVertices[1],massesV0));
322 <<
" TauErr_v " <<
m_CascadeTools->tauError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1],mass_v0)
323 <<
" TauErr_v0 " <<
m_V0Tools->tauError(cascadeVertices[0],cascadeVertices[1],massesV0,mass_v0));
324
325 ATH_MSG_DEBUG(
"CascadeTools main vert wrt PV " <<
" CascadeTools SV " <<
" V0Tools SV");
327 <<
", " <<
m_CascadeTools->a0z(moms[0],cascadeVertices[0],cascadeVertices[1])
328 <<
", " <<
m_V0Tools->a0z(cascadeVertices[0],cascadeVertices[1]));
330 <<
", " <<
m_CascadeTools->a0zError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
331 <<
", " <<
m_V0Tools->a0zError(cascadeVertices[0],cascadeVertices[1]));
333 <<
", " <<
m_CascadeTools->a0xy(moms[0],cascadeVertices[0],cascadeVertices[1])
334 <<
", " <<
m_V0Tools->a0xy(cascadeVertices[0],cascadeVertices[1]));
336 <<
", " <<
m_CascadeTools->a0xyError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
337 <<
", " <<
m_V0Tools->a0xyError(cascadeVertices[0],cascadeVertices[1]));
339 <<
", " <<
m_CascadeTools->a0(moms[0],cascadeVertices[0],cascadeVertices[1])
340 <<
", " <<
m_V0Tools->a0(cascadeVertices[0],cascadeVertices[1]));
342 <<
", " <<
m_CascadeTools->a0Error(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
343 <<
", " <<
m_V0Tools->a0Error(cascadeVertices[0],cascadeVertices[1]));
346 ATH_MSG_DEBUG(
"X0 " << primaryVertex->
x() <<
" Y0 " << primaryVertex->
y() <<
" Z0 " << primaryVertex->
z());
349 ATH_MSG_DEBUG(
"Rxy0 wrt PV " <<
m_V0Tools->rxy(cascadeVertices[0],primaryVertex) <<
" RxyErr0 wrt PV " <<
m_V0Tools->rxyError(cascadeVertices[0],primaryVertex));
350 ATH_MSG_DEBUG(
"Rxy1 wrt PV " <<
m_V0Tools->rxy(cascadeVertices[1],primaryVertex) <<
" RxyErr1 wrt PV " <<
m_V0Tools->rxyError(cascadeVertices[1],primaryVertex));
351 ATH_MSG_DEBUG(
"number of covariance matrices " << (
x->getCovariance()).size());
352 }
353
354
355
356 for (
auto x : cascadeinfoContainer)
delete x;
357
358 return StatusCode::SUCCESS;
359 }
#define ATH_CHECK
Evaluate an expression and check for errors.
ElementConstPointer cptr() const
Return a pointer to the referenced element.
bool isValid() const
Test to see if the link can be dereferenced.
#define ATH_MSG_WARNING(x)
#define BPHYS_CHECK(EXP)
Useful CHECK macro.
Gaudi::Property< int > m_PV_max
PublicToolHandle< Analysis::PrimaryVertexRefitter > m_pvRefitter
Gaudi::Property< int > m_v0_pid
SG::ReadHandleKey< xAOD::VertexContainer > m_vertexContainerKey
SG::ReadHandleKey< xAOD::VertexContainer > m_VxPrimaryCandidateName
Name of primary vertex container.
PublicToolHandle< DerivationFramework::CascadeTools > m_CascadeTools
PublicToolHandle< Trk::V0Tools > m_V0Tools
Gaudi::Property< int > m_jpsi_trk_pdg
StatusCode performSearch(std::vector< Trk::VxCascadeInfo * > &cascadeinfoContainer, const EventContext &ctx) const
SG::ReadHandleKey< xAOD::VertexContainer > m_vertexV0ContainerKey
SG::ReadHandleKey< xAOD::EventInfo > m_eventInfo_key
SG::WriteHandleKey< xAOD::VertexContainer > m_refPVContainerName
Gaudi::Property< size_t > m_PV_minNTracks
Gaudi::Property< bool > m_refitPV
Gaudi::Property< std::string > m_hypoName
name of the mass hypothesis.
Gaudi::Property< int > m_DoVertexType
SG::WriteHandleKeyArray< xAOD::VertexContainer > m_cascadeOutputsKeys
virtual bool isValid() override final
Can the handle be successfully dereferenced?
const_pointer_type cptr()
Dereference the pointer.
StatusCode record(std::unique_ptr< T > data)
Record a const object to the store.
pointer_type ptr()
Dereference the pointer.
virtual double pt() const override final
The transverse momentum ( ) of the particle.
float z() const
Returns the z position.
const TrackParticle * trackParticle(size_t i) const
Get the pointer to a given track that was used in vertex reco.
float y() const
Returns the y position.
float x() const
Returns the x position.
bool isElectron(const T &p)
SG::ReadCondHandle< T > makeHandle(const SG::ReadCondHandleKey< T > &key, const EventContext &ctx=Gaudi::Hive::currentContext())
VertexContainer_v1 VertexContainer
Definition of the current "Vertex container version".
Vertex_v1 Vertex
Define the latest version of the vertex class.