Skip to content

Commit 4bfefdf

Browse files
authored
[PWGCF] FemtoUniverse: Add option to randomize pair for DeltaEta-DeltaPhi* histograms in SH task (#15993)
1 parent 1638cd7 commit 4bfefdf

1 file changed

Lines changed: 22 additions & 6 deletions

File tree

PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackTrackSpherHarMultKtExtended.cxx

Lines changed: 22 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,7 @@ struct FemtoUniversePairTaskTrackTrackSpherHarMultKtExtended {
138138
Configurable<float> confPairFracSharedTPCclsMin{"confPairFracSharedTPCclsMin", 0.0, "Min. fraction of TPC shared clusters between two closed tracks"};
139139
Configurable<float> confPairFracSharedTPCclsMax{"confPairFracSharedTPCclsMax", 1.0, "Miax. fraction of TPC shared clusters between two closed tracks"};
140140
Configurable<float> confCPRChosenRadii{"confCPRChosenRadii", 0.80, "Delta Eta cut for Close Pair Rejection"};
141+
Configurable<bool> confRandomizeCPR{"confRandomizeCPR", false, "Fill Deta-Dphi histograms for PlusMinus pair in random order: p1, p2 or p2, p1"};
141142
Configurable<bool> confUseCCImCut{"confUseCCImCut", false, "Fill SH within specific quadrants of qout-qside"};
142143
Configurable<float> confMinqTcut{"confMinqTcut", 0.005, "Min. qT cut on filling SH"};
143144
Configurable<bool> confUse1stand3rd{"confUse1stand3rd", false, "Use first and third quadrants of qout-qside"};
@@ -622,18 +623,31 @@ struct FemtoUniversePairTaskTrackTrackSpherHarMultKtExtended {
622623
continue;
623624
}
624625

626+
auto part1 = p1;
627+
auto part2 = p2;
628+
629+
if (twotracksconfigs.confRandomizeCPR) {
630+
double rand;
631+
rand = randgen->Rndm();
632+
633+
if (rand > twotracksconfigs.confRandValue) {
634+
part1 = p2;
635+
part2 = p1;
636+
}
637+
}
638+
625639
if (twotracksconfigs.confIsCPR.value) {
626640
if (twotracksconfigs.confIsCPRatITS && twotracksconfigs.confCPRFracMax.value) {
627-
if (pairCloseRejection.isClosePairAtITS(p1, p2, magFieldTesla, femto_universe_container::EventType::same)) {
641+
if (pairCloseRejection.isClosePairAtITS(part1, part2, magFieldTesla, femto_universe_container::EventType::same)) {
628642
continue;
629643
}
630644
} else {
631645
if (twotracksconfigs.confIsCPRkT) {
632-
if (pairCloseRejection.isClosePairkT(p1, p2, femto_universe_container::EventType::same, kT, twotracksconfigs.confIsCircularCut, twotracksconfigs.confCPRDphiAvgOrDist, magFieldTesla, twotracksconfigs.confCPRDistMax, twotracksconfigs.confCPRFracMax, twotracksconfigs.confIsfilldEtadPhiTPCcls, pairFractionTPCsCls)) {
646+
if (pairCloseRejection.isClosePairkT(part1, part2, femto_universe_container::EventType::same, kT, twotracksconfigs.confIsCircularCut, twotracksconfigs.confCPRDphiAvgOrDist, magFieldTesla, twotracksconfigs.confCPRDistMax, twotracksconfigs.confCPRFracMax, twotracksconfigs.confIsfilldEtadPhiTPCcls, pairFractionTPCsCls)) {
633647
continue;
634648
}
635649
} else {
636-
if (pairCloseRejection.isClosePairFrac(p1, p2, magFieldTesla, femto_universe_container::EventType::same, twotracksconfigs.confCPRDphiAvgOrDist, twotracksconfigs.confCPRDistMax, twotracksconfigs.confCPRFracMax, twotracksconfigs.confIsCircularCut)) {
650+
if (pairCloseRejection.isClosePairFrac(part1, part2, magFieldTesla, femto_universe_container::EventType::same, twotracksconfigs.confCPRDphiAvgOrDist, twotracksconfigs.confCPRDistMax, twotracksconfigs.confCPRFracMax, twotracksconfigs.confIsCircularCut)) {
637651
continue;
638652
}
639653
}
@@ -1048,9 +1062,11 @@ struct FemtoUniversePairTaskTrackTrackSpherHarMultKtExtended {
10481062
auto part1 = p1;
10491063
auto part2 = p2;
10501064

1051-
if (rand > twotracksconfigs.confRandValue) {
1052-
part1 = p2;
1053-
part2 = p1;
1065+
if (!PairType::PlusMinus || twotracksconfigs.confRandomizeCPR) {
1066+
if (rand > twotracksconfigs.confRandValue) {
1067+
part1 = p2;
1068+
part2 = p1;
1069+
}
10541070
}
10551071

10561072
if (twotracksconfigs.confIsCPR.value) {

0 commit comments

Comments
 (0)