Skip to content

[PWGHF] correlatorFlowCharmHadronsReduced: add multiplicity selections #18225

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Marcellocosti wants to merge 4 commits into
base: master
Choose a base branch
from
+75 −17
Open

[PWGHF] correlatorFlowCharmHadronsReduced: add multiplicity selections #18225

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
a8be160
Multiplicity selections
Marcellocosti Oct 8, 2026
3f31f7f
Fix template
Marcellocosti Oct 8, 2026
6b60af2
Please consider the following formatting changes
alibuild Oct 8, 2026
7e7c9f4
Merge pull request #88 from alibuild/alibot-cleanup-18225
Marcellocosti Oct 8, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
92 changes: 75 additions & 17 deletions PWGHF/HFC/TableProducer/correlatorFlowCharmHadronsReduced.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -96,8 +96,13 @@ struct HfCorrelatorFlowCharmHadronsReduced {
Configurable<std::vector<double>> bdtScore0PtMaxs{"bdtScore0PtMaxs", std::vector<double>{0.1, 0.1, 0.1, 0.1, 0.1}, "pT-differential maximum score 0 for charm candidates"};
Configurable<std::vector<double>> bdtScore1PtMins{"bdtScore1PtMins", std::vector<double>{0.1, 0.1, 0.1, 0.1, 0.1}, "pT-differential minimum score 1 for charm candidates"};
Configurable<std::vector<double>> binsPtAssoc{"binsPtAssoc", std::vector<double>{0.3, 1., 2., 50.}, "pT bin limits for associated particles"};
Configurable<bool> selCentrality{"selCentrality", false, "Use centrality to select events"};
Configurable<bool> selMultiplicity{"selMultiplicity", false, "Use multiplicity to select events"};
Configurable<bool> countCharmCandsPerCollision{"countCharmCandsPerCollision", false, "Count charm candidates per collision"};
Configurable<float> centralityMin{"centralityMin", 0, "min. centrality"};
Configurable<float> centralityMax{"centralityMax", 10., "max. centrality"};
Configurable<float> multiplicityMin{"multiplicityMin", 0, "min. multiplicity"};
Configurable<float> multiplicityMax{"multiplicityMax", 10., "max. multiplicity"};
Configurable<float> deltaEtaAbsMin{"deltaEtaAbsMin", 0.5, "min. pair delta eta"};
Configurable<float> deltaEtaAbsMax{"deltaEtaAbsMax", 2., "max. pair delta eta"};
Configurable<float> dcaXYTrackMax{"dcaXYTrackMax", 1., "max. track DCA XY"};
Expand All @@ -106,7 +111,7 @@ struct HfCorrelatorFlowCharmHadronsReduced {
Configurable<int> itsNClsMin{"itsNClsMin", 1, "min. ITS clusters"};
Configurable<float> downSamplePairs{"downSamplePairs", 1., "Fraction of pairs to keep"};
Configurable<float> ptMaxForDownSample{"ptMaxForDownSample", 10., "Maximum pt for the application of the downsampling factor"};
Configurable<float> centMaxForDownSample{"centMaxForDownSample", 10., "Maximum centrality for the application of the downsampling factor"};
Configurable<float> centMultMaxForDownSample{"centMultMaxForDownSample", 10., "Maximum centrality for the application of the downsampling factor"};

SliceCache cache;

Expand All @@ -131,17 +136,30 @@ struct HfCorrelatorFlowCharmHadronsReduced {
ConfigurableAxis binsMultFT0M{"binsMultFT0M", {100, 0., 10000.}, "Multiplicity as FT0M signal amplitude"};
ConfigurableAxis binsCent{"binsCent", {100, 0., 100.}, "Centrality bins"};
ConfigurableAxis binsPosZ{"binsPosZ", {100, -10., 10.}, "Primary vertex z coordinate"};
ConfigurableAxis binsEta{"binsEta", {50, -2., 2.}, "Eta bins"};
ConfigurableAxis binsPhi{"binsPhi", {64, -o2::constants::math::PIHalf, 3. * o2::constants::math::PIHalf}, "Phi bins"};
ConfigurableAxis binsDeltaEta{"binsDeltaEta", {100, -2., 2.}, "Delta Eta bins"};
ConfigurableAxis binsDeltaPhi{"binsDeltaPhi", {64, -3., 3.}, "Delta Phi bins"};
ConfigurableAxis binsDeltaPhi{"binsDeltaPhi", {64, -o2::constants::math::PIHalf, 3. * o2::constants::math::PIHalf}, "Delta Phi bins"};
ConfigurableAxis binsMlOne{"binsMlOne", {100, 0., 1.}, "ML score index 1 bins"};
ConfigurableAxis binsMlTwo{"binsMlTwo", {100, 0., 1.}, "ML score index 2 bins"};

HistogramRegistry registry{"registry", {}, OutputObjHandlingPolicy::AnalysisObject};

void init(InitContext&)
{
if (selCentrality && selMultiplicity) {
LOGP(fatal, "You cannot select events based on both centrality and multiplicity! Please check your configuration!");
}
if (!selCentrality && !selMultiplicity) {
LOGP(fatal, "You must select events based on either centrality or multiplicity! Please check your configuration!");
}

// Fatal if selCentrality and process functions using multiplicity mixing, or selMultiplicity and process functions using centrality mixing
if (selCentrality && (doprocessSameEventCharmHadWMultMix || doprocessMixedEventCharmHadWMultMix || doprocessSameEventHadHadWMultMix || doprocessMixedEventHadHadWMultMix)) {
LOGP(fatal, "You cannot select events based on centrality and process functions using multiplicity mixing! Please check your configuration!");
}
if (selMultiplicity && (doprocessSameEventCharmHadWCentMix || doprocessMixedEventCharmHadWCentMix || doprocessSameEventHadHadWCentMix || doprocessMixedEventHadHadWCentMix)) {
LOGP(fatal, "You cannot select events based on multiplicity and process functions using centrality mixing! Please check your configuration!");
}

if ((doprocessSameEventCharmHadWCentMix && doprocessMixedEventCharmHadWMultMix) ||
(doprocessSameEventCharmHadWMultMix && doprocessMixedEventCharmHadWCentMix) ||
(doprocessSameEventHadHadWCentMix && doprocessMixedEventHadHadWMultMix) ||
Expand Down Expand Up @@ -185,6 +203,8 @@ struct HfCorrelatorFlowCharmHadronsReduced {
}
registry.add("hZVtxPoolBinSE", "z vertex SE", {HistType::kTH2F, {{axisPosZ}, {axisPoolBin}}});
registry.add("hZVtxPoolBinME", "z vertex ME", {HistType::kTH2F, {{axisPosZ}, {axisPoolBin}}});
registry.add("hCollisionsCentMultAll", "Collision centrality vs multiplicity", {HistType::kTH2F, {{axisCent}, {axisMultFT0M}}});
registry.add("hCollisionsCentMultSel", "Collision centrality vs multiplicity", {HistType::kTH2F, {{axisCent}, {axisMultFT0M}}});
registry.add("hPoolBinTrigSE", "Trigger candidates pool bin SE", {HistType::kTH1F, {axisPoolBin}});
registry.add("hPoolBinTrigME", "Trigger candidates pool bin ME", {HistType::kTH1F, {axisPoolBin}});
registry.add("hPoolBinAssocSE", "Associated particles pool bin SE", {HistType::kTH1F, {axisPoolBin}});
Expand All @@ -205,12 +225,26 @@ struct HfCorrelatorFlowCharmHadronsReduced {
registry.add("hSparseCorrelationsMECharmHad", "THn for ME Charm-Had correlations", HistType::kTHnSparseF, axes);
}
if (doprocessCharmTriggers) {
if (countCharmCandsPerCollision) {
axesTrigger.push_back({100, -0.5, 99.5, "nCharmCandidates"});
}
registry.add("hSparseTrigCandsCharm", "THn for Charm trigger candidates", HistType::kTHnSparseF, axesTrigger);
}
}
}
}

template <typename TColl>
bool isSelCollision(const TColl& collision)
{
if (selCentrality) {
return (collision.centrality() >= centralityMin && collision.centrality() <= centralityMax);
} else if (selMultiplicity) {
return (collision.multiplicity() >= multiplicityMin && collision.multiplicity() <= multiplicityMax);
}
return false;
}

/// Get the binning pool associated to the collision
/// \param collision is the collision
/// \param binPolicy is the binning policy for the correlation
Expand Down Expand Up @@ -265,9 +299,12 @@ struct HfCorrelatorFlowCharmHadronsReduced {
const TBinningType& binPolicy)
{
auto collision = pair.template hfcRedCorrColl_as<o2::aod::HfcRedCorrColls>();
if (collision.centrality() < centralityMin || collision.centrality() > centralityMax) {
registry.fill(HIST("hCollisionsCentMultAll"), collision.centrality(), collision.multiplicity());
if (!isSelCollision(collision)) {
return;
}
registry.fill(HIST("hCollisionsCentMultSel"), collision.centrality(), collision.multiplicity());

double const ptTrig = trigCand.ptTrig();
if constexpr (requires { trigCand.bdtScore0Trig(); }) { // ML selection on bkg score for Charm-Had case
if (!isSelBdtScoreCut(trigCand, ptTrig)) {
Expand All @@ -276,7 +313,8 @@ struct HfCorrelatorFlowCharmHadronsReduced {
}
if (downSamplePairs < 1.) {
float const pseudoRndm = ptTrig * 1000. - static_cast<int64_t>(ptTrig * 1000);
if (ptTrig < ptMaxForDownSample && collision.centrality() < centMaxForDownSample && pseudoRndm >= downSamplePairs) {
float collCentMult = selCentrality ? collision.centrality() : collision.multiplicity();
if (ptTrig < ptMaxForDownSample && collCentMult < centMultMaxForDownSample && pseudoRndm >= downSamplePairs) {
return;
}
}
Expand Down Expand Up @@ -317,10 +355,14 @@ struct HfCorrelatorFlowCharmHadronsReduced {
if (trigCands.size() == 0 || assocTracks.size() == 0) {
continue;
}
if (trigColl.centrality() < centralityMin || trigColl.centrality() > centralityMax ||
assocColl.centrality() < centralityMin || assocColl.centrality() > centralityMax) {
registry.fill(HIST("hCollisionsCentMultAll"), trigColl.centrality(), trigColl.multiplicity());
registry.fill(HIST("hCollisionsCentMultAll"), assocColl.centrality(), assocColl.multiplicity());
if (!isSelCollision(trigColl) || !isSelCollision(assocColl)) {
continue;
}
registry.fill(HIST("hCollisionsCentMultSel"), trigColl.centrality(), trigColl.multiplicity());
registry.fill(HIST("hCollisionsCentMultSel"), assocColl.centrality(), assocColl.multiplicity());

int const poolBinTrig = getPoolBin<true>(trigColl, binPolicy);
int const poolBinAssoc = getPoolBin<true>(assocColl, binPolicy);
if (poolBinAssoc != poolBinTrig) {
Expand All @@ -345,8 +387,10 @@ struct HfCorrelatorFlowCharmHadronsReduced {
double const ptAssoc = getPt(assocTrack);
if (downSamplePairs < 1.) {
float const pseudoRndm = ptAssoc * 1000. - static_cast<int64_t>(ptAssoc * 1000);
if (ptTrig < ptMaxForDownSample && trigColl.centrality() < centMaxForDownSample &&
assocColl.centrality() < centMaxForDownSample && pseudoRndm >= downSamplePairs) {
float trigCollCentMult = selCentrality ? trigColl.centrality() : trigColl.multiplicity();
float assocCollCentMult = selCentrality ? assocColl.centrality() : assocColl.multiplicity();
if (ptTrig < ptMaxForDownSample && trigCollCentMult < centMultMaxForDownSample &&
assocCollCentMult < centMultMaxForDownSample && pseudoRndm >= downSamplePairs) {
continue;
}
}
Expand Down Expand Up @@ -430,12 +474,15 @@ struct HfCorrelatorFlowCharmHadronsReduced {
Pair<TCollision, TTrigCand, TTrackAssoc, TBinningType> pairData{binPolicy, numberEventsMixed, -1, collisions, tracksTuple, &cache};

for (const auto& [c1, tracks1, c2, tracks2] : pairData) {
if (!isSelCollision(c1) || !isSelCollision(c2)) {
continue;
}
if (tracks1.size() == 0) {
continue;
}

int poolBin = binPolicy.getBin({c2.posZ(), c2.centrality()});
int poolBinTrigCand = binPolicy.getBin({c1.posZ(), c1.centrality()});
int poolBin = binPolicy.getBin({c2.posZ(), selCentrality ? c2.centrality() : c2.multiplicity()});
int poolBinTrigCand = binPolicy.getBin({c1.posZ(), selCentrality ? c1.centrality() : c1.multiplicity()});

if (poolBin != poolBinTrigCand) {
LOGF(info, "Error, poolBins are different");
Expand All @@ -446,7 +493,6 @@ struct HfCorrelatorFlowCharmHadronsReduced {
if (!isSelBdtScoreCut(trigCand, trigCand.ptTrig())) {
continue;
}
LOGF(info, "Mixed event tracks pair: (%d, %d) from events (%d, %d), track event: (%d, %d)", trigCand.index(), assTrk.index(), c1.index(), c2.index(), trigCand.hfcRedCorrCollId(), assTrk.hfcRedCorrCollId());

double deltaPhi = RecoDecay::constrainAngle(assTrk.phiAssoc() - trigCand.phiTrig(), -o2::constants::math::PIHalf);
double deltaEta = assTrk.etaAssoc() - trigCand.etaTrig();
Expand Down Expand Up @@ -476,7 +522,7 @@ struct HfCorrelatorFlowCharmHadronsReduced {
fillSameEvent<false, true>(pair, trigCand, binPolicyPosZMult);
}
}
PROCESS_SWITCH(HfCorrelatorFlowCharmHadronsReduced, processSameEventCharmHadWMultMix, "Process Same Event for Charm-Had with multiplicity pools", true);
PROCESS_SWITCH(HfCorrelatorFlowCharmHadronsReduced, processSameEventCharmHadWMultMix, "Process Same Event for Charm-Had with multiplicity pools", false);

void processSameEventHadHadWMultMix(SameEvtPairsHadHad::iterator const& pair,
aod::HfcRedTrigTracks const&,
Expand Down Expand Up @@ -595,15 +641,27 @@ struct HfCorrelatorFlowCharmHadronsReduced {
void processCharmTriggers(aod::HfcRedTrigCharms const& trigCands,
aod::HfcRedCorrColls const&)
{

std::unordered_map<int, int> nCharmCandidatesPerCollision;
for (const auto& cand : trigCands) {
auto collision = cand.template hfcRedCorrColl_as<o2::aod::HfcRedCorrColls>();
++nCharmCandidatesPerCollision[collision.globalIndex()];
}

for (const auto& trigCand : trigCands) {
auto collision = trigCand.template hfcRedCorrColl_as<o2::aod::HfcRedCorrColls>();
if (collision.centrality() < centralityMin || collision.centrality() > centralityMax) {
if (!isSelCollision(collision)) {
continue;
}
if (!isSelBdtScoreCut(trigCand, trigCand.ptTrig())) {
continue;
}
registry.fill(HIST("hSparseTrigCandsCharm"), trigCand.invMassTrig(), trigCand.ptTrig(), trigCand.bdtScore0Trig(), trigCand.bdtScore1Trig());
const auto nCharmCandidates = nCharmCandidatesPerCollision[collision.globalIndex()];
if (countCharmCandsPerCollision) {
registry.fill(HIST("hSparseTrigCandsCharm"), trigCand.invMassTrig(), trigCand.ptTrig(), trigCand.bdtScore0Trig(), trigCand.bdtScore1Trig(), nCharmCandidates);
} else {
registry.fill(HIST("hSparseTrigCandsCharm"), trigCand.invMassTrig(), trigCand.ptTrig(), trigCand.bdtScore0Trig(), trigCand.bdtScore1Trig());
}
}
}
PROCESS_SWITCH(HfCorrelatorFlowCharmHadronsReduced, processCharmTriggers, "Process charm trigger info", false);
Expand All @@ -615,7 +673,7 @@ struct HfCorrelatorFlowCharmHadronsReduced {
BinningCentPosZ binPolicyPosZCent{{zPoolBins, centPoolBins}};

for (const auto& collision : collisions) {
if (collision.centrality() < centralityMin || collision.centrality() > centralityMax) {
if (!isSelCollision(collision)) {
continue;
}
int poolBin = binPolicyPosZCent.getBin({collision.posZ(), collision.centrality()});
Expand Down
Loading