Skip to content

[PWGHF] Refactor mixed event handling for track selection in Charm Track femto task #17191

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
zhangbiao-phy merged 2 commits into from
Jul 24, 2026
+109 −76
Merged

[PWGHF] Refactor mixed event handling for track selection in Charm Track femto task #17191

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
b6de1a3
Refactor mixed event handling for track selection
zhangbiao-phy Jul 24, 2026
55c4702
Fix the errors
zhangbiao-phy Jul 24, 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
185 changes: 109 additions & 76 deletions PWGHF/HFC/Tasks/taskCharmHadronsTrackFemtoDream.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -382,10 +382,9 @@ struct HfTaskCharmHadronsTrackFemtoDream {
if (cand.candidateSelFlag() == 1) {
invMass = cand.m(std::array{MassPiPlus, MassKPlus});
return invMass;
} else {
invMass = cand.m(std::array{MassKPlus, MassPiPlus});
return invMass;
}
invMass = cand.m(std::array{MassKPlus, MassPiPlus});
return invMass;
} else if constexpr (Channel == DecayChannel::DstarToD0Pi) { // D* → D0π (PDG: 413)
float mDstar = 0.f;
float mD0 = 0.f;
Expand All @@ -398,9 +397,8 @@ struct HfTaskCharmHadronsTrackFemtoDream {
}
if (ReturnDaughMass) {
return mD0;
} else {
return mDstar - mD0;
}
return mDstar - mD0;
} else if constexpr (Channel == DecayChannel::XicToXiPiPi) {
invMass = cand.m(std::array{MassXiMinus, MassPiPlus, MassPiPlus});
return invMass;
Expand Down Expand Up @@ -621,7 +619,7 @@ struct HfTaskCharmHadronsTrackFemtoDream {
}

template <bool IsMc, DecayChannel Channel, typename CollisionType, typename PartitionType1, typename PartitionType2, typename TableTracks, typename BinningType>
void doMixedEvent(CollisionType const& cols, PartitionType1& charms, PartitionType2& trks, TableTracks const& parts, BinningType policy)
void doMixedEvent(CollisionType const& cols, PartitionType1& charms, PartitionType2& trks, TableTracks const& parts, BinningType const& policy)
{
processType = 2; // for mixed event
// Mixed events that contain the pair of interest
Expand Down Expand Up @@ -912,23 +910,26 @@ struct HfTaskCharmHadronsTrackFemtoDream {
auto sliceCharmHad = partitionCharmHadron3Prong->sliceByCached(aod::femtodreamparticle::fdCollisionId, col.globalIndex(), cache);
if (fillTableWithCharm.value && sliceCharmHad.size() == 0) {
continue;
} else {
fillTables<false, DecayChannel::LcToPKPi>(col, sliceTrk1, sliceCharmHad);
}
fillTables<false, DecayChannel::LcToPKPi>(col, sliceTrk1, sliceCharmHad);
if (sliceCharmHad.size() > 0 && sliceTrk1.size() > 0) {
doSameEvent<false, DecayChannel::LcToPKPi, FilteredCharmCand3Prongs>(sliceCharmHad, sliceTrk1, parts, col);
}
}
if (mixSetting.doMixEvent) {
auto* partitionTrk1Selected = &partitionTrk1;
if (trackSel.pdgCodeTrack1.value == kKPlus) {
partitionTrk1Selected = &partitionTrk1Ka;
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<false, DecayChannel::LcToPKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, partitionTrk1, parts, colBinningMult);
doMixedEvent<false, DecayChannel::LcToPKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<false, DecayChannel::LcToPKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, partitionTrk1, parts, colBinningMultPercentile);
doMixedEvent<false, DecayChannel::LcToPKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<false, DecayChannel::LcToPKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, partitionTrk1, parts, colBinningMultMultPercentile);
doMixedEvent<false, DecayChannel::LcToPKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
Expand All @@ -954,23 +955,26 @@ struct HfTaskCharmHadronsTrackFemtoDream {

if (fillTableWithCharm.value && sliceCharmHad.size() == 0) {
continue;
} else {
fillTables<false, DecayChannel::DplusToPiKPi>(col, sliceTrk1, sliceCharmHad);
}
fillTables<false, DecayChannel::DplusToPiKPi>(col, sliceTrk1, sliceCharmHad);
if (sliceCharmHad.size() > 0 && sliceTrk1.size() > 0) {
doSameEvent<false, DecayChannel::DplusToPiKPi, FilteredCharmCand3Prongs>(sliceCharmHad, sliceTrk1, parts, col);
}
}
if (mixSetting.doMixEvent) {
auto* partitionTrk1Selected = &partitionTrk1;
if (trackSel.pdgCodeTrack1.value == kKPlus) {
partitionTrk1Selected = &partitionTrk1Ka;
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<false, DecayChannel::DplusToPiKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, partitionTrk1, parts, colBinningMult);
doMixedEvent<false, DecayChannel::DplusToPiKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<false, DecayChannel::DplusToPiKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, partitionTrk1, parts, colBinningMultPercentile);
doMixedEvent<false, DecayChannel::DplusToPiKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<false, DecayChannel::DplusToPiKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, partitionTrk1, parts, colBinningMultMultPercentile);
doMixedEvent<false, DecayChannel::DplusToPiKPi, FilteredCollisions>(cols, partitionCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
Expand All @@ -995,23 +999,26 @@ struct HfTaskCharmHadronsTrackFemtoDream {
auto sliceCharmHad = partitionCharmHadron2Prong->sliceByCached(aod::femtodreamparticle::fdCollisionId, col.globalIndex(), cache);
if (fillTableWithCharm.value && sliceCharmHad.size() == 0) {
continue;
} else {
fillTables<false, DecayChannel::D0ToPiK>(col, sliceTrk1, sliceCharmHad);
}
fillTables<false, DecayChannel::D0ToPiK>(col, sliceTrk1, sliceCharmHad);
if (sliceCharmHad.size() > 0 && sliceTrk1.size() > 0) {
doSameEvent<false, DecayChannel::D0ToPiK, FilteredCharmCand2Prongs>(sliceCharmHad, sliceTrk1, parts, col);
}
}
if (mixSetting.doMixEvent) {
auto* partitionTrk1Selected = &partitionTrk1;
if (trackSel.pdgCodeTrack1.value == kKPlus) {
partitionTrk1Selected = &partitionTrk1Ka;
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<false, DecayChannel::D0ToPiK, FilteredCollisions>(cols, partitionCharmHadron2Prong, partitionTrk1, parts, colBinningMult);
doMixedEvent<false, DecayChannel::D0ToPiK, FilteredCollisions>(cols, partitionCharmHadron2Prong, *partitionTrk1Selected, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<false, DecayChannel::D0ToPiK, FilteredCollisions>(cols, partitionCharmHadron2Prong, partitionTrk1, parts, colBinningMultPercentile);
doMixedEvent<false, DecayChannel::D0ToPiK, FilteredCollisions>(cols, partitionCharmHadron2Prong, *partitionTrk1Selected, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<false, DecayChannel::D0ToPiK, FilteredCollisions>(cols, partitionCharmHadron2Prong, partitionTrk1, parts, colBinningMultMultPercentile);
doMixedEvent<false, DecayChannel::D0ToPiK, FilteredCollisions>(cols, partitionCharmHadron2Prong, *partitionTrk1Selected, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
Expand All @@ -1036,23 +1043,26 @@ struct HfTaskCharmHadronsTrackFemtoDream {
auto sliceCharmHad = partitionCharmHadronDstar->sliceByCached(aod::femtodreamparticle::fdCollisionId, col.globalIndex(), cache);
if (fillTableWithCharm.value && sliceCharmHad.size() == 0) {
continue;
} else {
fillTables<false, DecayChannel::DstarToD0Pi>(col, sliceTrk1, sliceCharmHad);
}
fillTables<false, DecayChannel::DstarToD0Pi>(col, sliceTrk1, sliceCharmHad);
if (sliceCharmHad.size() > 0 && sliceTrk1.size() > 0) {
doSameEvent<false, DecayChannel::DstarToD0Pi, FilteredCharmCandDstars>(sliceCharmHad, sliceTrk1, parts, col);
}
}
if (mixSetting.doMixEvent) {
auto* partitionTrk1Selected = &partitionTrk1;
if (trackSel.pdgCodeTrack1.value == kKPlus) {
partitionTrk1Selected = &partitionTrk1Ka;
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<false, DecayChannel::DstarToD0Pi, FilteredCollisions>(cols, partitionCharmHadronDstar, partitionTrk1, parts, colBinningMult);
doMixedEvent<false, DecayChannel::DstarToD0Pi, FilteredCollisions>(cols, partitionCharmHadronDstar, *partitionTrk1Selected, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<false, DecayChannel::DstarToD0Pi, FilteredCollisions>(cols, partitionCharmHadronDstar, partitionTrk1, parts, colBinningMultPercentile);
doMixedEvent<false, DecayChannel::DstarToD0Pi, FilteredCollisions>(cols, partitionCharmHadronDstar, *partitionTrk1Selected, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<false, DecayChannel::DstarToD0Pi, FilteredCollisions>(cols, partitionCharmHadronDstar, partitionTrk1, parts, colBinningMultMultPercentile);
doMixedEvent<false, DecayChannel::DstarToD0Pi, FilteredCollisions>(cols, partitionCharmHadronDstar, *partitionTrk1Selected, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
Expand All @@ -1075,9 +1085,8 @@ struct HfTaskCharmHadronsTrackFemtoDream {
auto sliceCharmHad = partitionCharmHadron3ProngXic->sliceByCached(aod::femtodreamparticle::fdCollisionId, col.globalIndex(), cache);
if (fillTableWithCharm.value && sliceCharmHad.size() == 0) {
continue;
} else {
fillTables<false, DecayChannel::XicToXiPiPi>(col, sliceTrk1, sliceCharmHad);
}
fillTables<false, DecayChannel::XicToXiPiPi>(col, sliceTrk1, sliceCharmHad);
if (sliceCharmHad.size() > 0 && sliceTrk1.size() > 0) {
doSameEvent<false, DecayChannel::XicToXiPiPi, FilteredCharmCand3ProngsXic>(sliceCharmHad, sliceTrk1, parts, col);
}
Expand Down Expand Up @@ -1123,18 +1132,24 @@ struct HfTaskCharmHadronsTrackFemtoDream {
}
doSameEvent<true, DecayChannel::LcToPKPi, FilteredCharmMcCand3Prongs>(sliceMcCharmHad, sliceMcTrk1, parts, col);
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<true, DecayChannel::LcToPKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, partitionMcTrk1, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<true, DecayChannel::LcToPKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, partitionMcTrk1, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<true, DecayChannel::LcToPKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, partitionMcTrk1, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
if (mixSetting.doMixEvent) {
auto* partitionTrk1Selected = &partitionMcTrk1;
if (trackSel.pdgCodeTrack1.value == kKPlus) {
partitionTrk1Selected = &partitionMcTrk1Ka;
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<true, DecayChannel::LcToPKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<true, DecayChannel::LcToPKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<true, DecayChannel::LcToPKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
}
}
}
PROCESS_SWITCH(HfTaskCharmHadronsTrackFemtoDream, processMcLcTrk, "Enable processing LcToPKPi and Tracks correlation for Monte Carlo", false);
Expand All @@ -1158,18 +1173,24 @@ struct HfTaskCharmHadronsTrackFemtoDream {
}
doSameEvent<true, DecayChannel::DplusToPiKPi, FilteredCharmMcCand3Prongs>(sliceMcCharmHad, sliceMcTrk1, parts, col);
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<true, DecayChannel::DplusToPiKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, partitionMcTrk1, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<true, DecayChannel::DplusToPiKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, partitionMcTrk1, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<true, DecayChannel::DplusToPiKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, partitionMcTrk1, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
if (mixSetting.doMixEvent) {
auto* partitionTrk1Selected = &partitionMcTrk1;
if (trackSel.pdgCodeTrack1.value == kKPlus) {
partitionTrk1Selected = &partitionMcTrk1Ka;
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<true, DecayChannel::DplusToPiKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<true, DecayChannel::DplusToPiKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<true, DecayChannel::DplusToPiKPi, FilteredMcColisions>(cols, partitionMcCharmHadron3Prong, *partitionTrk1Selected, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
}
}
}
PROCESS_SWITCH(HfTaskCharmHadronsTrackFemtoDream, processMcDplusTrk, "Enable processing DplusToPiKPi and Tracks correlation for Monte Carlo", false);
Expand All @@ -1193,18 +1214,24 @@ struct HfTaskCharmHadronsTrackFemtoDream {
}
doSameEvent<true, DecayChannel::D0ToPiK, FilteredCharmMcCand2Prongs>(sliceMcCharmHad, sliceMcTrk1, parts, col);
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<true, DecayChannel::D0ToPiK, FilteredMcColisions>(cols, partitionMcCharmHadron2Prong, partitionMcTrk1, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<true, DecayChannel::D0ToPiK, FilteredMcColisions>(cols, partitionMcCharmHadron2Prong, partitionMcTrk1, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<true, DecayChannel::D0ToPiK, FilteredMcColisions>(cols, partitionMcCharmHadron2Prong, partitionMcTrk1, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
if (mixSetting.doMixEvent) {
auto* partitionTrk1Selected = &partitionMcTrk1;
if (trackSel.pdgCodeTrack1.value == kKPlus) {
partitionTrk1Selected = &partitionMcTrk1Ka;
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<true, DecayChannel::D0ToPiK, FilteredMcColisions>(cols, partitionMcCharmHadron2Prong, *partitionTrk1Selected, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<true, DecayChannel::D0ToPiK, FilteredMcColisions>(cols, partitionMcCharmHadron2Prong, *partitionTrk1Selected, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<true, DecayChannel::D0ToPiK, FilteredMcColisions>(cols, partitionMcCharmHadron2Prong, *partitionTrk1Selected, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
}
}
}
PROCESS_SWITCH(HfTaskCharmHadronsTrackFemtoDream, processMcD0Trk, "Enable processing D0ToPiK and Tracks correlation for Monte Carlo", false);
Expand All @@ -1228,18 +1255,24 @@ struct HfTaskCharmHadronsTrackFemtoDream {
}
doSameEvent<true, DecayChannel::DstarToD0Pi, FilteredCharmMcCandDstars>(sliceMcCharmHad, sliceMcTrk1, parts, col);
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<true, DecayChannel::DstarToD0Pi, FilteredMcColisions>(cols, partitionMcCharmHadronDstar, partitionMcTrk1, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<true, DecayChannel::DstarToD0Pi, FilteredMcColisions>(cols, partitionMcCharmHadronDstar, partitionMcTrk1, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<true, DecayChannel::DstarToD0Pi, FilteredMcColisions>(cols, partitionMcCharmHadronDstar, partitionMcTrk1, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
if (mixSetting.doMixEvent) {
auto* partitionTrk1Selected = &partitionMcTrk1;
if (trackSel.pdgCodeTrack1.value == kKPlus) {
partitionTrk1Selected = &partitionMcTrk1Ka;
}
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<true, DecayChannel::DstarToD0Pi, FilteredMcColisions>(cols, partitionMcCharmHadronDstar, *partitionTrk1Selected, parts, colBinningMult);
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<true, DecayChannel::DstarToD0Pi, FilteredMcColisions>(cols, partitionMcCharmHadronDstar, *partitionTrk1Selected, parts, colBinningMultPercentile);
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<true, DecayChannel::DstarToD0Pi, FilteredMcColisions>(cols, partitionMcCharmHadronDstar, *partitionTrk1Selected, parts, colBinningMultMultPercentile);
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
}
}
}
PROCESS_SWITCH(HfTaskCharmHadronsTrackFemtoDream, processMcDstarTrk, "Enable processing DstarToD0Pi and Tracks correlation for Monte Carlo", false);
Expand Down
Loading