76 std::vector<Trk::VxCascadeInfo*> cascadeinfoContainer;
77 constexpr
int topoN = 2;
78 std::array<SG::WriteHandle<xAOD::VertexContainer>, topoN> Vtxwritehandles;
81 for (
int i =0;
i<topoN;
i++) {
83 ATH_CHECK( Vtxwritehandles[
i].record (std::make_unique<xAOD::VertexContainer>(),
84 std::make_unique<xAOD::VertexAuxContainer>()) );
91 if (!pvContainer.isValid()) {
93 return StatusCode::FAILURE;
97 if (pvContainer->size()==0){
98 ATH_MSG_WARNING(
"You have no primary vertices: " << pvContainer->size());
99 return StatusCode::RECOVERABLE;
110 ATH_CHECK(refPvContainer.
record(std::make_unique<xAOD::VertexContainer>(), std::make_unique<xAOD::VertexAuxContainer>()));
137 ATH_MSG_DEBUG(
"cascadeinfoContainer size " << cascadeinfoContainer.size());
141 if (!jpsiContainer.isValid()) {
143 return StatusCode::FAILURE;
146 if (!v0Container.isValid()) {
148 return StatusCode::FAILURE;
155 return StatusCode::FAILURE;
176 const std::vector<xAOD::Vertex*> &cascadeVertices =
x->vertices();
177 if(cascadeVertices.size()!=topoN)
179 if(cascadeVertices[0] ==
nullptr || cascadeVertices[1] ==
nullptr)
ATH_MSG_ERROR(
"Error null vertex");
181 for(
int i =0;
i<topoN;
i++) Vtxwritehandles[
i]->push_back(cascadeVertices[
i]);
183 x->setSVOwnership(
false);
184 const auto mainVertex = cascadeVertices[1];
186 const std::vector< std::vector<TLorentzVector> > &moms =
x->getParticleMoms();
189 std::vector<const xAOD::Vertex*> verticestoLink;
190 verticestoLink.push_back(cascadeVertices[0]);
196 const xAOD::Vertex* jpsiVertex = BPhysPVCascadeTools::FindVertex<2>(jpsiContainer.cptr(), cascadeVertices[1]);
197 ATH_MSG_DEBUG(
"1 pt Jpsi tracks " << cascadeVertices[1]->trackParticle(0)->
pt() <<
", " << cascadeVertices[1]->trackParticle(1)->
pt());
201 const xAOD::Vertex* v0Vertex = BPhysPVCascadeTools::FindVertex<2>(v0Container.cptr(), cascadeVertices[0]);;
202 ATH_MSG_DEBUG(
"1 pt V0 tracks " << cascadeVertices[0]->trackParticle(0)->
pt() <<
", " << cascadeVertices[0]->trackParticle(1)->
pt());
206 std::vector<const xAOD::Vertex*> jpsiVerticestoLink;
207 if (jpsiVertex) jpsiVerticestoLink.push_back(jpsiVertex);
212 std::vector<const xAOD::Vertex*> v0VerticestoLink;
213 if (v0Vertex) v0VerticestoLink.push_back(v0Vertex);
221 std::vector<double> massesJpsi(2, mass_track);
222 std::vector<double> massesV0;
223 std::vector<double> Masses(2, mass_track);
258 PtErr_decor(*mainVertex) =
m_CascadeTools->pTError(moms[1],
x->getCovariance()[1]);
260 chi2_decor(*mainVertex) =
x->fitChi2();
261 ndof_decor(*mainVertex) =
x->nDoF();
264 if (
m_refitPV) { refPVContainer = refPvContainer.
ptr(); }
271 Mass_svdecor(*mainVertex) =
m_CascadeTools->invariantMass(moms[0]);
272 MassErr_svdecor(*mainVertex) =
m_CascadeTools->invariantMassError(moms[0],
x->getCovariance()[0]);
274 PtErr_svdecor(*mainVertex) =
m_CascadeTools->pTError(moms[0],
x->getCovariance()[0]);
275 Lxy_svdecor(*mainVertex) =
m_CascadeTools->lxy(moms[0],cascadeVertices[0],cascadeVertices[1]);
276 LxyErr_svdecor(*mainVertex) =
m_CascadeTools->lxyError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1]);
277 Tau_svdecor(*mainVertex) =
m_CascadeTools->tau(moms[0],cascadeVertices[0],cascadeVertices[1]);
278 TauErr_svdecor(*mainVertex) =
m_CascadeTools->tauError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1]);
282 <<
" chi2_1 " <<
m_V0Tools->chisq(cascadeVertices[0])
283 <<
" chi2_2 " <<
m_V0Tools->chisq(cascadeVertices[1])
287 <<
" error " <<
m_V0Tools->invariantMassError(cascadeVertices[0],massesV0)
288 <<
" mass_J " <<
m_V0Tools->invariantMass(cascadeVertices[1],massesJpsi)
289 <<
" error " <<
m_V0Tools->invariantMassError(cascadeVertices[1],massesJpsi));
293 double Mass_B_err =
m_CascadeTools->invariantMassError(moms[1],
x->getCovariance()[1]);
294 double Mass_V0_err =
m_CascadeTools->invariantMassError(moms[0],
x->getCovariance()[0]);
295 ATH_MSG_DEBUG(
"Mass_B " << Mass_B <<
" Mass_V0 " << Mass_V0);
296 ATH_MSG_DEBUG(
"Mass_B_err " << Mass_B_err <<
" Mass_V0_err " << Mass_V0_err);
298 double mprob_V0 =
m_CascadeTools->massProbability(mass_v0,Mass_V0,Mass_V0_err);
299 ATH_MSG_DEBUG(
"mprob_B " << mprob_B <<
" mprob_V0 " << mprob_V0);
302 <<
" Mass_v0 " <<
m_CascadeTools->invariantMass(moms[0],massesV0));
304 <<
" Mass_v0_err " <<
m_CascadeTools->invariantMassError(moms[0],
x->getCovariance()[0],massesV0));
307 <<
" pt_v0 " <<
m_V0Tools->pT(cascadeVertices[0]));
309 <<
" ptErr_v " <<
m_CascadeTools->pTError(moms[0],
x->getCovariance()[0])
310 <<
" ptErr_v0 " <<
m_V0Tools->pTError(cascadeVertices[0]));
314 <<
" lxyErr_v " <<
m_CascadeTools->lxyError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
315 <<
" lxyErr_v0 " <<
m_V0Tools->lxyError(cascadeVertices[0],cascadeVertices[1]));
317 <<
" tau_v0 " <<
m_V0Tools->tau(cascadeVertices[0],cascadeVertices[1],massesV0));
319 <<
" tau_v " <<
m_CascadeTools->tau(moms[0],cascadeVertices[0],cascadeVertices[1])
320 <<
" tau_V " <<
m_CascadeTools->tau(moms[0],cascadeVertices[0],cascadeVertices[1],mass_v0));
322 <<
" tauErr_v " <<
m_CascadeTools->tauError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
323 <<
" tauErr_v0 " <<
m_V0Tools->tauError(cascadeVertices[0],cascadeVertices[1],massesV0));
325 <<
" TauErr_v " <<
m_CascadeTools->tauError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1],mass_v0)
326 <<
" TauErr_v0 " <<
m_V0Tools->tauError(cascadeVertices[0],cascadeVertices[1],massesV0,mass_v0));
328 ATH_MSG_DEBUG(
"CascadeTools main vert wrt PV " <<
" CascadeTools SV " <<
" V0Tools SV");
330 <<
", " <<
m_CascadeTools->a0z(moms[0],cascadeVertices[0],cascadeVertices[1])
331 <<
", " <<
m_V0Tools->a0z(cascadeVertices[0],cascadeVertices[1]));
333 <<
", " <<
m_CascadeTools->a0zError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
334 <<
", " <<
m_V0Tools->a0zError(cascadeVertices[0],cascadeVertices[1]));
336 <<
", " <<
m_CascadeTools->a0xy(moms[0],cascadeVertices[0],cascadeVertices[1])
337 <<
", " <<
m_V0Tools->a0xy(cascadeVertices[0],cascadeVertices[1]));
339 <<
", " <<
m_CascadeTools->a0xyError(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
340 <<
", " <<
m_V0Tools->a0xyError(cascadeVertices[0],cascadeVertices[1]));
342 <<
", " <<
m_CascadeTools->a0(moms[0],cascadeVertices[0],cascadeVertices[1])
343 <<
", " <<
m_V0Tools->a0(cascadeVertices[0],cascadeVertices[1]));
345 <<
", " <<
m_CascadeTools->a0Error(moms[0],
x->getCovariance()[0],cascadeVertices[0],cascadeVertices[1])
346 <<
", " <<
m_V0Tools->a0Error(cascadeVertices[0],cascadeVertices[1]));
349 ATH_MSG_DEBUG(
"X0 " << primaryVertex->
x() <<
" Y0 " << primaryVertex->
y() <<
" Z0 " << primaryVertex->
z());
352 ATH_MSG_DEBUG(
"Rxy0 wrt PV " <<
m_V0Tools->rxy(cascadeVertices[0],primaryVertex) <<
" RxyErr0 wrt PV " <<
m_V0Tools->rxyError(cascadeVertices[0],primaryVertex));
353 ATH_MSG_DEBUG(
"Rxy1 wrt PV " <<
m_V0Tools->rxy(cascadeVertices[1],primaryVertex) <<
" RxyErr1 wrt PV " <<
m_V0Tools->rxyError(cascadeVertices[1],primaryVertex));
354 ATH_MSG_DEBUG(
"number of covariance matrices " << (
x->getCovariance()).size());
359 for (
auto x : cascadeinfoContainer)
delete x;
361 return StatusCode::SUCCESS;