Skip to content

PWGHF: Constrain the mixed-event strategy and adjust the structure #7945

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 8 commits into from
Oct 12, 2024
Merged

PWGHF: Constrain the mixed-event strategy and adjust the structure #7945

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
650b326
Constrain the mixed-event strategy and adjust the structure
zhangbiao-phy Oct 10, 2024
900f0c6
fix comment
zhangbiao-phy Oct 10, 2024
1a53d2c
Require containing the pair of interest in the event for mixing
zhangbiao-phy Oct 10, 2024
fe3d7d3
Update taskCharmHadronsFemtoDream.cxx
zhangbiao-phy Oct 11, 2024
f60b81c
Update taskCharmHadronsFemtoDream.cxx
zhangbiao-phy Oct 11, 2024
46dbc6e
Update taskCharmHadronsFemtoDream.cxx
zhangbiao-phy Oct 11, 2024
6b6f773
Please consider the following formatting changes
alibuild Oct 11, 2024
494bed4
Merge pull request #6 from alibuild/alibot-cleanup-7945
zhangbiao-phy Oct 11, 2024
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
54 changes: 31 additions & 23 deletions PWGHF/HFC/Tasks/taskCharmHadronsFemtoDream.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -125,8 +125,8 @@ struct HfTaskCharmHadronsFemtoDream {
FemtoDreamContainer<femtoDreamContainer::EventType::same, femtoDreamContainer::Observable::kstar> sameEventCont;
FemtoDreamContainer<femtoDreamContainer::EventType::mixed, femtoDreamContainer::Observable::kstar> mixedEventCont;
FemtoDreamPairCleaner<aod::femtodreamparticle::ParticleType::kTrack, aod::femtodreamparticle::ParticleType::kCharmHadron> pairCleaner;
FemtoDreamDetaDphiStar<aod::femtodreamparticle::ParticleType::kTrack, aod::femtodreamparticle::ParticleType::kCharmHadron> pairCloseRejection;

FemtoDreamDetaDphiStar<aod::femtodreamparticle::ParticleType::kTrack, aod::femtodreamparticle::ParticleType::kCharmHadron> pairCloseRejectionSE;
FemtoDreamDetaDphiStar<aod::femtodreamparticle::ParticleType::kTrack, aod::femtodreamparticle::ParticleType::kCharmHadron> pairCloseRejectionME;
Filter eventMultiplicity = aod::femtodreamcollision::multNtr >= eventSel.multMin && aod::femtodreamcollision::multNtr <= eventSel.multMax;
Filter eventMultiplicityPercentile = aod::femtodreamcollision::multV0M >= eventSel.multPercentileMin && aod::femtodreamcollision::multV0M <= eventSel.multPercentileMax;
Filter hfCandSelFilter = aod::fdhf::candidateSelFlag >= charmHadCandSel.value;
Expand Down Expand Up @@ -205,7 +205,8 @@ struct HfTaskCharmHadronsFemtoDream {
mixedEventCont.setPDGCodes(pdgCodeTrack1, charmHadPDGCode);
pairCleaner.init(&registry);
if (useCPR.value) {
pairCloseRejection.init(&registry, &registry, cprDeltaPhiMax.value, cprDeltaEtaMax.value, cprPlotPerRadii.value);
pairCloseRejectionSE.init(&registry, &registry, cprDeltaPhiMax.value, cprDeltaEtaMax.value, cprPlotPerRadii.value, 1);
pairCloseRejectionME.init(&registry, &registry, cprDeltaPhiMax.value, cprDeltaEtaMax.value, cprPlotPerRadii.value, 2);
}
}

Expand All @@ -221,6 +222,17 @@ struct HfTaskCharmHadronsFemtoDream {
}

for (auto const& [p1, p2] : combinations(CombinationsFullIndexPolicy(sliceTrk1, sliceCharmHad))) {

if (useCPR.value) {
if (pairCloseRejectionSE.isClosePair(p1, p2, parts, col.magField())) {
continue;
}
}

if (!pairCleaner.isCleanPair(p1, p2, parts)) {
continue;
}

// proton track charge
float chargeTrack = 0.;
if ((p1.cut() & 2) == 2) {
Expand All @@ -247,16 +259,6 @@ struct HfTaskCharmHadronsFemtoDream {
// partSign = 1 << 1;
// }

if (useCPR.value) {
if (pairCloseRejection.isClosePair(p1, p2, parts, col.magField())) {
continue;
}
}

if (!pairCleaner.isCleanPair(p1, p2, parts)) {
continue;
}

float invMass;
if (p2.candidateSelFlag() == 1) {
invMass = p2.m(std::array{o2::constants::physics::MassProton, o2::constants::physics::MassKPlus, o2::constants::physics::MassPiPlus});
Expand Down Expand Up @@ -303,14 +305,28 @@ struct HfTaskCharmHadronsFemtoDream {
template <bool isMc, typename CollisionType, typename PartType, typename PartitionType1, typename PartitionType2, typename BinningType>
void doMixedEvent(CollisionType const& cols, PartType const& parts, PartitionType1& part1, PartitionType2& part2, BinningType policy)
{
processType = 1 << 1; // for mixed event

for (auto const& [collision1, collision2] : soa::selfCombinations(policy, mixingDepth.value, -1, cols, cols)) {
// Mixed events that contain the pair of interest
processType = 2; // for mixed event

for (auto const& [collision1, collision2] : combinations(soa::CombinationsBlockFullSameIndexPolicy(policy, mixingDepth.value, -1, cols, cols))) {
// make sure that tracks in the same events are not mixed
if (collision1.globalIndex() == collision2.globalIndex()) {
continue;
}
auto sliceTrk1 = part1->sliceByCached(aod::femtodreamparticle::fdCollisionId, collision1.globalIndex(), cache);
auto sliceCharmHad = part2->sliceByCached(aod::femtodreamparticle::fdCollisionId, collision2.globalIndex(), cache);
for (auto& [p1, p2] : combinations(CombinationsFullIndexPolicy(sliceTrk1, sliceCharmHad))) {

if (useCPR.value) {
if (pairCloseRejectionME.isClosePair(p1, p2, parts, collision1.magField())) {
continue;
}
}
if (!pairCleaner.isCleanPair(p1, p2, parts)) {
continue;
}

float chargeTrack = 0.;
if ((p1.cut() & 2) == 2) {
chargeTrack = PositiveCharge;
Expand Down Expand Up @@ -361,14 +377,6 @@ struct HfTaskCharmHadronsFemtoDream {
charmHadMc,
originType);

if (useCPR.value) {
if (pairCloseRejection.isClosePair(p1, p2, parts, collision1.magField())) {
continue;
}
}
if (!pairCleaner.isCleanPair(p1, p2, parts)) {
continue;
}
// if constexpr (!isMc) mixedEventCont.setPair<isMc, true>(p1, p2, collision1.multNtr(), collision1.multV0M(), use4D, extendedPlots, smearingByOrigin);
mixedEventCont.setPair<isMc, true>(p1, p2, collision1.multNtr(), collision1.multV0M(), use4D, extendedPlots, smearingByOrigin);
}
Expand Down