[PWGHF] Adding Ds prompt selection (#11939)

scattaru · web-flow · commit 4fcdc357e630 · 2025-07-08T09:34:05.000+02:00
diff --git a/PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx b/PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx
@@ -714,16 +714,18 @@ struct HfCorrelatorDsHadrons {
                 continue;
               }
 
-              registry.fill(HIST("hCorrAllPrimaryParticles"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
-              if (std::abs(particleAssoc.pdgCode()) == kPiPlus) {
-                registry.fill(HIST("hCorrAllPrimaryHadrons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
-                registry.fill(HIST("hCorrAllPrimaryPions"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
-              } else if (std::abs(particleAssoc.pdgCode()) == kKPlus) {
-                registry.fill(HIST("hCorrAllPrimaryHadrons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
-                registry.fill(HIST("hCorrAllPrimaryKaons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
-              } else if (std::abs(particleAssoc.pdgCode()) == kProton) {
-                registry.fill(HIST("hCorrAllPrimaryHadrons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
-                registry.fill(HIST("hCorrAllPrimaryProtons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
+              if (isDsPrompt) {
+                registry.fill(HIST("hCorrAllPrimaryParticles"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
+                if (std::abs(particleAssoc.pdgCode()) == kPiPlus) {
+                  registry.fill(HIST("hCorrAllPrimaryHadrons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
+                  registry.fill(HIST("hCorrAllPrimaryPions"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
+                } else if (std::abs(particleAssoc.pdgCode()) == kKPlus) {
+                  registry.fill(HIST("hCorrAllPrimaryHadrons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
+                  registry.fill(HIST("hCorrAllPrimaryKaons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
+                } else if (std::abs(particleAssoc.pdgCode()) == kProton) {
+                  registry.fill(HIST("hCorrAllPrimaryHadrons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
+                  registry.fill(HIST("hCorrAllPrimaryProtons"), getDeltaPhi(particleAssoc.phi(), particle.phi()), particle.pt(), particleAssoc.pt());
+                }
               }
 
               // trackOrigin = RecoDecay::getCharmHadronOrigin(mcParticles, particleAssoc, true);
diff --git a/PWGHF/HFC/Tasks/taskCorrelationDsHadrons.cxx b/PWGHF/HFC/Tasks/taskCorrelationDsHadrons.cxx
@@ -749,14 +749,15 @@ struct HfTaskCorrelationDsHadrons {
                 } else {
                   registry.fill(HIST("hPtCandMcGenPrompt"), mcParticle.pt());
                 }
-                if (mcParticle.originMcGen() == RecoDecay::OriginType::NonPrompt) {
-                  if (useHighDimHistoForEff) {
-                    registry.fill(HIST("hPtCandMcGenNonPrompt"), mcParticle.pt(), collision.numContrib());
-                  } else {
-                    registry.fill(HIST("hPtCandMcGenNonPrompt"), mcParticle.pt());
-                  }
+              }
+              if (mcParticle.originMcGen() == RecoDecay::OriginType::NonPrompt) {
+                if (useHighDimHistoForEff) {
+                  registry.fill(HIST("hPtCandMcGenNonPrompt"), mcParticle.pt(), collision.numContrib());
+                } else {
+                  registry.fill(HIST("hPtCandMcGenNonPrompt"), mcParticle.pt());
                 }
               }
+
               bool isDaughterInAcceptance = true;
               auto daughters = mcParticle.template daughters_as<CandDsMcGen>();
               for (const auto& daughter : daughters) {
@@ -786,7 +787,7 @@ struct HfTaskCorrelationDsHadrons {
               outputMl[iclass] = candidate.mlProbDsToPiKK()[classMl->at(iclass)];
             }
           }
-          if (outputMl[0] < mlOutputPromptMin->at(o2::analysis::findBin(binsPtD, candidate.pt())) || outputMl[0] < mlOutputPromptMax->at(o2::analysis::findBin(binsPtD, candidate.pt())) || outputMl[2] > mlOutputBkg->at(o2::analysis::findBin(binsPtD, candidate.pt()))) {
+          if (outputMl[0] < mlOutputPromptMin->at(o2::analysis::findBin(binsPtD, candidate.pt())) || outputMl[0] > mlOutputPromptMax->at(o2::analysis::findBin(binsPtD, candidate.pt())) || outputMl[2] > mlOutputBkg->at(o2::analysis::findBin(binsPtD, candidate.pt()))) {
             continue;
           }
 
@@ -865,7 +866,7 @@ struct HfTaskCorrelationDsHadrons {
           outputMl[iclass] = candidate.mlProbDsToPiKK()[classMl->at(iclass)];
         }
       }
-      if (outputMl[0] < mlOutputPromptMin->at(o2::analysis::findBin(binsPtD, candidate.pt())) || outputMl[0] < mlOutputPromptMax->at(o2::analysis::findBin(binsPtD, candidate.pt())) || outputMl[2] > mlOutputBkg->at(o2::analysis::findBin(binsPtD, candidate.pt()))) {
+      if (outputMl[0] < mlOutputPromptMin->at(o2::analysis::findBin(binsPtD, candidate.pt())) || outputMl[0] > mlOutputPromptMax->at(o2::analysis::findBin(binsPtD, candidate.pt())) || outputMl[2] > mlOutputBkg->at(o2::analysis::findBin(binsPtD, candidate.pt()))) {
         continue;
       }
       auto collision = candidate.template collision_as<soa::Join<aod::Collisions, aod::FT0Mults, aod::EvSels>>();

Original file line number	Diff line number	Diff line change
`@@ -749,14 +749,15 @@ struct HfTaskCorrelationDsHadrons {`
`749`	`749`	`} else {`
`750`	`750`	`registry.fill(HIST("hPtCandMcGenPrompt"), mcParticle.pt());`
`751`	`751`	`}`
`752`		`- if (mcParticle.originMcGen() == RecoDecay::OriginType::NonPrompt) {`
`753`		`- if (useHighDimHistoForEff) {`
`754`		`- registry.fill(HIST("hPtCandMcGenNonPrompt"), mcParticle.pt(), collision.numContrib());`
`755`		`- } else {`
`756`		`- registry.fill(HIST("hPtCandMcGenNonPrompt"), mcParticle.pt());`
`757`		`- }`
	`752`	`+ }`
	`753`	`+ if (mcParticle.originMcGen() == RecoDecay::OriginType::NonPrompt) {`
	`754`	`+ if (useHighDimHistoForEff) {`
	`755`	`+ registry.fill(HIST("hPtCandMcGenNonPrompt"), mcParticle.pt(), collision.numContrib());`
	`756`	`+ } else {`
	`757`	`+ registry.fill(HIST("hPtCandMcGenNonPrompt"), mcParticle.pt());`
`758`	`758`	`}`
`759`	`759`	`}`
	`760`	`+`
`760`	`761`	`bool isDaughterInAcceptance = true;`
`761`	`762`	`auto daughters = mcParticle.template daughters_as<CandDsMcGen>();`
`762`	`763`	`for (const auto& daughter : daughters) {`
`@@ -786,7 +787,7 @@ struct HfTaskCorrelationDsHadrons {`
`786`	`787`	`outputMl[iclass] = candidate.mlProbDsToPiKK()[classMl->at(iclass)];`
`787`	`788`	`}`
`788`	`789`	`}`
`789`		`- if (outputMl[0] < mlOutputPromptMin->at(o2::analysis::findBin(binsPtD, candidate.pt())) \|\| outputMl[0] < mlOutputPromptMax->at(o2::analysis::findBin(binsPtD, candidate.pt())) \|\| outputMl[2] > mlOutputBkg->at(o2::analysis::findBin(binsPtD, candidate.pt()))) {`
	`790`	`+ if (outputMl[0] < mlOutputPromptMin->at(o2::analysis::findBin(binsPtD, candidate.pt())) \|\| outputMl[0] > mlOutputPromptMax->at(o2::analysis::findBin(binsPtD, candidate.pt())) \|\| outputMl[2] > mlOutputBkg->at(o2::analysis::findBin(binsPtD, candidate.pt()))) {`
`790`	`791`	`continue;`
`791`	`792`	`}`
`792`	`793`
`@@ -865,7 +866,7 @@ struct HfTaskCorrelationDsHadrons {`
`865`	`866`	`outputMl[iclass] = candidate.mlProbDsToPiKK()[classMl->at(iclass)];`
`866`	`867`	`}`
`867`	`868`	`}`
`868`		`- if (outputMl[0] < mlOutputPromptMin->at(o2::analysis::findBin(binsPtD, candidate.pt())) \|\| outputMl[0] < mlOutputPromptMax->at(o2::analysis::findBin(binsPtD, candidate.pt())) \|\| outputMl[2] > mlOutputBkg->at(o2::analysis::findBin(binsPtD, candidate.pt()))) {`
	`869`	`+ if (outputMl[0] < mlOutputPromptMin->at(o2::analysis::findBin(binsPtD, candidate.pt())) \|\| outputMl[0] > mlOutputPromptMax->at(o2::analysis::findBin(binsPtD, candidate.pt())) \|\| outputMl[2] > mlOutputBkg->at(o2::analysis::findBin(binsPtD, candidate.pt()))) {`
`869`	`870`	`continue;`
`870`	`871`	`}`
`871`	`872`	`auto collision = candidate.template collision_as<soa::Join<aod::Collisions, aod::FT0Mults, aod::EvSels>>();`