Skip to content

Commit 8420c8c

Browse files
ataviragatavirag
committed
[PWGHF] Add variables to derivedDataCreatorBplusToD0Pi.cxx
1 parent ef0039e commit 8420c8c

File tree

2 files changed

+58
-0
lines changed

2 files changed

+58
-0
lines changed

‎PWGHF/DataModel/DerivedTables.h‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -610,9 +610,14 @@ DECLARE_SOA_TABLE_STAGED(HfBplusPars, "HFBPPAR", //! Table with candidate proper
610610

611611
DECLARE_SOA_TABLE_STAGED(HfBplusParD0s, "HFBPPARD0", //! Table with D0 candidate properties used for selection of B+
612612
hf_cand_par_charm::CpaCharm,
613+
hf_cand_par_charm::DecayLengthXYCharm,
614+
hf_cand_par_charm::DecayLengthNormalisedCharm,
615+
hf_cand_par_charm::DecayLengthXYNormalisedCharm,
613616
hf_cand_par_charm::DecayLengthCharm,
614617
hf_cand_par_charm::ImpactParameter0Charm,
615618
hf_cand_par_charm::ImpactParameter1Charm,
619+
hf_cand_par_charm::ImpactParameterNormalised0Charm,
620+
hf_cand_par_charm::ImpactParameterNormalised1Charm,
616621
hf_cand_par_charm::ImpactParameterProductCharm,
617622
hf_cand_par_charm::NSigTpcPiExpPiCharm,
618623
hf_cand_par_charm::NSigTofPiExpPiCharm,
@@ -627,6 +632,21 @@ DECLARE_SOA_TABLE_STAGED(HfBplusParD0s, "HFBPPARD0", //! Table with D0 candidate
627632
hf_cand_par_charm::NSigTofKaExpKaCharm,
628633
hf_cand_par_charm::NSigTpcTofKaExpKaCharm);
629634

635+
DECLARE_SOA_TABLE_STAGED(HfBplusParD0Es, "HFBPPARD0E", //! Table with additional D0 candidate properties used for selection of B+
636+
hf_cand_par_charm::Chi2PCACharm,
637+
hf_cand_par_charm::NProngsContributorsPVCharm,
638+
hf_cand_par_charm::InvMassCharm,
639+
hf_cand_par_charm::MaxNormalisedDeltaIPCharm,
640+
hf_cand_par_charm::PxProng0Charm,
641+
hf_cand_par_charm::PyProng0Charm,
642+
hf_cand_par_charm::PzProng0Charm,
643+
hf_cand_par_charm::PxProng1Charm,
644+
hf_cand_par_charm::PyProng1Charm,
645+
hf_cand_par_charm::PzProng1Charm,
646+
hf_cand_par_charm::PtProng0Charm,
647+
hf_cand_par_charm::PtProng1Charm
648+
);
649+
630650
DECLARE_SOA_TABLE_STAGED(HfBplusParEs, "HFBPPARE", //! Table with additional candidate properties used for selection
631651
hf_cand::XSecondaryVertex,
632652
hf_cand::YSecondaryVertex,
@@ -638,6 +658,10 @@ DECLARE_SOA_TABLE_STAGED(HfBplusParEs, "HFBPPARE", //! Table with additional can
638658
hf_cand::PxProng1,
639659
hf_cand::PyProng1,
640660
hf_cand::PzProng1,
661+
hf_cand_par::PProng0,
662+
hf_cand::PxProng0,
663+
hf_cand::PyProng0,
664+
hf_cand::PzProng0,
641665
hf_cand::ErrorImpactParameter1,
642666
hf_cand_par::CosThetaStar,
643667
hf_cand_par::Ct,

‎PWGHF/TableProducer/derivedDataCreatorBplusToD0Pi.cxx‎

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ struct HfDerivedDataCreatorBplusToD0Pi {
7272
// Candidates
7373
Produces<o2::aod::HfBplusPars> rowCandidatePar;
7474
Produces<o2::aod::HfBplusParD0s> rowCandidateParD0;
75+
Produces<o2::aod::HfBplusParD0Es> rowCandidateParD0E;
7576
Produces<o2::aod::HfBplusParEs> rowCandidateParE;
7677
Produces<o2::aod::HfBplusSels> rowCandidateSel;
7778
Produces<o2::aod::HfBplusMls> rowCandidateMl;
@@ -83,6 +84,7 @@ struct HfDerivedDataCreatorBplusToD0Pi {
8384
HfConfigurableDerivedData confDerData;
8485
Configurable<bool> fillCandidatePar{"fillCandidatePar", true, "Fill candidate parameters"};
8586
Configurable<bool> fillCandidateParD0{"fillCandidateParD0", true, "Fill D0 candidate parameters"};
87+
Configurable<bool> fillCandidateParD0E{"fillCandidateParD0E", true, "Fill additional D0 candidate parameters"};
8688
Configurable<bool> fillCandidateParE{"fillCandidateParE", true, "Fill candidate extended parameters"};
8789
Configurable<bool> fillCandidateSel{"fillCandidateSel", true, "Fill candidate selection flags"};
8890
Configurable<bool> fillCandidateMl{"fillCandidateMl", true, "Fill candidate selection ML scores"};
@@ -181,8 +183,13 @@ struct HfDerivedDataCreatorBplusToD0Pi {
181183
rowCandidateParD0(
182184
prongCharm.cpa(),
183185
prongCharm.decayLength(),
186+
prongCharm.decayLengthXY(),
187+
prongCharm.decayLengthNormalised(),
188+
prongCharm.decayLengthXYNormalised(),
184189
prongCharm.impactParameter0(),
185190
prongCharm.impactParameter1(),
191+
prongCharm.impactParameterNormalised0(),
192+
prongCharm.impactParameterNormalised1(),
186193
prongCharm.impactParameterProduct(),
187194
sigmas[HfProngSpecies::Pion][HfProngSpecies::Pion][0],
188195
sigmas[HfProngSpecies::Pion][HfProngSpecies::Pion][1],
@@ -197,6 +204,28 @@ struct HfDerivedDataCreatorBplusToD0Pi {
197204
sigmas[HfProngSpecies::Kaon][HfProngSpecies::Kaon][1],
198205
sigmas[HfProngSpecies::Kaon][HfProngSpecies::Kaon][2]);
199206
}
207+
if (fillCandidateParD0E) {
208+
float invMassD0;
209+
if (candFlag == 0) {
210+
invMassD0 = HfHelper::invMassD0ToPiK(prongCharm);
211+
} else if (candFlag == 1) {
212+
invMassD0 = HfHelper::invMassD0barToKPi(prongCharm);
213+
}
214+
rowCandidateParD0E(
215+
prongCharm.chi2PCA(),
216+
prongCharm.nProngsContributorsPV(),
217+
invMassD0,
218+
prongCharm.maxNormalisedDeltaIP(),
219+
prongCharm.pxProng0(),
220+
prongCharm.pyProng0(),
221+
prongCharm.pzProng0(),
222+
prongCharm.pxProng1(),
223+
prongCharm.pyProng1(),
224+
prongCharm.pzProng1(),
225+
prongCharm.ptProng0(),
226+
prongCharm.ptProng1()
227+
);
228+
}
200229
if (fillCandidateParE) {
201230
rowCandidateParE(
202231
candidate.xSecondaryVertex(),
@@ -209,6 +238,10 @@ struct HfDerivedDataCreatorBplusToD0Pi {
209238
candidate.pxProng1(),
210239
candidate.pyProng1(),
211240
candidate.pzProng1(),
241+
RecoDecay::p(candidate.pxProng0(), candidate.pyProng0(), candidate.pzProng0()),
242+
candidate.pxProng0(),
243+
candidate.pyProng0(),
244+
candidate.pzProng0(),
212245
candidate.errorImpactParameter1(),
213246
HfHelper::cosThetaStarBplus(candidate),
214247
ct);
@@ -277,6 +310,7 @@ struct HfDerivedDataCreatorBplusToD0Pi {
277310
reserveTable(rowCandidatePar, fillCandidatePar, sizeTableCand);
278311
reserveTable(rowCandidateParD0, fillCandidateParD0, sizeTableCand);
279312
reserveTable(rowCandidateParE, fillCandidateParE, sizeTableCand);
313+
reserveTable(rowCandidateParD0E, fillCandidateParD0E, sizeTableCand);
280314
reserveTable(rowCandidateSel, fillCandidateSel, sizeTableCand);
281315
reserveTable(rowCandidateMl, fillCandidateMl, sizeTableCand);
282316
reserveTable(rowCandidateMlD0, fillCandidateMlD0, sizeTableCand);

0 commit comments

Comments
 (0)