|
19 | 19 | #include "PWGEM/Dilepton/Utils/EMTrackUtilities.h" |
20 | 20 | #include "PWGEM/Dilepton/Utils/PairUtilities.h" |
21 | 21 |
|
22 | | -#include "CommonConstants/PhysicsConstants.h" |
23 | | -#include "Framework/DataTypes.h" |
24 | | -#include "Framework/Logger.h" |
| 22 | +#include "Common/Core/RecoDecay.h" |
25 | 23 |
|
26 | | -#include "Math/Vector4D.h" |
27 | | -#include "TNamed.h" |
| 24 | +#include <CommonConstants/PhysicsConstants.h> |
| 25 | +#include <Framework/Logger.h> |
| 26 | +#include <MathUtils/Utils.h> |
| 27 | + |
| 28 | +#include <Math/Vector4D.h> // IWYU pragma: keep (do not replace with Math/Vector4Dfwd.h) |
| 29 | +#include <Math/Vector4Dfwd.h> |
| 30 | +#include <TNamed.h> |
| 31 | + |
| 32 | +#include <Rtypes.h> |
28 | 33 |
|
29 | 34 | #include <algorithm> |
| 35 | +#include <cmath> |
| 36 | +#include <cstdint> |
| 37 | +#include <functional> |
30 | 38 | #include <set> |
31 | | -#include <string> |
32 | 39 | #include <utility> |
33 | 40 | #include <vector> |
34 | 41 |
|
35 | | -using namespace o2::aod::pwgem::dilepton::utils::emtrackutil; |
36 | | -using namespace o2::aod::pwgem::dilepton::utils::pairutil; |
| 42 | +#include <math.h> |
37 | 43 |
|
38 | 44 | class DielectronCut : public TNamed |
39 | 45 | { |
@@ -108,9 +114,9 @@ class DielectronCut : public TNamed |
108 | 114 | ROOT::Math::PtEtaPhiMVector v2(t2.pt(), t2.eta(), t2.phi(), o2::constants::physics::MassElectron); |
109 | 115 | ROOT::Math::PtEtaPhiMVector v12 = v1 + v2; |
110 | 116 |
|
111 | | - float dca_ee_3d = pairDCAQuadSum(dca3DinSigma(t1), dca3DinSigma(t2)); |
112 | | - float phiv = getPhivPair(t1.px(), t1.py(), t1.pz(), t2.px(), t2.py(), t2.pz(), t1.sign(), t2.sign(), bz); |
113 | | - float opAng = getOpeningAngle(t1.px(), t1.py(), t1.pz(), t2.px(), t2.py(), t2.pz()); |
| 117 | + float dca_ee_3d = o2::aod::pwgem::dilepton::utils::pairutil::pairDCAQuadSum(o2::aod::pwgem::dilepton::utils::emtrackutil::dca3DinSigma(t1), o2::aod::pwgem::dilepton::utils::emtrackutil::dca3DinSigma(t2)); |
| 118 | + float phiv = o2::aod::pwgem::dilepton::utils::pairutil::getPhivPair(t1.px(), t1.py(), t1.pz(), t2.px(), t2.py(), t2.pz(), t1.sign(), t2.sign(), bz); |
| 119 | + float opAng = o2::aod::pwgem::dilepton::utils::pairutil::getOpeningAngle(t1.px(), t1.py(), t1.pz(), t2.px(), t2.py(), t2.pz()); |
114 | 120 |
|
115 | 121 | if (v12.M() < mMinMee || mMaxMee < v12.M()) { |
116 | 122 | return false; |
@@ -457,7 +463,7 @@ class DielectronCut : public TNamed |
457 | 463 | return mMinChi2PerClusterTPC < track.tpcChi2NCl() && track.tpcChi2NCl() < mMaxChi2PerClusterTPC; |
458 | 464 |
|
459 | 465 | case DielectronCuts::kDCA3Dsigma: |
460 | | - return mMinDca3D < dca3DinSigma(track) && dca3DinSigma(track) < mMaxDca3D; // in sigma for single leg |
| 466 | + return mMinDca3D < o2::aod::pwgem::dilepton::utils::emtrackutil::dca3DinSigma(track) && o2::aod::pwgem::dilepton::utils::emtrackutil::dca3DinSigma(track) < mMaxDca3D; // in sigma for single leg |
461 | 467 |
|
462 | 468 | case DielectronCuts::kDCAxy: |
463 | 469 | return std::fabs(track.dcaXY()) < ((mMaxDcaXYPtDep) ? mMaxDcaXYPtDep(track.pt()) : mMaxDcaXY); |
|
0 commit comments