Skip to content

Commit f2083b3

Browse files
authored
[PWGEM/Dilepton] add phi protection in EMTrack.h (#13867)
1 parent 2e8a270 commit f2083b3

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

PWGEM/Dilepton/Utils/EMTrack.h

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -184,8 +184,11 @@ class EMPair : public EMTrack
184184
void setPositiveLegPxPyPzM(float px, float py, float pz, float m)
185185
{
186186
float pt = std::sqrt(px * px + py * py);
187-
float eta = std::atanh(pz / sqrt(std::pow(px, 2) + std::pow(py, 2) + std::pow(pz, 2)));
187+
float eta = std::atanh(pz / std::sqrt(std::pow(px, 2) + std::pow(py, 2) + std::pow(pz, 2)));
188188
float phi = std::atan2(py, px);
189+
if (phi < 0.f) {
190+
phi += 2.f * M_PI;
191+
}
189192

190193
fVPos.SetPt(pt);
191194
fVPos.SetEta(eta);
@@ -195,8 +198,11 @@ class EMPair : public EMTrack
195198
void setNegativeLegPxPyPzM(float px, float py, float pz, float m)
196199
{
197200
float pt = std::sqrt(px * px + py * py);
198-
float eta = std::atanh(pz / std::sqrt(pow(px, 2) + std::pow(py, 2) + std::pow(pz, 2)));
201+
float eta = std::atanh(pz / std::sqrt(std::pow(px, 2) + std::pow(py, 2) + std::pow(pz, 2)));
199202
float phi = std::atan2(py, px);
203+
if (phi < 0.f) {
204+
phi += 2.f * M_PI;
205+
}
200206

201207
fVNeg.SetPt(pt);
202208
fVNeg.SetEta(eta);

0 commit comments

Comments
 (0)