Skip to content

[PWGHF] Add process function for MC D+V0 femtoscopy #16924

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
fgrosa merged 4 commits into from
Jul 5, 2026
Merged

[PWGHF] Add process function for MC D+V0 femtoscopy #16924

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
393236f
Add process function for MC D+V0 femtoscopy
fgrosa Jul 5, 2026
deb52a8
Please consider the following formatting changes
alibuild Jul 5, 2026
1a9e135
Merge pull request #155 from alibuild/alibot-cleanup-16924
fgrosa Jul 5, 2026
4201c4e
Fix details
fgrosa Jul 5, 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
121 changes: 98 additions & 23 deletions PWGHF/HFC/Tasks/taskCharmHadronsV0FemtoDream.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -174,10 +174,10 @@ struct HfTaskCharmHadronsV0FemtoDream {
using FilteredCollisions = soa::Filtered<soa::Join<FDCollisions, FDColMasks>>;
using FilteredCollision = FilteredCollisions::iterator;

using FilteredMcColisions = soa::Filtered<soa::Join<aod::FDCollisions, FDColMasks, aod::FDMCCollLabels>>;
using FilteredMcColision = FilteredMcColisions::iterator;
using FilteredMcCollisions = soa::Filtered<soa::Join<aod::FDCollisions, FDColMasks, aod::FDMCCollLabels>>;
using FilteredMcColision = FilteredMcCollisions::iterator;

using FilteredFDMcParts = soa::Filtered<soa::Join<aod::FDParticles, aod::FDParticlesIndex, aod::FDExtParticles, aod::FDMCLabels, aod::FDExtMCLabels>>;
using FilteredFDMcParts = soa::Filtered<soa::Join<aod::FDParticles, aod::FDParticlesIndex, aod::FDExtParticles, aod::FDMCLabels, aod::FDExtMCLabels, aod::FDTrkTimeStamp>>;
using FilteredFDMcPart = FilteredFDMcParts::iterator;

using FDV0Particles = soa::Join<aod::FDParticles, aod::FDExtParticles, aod::FDParticlesIndex, aod::FDTrkTimeStamp>;
Expand All @@ -189,11 +189,12 @@ struct HfTaskCharmHadronsV0FemtoDream {
Filter hfMcSelFilter = (nabs(aod::fdhf::flagMc) == charmSel.charmHadMcSel);

Preslice<FDV0Particles> perCol = aod::femtodreamparticle::fdCollisionId;
Preslice<FilteredFDMcParts> perColMc = aod::femtodreamparticle::fdCollisionId;
Preslice<FilteredCharmCand3Prongs> perHf3ProngByCol = aod::femtodreamparticle::fdCollisionId;
Preslice<FilteredCharmCand2Prongs> perHf2ProngByCol = aod::femtodreamparticle::fdCollisionId;
Preslice<FilteredCharmCandDstars> perHfDstarByCol = aod::femtodreamparticle::fdCollisionId;

/// Partition for particle Lambda
/// Partitions for particle Lambda
Partition<FDV0Particles> partitionLambda = (aod::femtodreamparticle::partType == uint8_t(aod::femtodreamparticle::ParticleType::kV0)) &&
((aod::femtodreamparticle::cut & v0Sel.cutBit) == v0Sel.cutBit) &&
(aod::femtodreamparticle::pt > v0Sel.ptV0Min) &&
Expand All @@ -205,7 +206,18 @@ struct HfTaskCharmHadronsV0FemtoDream {
(aod::femtodreamparticle::mAntiLambda > v0Sel.invMassAntiV0Min) &&
(aod::femtodreamparticle::mAntiLambda < v0Sel.invMassAntiV0Max);

/// Partition for particle K0Short
Partition<FilteredFDMcParts> partitionMcLambda = (aod::femtodreamparticle::partType == uint8_t(aod::femtodreamparticle::ParticleType::kV0)) &&
((aod::femtodreamparticle::cut & v0Sel.cutBit) == v0Sel.cutBit) &&
(aod::femtodreamparticle::pt > v0Sel.ptV0Min) &&
(aod::femtodreamparticle::pt < v0Sel.ptV0Max) &&
(aod::femtodreamparticle::eta > v0Sel.etaV0Min) &&
(aod::femtodreamparticle::eta < v0Sel.etaV0Max) &&
(aod::femtodreamparticle::mLambda > v0Sel.invMassV0Min) &&
(aod::femtodreamparticle::mLambda < v0Sel.invMassV0Max) &&
(aod::femtodreamparticle::mAntiLambda > v0Sel.invMassAntiV0Min) &&
(aod::femtodreamparticle::mAntiLambda < v0Sel.invMassAntiV0Max);

/// Partitions for particle K0Short
Partition<FDV0Particles> partitionK0Short = (aod::femtodreamparticle::partType == uint8_t(aod::femtodreamparticle::ParticleType::kV0K0Short)) &&
((aod::femtodreamparticle::cut & v0Sel.cutBit) == v0Sel.cutBit) &&
(aod::femtodreamparticle::pt > v0Sel.ptV0Min) &&
Expand All @@ -217,6 +229,17 @@ struct HfTaskCharmHadronsV0FemtoDream {
(aod::femtodreamparticle::mAntiLambda > v0Sel.invMassAntiV0Min) &&
(aod::femtodreamparticle::mAntiLambda < v0Sel.invMassAntiV0Max);

Partition<FilteredFDMcParts> partitionMcK0Short = (aod::femtodreamparticle::partType == uint8_t(aod::femtodreamparticle::ParticleType::kV0K0Short)) &&
((aod::femtodreamparticle::cut & v0Sel.cutBit) == v0Sel.cutBit) &&
(aod::femtodreamparticle::pt > v0Sel.ptV0Min) &&
(aod::femtodreamparticle::pt < v0Sel.ptV0Max) &&
(aod::femtodreamparticle::eta > v0Sel.etaV0Min) &&
(aod::femtodreamparticle::eta < v0Sel.etaV0Max) &&
(aod::femtodreamparticle::mLambda > v0Sel.invMassV0Min) &&
(aod::femtodreamparticle::mLambda < v0Sel.invMassV0Max) &&
(aod::femtodreamparticle::mAntiLambda > v0Sel.invMassAntiV0Min) &&
(aod::femtodreamparticle::mAntiLambda < v0Sel.invMassAntiV0Max);

/// Partition for particle 2
Partition<FilteredCharmCand3Prongs> partitionCharmHadron3Prong = aod::fdhf::bdtBkg < charmSel.charmHadBkgBDTmax && aod::fdhf::bdtFD < charmSel.charmHadFdBDTmax && aod::fdhf::bdtFD > charmSel.charmHadFdBDTmin&& aod::fdhf::bdtPrompt<charmSel.charmHadPromptBDTmax && aod::fdhf::bdtPrompt> charmSel.charmHadPromptBDTmin;
Partition<FilteredCharmCand2Prongs> partitionCharmHadron2Prong = aod::fdhf::bdtBkg < charmSel.charmHadBkgBDTmax && aod::fdhf::bdtFD < charmSel.charmHadFdBDTmax && aod::fdhf::bdtFD > charmSel.charmHadFdBDTmin&& aod::fdhf::bdtPrompt<charmSel.charmHadPromptBDTmax && aod::fdhf::bdtPrompt> charmSel.charmHadPromptBDTmin;
Expand Down Expand Up @@ -381,10 +404,9 @@ struct HfTaskCharmHadronsV0FemtoDream {
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 @@ -397,9 +419,8 @@ struct HfTaskCharmHadronsV0FemtoDream {
}
if (ReturnDaughMass) {
return mD0;
} else {
return mDstar - mD0;
}
return mDstar - mD0;
}
// Add more channels as needed
return 0.f;
Expand Down Expand Up @@ -502,7 +523,7 @@ struct HfTaskCharmHadronsV0FemtoDream {
}

template <bool IsMc, DecayChannel Channel, typename CollisionType, typename PartitionType1, typename PartitionType2, typename FDParticles, typename BinningType>
void doMixedEvent(CollisionType const& cols, PartitionType1& charms, PartitionType2& v0s, FDParticles const& /*femtoParts*/, BinningType policy)
void doMixedEvent(CollisionType const& cols, PartitionType1& charms, PartitionType2& v0s, FDParticles const& /*femtoParts*/, BinningType const& policy)
{
processType = 2; // for mixed event
// Mixed events that contain the pair of interest
Expand Down Expand Up @@ -717,9 +738,9 @@ struct HfTaskCharmHadronsV0FemtoDream {
}
}

template <DecayChannel Channel, typename CharmPart>
void runMixing(FilteredCollisions const& cols,
FDV0Particles const& parts,
template <bool IsMc, DecayChannel Channel, typename Cols, typename V0Part, typename CharmPart>
void runMixing(Cols const& cols,
V0Part const& parts,
CharmPart& charmPart)
{
if (!mixSetting.doMixEvent) {
Expand All @@ -728,23 +749,43 @@ struct HfTaskCharmHadronsV0FemtoDream {
auto run = [&](auto& v0Part) {
switch (mixSetting.mixingBinPolicy) {
case femtodreamcollision::kMult:
doMixedEvent<false, Channel, FilteredCollisions>(cols, charmPart, v0Part, parts, colBinningMult);
if constexpr (IsMc) {
doMixedEvent<IsMc, Channel, FilteredMcCollisions>(cols, charmPart, v0Part, parts, colBinningMult);
} else {
doMixedEvent<IsMc, Channel, FilteredCollisions>(cols, charmPart, v0Part, parts, colBinningMult);
}
break;
case femtodreamcollision::kMultPercentile:
doMixedEvent<false, Channel, FilteredCollisions>(cols, charmPart, v0Part, parts, colBinningMultPercentile);
if constexpr (IsMc) {
doMixedEvent<IsMc, Channel, FilteredMcCollisions>(cols, charmPart, v0Part, parts, colBinningMultPercentile);
} else {
doMixedEvent<IsMc, Channel, FilteredCollisions>(cols, charmPart, v0Part, parts, colBinningMultPercentile);
}
break;
case femtodreamcollision::kMultMultPercentile:
doMixedEvent<false, Channel, FilteredCollisions>(cols, charmPart, v0Part, parts, colBinningMultMultPercentile);
if constexpr (IsMc) {
doMixedEvent<IsMc, Channel, FilteredMcCollisions>(cols, charmPart, v0Part, parts, colBinningMultMultPercentile);
} else {
doMixedEvent<IsMc, Channel, FilteredCollisions>(cols, charmPart, v0Part, parts, colBinningMultMultPercentile);
}
break;
default:
LOG(fatal) << "Invalid binning policiy specifed. Breaking...";
}
};

if (v0Sel.pdgCodeV0 == kLambda0) {
run(partitionLambda);
if constexpr (IsMc) {
run(partitionMcLambda);
} else {
run(partitionLambda);
}
} else if (v0Sel.pdgCodeV0 == kK0Short) {
run(partitionK0Short);
if constexpr (IsMc) {
run(partitionMcK0Short);
} else {
run(partitionK0Short);
}
} else {
LOG(fatal) << "Unsupported V0 PDG: " << v0Sel.pdgCodeV0 << " (allowed: 3122, 310) for mixed-events";
}
Expand Down Expand Up @@ -781,7 +822,7 @@ struct HfTaskCharmHadronsV0FemtoDream {
LOG(fatal) << "Unsupported V0 PDG: " << v0Sel.pdgCodeV0 << " (allowed: 3122, 310) for same events";
}
}
runMixing<DecayChannel::LcToPKPi>(cols, parts, partitionCharmHadron3Prong);
runMixing<false, DecayChannel::LcToPKPi>(cols, parts, partitionCharmHadron3Prong);
}
PROCESS_SWITCH(HfTaskCharmHadronsV0FemtoDream, processDataLcV0, "Enable processing LcToPKPi and V0 correlation", false);

Expand Down Expand Up @@ -815,7 +856,7 @@ struct HfTaskCharmHadronsV0FemtoDream {
LOG(fatal) << "Unsupported V0 PDG: " << v0Sel.pdgCodeV0 << " (allowed: 3122, 310)";
}
}
runMixing<DecayChannel::DplusToPiKPi>(cols, parts, partitionCharmHadron3Prong);
runMixing<false, DecayChannel::DplusToPiKPi>(cols, parts, partitionCharmHadron3Prong);
}
PROCESS_SWITCH(HfTaskCharmHadronsV0FemtoDream, processDataDplusV0, "Enable processing DplusToPiKPi and V0 correlation", false);

Expand Down Expand Up @@ -849,7 +890,7 @@ struct HfTaskCharmHadronsV0FemtoDream {
LOG(fatal) << "Unsupported V0 PDG: " << v0Sel.pdgCodeV0 << " (allowed: 3122, 310)";
}
}
runMixing<DecayChannel::D0ToPiK>(cols, parts, partitionCharmHadron2Prong);
runMixing<false, DecayChannel::D0ToPiK>(cols, parts, partitionCharmHadron2Prong);
}
PROCESS_SWITCH(HfTaskCharmHadronsV0FemtoDream, processDataD0V0, "Enable processing D0ToPiK and V0 correlation", false);

Expand Down Expand Up @@ -884,9 +925,43 @@ struct HfTaskCharmHadronsV0FemtoDream {
LOG(fatal) << "Unsupported V0 PDG: " << v0Sel.pdgCodeV0 << " (allowed: 3122, 310)";
}
}
runMixing<DecayChannel::DstarToD0Pi>(cols, parts, partitionCharmHadronDstar);
runMixing<false, DecayChannel::DstarToD0Pi>(cols, parts, partitionCharmHadronDstar);
}
PROCESS_SWITCH(HfTaskCharmHadronsV0FemtoDream, processDataDstarV0, "Enable processing DstarToD0Pi and V0 correlation", false);

void processMcDplusV0(FilteredMcCollisions const& cols,
FilteredFDMcParts const& parts,
o2::aod::FDMCParticles const&,
o2::aod::FDExtMCParticles const&,
FilteredCharmMcCand3Prongs const&)
{
for (const auto& col : cols) {
eventHisto.fillQA(col);
auto sliceMcCharmHad = partitionMcCharmHadron3Prong->sliceByCached(aod::femtodreamparticle::fdCollisionId, col.globalIndex(), cache);
if (fillTableWithCharm.value && sliceMcCharmHad.size() == 0) {
continue;
}
if (v0Sel.pdgCodeV0 == kLambda0) {
auto sliceMcV0 = partitionMcLambda->sliceByCached(aod::femtodreamparticle::fdCollisionId, col.globalIndex(), cache);
fillTables<true, DecayChannel::DplusToPiKPi>(col, sliceMcV0, sliceMcCharmHad, parts);
if (sliceMcCharmHad.size() > 0 && sliceMcV0.size() > 0) {
doSameEvent<true, DecayChannel::DplusToPiKPi, FilteredCharmMcCand3Prongs>(sliceMcCharmHad, sliceMcV0, parts, col);
}

} else if (v0Sel.pdgCodeV0 == kK0Short) {
auto sliceMcV0 = partitionMcK0Short->sliceByCached(aod::femtodreamparticle::fdCollisionId, col.globalIndex(), cache);
fillTables<true, DecayChannel::DplusToPiKPi>(col, sliceMcV0, sliceMcCharmHad, parts);

if (sliceMcCharmHad.size() > 0 && sliceMcV0.size() > 0) {
doSameEvent<true, DecayChannel::DplusToPiKPi, FilteredCharmMcCand3Prongs>(sliceMcCharmHad, sliceMcV0, parts, col);
}
} else {
LOG(fatal) << "Unsupported V0 PDG: " << v0Sel.pdgCodeV0 << " (allowed: 3122, 310)";
}
}
runMixing<true, DecayChannel::DplusToPiKPi>(cols, parts, partitionMcCharmHadron3Prong);
}
PROCESS_SWITCH(HfTaskCharmHadronsV0FemtoDream, processMcDplusV0, "Enable processing DplusToPiKPi and V0 correlation for Monte Carlo", false);
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
Expand Down
Loading