Skip to content

[PWGHF] new processes in correlator flow #13781

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

Merged
stefanopolitano merged 4 commits into from
Nov 13, 2025
Merged

[PWGHF] new processes in correlator flow #13781

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
0e018fd
Added base processes for correlation studies
scattaru Nov 10, 2025
e6181a7
Fix histogram
scattaru Nov 11, 2025
e683a3f
Fix histogram
scattaru Nov 11, 2025
fbf9f35
Update correlatorDsHadrons.cxx
scattaru Nov 12, 2025
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
18 changes: 6 additions & 12 deletions PWGHF/HFC/TableProducer/correlatorDsHadrons.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -280,12 +280,9 @@ struct HfCorrelatorDsHadrons {
registry.add("hDsPoolBin", "Ds candidates pool bin", {HistType::kTH1F, {axisPoolBin}});
registry.add("hTracksPoolBin", "Particles associated pool bin", {HistType::kTH1F, {axisPoolBin}});
if (pidTrkApplied) {
registry.add("hTpcNSigmaPIDpion", "n sigma tpc for pion hypothesis", {HistType::kTH2F, {{axisPid}, {axisPtHadron}}});
registry.add("hTpcNSigmaPIDkaon", "n sigma tpc for kaon hypothesis", {HistType::kTH2F, {{axisPid}, {axisPtHadron}}});
registry.add("hTpcNSigmaPIDproton", "n sigma tpc for proton hypothesis", {HistType::kTH2F, {{axisPid}, {axisPtHadron}}});
registry.add("hTofNSigmaPIDpion", "n sigma tof for pion hypothesis", {HistType::kTH2F, {{axisPid}, {axisPtHadron}}});
registry.add("hTofNSigmaPIDkaon", "n sigma tof for kaon hypothesis", {HistType::kTH2F, {{axisPid}, {axisPtHadron}}});
registry.add("hTofNSigmaPIDproton", "n sigma tof for proton hypothesis", {HistType::kTH2F, {{axisPid}, {axisPtHadron}}});
registry.add("hTpcTofNSigmaPidPion", "n sigma tpc and tof for pion hypothesis", {HistType::kTH3F, {{axisPid}, {axisPid}, {axisPtHadron}}});
registry.add("hTpcTofNSigmaPidKaon", "n sigma tpc and tof for kaon hypothesis", {HistType::kTH3F, {{axisPid}, {axisPid}, {axisPtHadron}}});
registry.add("hTpcTofNSigmaPidProton", "n sigma tpc and tof for proton hypothesis", {HistType::kTH3F, {{axisPid}, {axisPid}, {axisPtHadron}}});
}
}
// Histograms for MC Reco analysis
Expand Down Expand Up @@ -840,12 +837,9 @@ struct HfCorrelatorDsHadrons {
if (!passPIDSelection(track, trkPIDspecies, pidTPCMax, pidTOFMax, tofPIDThreshold, forceTOF)) {
continue;
}
registry.fill(HIST("hTpcNSigmaPIDpion"), track.tpcNSigmaPi(), track.pt());
registry.fill(HIST("hTpcNSigmaPIDkaon"), track.tpcNSigmaKa(), track.pt());
registry.fill(HIST("hTpcNSigmaPIDproton"), track.tpcNSigmaPr(), track.pt());
registry.fill(HIST("hTofNSigmaPIDpion"), track.tofNSigmaPi(), track.pt());
registry.fill(HIST("hTofNSigmaPIDkaon"), track.tofNSigmaKa(), track.pt());
registry.fill(HIST("hTofNSigmaPIDproton"), track.tofNSigmaPr(), track.pt());
registry.fill(HIST("hTpcTofNSigmaPidPion"), track.tpcNSigmaPi(), track.tofNSigmaPi(), track.pt());
registry.fill(HIST("hTpcTofNSigmaPidKaon"), track.tpcNSigmaKa(), track.tofNSigmaKa(), track.pt());
registry.fill(HIST("hTpcTofNSigmaPidProton"), track.tpcNSigmaPr(), track.tofNSigmaPr(), track.pt());
}
assocTrackReduced(indexHfcReducedCollision, track.globalIndex(), track.phi(), track.eta(), track.pt() * track.sign());
assocTrackSelInfo(indexHfcReducedCollision, track.tpcNClsCrossedRows(), track.itsClusterMap(), track.itsNCls(), track.dcaXY(), track.dcaZ());
Expand Down
122 changes: 120 additions & 2 deletions PWGHF/HFC/TableProducer/correlatorFlowCharmHadronsReduced.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -199,9 +199,9 @@ struct HfCorrelatorFlowCharmHadronsReduced {
} else {
axes.insert(axes.end(), {axisInvMass});
// axes.insert(axes.end(), {axisInvMass, axisMlOne, axisMlTwo});
if (doprocessSameEventCharmHadWCentMix || doprocessSameEventCharmHadWMultMix) {
if (doprocessSameEventCharmHadWCentMix || doprocessSameEventCharmHadWMultMix || doprocessSameEventCharmHadWCentMixBase) {
registry.add("hSparseCorrelationsSECharmHad", "THn for SE Charm-Had correlations", HistType::kTHnSparseF, axes);
} else if (doprocessMixedEventCharmHadWCentMix || doprocessMixedEventCharmHadWMultMix) {
} else if (doprocessMixedEventCharmHadWCentMix || doprocessMixedEventCharmHadWMultMix || doprocessMixedEventCharmHadWCentMixBase) {
registry.add("hSparseCorrelationsMECharmHad", "THn for ME Charm-Had correlations", HistType::kTHnSparseF, axes);
}
if (doprocessCharmTriggers) {
Expand Down Expand Up @@ -375,6 +375,93 @@ struct HfCorrelatorFlowCharmHadronsReduced {
}
}

/// Correlations for Same Event pairs
/// \param poolBin collision pool bin based on multiplicity and z-vertex position
/// \param trigCandsThisColl are the selected trigger candidates in the collision
/// \param assocTracksThisColl are the selected associated tracks in the collision
template <bool FillSparses, typename TTrigCand, typename TTrackAssoc>
void doCorrelationsSameEvent(int poolBin,
const TTrigCand& trigCandsThisColl,
const TTrackAssoc& assocTracksThisColl)
{
for (const auto& trigCand : trigCandsThisColl) {
double const ptTrig = trigCand.ptTrig();
if constexpr (requires { trigCand.bdtScore0Trig(); }) { // ML selection on bkg score for Charm-Had case
if (!isSelBdtScoreCut(trigCand, ptTrig)) {
continue;
}
}

for (const auto& assTrk : assocTracksThisColl) {
// TODO: Remove Ds daughters
/*if (assTrk.originTrackId() == candidate.prong0Id() ||
assTrk.originTrackId() == candidate.prong1Id() ||
assTrk.originTrackId() == candidate.prong2Id()) {
continue;
}*/
// TODO: DCA cut
double deltaPhi = RecoDecay::constrainAngle(assTrk.phiAssoc() - trigCand.phiTrig(), -o2::constants::math::PIHalf);
double deltaEta = assTrk.etaAssoc() - trigCand.etaTrig();
if constexpr (FillSparses) {
if constexpr (requires { trigCand.bdtScore0Trig(); }) { // Separate Charm-Had and Had-Had cases
registry.fill(HIST("hSparseCorrelationsSECharmHad"), poolBin, ptTrig, assTrk.ptAssoc(), deltaEta,
deltaPhi, trigCand.invMassTrig());
} else {
registry.fill(HIST("hSparseCorrelationsSEHadHad"), poolBin, ptTrig, assTrk.ptAssoc(), deltaEta, deltaPhi);
}
}
}
}
}

/// Correlations for Mixed Event pairs
/// \param collisions are the selected collisions
/// \param trigCands are the trigger candidates
/// \param assocTracks are the associated tracks
/// \param binPolicy is the binning policy for the correlation
template <bool FillSparses, typename TCollision, typename TTrigCand, typename TTrackAssoc, typename TBinningType>
void doCorrelationsMixedEvent(const TCollision& collisions,
const TTrigCand& trigCands,
const TTrackAssoc& assocTracks,
TBinningType binPolicy)
{
auto tracksTuple = std::make_tuple(trigCands, assocTracks);

Pair<TCollision, TTrigCand, TTrackAssoc, TBinningType> pairData{binPolicy, numberEventsMixed, -1, collisions, tracksTuple, &cache};

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

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

if (poolBin != poolBinTrigCand) {
LOGF(info, "Error, poolBins are different");
continue;
}

for (const auto& [trigCand, assTrk] : o2::soa::combinations(o2::soa::CombinationsFullIndexPolicy(tracks1, tracks2))) {
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();
if constexpr (FillSparses) {
if constexpr (requires { trigCand.bdtScore0Trig(); }) { // Separate Charm-Had and Had-Had cases
registry.fill(HIST("hSparseCorrelationsMECharmHad"), poolBin, trigCand.ptTrig(), assTrk.ptAssoc(), deltaEta,
deltaPhi, trigCand.invMassTrig());
} else {
registry.fill(HIST("hSparseCorrelationsMEHadHad"), poolBin, trigCand.ptTrig(), assTrk.ptAssoc(), deltaEta, deltaPhi);
}
}
}
}
}

void processSameEventCharmHadWMultMix(SameEvtPairsChHad::iterator const& pair,
aod::HfcRedTrigCharms const&,
aod::HfcRedCorrColls const&)
Expand Down Expand Up @@ -520,6 +607,37 @@ struct HfCorrelatorFlowCharmHadronsReduced {
}
}
PROCESS_SWITCH(HfCorrelatorFlowCharmHadronsReduced, processCharmTriggers, "Process charm trigger info", false);

void processSameEventCharmHadWCentMixBase(aod::HfcRedCorrColls const& collisions,
TrigCharmCands const& candidates,
AssocTracks const& tracks)
{
BinningCentPosZ binPolicyPosZCent{{zPoolBins, centPoolBins}, true};

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

auto thisCollId = collision.globalIndex();
auto candsThisColl = candidates.sliceBy(trigCharmCandsPerCol, thisCollId);
auto tracksThisColl = tracks.sliceBy(assocTracksPerCol, thisCollId);

doCorrelationsSameEvent<true>(poolBin, candsThisColl, tracksThisColl);
}
}
PROCESS_SWITCH(HfCorrelatorFlowCharmHadronsReduced, processSameEventCharmHadWCentMixBase, "Process Same Event base", false);

void processMixedEventCharmHadWCentMixBase(aod::HfcRedCorrColls const& collisions,
TrigCharmCands const& candidates,
AssocTracks const& tracks)
{
BinningCentPosZ binPolicyPosZCent{{zPoolBins, centPoolBins}, true};

doCorrelationsMixedEvent<true>(collisions, candidates, tracks, binPolicyPosZCent);
}
PROCESS_SWITCH(HfCorrelatorFlowCharmHadronsReduced, processMixedEventCharmHadWCentMixBase, "Process Mixed Event base", false);
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
Expand Down
Loading