Skip to content

Commit 70712a3

Browse files
lubynetslubynets
committed
add N PV tracks for each candidate
1 parent bdbc872 commit 70712a3

File tree

1 file changed

+17
-11
lines changed

1 file changed

+17
-11
lines changed

‎PWGHF/TableProducer/treeCreatorLcToPKPi.cxx‎

Lines changed: 17 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ DECLARE_SOA_COLUMN(ErrP, errP, float); //! mom
131131
DECLARE_SOA_COLUMN(ErrPt, errPt, float); //! transverse momentum error
132132
DECLARE_SOA_COLUMN(IsSelected, isSelected, int); //! flag whether candidate was selected in candidateSelectorLc task
133133
DECLARE_SOA_COLUMN(SigBgStatus, sigBgStatus, int); //! 0 bg, 1 prompt, 2 non-prompt, 3 wrong order of prongs, -1 default value (impossible, should not be the case), -999 for data
134+
DECLARE_SOA_COLUMN(MultNTracksPV, multNTracksPV, int);
134135
} // namespace kf
135136

136137
namespace mc_match
@@ -161,7 +162,8 @@ DECLARE_SOA_TABLE(HfCandLcKFs, "AOD", "HFCANDLCKF",
161162
kf::Chi2GeoProtonKaon, kf::Chi2GeoProtonPion, kf::Chi2GeoPionKaon,
162163
kf::Chi2Geo, kf::Chi2Topo, kf::DecayLength, kf::DecayLengthError, kf::DecayLengthNormalised, kf::T, kf::ErrT,
163164
kf::MassInv, kf::P, kf::Pt, kf::ErrP, kf::ErrPt,
164-
kf::IsSelected, kf::SigBgStatus);
165+
kf::IsSelected, kf::SigBgStatus,
166+
kf::MultNTracksPV);
165167

166168
DECLARE_SOA_TABLE(HfCandLcLites, "AOD", "HFCANDLCLITE",
167169
collision::PosX,
@@ -478,6 +480,7 @@ struct HfTreeCreatorLcToPKPi {
478480
auto trackPos1 = candidate.template prong0_as<soa::Join<TracksWPid, o2::aod::McTrackLabels>>(); // positive daughter (negative for the antiparticles)
479481
auto trackNeg = candidate.template prong1_as<soa::Join<TracksWPid, o2::aod::McTrackLabels>>(); // negative daughter (positive for the antiparticles)
480482
auto trackPos2 = candidate.template prong2_as<soa::Join<TracksWPid, o2::aod::McTrackLabels>>(); // positive daughter (negative for the antiparticles)
483+
auto collision = candidate.template collision_as<Colls>();
481484
auto fillTable = [&](int CandFlag) {
482485
double pseudoRndm = trackPos1.pt() * 1000. - static_cast<int64_t>(trackPos1.pt() * 1000);
483486
const int FunctionSelection = CandFlag == 0 ? candidate.isSelLcToPKPi() : candidate.isSelLcToPiKP();
@@ -664,7 +667,8 @@ struct HfTreeCreatorLcToPKPi {
664667
chi2GeoProtonKaon, chi2GeoProtonPion, chi2GeoPionKaon,
665668
chi2Geo, chi2Topo, l, dl, l / dl, t, deltaT,
666669
mass, p, pt, deltaP, deltaPt,
667-
FunctionSelection, sigbgstatus);
670+
FunctionSelection, sigbgstatus,
671+
collision.multNTracksPV());
668672
}
669673
if (fillCandidateMcTable) {
670674
float p, pt, svX, svY, svZ, pvX, pvY, pvZ, l, t;
@@ -747,7 +751,7 @@ struct HfTreeCreatorLcToPKPi {
747751
/// \param particles Generated particle table
748752
/// \param tracks Track table
749753
/// \param bcs Bunch-crossing table
750-
void processMcNoCentralityWithDCAFitterN(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::PVMultZeqs> const& collisions,
754+
void processMcNoCentralityWithDCAFitterN(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::PVMultZeqs, aod::PVMults> const& collisions,
751755
aod::McCollisions const& mcCollisions,
752756
soa::Join<aod::HfCand3Prong, aod::HfCand3ProngMcRec, aod::HfSelLc> const& candidates,
753757
soa::Join<aod::McParticles, aod::HfCand3ProngMcGen> const& particles,
@@ -763,7 +767,7 @@ struct HfTreeCreatorLcToPKPi {
763767
/// \param candidates Lc->pKpi candidate table
764768
/// \param tracks Track table
765769
/// \param bcs Bunch-crossing table
766-
void processMcWithCentralityWithDCAFitterN(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::PVMultZeqs, Cents> const& collisions,
770+
void processMcWithCentralityWithDCAFitterN(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::PVMultZeqs, Cents, aod::PVMults> const& collisions,
767771
aod::McCollisions const& mcCollisions,
768772
soa::Join<aod::HfCand3Prong, aod::HfCand3ProngMcRec, aod::HfSelLc> const& candidates,
769773
soa::Join<aod::McParticles, aod::HfCand3ProngMcGen> const& particles,
@@ -780,7 +784,7 @@ struct HfTreeCreatorLcToPKPi {
780784
/// \param particles Generated particle table
781785
/// \param tracks Track table
782786
/// \param bcs Bunch-crossing table
783-
void processMcNoCentralityWithKFParticle(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::PVMultZeqs> const& collisions,
787+
void processMcNoCentralityWithKFParticle(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::PVMultZeqs, aod::PVMults> const& collisions,
784788
aod::McCollisions const& mcCollisions,
785789
soa::Join<aod::HfCand3Prong, aod::HfCand3ProngMcRec, aod::HfSelLc, aod::HfCand3ProngKF> const& candidates,
786790
soa::Join<aod::McParticles, aod::HfCand3ProngMcGen> const& particles,
@@ -796,7 +800,7 @@ struct HfTreeCreatorLcToPKPi {
796800
/// \param candidates Lc->pKpi candidate table
797801
/// \param tracks Track table
798802
/// \param bcs Bunch-crossing table
799-
void processMcWithCentralityWithKFParticle(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::PVMultZeqs, Cents> const& collisions,
803+
void processMcWithCentralityWithKFParticle(soa::Join<aod::Collisions, aod::McCollisionLabels, aod::PVMultZeqs, Cents, aod::PVMults> const& collisions,
800804
aod::McCollisions const& mcCollisions,
801805
soa::Join<aod::HfCand3Prong, aod::HfCand3ProngMcRec, aod::HfSelLc, aod::HfCand3ProngKF> const& candidates,
802806
soa::Join<aod::McParticles, aod::HfCand3ProngMcGen> const& particles,
@@ -867,6 +871,7 @@ struct HfTreeCreatorLcToPKPi {
867871
auto trackPos1 = candidate.template prong0_as<TracksWPid>(); // positive daughter (negative for the antiparticles)
868872
auto trackNeg = candidate.template prong1_as<TracksWPid>(); // negative daughter (positive for the antiparticles)
869873
auto trackPos2 = candidate.template prong2_as<TracksWPid>(); // positive daughter (negative for the antiparticles)
874+
auto collision = candidate.template collision_as<Colls>();
870875
auto fillTable = [&](int CandFlag) {
871876
double pseudoRndm = trackPos1.pt() * 1000. - static_cast<int64_t>(trackPos1.pt() * 1000);
872877
const int FunctionSelection = CandFlag == 0 ? candidate.isSelLcToPKPi() : candidate.isSelLcToPiKP();
@@ -1052,7 +1057,8 @@ struct HfTreeCreatorLcToPKPi {
10521057
chi2Geo_proton_kaon, chi2Geo_proton_pion, chi2Geo_pion_kaon,
10531058
chi2Geo, chi2Topo, l, dl, l / dl, T, deltaT,
10541059
mass, p, pt, deltaP, deltaPt,
1055-
FunctionSelection, UndefValueInt);
1060+
FunctionSelection, UndefValueInt,
1061+
collision.multNTracksPV());
10561062
}
10571063
}
10581064
};
@@ -1067,7 +1073,7 @@ struct HfTreeCreatorLcToPKPi {
10671073
/// \param candidates Lc->pKpi candidate table
10681074
/// \param tracks Track table
10691075
/// \param bcs Bunch-crossing table
1070-
void processDataNoCentralityWithDCAFitterN(soa::Join<aod::Collisions, aod::PVMultZeqs> const& collisions,
1076+
void processDataNoCentralityWithDCAFitterN(soa::Join<aod::Collisions, aod::PVMultZeqs, aod::PVMults> const& collisions,
10711077
soa::Join<aod::HfCand3Prong, aod::HfSelLc> const& candidates,
10721078
TracksWPid const& tracks, aod::BCs const& bcs)
10731079
{
@@ -1080,7 +1086,7 @@ struct HfTreeCreatorLcToPKPi {
10801086
/// \param candidates Lc->pKpi candidate table
10811087
/// \param tracks Track table
10821088
/// \param bcs Bunch-crossing table
1083-
void processDataWithCentralityWithDCAFitterN(soa::Join<aod::Collisions, aod::PVMultZeqs, Cents> const& collisions,
1089+
void processDataWithCentralityWithDCAFitterN(soa::Join<aod::Collisions, aod::PVMultZeqs, Cents, aod::PVMults> const& collisions,
10841090
soa::Join<aod::HfCand3Prong, aod::HfSelLc> const& candidates,
10851091
TracksWPid const& tracks, aod::BCs const& bcs)
10861092
{
@@ -1093,7 +1099,7 @@ struct HfTreeCreatorLcToPKPi {
10931099
/// \param candidates Lc->pKpi candidate table
10941100
/// \param tracks Track table
10951101
/// \param bcs Bunch-crossing table
1096-
void processDataNoCentralityWithKFParticle(soa::Join<aod::Collisions, aod::PVMultZeqs> const& collisions,
1102+
void processDataNoCentralityWithKFParticle(soa::Join<aod::Collisions, aod::PVMultZeqs, aod::PVMults> const& collisions,
10971103
soa::Join<aod::HfCand3Prong, aod::HfSelLc, aod::HfCand3ProngKF> const& candidates,
10981104
TracksWPid const& tracks, aod::BCs const& bcs)
10991105
{
@@ -1106,7 +1112,7 @@ struct HfTreeCreatorLcToPKPi {
11061112
/// \param candidates Lc->pKpi candidate table
11071113
/// \param tracks Track table
11081114
/// \param bcs Bunch-crossing table
1109-
void processDataWithCentralityWithKFParticle(soa::Join<aod::Collisions, aod::PVMultZeqs, Cents> const& collisions,
1115+
void processDataWithCentralityWithKFParticle(soa::Join<aod::Collisions, aod::PVMultZeqs, Cents, aod::PVMults> const& collisions,
11101116
soa::Join<aod::HfCand3Prong, aod::HfSelLc, aod::HfCand3ProngKF> const& candidates,
11111117
TracksWPid const& tracks, aod::BCs const& bcs)
11121118
{

0 commit comments

Comments
 (0)