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

jikim1290 · junleekim · web-flow · commit aee56dc01c48 · 2024-12-10T22:04:18.000+01:00
Co-authored-by: junleekim &lt;junleekim@maccbookjunlee2024.dyndns.cern.ch&gt;
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);
             }
           }

Original file line number	Diff line number	Diff line change
`@@ -267,6 +267,10 @@ struct CorrelationTask {`
`267`	`267`	`for (const auto& track2 : tracks2) {`
`268`	`268`	`if constexpr (std::experimental::is_detected<HasInvMass, typename TTracks2::iterator>::value && std::experimental::is_detected<HasDecay, typename TTracks2::iterator>::value) {`
`269`	`269`	`if (doprocessSame2Prong2Prong) {`
	`270`	`+ if (cfgDecayParticleMask != 0 && (cfgDecayParticleMask & (1u << static_cast<uint32_t>(track1.decay()))) == 0u)`
	`271`	`+ continue;`
	`272`	`+ if ((track1.decay() != 0) \|\| (track2.decay() != 1)) // D0 in trk1, D0bar in trk2`
	`273`	`+ continue;`
`270`	`274`	`registry.fill(HIST("invMassTwoPart"), track1.invMass(), track2.invMass(), track1.pt(), track2.pt(), multiplicity);`
`271`	`275`	`}`
`272`	`276`	`}`