Skip to content

[PWGCF] FemtoUniverse V0 task: Refactor separate mass cut for lambdas and antilambdas #15046

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
alicja-pp wants to merge 1 commit into
base: master
Choose a base branch
from
+25 −90
Open

[PWGCF] FemtoUniverse V0 task: Refactor separate mass cut for lambdas and antilambdas #15046

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
115 changes: 25 additions & 90 deletions PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackV0Extended.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@ struct FemtoUniversePairTaskTrackV0Extended {
Configurable<float> confHPtChildPion{"confHPtChildPion", 10.0f, "Higher limit for pt of children pions"};
Configurable<float> confLPtChildPion{"confLPtChildPion", 0.f, "Lower limit for pt of children pions"};
Configurable<bool> confV0DuplCosPA{"confV0DuplCosPA", false, "Use cosPA instead of inv. mass as a deciding factor in rejecting a V0 in V0V0 pairs"};
Configurable<bool> confAddInvMassCheck{"confAddInvMassCheck", false, "Apply additional cut separate for mLambda and mAntiLambda"};
Configurable<bool> confSeparateInvMassCheck{"confSeparateInvMassCheck", false, "Apply additional cut separate for mLambda and mAntiLambda"};
} ConfV0Selection;

/// Partition for particle 2 using extended table
Expand Down Expand Up @@ -210,10 +210,17 @@ struct FemtoUniversePairTaskTrackV0Extended {
}
}

bool invMLambda(float invMassLambda, float invMassAntiLambda)
bool invMLambda(float invMassLambda, float invMassAntiLambda, int V0Type)
{
if ((invMassLambda < ConfV0Selection.confV0InvMassLowLimit || invMassLambda > ConfV0Selection.confV0InvMassUpLimit) && (invMassAntiLambda < ConfV0Selection.confV0InvMassLowLimit || invMassAntiLambda > ConfV0Selection.confV0InvMassUpLimit)) {
return false;
if (ConfV0Selection.confSeparateInvMassCheck) {
const float pMass = V0Type ? invMassAntiLambda : invMassLambda;
if (pMass < ConfV0Selection.confV0InvMassLowLimit || pMass > ConfV0Selection.confV0InvMassUpLimit) {
return false;
}
} else {
if ((invMassLambda < ConfV0Selection.confV0InvMassLowLimit || invMassLambda > ConfV0Selection.confV0InvMassUpLimit) && (invMassAntiLambda < ConfV0Selection.confV0InvMassLowLimit || invMassAntiLambda > ConfV0Selection.confV0InvMassUpLimit)) {
return false;
}
}
return true;
}
Expand Down Expand Up @@ -417,7 +424,7 @@ struct FemtoUniversePairTaskTrackV0Extended {

/// Histogramming same event
for (const auto& part : groupPartsTwo) {
if (!invMLambda(part.mLambda(), part.mAntiLambda()))
if (!invMLambda(part.mLambda(), part.mAntiLambda(), ConfV0Selection.confV0Type1))
continue;
const auto& posChild = parts.iteratorAt(part.globalIndex() - 2 - parts.begin().globalIndex());
const auto& negChild = parts.iteratorAt(part.globalIndex() - 1 - parts.begin().globalIndex());
Expand All @@ -426,13 +433,6 @@ struct FemtoUniversePairTaskTrackV0Extended {
continue;
}

if (ConfV0Selection.confAddInvMassCheck) {
const float pMass = ConfV0Selection.confV0Type1 ? part.mAntiLambda() : part.mLambda();
if (pMass < ConfV0Selection.confV0InvMassLowLimit || pMass > ConfV0Selection.confV0InvMassUpLimit) {
continue;
}
}

/// Daughters that do not pass this condition are not selected
if constexpr (std::experimental::is_detected<hasSigma, typename PartType::iterator>::value) {
if (!isParticleTPC(posChild, V0ChildTable[ConfV0Selection.confV0Type1][0]) || !isParticleTPC(negChild, V0ChildTable[ConfV0Selection.confV0Type1][1]))
Expand Down Expand Up @@ -492,7 +492,7 @@ struct FemtoUniversePairTaskTrackV0Extended {
/// Now build the combinations
for (const auto& [p1, p2] : combinations(CombinationsFullIndexPolicy(groupPartsOne, groupPartsTwo))) {
// Lambda invariant mass cut
if (!invMLambda(p2.mLambda(), p2.mAntiLambda()))
if (!invMLambda(p2.mLambda(), p2.mAntiLambda(), ConfV0Selection.confV0Type1))
continue;
/// PID using stored binned nsigma
if constexpr (std::experimental::is_detected<hasSigma, typename PartType::iterator>::value) {
Expand Down Expand Up @@ -534,13 +534,6 @@ struct FemtoUniversePairTaskTrackV0Extended {
}
}

if (ConfV0Selection.confAddInvMassCheck) {
const float pMass = ConfV0Selection.confV0Type1 ? p2.mAntiLambda() : p2.mLambda();
if (pMass < ConfV0Selection.confV0InvMassLowLimit || pMass > ConfV0Selection.confV0InvMassUpLimit) {
continue;
}
}

if (ConfCPR.confIsCPR.value) {
if (pairCloseRejection.isClosePair(p1, p2, parts, magFieldTesla, femto_universe_container::EventType::same)) {
continue;
Expand Down Expand Up @@ -571,7 +564,7 @@ struct FemtoUniversePairTaskTrackV0Extended {

/// Histogramming same event for first V0 particle
for (const auto& part : groupPartsTwo) {
if (!invMLambda(part.mLambda(), part.mAntiLambda()))
if (!invMLambda(part.mLambda(), part.mAntiLambda(), ConfV0Selection.confV0Type1))
continue;
const auto& posChild = parts.iteratorAt(part.globalIndex() - 2 - parts.begin().globalIndex());
const auto& negChild = parts.iteratorAt(part.globalIndex() - 1 - parts.begin().globalIndex());
Expand All @@ -586,12 +579,6 @@ struct FemtoUniversePairTaskTrackV0Extended {
if (!isParticleTOF(posChild, V0ChildTable[ConfV0Selection.confV0Type1][0]) || !isParticleTOF(negChild, V0ChildTable[ConfV0Selection.confV0Type1][1]))
continue;

if (ConfV0Selection.confAddInvMassCheck) {
const float pMass = ConfV0Selection.confV0Type1 ? part.mAntiLambda() : part.mLambda();
if (pMass < ConfV0Selection.confV0InvMassLowLimit || pMass > ConfV0Selection.confV0InvMassUpLimit) {
continue;
}
}
trackHistoV0Type1.fillQABase<false, true>(part, HIST("V0Type1"));
posChildV0Type1.fillQABase<false, true>(posChild, HIST("posChildV0Type1"));
negChildV0Type1.fillQABase<false, true>(negChild, HIST("negChildV0Type1"));
Expand All @@ -618,7 +605,7 @@ struct FemtoUniversePairTaskTrackV0Extended {

/// Histogramming same event for second V0 particle
for (const auto& part : groupPartsTwo) {
if (!invMLambda(part.mLambda(), part.mAntiLambda()))
if (!invMLambda(part.mLambda(), part.mAntiLambda(), ConfV0Selection.confV0Type2))
continue;
const auto& posChild = parts.iteratorAt(part.globalIndex() - 2 - parts.begin().globalIndex());
const auto& negChild = parts.iteratorAt(part.globalIndex() - 1 - parts.begin().globalIndex());
Expand All @@ -633,12 +620,6 @@ struct FemtoUniversePairTaskTrackV0Extended {
if (!isParticleTOF(posChild, V0ChildTable[ConfV0Selection.confV0Type2][0]) || !isParticleTOF(negChild, V0ChildTable[ConfV0Selection.confV0Type2][1]))
continue;

if (ConfV0Selection.confAddInvMassCheck) {
const float pMass = ConfV0Selection.confV0Type2 ? part.mAntiLambda() : part.mLambda();
if (pMass < ConfV0Selection.confV0InvMassLowLimit || pMass > ConfV0Selection.confV0InvMassUpLimit) {
continue;
}
}
trackHistoV0Type2.fillQABase<false, true>(part, HIST("V0Type2"));
posChildV0Type2.fillQABase<false, true>(posChild, HIST("posChildV0Type2"));
negChildV0Type2.fillQABase<false, true>(negChild, HIST("negChildV0Type2"));
Expand All @@ -665,20 +646,12 @@ struct FemtoUniversePairTaskTrackV0Extended {

auto pairDuplicateCheckFunc = [&](auto& p1, auto& p2) -> void {
// V0 inv mass cut for p1
if (!invMLambda(p1.mLambda(), p1.mAntiLambda()))
if (!invMLambda(p1.mLambda(), p1.mAntiLambda(), ConfV0Selection.confV0Type1))
return;
// V0 inv mass cut for p2
if (!invMLambda(p2.mLambda(), p2.mAntiLambda()))
if (!invMLambda(p2.mLambda(), p2.mAntiLambda(), ConfV0Selection.confV0Type2))
return;

if (ConfV0Selection.confAddInvMassCheck) {
const float p1Mass = ConfV0Selection.confV0Type1 ? p1.mAntiLambda() : p1.mLambda();
const float p2Mass = ConfV0Selection.confV0Type2 ? p2.mAntiLambda() : p2.mLambda();
if (p1Mass < ConfV0Selection.confV0InvMassLowLimit || p1Mass > ConfV0Selection.confV0InvMassUpLimit || p2Mass < ConfV0Selection.confV0InvMassLowLimit || p2Mass > ConfV0Selection.confV0InvMassUpLimit) {
return;
}
}

// track cleaning & checking for duplicate pairs
if (!pairCleanerV0.isCleanPair(p1, p2, parts)) {
// mark for rejection the cascade that shares a daughter with another cascade and has an invariant mass further from default value
Expand All @@ -702,10 +675,10 @@ struct FemtoUniversePairTaskTrackV0Extended {
if (v0Duplicates.contains(p1.globalIndex()) || v0Duplicates.contains(p2.globalIndex()))
return false;
// Lambda invariant mass cut for p1
if (!invMLambda(p1.mLambda(), p1.mAntiLambda()))
if (!invMLambda(p1.mLambda(), p1.mAntiLambda(), ConfV0Selection.confV0Type1))
return false;
// Lambda invariant mass cut for p2
if (!invMLambda(p2.mLambda(), p2.mAntiLambda()))
if (!invMLambda(p2.mLambda(), p2.mAntiLambda(), ConfV0Selection.confV0Type2))
return false;

const auto& posChild1 = parts.iteratorAt(p1.globalIndex() - 2 - parts.begin().globalIndex());
Expand Down Expand Up @@ -756,14 +729,6 @@ struct FemtoUniversePairTaskTrackV0Extended {
}
}

if (ConfV0Selection.confAddInvMassCheck) {
const float p1Mass = ConfV0Selection.confV0Type1 ? p1.mAntiLambda() : p1.mLambda();
const float p2Mass = ConfV0Selection.confV0Type2 ? p2.mAntiLambda() : p2.mLambda();
if (p1Mass < ConfV0Selection.confV0InvMassLowLimit || p1Mass > ConfV0Selection.confV0InvMassUpLimit || p2Mass < ConfV0Selection.confV0InvMassLowLimit || p2Mass > ConfV0Selection.confV0InvMassUpLimit) {
return false;
}
}

if (ConfCPR.confIsCPR.value) {
if (ConfCPR.confRectV0V0CPR && pairCloseRejectionV0.isClosePair<true>(p1, p2, parts, magFieldTesla, femto_universe_container::EventType::same)) {
return false;
Expand Down Expand Up @@ -984,7 +949,7 @@ struct FemtoUniversePairTaskTrackV0Extended {

for (const auto& [p1, p2] : combinations(CombinationsFullIndexPolicy(groupPartsOne, groupPartsTwo))) {
// Lambda invariant mass cut
if (!invMLambda(p2.mLambda(), p2.mAntiLambda()))
if (!invMLambda(p2.mLambda(), p2.mAntiLambda(), ConfV0Selection.confV0Type1))
continue;
/// PID using stored binned nsigma
if constexpr (std::experimental::is_detected<hasSigma, typename PartType::iterator>::value) {
Expand Down Expand Up @@ -1019,13 +984,6 @@ struct FemtoUniversePairTaskTrackV0Extended {
}
}

if (ConfV0Selection.confAddInvMassCheck) {
const float pMass = ConfV0Selection.confV0Type1 ? p2.mAntiLambda() : p2.mLambda();
if (pMass < ConfV0Selection.confV0InvMassLowLimit || pMass > ConfV0Selection.confV0InvMassUpLimit) {
continue;
}
}

// track cleaning
if (!pairCleaner.isCleanPair(p1, p2, parts)) {
continue;
Expand Down Expand Up @@ -1084,11 +1042,11 @@ struct FemtoUniversePairTaskTrackV0Extended {

for (const auto& [p1, p2] : combinations(CombinationsFullIndexPolicy(groupPartsOne, groupPartsTwo))) {
// Lambda invariant mass cut for p1
if (!invMLambda(p1.mLambda(), p1.mAntiLambda())) {
if (!invMLambda(p1.mLambda(), p1.mAntiLambda(), ConfV0Selection.confV0Type1)) {
continue;
}
// Lambda invariant mass cut for p2
if (!invMLambda(p2.mLambda(), p2.mAntiLambda())) {
if (!invMLambda(p2.mLambda(), p2.mAntiLambda(), ConfV0Selection.confV0Type2)) {
continue;
}

Expand Down Expand Up @@ -1141,14 +1099,6 @@ struct FemtoUniversePairTaskTrackV0Extended {
}
}

if (ConfV0Selection.confAddInvMassCheck) {
const float p1Mass = ConfV0Selection.confV0Type1 ? p1.mAntiLambda() : p1.mLambda();
const float p2Mass = ConfV0Selection.confV0Type2 ? p2.mAntiLambda() : p2.mLambda();
if (p1Mass < ConfV0Selection.confV0InvMassLowLimit || p1Mass > ConfV0Selection.confV0InvMassUpLimit || p2Mass < ConfV0Selection.confV0InvMassLowLimit || p2Mass > ConfV0Selection.confV0InvMassUpLimit) {
continue;
}
}

// track cleaning
if (!pairCleanerV0.isCleanPair(p1, p2, parts)) {
continue;
Expand Down Expand Up @@ -1375,7 +1325,7 @@ struct FemtoUniversePairTaskTrackV0Extended {

for (const auto& [p1, p2] : combinations(CombinationsFullIndexPolicy(groupPartsOne, groupPartsTwo))) {
// Lambda invariant mass cut
if (!invMLambda(p2.mLambda(), p2.mAntiLambda()))
if (!invMLambda(p2.mLambda(), p2.mAntiLambda(), ConfV0Selection.confV0Type1))
continue;
/// PID using stored binned nsigma
if (!isParticleCombined(p1, ConfTrkSelection.confTrackChoicePartOne))
Expand All @@ -1393,13 +1343,6 @@ struct FemtoUniversePairTaskTrackV0Extended {
if (!isParticleTOF(posChild, V0ChildTable[ConfV0Selection.confV0Type1][0]) || !isParticleTOF(negChild, V0ChildTable[ConfV0Selection.confV0Type1][1]))
continue;

if (ConfV0Selection.confAddInvMassCheck) {
const float pMass = ConfV0Selection.confV0Type1 ? p2.mAntiLambda() : p2.mLambda();
if (pMass < ConfV0Selection.confV0InvMassLowLimit || pMass > ConfV0Selection.confV0InvMassUpLimit) {
continue;
}
}

// track cleaning
if (!pairCleaner.isCleanPair(p1, p2, parts)) {
continue;
Expand Down Expand Up @@ -1452,11 +1395,11 @@ struct FemtoUniversePairTaskTrackV0Extended {

for (const auto& [p1, p2] : combinations(CombinationsFullIndexPolicy(groupPartsOne, groupPartsTwo))) {
// Lambda invariant mass cut for p1
if (!invMLambda(p1.mLambda(), p1.mAntiLambda())) {
if (!invMLambda(p1.mLambda(), p1.mAntiLambda(), ConfV0Selection.confV0Type1)) {
continue;
}
// Lambda invariant mass cut for p2
if (!invMLambda(p2.mLambda(), p2.mAntiLambda())) {
if (!invMLambda(p2.mLambda(), p2.mAntiLambda(), ConfV0Selection.confV0Type2)) {
continue;
}

Expand Down Expand Up @@ -1484,14 +1427,6 @@ struct FemtoUniversePairTaskTrackV0Extended {
if (!isParticleTOF(negChild2, V0ChildTable[ConfV0Selection.confV0Type1][0]) || !isParticleTOF(negChild2, V0ChildTable[ConfV0Selection.confV0Type1][1]))
continue;

if (ConfV0Selection.confAddInvMassCheck) {
const float p1Mass = ConfV0Selection.confV0Type1 ? p1.mAntiLambda() : p1.mLambda();
const float p2Mass = ConfV0Selection.confV0Type2 ? p2.mAntiLambda() : p2.mLambda();
if (p1Mass < ConfV0Selection.confV0InvMassLowLimit || p1Mass > ConfV0Selection.confV0InvMassUpLimit || p2Mass < ConfV0Selection.confV0InvMassLowLimit || p2Mass > ConfV0Selection.confV0InvMassUpLimit) {
continue;
}
}

// track cleaning
if (!pairCleanerV0.isCleanPair(p1, p2, parts)) {
continue;
Expand Down
Loading