Skip to content

Commit 53cd7b0

Browse files
davidrohrdavidrohr
committed
GPU: Move some functionality to CAMath
1 parent d2e470b commit 53cd7b0

File tree

8 files changed

+18
-20
lines changed

8 files changed

+18
-20
lines changed

‎GPU/Common/GPUCommonMath.h‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,7 @@ class GPUCommonMath
9595
GPUhdni() constexpr static float Copysign(float x, float y) { return GPUCA_CHOICE(std::copysignf(x, y), copysignf(x, y), copysign(x, y)); }
9696
GPUd() constexpr static float TwoPi() { return 6.2831853f; }
9797
GPUd() constexpr static float Pi() { return 3.1415927f; }
98+
GPUd() constexpr static float Deg2Rad() { return Pi() / 180.f; }
9899
GPUd() constexpr static float Round(float x);
99100
GPUd() constexpr static float Floor(float x) { return GPUCA_CHOICE(floorf(x), floorf(x), floor(x)); }
100101
GPUd() static uint32_t Float2UIntReint(const float& x);

‎GPU/GPUTracking/DataTypes/GPUTPCGeometry.h‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
#define GPUTPCGEOMETRY_H
1717

1818
#include "GPUCommonDef.h"
19+
#include "GPUCommonMath.h"
1920

2021
#if !defined(GPUCA_NSECTORS) && !defined(GPUCA_ROW_COUNT)
2122
#include "DataFormatsTPC/Constants.h"
@@ -145,7 +146,7 @@ class GPUTPCGeometry // TODO: Make values constexpr
145146
return (250.f - v) * FACTOR_Z2T; // Used in compression, must remain constant at 250cm
146147
}
147148

148-
GPUd() static constexpr float kSectAngle() { return 2 * M_PI / 18.f; }
149+
GPUd() static constexpr float kSectAngle() { return 2 * CAMath::Pi() / 18.f; }
149150
};
150151

151152
} // namespace o2::gpu

‎GPU/GPUTracking/Merger/GPUTPCGMTrackParam.cxx‎

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -284,8 +284,7 @@ GPUdii() int32_t GPUTPCGMTrackParam::FitHit(GPUTPCGMMerger& GPUrestrict() merger
284284
uncorrectedY = AttachClusters(merger, cluster.sector, cluster.row, iTrk, track.Leg() == 0, prop);
285285
}
286286

287-
static constexpr float kDeg2Rad = M_PI / 180.f;
288-
const float maxSinForUpdate = CAMath::Sin(70.f * kDeg2Rad);
287+
const float maxSinForUpdate = CAMath::Sin(70.f * CAMath::Deg2Rad());
289288
const bool sinPhiErr = mNDF > 0 && CAMath::Abs(prop.GetSinPhi0()) >= maxSinForUpdate;
290289
if (mNDF >= 0 && (mC[0] > param.rec.tpc.trackFitCovLimit || mC[2] > param.rec.tpc.trackFitCovLimit)) {
291290
return 1; // bad chi2 for the whole track, stop the fit
@@ -533,13 +532,11 @@ GPUdii() void GPUTPCGMTrackParam::DodEdx(GPUdEdx& GPUrestrict() dEdx, GPUdEdx& G
533532

534533
GPUdni() void GPUTPCGMTrackParam::MoveToReference(GPUTPCGMPropagator& GPUrestrict() prop, const GPUParam& GPUrestrict() param, float& GPUrestrict() Alpha)
535534
{
536-
static constexpr float kDeg2Rad = M_PI / 180.f;
537-
538535
if (param.rec.tpc.trackReferenceX <= 500) {
539536
GPUTPCGMTrackParam save = *this;
540537
float saveAlpha = Alpha;
541538
for (int32_t attempt = 0; attempt < 3; attempt++) {
542-
float dAngle = CAMath::Round(CAMath::ATan2(mP[0], mX) / kDeg2Rad / 20.f) * GPUTPCGeometry::kSectAngle();
539+
float dAngle = CAMath::Round(CAMath::ATan2(mP[0], mX) / CAMath::Deg2Rad() / 20.f) * GPUTPCGeometry::kSectAngle();
543540
Alpha += dAngle;
544541
if (prop.PropagateToXAlpha(param.rec.tpc.trackReferenceX, Alpha, 0)) {
545542
break;
@@ -553,7 +550,7 @@ GPUdni() void GPUTPCGMTrackParam::MoveToReference(GPUTPCGMPropagator& GPUrestric
553550
Alpha = saveAlpha;
554551
}
555552
if (CAMath::Abs(mP[0]) > mX * CAMath::Tan(GPUTPCGeometry::kSectAngle() / 2.f)) {
556-
float dAngle = CAMath::Round(CAMath::ATan2(mP[0], mX) / kDeg2Rad / 20.f) * GPUTPCGeometry::kSectAngle();
553+
float dAngle = CAMath::Round(CAMath::ATan2(mP[0], mX) / CAMath::Deg2Rad() / 20.f) * GPUTPCGeometry::kSectAngle();
557554
Rotate(dAngle);
558555
ConstrainSinPhi();
559556
Alpha += dAngle;

‎GPU/GPUTracking/Refit/GPUTrackingRefit.cxx‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -398,11 +398,10 @@ GPUd() int32_t GPUTrackingRefit::RefitTrack(T& trkX, bool outward, bool resetCov
398398
trk.NormalizeAlpha(alpha);
399399
prop.SetAlpha(alpha);
400400
} else if constexpr (std::is_same_v<S, TrackParCov>) {
401-
static constexpr float kDeg2Rad = M_PI / 180.f;
402401
if (mPparam->rec.tpc.trackReferenceX <= 500) {
403402
if (prop->PropagateToXBxByBz(trk, mPparam->rec.tpc.trackReferenceX)) {
404403
if (CAMath::Abs(trk.getY()) > trk.getX() * CAMath::Tan(GPUTPCGeometry::kSectAngle() / 2.f)) {
405-
float newAlpha = trk.getAlpha() + CAMath::Round(CAMath::ATan2(trk.getY(), trk.getX()) / kDeg2Rad / 20.f) * GPUTPCGeometry::kSectAngle();
404+
float newAlpha = trk.getAlpha() + CAMath::Round(CAMath::ATan2(trk.getY(), trk.getX()) / CAMath::Deg2Rad() / 20.f) * GPUTPCGeometry::kSectAngle();
406405
GPUTPCGMTrackParam::NormalizeAlpha(newAlpha);
407406
trk.rotate(newAlpha) && prop->PropagateToXBxByBz(trk, mPparam->rec.tpc.trackReferenceX);
408407
}

‎GPU/GPUTracking/TRDTracking/GPUTRDInterfaces.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ class trackInterface<GPUTPCGMTrackParam> : public GPUTPCGMTrackParam
147147
GPUd() float getSnp() const { return GetSinPhi(); }
148148
GPUd() float getTgl() const { return GetDzDs(); }
149149
GPUd() float getQ2Pt() const { return GetQPt(); }
150-
GPUd() float getEta() const { return -CAMath::Log(CAMath::Tan(0.5f * (0.5f * M_PI - CAMath::ATan(getTgl())))); }
150+
GPUd() float getEta() const { return -CAMath::Log(CAMath::Tan(0.5f * (0.5f * CAMath::Pi() - CAMath::ATan(getTgl())))); }
151151
GPUd() float getPt() const { return CAMath::Abs(getQ2Pt()) > 0 ? CAMath::Abs(1.f / getQ2Pt()) : 99999.f; }
152152
GPUd() float getSigmaY2() const { return GetErr2Y(); }
153153
GPUd() float getSigmaZ2() const { return GetErr2Z(); }

‎GPU/GPUTracking/qa/GPUQA.cxx‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@ static const constexpr int32_t COLORS_HEX[COLORCOUNT] = {0xB03030, 0x00A000, 0x0
189189
static const constexpr int32_t CONFIG_DASHED_MARKERS = 0;
190190

191191
static const constexpr float AXES_MIN[5] = {-Y_MAX, -Z_MAX, 0.f, -ETA_MAX, PT_MIN};
192-
static const constexpr float AXES_MAX[5] = {Y_MAX, Z_MAX, 2.f * M_PI, ETA_MAX, PT_MAX};
192+
static const constexpr float AXES_MAX[5] = {Y_MAX, Z_MAX, 2.f * CAMath::Pi(), ETA_MAX, PT_MAX};
193193
static const constexpr int32_t AXIS_BINS[5] = {51, 51, 144, 31, 50};
194194
static const constexpr int32_t RES_AXIS_BINS[] = {1017, 113}; // Consecutive bin sizes, histograms are binned down until the maximum entry is 50, each bin size should evenly divide its predecessor.
195195
static const constexpr float RES_AXES[5] = {1., 1., 0.03, 0.03, 1.0};
@@ -1163,12 +1163,12 @@ void GPUQA::RunQA(bool matchOnly, const std::vector<o2::tpc::TrackTPC>* tracksEx
11631163
const mcInfo_t& info = GetMCTrack(i, iCol);
11641164
additionalMCParameters& mc2 = mMCParam[iCol][i];
11651165
mc2.pt = std::sqrt(info.pX * info.pX + info.pY * info.pY);
1166-
mc2.phi = M_PI + std::atan2(-info.pY, -info.pX);
1166+
mc2.phi = CAMath::Pi() + std::atan2(-info.pY, -info.pX);
11671167
float p = info.pX * info.pX + info.pY * info.pY + info.pZ * info.pZ;
11681168
if (p < 1e-18) {
11691169
mc2.theta = mc2.eta = 0.f;
11701170
} else {
1171-
mc2.theta = info.pZ == 0 ? (M_PI / 2) : (std::acos(info.pZ / std::sqrt(p)));
1171+
mc2.theta = info.pZ == 0 ? (CAMath::Pi() / 2) : (std::acos(info.pZ / std::sqrt(p)));
11721172
mc2.eta = -std::log(std::tan(0.5 * mc2.theta));
11731173
}
11741174
if (mConfig.writeMCLabels) {
@@ -1222,10 +1222,10 @@ void GPUQA::RunQA(bool matchOnly, const std::vector<o2::tpc::TrackTPC>* tracksEx
12221222
}
12231223

12241224
float alpha = std::atan2(info.y, info.x);
1225-
alpha /= M_PI / 9.f;
1225+
alpha /= CAMath::Pi() / 9.f;
12261226
alpha = std::floor(alpha);
1227-
alpha *= M_PI / 9.f;
1228-
alpha += M_PI / 18.f;
1227+
alpha *= CAMath::Pi() / 9.f;
1228+
alpha += CAMath::Pi() / 18.f;
12291229

12301230
float c = std::cos(alpha);
12311231
float s = std::sin(alpha);

‎GPU/GPUTracking/qa/genEvents.cxx‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -200,8 +200,8 @@ int32_t genEvents::GenerateEvent(const GPUParam& param, const char* filename)
200200
double eta = gRandom->Uniform(-1.5, 1.5);
201201

202202
double theta = 2 * std::atan(1. / exp(eta));
203-
double lambda = theta - M_PI / 2;
204-
// double theta = gRandom->Uniform(-60,60)*M_PI/180.;
203+
double lambda = theta - CAMath::Pi() / 2;
204+
// double theta = gRandom->Uniform(-60,60)*CAMath::Pi()/180.;
205205
double pt = .08 * std::pow(10, gRandom->Uniform(0, 2.2));
206206

207207
double q = 1.;

‎GPU/GPUTracking/qa/genEvents.h‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
#define GENEVENTS_H
1717

1818
#include "GPUCommonDef.h"
19-
#include <cmath>
19+
#include "GPUCommonMath.h"
2020

2121
namespace o2::gpu
2222
{
@@ -66,7 +66,7 @@ class genEvents
6666
uint32_t id;
6767
};
6868

69-
const double mTwoPi = 2 * M_PI;
69+
const double mTwoPi = 2 * CAMath::Pi();
7070
const double mSectorDAngle = mTwoPi / 18.;
7171
const double mSectorAngleOffset = mSectorDAngle / 2;
7272

0 commit comments

Comments
 (0)