Skip to content

Commit aca9a5c

Browse files
authored
[PWGCF] FU: adding colCheck (D0 MC process), mother PDG check (eff-base), fixing a bug in PurityQA process (D0 task) (#17493)
1 parent b423639 commit aca9a5c

3 files changed

Lines changed: 112 additions & 151 deletions

File tree

PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2892,6 +2892,8 @@ struct FemtoUniverseProducerTask {
28922892
aod::BCsWithTimestamps const&,
28932893
aod::McParticles const& mcParts)
28942894
{
2895+
std::set<int> mcColIds;
2896+
mcColIds.clear();
28952897
// MC Reco
28962898
for (const auto& col : collisions) {
28972899
auto groupedTracks = tracks.sliceBy(perCollisionTracks, col.globalIndex());
@@ -2901,12 +2903,16 @@ struct FemtoUniverseProducerTask {
29012903
// fill the tables
29022904
const auto colcheck = fillCollisions<true>(col, tracks);
29032905
if (colcheck) {
2906+
mcColIds.insert(col.mcCollisionId());
29042907
fillTracks<true>(groupedTracks);
29052908
fillD0D0barMcMl<true>(col, groupedTracks, groupedD0s, mcParts);
29062909
}
29072910
}
29082911
// MC Truth
29092912
for (const auto& mccol : mccols) {
2913+
if (confCollMCTruthOnlyReco && !mcColIds.contains(mccol.globalIndex())) {
2914+
continue;
2915+
}
29102916
auto groupedMCParticles = hfMcGenCands.sliceBy(mcPartPerMcColl, mccol.globalIndex());
29112917
auto groupedCollisions = collisions.sliceBy(recoCollsPerMCColl, mccol.globalIndex());
29122918
fillMCTruthCollisions(groupedCollisions, groupedMCParticles); // fills the reco collisions for mc collision

0 commit comments

Comments
 (0)