Skip to content

Commit e6fc7b7

Browse files
committed
revert to TLorentzVector
1 parent 1e76f31 commit e6fc7b7

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

PWGLF/Tasks/Resonances/doubleResonanceScan.cxx

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
///
1717

1818
#include <vector>
19+
#include <TLorentzVector.h>
1920

2021
#include "Framework/AnalysisTask.h"
2122
#include "Framework/ASoAHelpers.h"
@@ -387,9 +388,9 @@ struct DoubleResonanceScan {
387388
return selectedIndices;
388389
}
389390

390-
bool isPairSelected(const ROOT::Math::LorentzVector& lv1, const ROOT::Math::LorentzVector& lv2, int pairType = 0)
391+
bool isPairSelected(const TLorentzVector & lv1, const TLorentzVector & lv2, int pairType = 0)
391392
{
392-
ROOT::Math::LorentzVector lvSum = lv1 + lv2;
393+
TLorentzVector lvSum = lv1 + lv2;
393394
// Mass window cut
394395
auto pairMass = lvSum.M();
395396
auto pairMassesLow = PairCuts.cfgPairMassesLow.value;
@@ -431,7 +432,7 @@ struct DoubleResonanceScan {
431432
int pairType)
432433
{
433434
std::vector<std::pair<int, int>> selectedPairs;
434-
ROOT::Math::LorentzVector lv1, lv2;
435+
TLorentzVector lv1, lv2;
435436
for (const auto& indexA : indicesA) {
436437
for (const auto& indexB : indicesB) {
437438
if (indexA == indexB) {
@@ -453,15 +454,15 @@ struct DoubleResonanceScan {
453454
return selectedPairs;
454455
}
455456

456-
bool isResoSelected(const ROOT::Math::LorentzVector& par1, const ROOT::Math::LorentzVector& pair2)
457+
bool isResoSelected(const TLorentzVector & par1, const TLorentzVector & pair2)
457458
{
458459
// Opening angle (3D)
459460
double oa = par1.Vect().Angle(pair2.Vect());
460461
if (oa < ResoCuts.cfgPairOALow || oa > ResoCuts.cfgPairOAHigh) {
461462
return false;
462463
}
463464
// Rapidity cut
464-
ROOT::Math::LorentzVector lvTotal = par1 + pair2;
465+
TLorentzVector lvTotal = par1 + pair2;
465466
if (lvTotal.Rapidity() < -0.5 || lvTotal.Rapidity() > 0.5) {
466467
return false;
467468
}
@@ -501,7 +502,7 @@ struct DoubleResonanceScan {
501502
auto t3 = dTracks.rawIteratorAt(j1);
502503
auto t4 = dTracks.rawIteratorAt(j2);
503504

504-
ROOT::Math::LorentzVector lv1, lv2, lv3, lv4, lvPair1, lvPair2, lvTotal, lResonanceRot;
505+
TLorentzVector lv1, lv2, lv3, lv4, lvPair1, lvPair2, lvTotal, lResonanceRot;
505506
lv1.SetXYZM(t1.px(), t1.py(), t1.pz(), mass1);
506507
lv2.SetXYZM(t2.px(), t2.py(), t2.pz(), mass2);
507508
lv3.SetXYZM(t3.px(), t3.py(), t3.pz(), mass3);

0 commit comments

Comments
 (0)