Skip to content

Commit 464cb1f

Browse files
committed
Move gamma pair rejection function to header file
1 parent d8b6827 commit 464cb1f

File tree

3 files changed

+34
-59
lines changed

3 files changed

+34
-59
lines changed

PWGCF/FemtoUniverse/Core/FemtoUniverseDetaDphiStar.h

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -694,6 +694,31 @@ class FemtoUniverseDetaDphiStar
694694
}
695695
}
696696

697+
template <typename PartType>
698+
bool isGammaPair(PartType track1, PartType track2, double maxEEMinv, double maxDTheta)
699+
{
700+
double me = o2::constants::physics::MassElectron;
701+
702+
double magTrack1 = track1.px() * track1.px() + track1.py() * track1.py() + track1.pz() * track1.pz();
703+
double magTrack2 = track2.px() * track2.px() + track2.py() * track2.py() + track2.pz() * track2.pz();
704+
double dotTr1Tr2 = track1.px() * track2.px() + track1.py() * track2.py() + track1.pz() * track2.pz();
705+
706+
if ((track1.sign() * track2.sign()) < 0.0) {
707+
double theta1 = track1.theta();
708+
double theta2 = track2.theta();
709+
double dtheta = TMath::Abs(theta1 - theta2);
710+
711+
double e1 = TMath::Sqrt(me * me + magTrack1);
712+
double e2 = TMath::Sqrt(me * me + magTrack2);
713+
714+
double minv = 2 * me * me + 2 * (e1 * e2 - dotTr1Tr2);
715+
if ((TMath::Abs(minv) < maxEEMinv) && (dtheta < maxDTheta)) {
716+
return true;
717+
}
718+
}
719+
return false;
720+
}
721+
697722
/// Check if pair is close or not
698723
template <typename Part>
699724
void ClosePairqLCMS(Part const& part1, Part const& part2, float lmagfield, uint8_t ChosenEventType, double qlcms) // add typename Parts and variable parts for adding MClabels

PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackTrackMultKtExtended.cxx

Lines changed: 7 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -339,31 +339,6 @@ struct FemtoUniversePairTaskTrackTrackMultKtExtended {
339339
return false;
340340
}
341341

342-
template <typename PartType>
343-
bool rejectGammaPair(PartType track1, PartType track2)
344-
{
345-
double me = o2::constants::physics::MassElectron;
346-
347-
double magTrack1 = track1.px() * track1.px() + track1.py() * track1.py() + track1.pz() * track1.pz();
348-
double magTrack2 = track2.px() * track2.px() + track2.py() * track2.py() + track2.pz() * track2.pz();
349-
double dotTr1Tr2 = track1.px() * track2.px() + track1.py() * track2.py() + track1.pz() * track2.pz();
350-
351-
if ((track1.sign() * track2.sign()) < 0.0) {
352-
double theta1 = track1.theta();
353-
double theta2 = track2.theta();
354-
double dtheta = TMath::Abs(theta1 - theta2);
355-
356-
double e1 = TMath::Sqrt(me * me + magTrack1);
357-
double e2 = TMath::Sqrt(me * me + magTrack2);
358-
359-
double minv = 2 * me * me + 2 * (e1 * e2 - dotTr1Tr2);
360-
if ((TMath::Abs(minv) < confMaxEEMinv) && (dtheta < confMaxDTheta)) {
361-
return false;
362-
}
363-
}
364-
return true;
365-
}
366-
367342
void init(InitContext&)
368343
{
369344
eventHisto.init(&qaRegistry);
@@ -509,7 +484,12 @@ struct FemtoUniversePairTaskTrackTrackMultKtExtended {
509484
continue;
510485
}
511486

512-
if (confRejectGammaPair && !rejectGammaPair(p1, p2)) {
487+
if (confRejectGammaPair && pairCloseRejection.isGammaPair(p1, p2, confMaxEEMinv, confMaxDTheta)) {
488+
continue;
489+
}
490+
491+
// track cleaning
492+
if (!pairCleaner.isCleanPair(p1, p2, parts)) {
513493
continue;
514494
}
515495

@@ -533,11 +513,6 @@ struct FemtoUniversePairTaskTrackTrackMultKtExtended {
533513
}
534514
}
535515

536-
// track cleaning
537-
if (!pairCleaner.isCleanPair(p1, p2, parts)) {
538-
continue;
539-
}
540-
541516
float kstar = FemtoUniverseMath::getkstar(p1, mass1, p2, mass2);
542517
float kT = FemtoUniverseMath::getkT(p1, mass1, p2, mass2);
543518

@@ -706,7 +681,7 @@ struct FemtoUniversePairTaskTrackTrackMultKtExtended {
706681
continue;
707682
}
708683

709-
if (confRejectGammaPair && !rejectGammaPair(p1, p2)) {
684+
if (confRejectGammaPair && pairCloseRejection.isGammaPair(p1, p2, confMaxEEMinv, confMaxDTheta)) {
710685
continue;
711686
}
712687

PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackTrackSpherHarMultKtExtended.cxx

Lines changed: 2 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -415,31 +415,6 @@ struct femtoUniversePairTaskTrackTrackSpherHarMultKtExtended {
415415
return false;
416416
}
417417

418-
template <typename PartType>
419-
bool rejectGammaPair(PartType track1, PartType track2)
420-
{
421-
double me = o2::constants::physics::MassElectron;
422-
423-
double magTrack1 = track1.px() * track1.px() + track1.py() * track1.py() + track1.pz() * track1.pz();
424-
double magTrack2 = track2.px() * track2.px() + track2.py() * track2.py() + track2.pz() * track2.pz();
425-
double dotTr1Tr2 = track1.px() * track2.px() + track1.py() * track2.py() + track1.pz() * track2.pz();
426-
427-
if ((track1.sign() * track2.sign()) < 0.0) {
428-
double theta1 = track1.theta();
429-
double theta2 = track2.theta();
430-
double dtheta = TMath::Abs(theta1 - theta2);
431-
432-
double e1 = TMath::Sqrt(me * me + magTrack1);
433-
double e2 = TMath::Sqrt(me * me + magTrack2);
434-
435-
double minv = 2 * me * me + 2 * (e1 * e2 - dotTr1Tr2);
436-
if ((TMath::Abs(minv) < confMaxEEMinv) && (dtheta < confMaxDTheta)) {
437-
return false;
438-
}
439-
}
440-
return true;
441-
}
442-
443418
void init(InitContext&)
444419
{
445420
eventHisto.init(&qaRegistry);
@@ -565,7 +540,7 @@ struct femtoUniversePairTaskTrackTrackSpherHarMultKtExtended {
565540
continue;
566541
}
567542

568-
if (confRejectGammaPair && !rejectGammaPair(p1, p2)) {
543+
if (confRejectGammaPair && pairCloseRejection.isGammaPair(p1, p2, confMaxEEMinv, confMaxDTheta)) {
569544
continue;
570545
}
571546

@@ -971,7 +946,7 @@ struct femtoUniversePairTaskTrackTrackSpherHarMultKtExtended {
971946
continue;
972947
}
973948

974-
if (confRejectGammaPair && !rejectGammaPair(p1, p2)) {
949+
if (confRejectGammaPair && pairCloseRejection.isGammaPair(p1, p2, confMaxEEMinv, confMaxDTheta)) {
975950
continue;
976951
}
977952

0 commit comments

Comments
 (0)