Skip to content

Commit 4988e10

Browse files
fgrosafgrosa
committed
Fix track type
1 parent d3f5fe9 commit 4988e10

1 file changed

Lines changed: 35 additions & 32 deletions

File tree

‎PWGHF/TableProducer/candidateCreatorXicToXiPiPi.cxx‎

Lines changed: 35 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,6 @@ struct HfCandidateCreatorXicToXiPiPi {
146146
using CascFull = soa::Join<aod::CascDatas, aod::CascCovs>;
147147
using KFCascadesLinked = soa::Join<aod::Cascades, aod::KFCascDataLink>;
148148
using KFCascFull = soa::Join<aod::KFCascDatas, aod::KFCascCovs>;
149-
using TracksWCovDcaPidPrPi = soa::Join<aod::TracksWCovDca, aod::TracksPidPr, aod::TracksPidPi>;
150149
using TracksWCovExtraPidPrPi = soa::Join<aod::TracksWCovExtra, aod::TracksPidPr, aod::TracksPidPi>;
151150

152151
HistogramRegistry registry{"registry"};
@@ -292,7 +291,7 @@ struct HfCandidateCreatorXicToXiPiPi {
292291
aod::HfCascLf3Prongs const& rowsTrackIndexXicPlus,
293292
CascadesLinked const&,
294293
CascFull const&,
295-
TracksWCovDcaPidPrPi const&,
294+
TracksWCovExtraPidPrPi const&,
296295
aod::BCsWithTimestamps const&)
297296
{
298297
// loop over triplets of track indices
@@ -315,8 +314,8 @@ struct HfCandidateCreatorXicToXiPiPi {
315314
continue;
316315
}
317316
auto casc = cascAodElement.cascData_as<CascFull>();
318-
auto trackCharmBachelor0 = rowTrackIndexXicPlus.prong0_as<TracksWCovDcaPidPrPi>();
319-
auto trackCharmBachelor1 = rowTrackIndexXicPlus.prong1_as<TracksWCovDcaPidPrPi>();
317+
auto trackCharmBachelor0 = rowTrackIndexXicPlus.prong0_as<TracksWCovExtraPidPrPi>();
318+
auto trackCharmBachelor1 = rowTrackIndexXicPlus.prong1_as<TracksWCovExtraPidPrPi>();
320319

321320
// preselect cascade candidates
322321
if (doCascadePreselection) {
@@ -376,16 +375,17 @@ struct HfCandidateCreatorXicToXiPiPi {
376375
auto trackParCovCharmBachelor1 = getTrackParCov(trackCharmBachelor1);
377376

378377
// if enabled, apply selections of software trigger
379-
float pPiFromLambda{}, pPrFromLambda{}, pPionFromXi{}, nSigTpcBachelorPi{}, nSigTofBachelorPi{}, nSigTpcPiFromLambda{}, nSigTofPiFromLambda{}, nSigTpcPrFromLambda{}, nSigTofPrFromLambda{};
378+
float pPiFromLambda{}, pPrFromLambda{}, pPionFromXi{}, ptPionFromXi{}, nSigTpcBachelorPi{}, nSigTofBachelorPi{}, nSigTpcPiFromLambda{}, nSigTofPiFromLambda{}, nSigTpcPrFromLambda{}, nSigTofPrFromLambda{};
380379
if (softTrigCuts.applySoftwareTrigSelections) {
381380
// get PID information already here
382-
auto trackPionFromXi = casc.bachelor_as<TracksWCovDcaPidPrPi>();
383-
auto trackPosLambdaDaughter = casc.posTrack_as<TracksWCovDcaPidPrPi>();
384-
auto trackNegLambdaDaughter = casc.negTrack_as<TracksWCovDcaPidPrPi>();
381+
auto trackPionFromXi = casc.bachelor_as<TracksWCovExtraPidPrPi>();
382+
auto trackPosLambdaDaughter = casc.posTrack_as<TracksWCovExtraPidPrPi>();
383+
auto trackNegLambdaDaughter = casc.negTrack_as<TracksWCovExtraPidPrPi>();
385384
nSigTpcBachelorPi = trackPionFromXi.tpcNSigmaPi();
386385
nSigTofBachelorPi = trackPionFromXi.tofNSigmaPi();
387386
pPionFromXi = trackPionFromXi.p();
388-
bool hasTofBachelorPi = trackPionFromXi.has_TOF();
387+
ptPionFromXi = trackPionFromXi.pt();
388+
bool hasTofBachelorPi = trackPionFromXi.hasTOF();
389389
bool hasTofPiFromLambda = false;
390390
bool hasTofPrFromLambda = false;
391391
if (signXic == +1) {
@@ -482,13 +482,14 @@ struct HfCandidateCreatorXicToXiPiPi {
482482
float const nSigTofPiFromXicPlus1 = trackCharmBachelor1.tofNSigmaPi();
483483
if (!softTrigCuts.applySoftwareTrigSelections) {
484484
// Bachelor pion
485-
auto trackPionFromXi = casc.bachelor_as<TracksWCovDcaPidPrPi>();
485+
auto trackPionFromXi = casc.bachelor_as<TracksWCovExtraPidPrPi>();
486486
nSigTpcBachelorPi = trackPionFromXi.tpcNSigmaPi();
487487
nSigTofBachelorPi = trackPionFromXi.tofNSigmaPi();
488488
pPionFromXi = trackPionFromXi.p();
489+
ptPionFromXi = trackPionFromXi.pt();
489490
// Lambda daughters
490-
auto trackPosLambdaDaughter = casc.posTrack_as<TracksWCovDcaPidPrPi>();
491-
auto trackNegLambdaDaughter = casc.negTrack_as<TracksWCovDcaPidPrPi>();
491+
auto trackPosLambdaDaughter = casc.posTrack_as<TracksWCovExtraPidPrPi>();
492+
auto trackNegLambdaDaughter = casc.negTrack_as<TracksWCovExtraPidPrPi>();
492493
if (signXic == +1) {
493494
pPiFromLambda = trackNegLambdaDaughter.p();
494495
nSigTpcPiFromLambda = trackNegLambdaDaughter.tpcNSigmaPi();
@@ -546,7 +547,7 @@ struct HfCandidateCreatorXicToXiPiPi {
546547
impactParameterCasc.getY(), impactParameter0.getY(), impactParameter1.getY(),
547548
std::sqrt(impactParameterCasc.getSigmaY2()), std::sqrt(impactParameter0.getSigmaY2()), std::sqrt(impactParameter1.getSigmaY2()),
548549
/*cascade specific columns*/
549-
pPionFromXi, pPiFromLambda, pPrFromLambda, trackPionFromXi.pt(),
550+
pPionFromXi, pPiFromLambda, pPrFromLambda, ptPionFromXi,
550551
cpaXi, cpaXYXi, cpaLambda, cpaXYLambda, cpaLambdaToXi, cpaXYLambdaToXi,
551552
casc.mXi(), casc.mLambda(), massXiPi0, massXiPi1,
552553
radiusLambda,
@@ -598,33 +599,34 @@ struct HfCandidateCreatorXicToXiPiPi {
598599
float pPiFromLambda{}, pPrFromLambda{}, pPionFromXi{}, nSigTpcBachelorPi{}, nSigTofBachelorPi{}, nSigTpcPiFromLambda{}, nSigTofPiFromLambda{}, nSigTpcPrFromLambda{}, nSigTofPrFromLambda{};
599600
if (softTrigCuts.applySoftwareTrigSelections) {
600601
// get PID information already here
601-
auto trackPionFromXi = casc.bachelor_as<TracksWCovDcaPidPrPi>();
602-
auto trackPosLambdaDaughter = casc.posTrack_as<TracksWCovDcaPidPrPi>();
603-
auto trackNegLambdaDaughter = casc.negTrack_as<TracksWCovDcaPidPrPi>();
602+
auto trackPionFromXi = casc.bachelor_as<TracksWCovExtraPidPrPi>();
603+
auto trackPosLambdaDaughter = casc.posTrack_as<TracksWCovExtraPidPrPi>();
604+
auto trackNegLambdaDaughter = casc.negTrack_as<TracksWCovExtraPidPrPi>();
604605
nSigTpcBachelorPi = trackPionFromXi.tpcNSigmaPi();
605606
nSigTofBachelorPi = trackPionFromXi.tofNSigmaPi();
606607
pPionFromXi = trackPionFromXi.p();
607-
bool hasTofBachelorPi = trackPionFromXi.has_TOF();
608+
ptPionFromXi = trackPionFromXi.pt();
609+
bool hasTofBachelorPi = trackPionFromXi.hasTOF();
608610
bool hasTofPiFromLambda = false;
609611
bool hasTofPrFromLambda = false;
610612
if (signXic == +1) {
611613
pPiFromLambda = trackNegLambdaDaughter.p();
612614
nSigTpcPiFromLambda = trackNegLambdaDaughter.tpcNSigmaPi();
613615
nSigTofPiFromLambda = trackNegLambdaDaughter.tofNSigmaPi();
614-
hasTofPiFromLambda = trackNegLambdaDaughter.has_TOF();
616+
hasTofPiFromLambda = trackNegLambdaDaughter.hasTOF();
615617
pPrFromLambda = trackPosLambdaDaughter.p();
616618
nSigTpcPrFromLambda = trackPosLambdaDaughter.tpcNSigmaPr();
617619
nSigTofPrFromLambda = trackPosLambdaDaughter.tofNSigmaPr();
618-
hasTofPrFromLambda = trackPosLambdaDaughter.has_TOF();
620+
hasTofPrFromLambda = trackPosLambdaDaughter.hasTOF();
619621
} else {
620622
pPiFromLambda = trackPosLambdaDaughter.p();
621623
nSigTpcPiFromLambda = trackPosLambdaDaughter.tpcNSigmaPi();
622624
nSigTofPiFromLambda = trackPosLambdaDaughter.tofNSigmaPi();
623-
hasTofPiFromLambda = trackPosLambdaDaughter.has_TOF();
625+
hasTofPiFromLambda = trackPosLambdaDaughter.hasTOF();
624626
pPrFromLambda = trackNegLambdaDaughter.p();
625627
nSigTpcPrFromLambda = trackNegLambdaDaughter.tpcNSigmaPr();
626628
nSigTofPrFromLambda = trackNegLambdaDaughter.tofNSigmaPr();
627-
hasTofPrFromLambda = trackNegLambdaDaughter.has_TOF();
629+
hasTofPrFromLambda = trackNegLambdaDaughter.hasTOF();
628630
}
629631
auto trackParCovCharmBachelor0 = getTrackParCov(trackCharmBachelor0);
630632
auto trackParCovCharmBachelor1 = getTrackParCov(trackCharmBachelor1);
@@ -808,31 +810,32 @@ struct HfCandidateCreatorXicToXiPiPi {
808810
float const nSigTofPiFromXicPlus1 = trackCharmBachelor1.tofNSigmaPi();
809811
if (!softTrigCuts.applySoftwareTrigSelections) {
810812
// Bachelor pion
811-
auto trackPionFromXi = casc.bachelor_as<TracksWCovDcaPidPrPi>();
813+
auto trackPionFromXi = casc.bachelor_as<TracksWCovExtraPidPrPi>();
812814
nSigTpcBachelorPi = trackPionFromXi.tpcNSigmaPi();
813815
nSigTofBachelorPi = trackPionFromXi.tofNSigmaPi();
814816
pPionFromXi = trackPionFromXi.p();
817+
ptPionFromXi = trackPionFromXi.pt();
815818
// Lambda daughters
816-
auto trackPosLambdaDaughter = casc.posTrack_as<TracksWCovDcaPidPrPi>();
817-
auto trackNegLambdaDaughter = casc.negTrack_as<TracksWCovDcaPidPrPi>();
819+
auto trackPosLambdaDaughter = casc.posTrack_as<TracksWCovExtraPidPrPi>();
820+
auto trackNegLambdaDaughter = casc.negTrack_as<TracksWCovExtraPidPrPi>();
818821
if (signXic == +1) {
819822
pPiFromLambda = trackNegLambdaDaughter.p();
820823
nSigTpcPiFromLambda = trackNegLambdaDaughter.tpcNSigmaPi();
821824
nSigTofPiFromLambda = trackNegLambdaDaughter.tofNSigmaPi();
822-
hasTofPiFromLambda = trackNegLambdaDaughter.has_TOF();
825+
hasTofPiFromLambda = trackNegLambdaDaughter.hasTOF();
823826
pPrFromLambda = trackPosLambdaDaughter.p();
824827
nSigTpcPrFromLambda = trackPosLambdaDaughter.tpcNSigmaPr();
825828
nSigTofPrFromLambda = trackPosLambdaDaughter.tofNSigmaPr();
826-
hasTofPrFromLambda = trackPosLambdaDaughter.has_TOF();
829+
hasTofPrFromLambda = trackPosLambdaDaughter.hasTOF();
827830
} else {
828831
pPiFromLambda = trackPosLambdaDaughter.p();
829832
nSigTpcPiFromLambda = trackPosLambdaDaughter.tpcNSigmaPi();
830833
nSigTofPiFromLambda = trackPosLambdaDaughter.tofNSigmaPi();
831-
hasTofPiFromLambda = trackPosLambdaDaughter.has_TOF();
834+
hasTofPiFromLambda = trackPosLambdaDaughter.hasTOF();
832835
pPrFromLambda = trackNegLambdaDaughter.p();
833836
nSigTpcPrFromLambda = trackNegLambdaDaughter.tpcNSigmaPr();
834837
nSigTofPrFromLambda = trackNegLambdaDaughter.tofNSigmaPr();
835-
hasTofPrFromLambda = trackNegLambdaDaughter.has_TOF();
838+
hasTofPrFromLambda = trackNegLambdaDaughter.hasTOF();
836839
}
837840
}
838841

@@ -874,7 +877,7 @@ struct HfCandidateCreatorXicToXiPiPi {
874877
impactParameterXiXY, impactParameterPi0XY, impactParameterPi1XY,
875878
errImpactParameterXiXY, errImpactParameterPi0XY, errImpactParameterPi1XY,
876879
/*cascade specific columns*/
877-
pPionFromXi, pPiFromLambda, pPrFromLambda, trackPionFromXi.pt(),
880+
pPionFromXi, pPiFromLambda, pPrFromLambda, ptPionFromXi,
878881
cpaXi, cpaXYXi, cpaLambda, cpaXYLambda, cpaLambdaToXi, cpaXYLambdaToXi,
879882
massXi, casc.mLambda(), massXiPi0, massXiPi1,
880883
radiusLambda,
@@ -904,7 +907,7 @@ struct HfCandidateCreatorXicToXiPiPi {
904907
aod::HfCascLf3Prongs const& rowsTrackIndexXicPlus,
905908
CascadesLinked const& cascadesLinked,
906909
CascFull const& cascadesFull,
907-
TracksWCovDcaPidPrPi const& tracks,
910+
TracksWCovExtraPidPrPi const& tracks,
908911
aod::BCsWithTimestamps const& bcs)
909912
{
910913
runXicplusCreatorWithDcaFitter<o2::hf_centrality::CentralityEstimator::None>(collisions, rowsTrackIndexXicPlus, cascadesLinked, cascadesFull, tracks, bcs);
@@ -915,7 +918,7 @@ struct HfCandidateCreatorXicToXiPiPi {
915918
aod::HfCascLf3Prongs const& rowsTrackIndexXicPlus,
916919
CascadesLinked const& cascadesLinked,
917920
CascFull const& cascadesFull,
918-
TracksWCovDcaPidPrPi const& tracks,
921+
TracksWCovExtraPidPrPi const& tracks,
919922
aod::BCsWithTimestamps const& bcs)
920923
{
921924
runXicplusCreatorWithDcaFitter<o2::hf_centrality::CentralityEstimator::FT0C>(collisions, rowsTrackIndexXicPlus, cascadesLinked, cascadesFull, tracks, bcs);
@@ -926,7 +929,7 @@ struct HfCandidateCreatorXicToXiPiPi {
926929
aod::HfCascLf3Prongs const& rowsTrackIndexXicPlus,
927930
CascadesLinked const& cascadesLinked,
928931
CascFull const& cascadesFull,
929-
TracksWCovDcaPidPrPi const& tracks,
932+
TracksWCovExtraPidPrPi const& tracks,
930933
aod::BCsWithTimestamps const& bcs)
931934
{
932935
runXicplusCreatorWithDcaFitter<o2::hf_centrality::CentralityEstimator::FT0M>(collisions, rowsTrackIndexXicPlus, cascadesLinked, cascadesFull, tracks, bcs);

0 commit comments

Comments
 (0)