1616#ifndef PWGCF_FEMTO3D_CORE_FEMTO3DPAIRTASK_H_
1717#define PWGCF_FEMTO3D_CORE_FEMTO3DPAIRTASK_H_
1818
19- #define THETA (eta ) 2.0 * atan (exp(-eta))
19+ #define THETA (eta ) 2.0 * std:: atan (std:: exp(-eta))
2020// #include "Framework/ASoA.h"
2121// #include "Framework/DataTypes.h"
2222// #include "Framework/AnalysisDataModel.h"
3030#include " TVector3.h"
3131#include " TDatabasePDG.h"
3232
33+ #include " CommonConstants/PhysicsConstants.h"
3334#include " CommonConstants/MathConstants.h"
3435
35- double particle_mass (int PDGcode)
36+ double particle_mass (const int PDGcode)
3637{
37- // if(PDGcode == 2212) return TDatabasePDG::Instance()->GetParticle(2212)->Mass();
38- if (PDGcode == 1000010020 )
39- return 1.87561294257 ;
40- else
41- return TDatabasePDG::Instance ()->GetParticle (PDGcode)->Mass ();
38+ switch (std::abs (PDGcode)) {
39+ case o2::constants::physics::kDeuteron :
40+ return o2::constants::physics::MassDeuteron;
41+ case o2::constants::physics::kTriton :
42+ return o2::constants::physics::MassTriton;
43+ case o2::constants::physics::kHelium3 :
44+ return o2::constants::physics::MassHelium3;
45+ case 211 :
46+ return o2::constants::physics::MassPionCharged;
47+ case 321 :
48+ return o2::constants::physics::MassKaonCharged;
49+ case 2212 :
50+ return o2::constants::physics::MassProton;
51+ default :
52+ break ;
53+ }
54+ return TDatabasePDG::Instance ()->GetParticle (PDGcode)->Mass ();
4255}
4356
4457// for the variable binning in 3D DCA histos in the PairMC task
@@ -72,7 +85,7 @@ inline std::unique_ptr<double[]> calc_var_bins(const int& N, const float& xmax,
7285 bins[N - 1 ] = xmax;
7386
7487 for (int i = 1 ; i < 0.5 * N - 1 ; i++) {
75- bin_edge += winit * pow (q, i);
88+ bin_edge += winit * std:: pow (q, i);
7689 bins[0.5 * N - 1 - i] = -bin_edge;
7790 bins[0.5 * N + i] = bin_edge;
7891 }
@@ -295,9 +308,9 @@ float FemtoPair<TrackType>::GetAvgSep() const
295308 float res = 0.0 ;
296309
297310 for (const auto & radius : TPCradii) {
298- const float dRtrans = 2.0 * radius * sin (0.5 * GetPhiStarDiff (radius));
299- const float dRlong = 2.0 * radius * sin (0.5 * dtheta);
300- res += sqrt (dRtrans * dRtrans + dRlong * dRlong);
311+ const float dRtrans = 2.0 * radius * std:: sin (0.5 * GetPhiStarDiff (radius));
312+ const float dRlong = 2.0 * radius * std:: sin (0.5 * dtheta);
313+ res += std:: sqrt (dRtrans * dRtrans + dRlong * dRlong);
301314 }
302315
303316 return 100.0 * res / TPCradii.size ();
@@ -315,7 +328,7 @@ float FemtoPair<TrackType>::GetAvgPhiStarDiff() const
315328
316329 for (const auto & radius : TPCradii) {
317330 const float dphi = GetPhiStarDiff (radius);
318- res += fabs (dphi) > o2::constants::math::PI ? (1.0 - 2.0 * o2::constants::math::PI / fabs (dphi)) * dphi : dphi;
331+ res += std:: fabs (dphi) > o2::constants::math::PI ? (1.0 - 2.0 * o2::constants::math::PI / std:: fabs (dphi)) * dphi : dphi;
319332 }
320333
321334 return res / TPCradii.size ();
0 commit comments