Skip to content

Commit 71e1e50

Browse files
committed
Temporary protection against snp>1 for imported TPC tracks
1 parent a89b606 commit 71e1e50

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

GPU/GPUTracking/Merger/GPUTPCGMO2Output.cxx

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -117,13 +117,19 @@ GPUdii() void GPUTPCGMO2Output::Thread<GPUTPCGMO2Output::output>(int nBlocks, in
117117

118118
GPUTPCGMMerger::tmpSort* GPUrestrict() trackSort = merger.TrackSortO2();
119119
uint2* GPUrestrict() tmpData = merger.ClusRefTmp();
120+
float const SNPThresh = 0.999990f;
120121

121122
for (int iTmp = get_global_id(0); iTmp < nTracks; iTmp += get_global_size(0)) {
122123
TrackTPC oTrack;
123124
const int i = trackSort[iTmp].x;
124-
125+
auto snpIn = tracks[i].GetParam().GetSinPhi();
126+
if (snpIn > SNPThresh) {
127+
snpIn = SNPThresh;
128+
} else if (snpIn < -SNPThresh) {
129+
snpIn = -SNPThresh;
130+
}
125131
oTrack.set(tracks[i].GetParam().GetX(), tracks[i].GetAlpha(),
126-
{tracks[i].GetParam().GetY(), tracks[i].GetParam().GetZ(), tracks[i].GetParam().GetSinPhi(), tracks[i].GetParam().GetDzDs(), tracks[i].GetParam().GetQPt()},
132+
{tracks[i].GetParam().GetY(), tracks[i].GetParam().GetZ(), snpIn, tracks[i].GetParam().GetDzDs(), tracks[i].GetParam().GetQPt()},
127133
{tracks[i].GetParam().GetCov(0),
128134
tracks[i].GetParam().GetCov(1), tracks[i].GetParam().GetCov(2),
129135
tracks[i].GetParam().GetCov(3), tracks[i].GetParam().GetCov(4), tracks[i].GetParam().GetCov(5),
@@ -136,9 +142,15 @@ GPUdii() void GPUTPCGMO2Output::Thread<GPUTPCGMO2Output::output>(int nBlocks, in
136142
oTrack.setdEdx(tracksdEdx[i]);
137143
}
138144

145+
auto snpOut = outerPar.P[2];
146+
if (snpOut > SNPThresh) {
147+
snpOut = SNPThresh;
148+
} else if (snpOut < -SNPThresh) {
149+
snpOut = -SNPThresh;
150+
}
139151
oTrack.setOuterParam(o2::track::TrackParCov(
140152
outerPar.X, outerPar.alpha,
141-
{outerPar.P[0], outerPar.P[1], outerPar.P[2], outerPar.P[3], outerPar.P[4]},
153+
{outerPar.P[0], outerPar.P[1], snpOut, outerPar.P[3], outerPar.P[4]},
142154
{outerPar.C[0], outerPar.C[1], outerPar.C[2], outerPar.C[3], outerPar.C[4], outerPar.C[5],
143155
outerPar.C[6], outerPar.C[7], outerPar.C[8], outerPar.C[9], outerPar.C[10], outerPar.C[11],
144156
outerPar.C[12], outerPar.C[13], outerPar.C[14]}));

0 commit comments

Comments
 (0)