[PWGCF] adding decay mask to second track (AliceO2Group#8897)

Co-authored-by: junleekim <[email protected]>
samrangy · Dec 10, 2024 · aee56dc · aee56dc
1 parent 2a9803e
commit aee56dc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/PWGCF/Tasks/correlations.cxx b/PWGCF/Tasks/correlations.cxx
@@ -267,6 +267,10 @@ struct CorrelationTask {
         for (const auto& track2 : tracks2) {
           if constexpr (std::experimental::is_detected<HasInvMass, typename TTracks2::iterator>::value && std::experimental::is_detected<HasDecay, typename TTracks2::iterator>::value) {
             if (doprocessSame2Prong2Prong) {
+              if (cfgDecayParticleMask != 0 && (cfgDecayParticleMask & (1u << static_cast<uint32_t>(track1.decay()))) == 0u)
+                continue;
+              if ((track1.decay() != 0) || (track2.decay() != 1)) // D0 in trk1, D0bar in trk2
+                continue;
               registry.fill(HIST("invMassTwoPart"), track1.invMass(), track2.invMass(), track1.pt(), track2.pt(), multiplicity);
             }
           }