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