Project
Loading...
Searching...
No Matches
TPCFastTransform.h
Go to the documentation of this file.
1// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
2// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
3// All rights not expressly granted are reserved.
4//
5// This software is distributed under the terms of the GNU General Public
6// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
7//
8// In applying this license CERN does not waive the privileges and immunities
9// granted to it by virtue of its status as an Intergovernmental Organization
10// or submit itself to any jurisdiction.
11
16
17#ifndef ALICEO2_GPUCOMMON_TPCFASTTRANSFORMATION_TPCFASTTRANSFORM_H
18#define ALICEO2_GPUCOMMON_TPCFASTTRANSFORMATION_TPCFASTTRANSFORM_H
19
20#include "GPUCommonDef.h"
21
22#include "FlatObject.h"
23#include "TPCFastTransformGeo.h"
24#include "TPCFastSpaceChargeCorrection.h"
25#include "GPUCommonMath.h"
26#include "GPUDebugStreamer.h"
27
28#if !defined(GPUCA_GPUCODE)
29#include <string>
30#endif // !GPUCA_GPUCODE
31
32namespace o2::tpc
33{
34template <class T>
35class SpaceCharge;
36}
37
38namespace o2::gpu
39{
40
42struct TPCSlowSpaceChargeCorrection {
43
44#if !defined(GPUCA_GPUCODE) && !defined(GPUCA_STANDALONE)
46 ~TPCSlowSpaceChargeCorrection();
47
49 void getCorrections(const float gx, const float gy, const float gz, const int32_t sector, float& gdxC, float& gdyC, float& gdzC) const;
50
51 o2::tpc::SpaceCharge<float>* mCorr{nullptr};
52#else
53 ~TPCSlowSpaceChargeCorrection() = default;
54
56 GPUdi() void getCorrections(const float gx, const float gy, const float gz, const int32_t sector, float& gdxC, float& gdyC, float& gdzC) const
57 {
58 gdxC = 0;
59 gdyC = 0;
60 gdzC = 0;
61 }
62#endif
63
64 ClassDefNV(TPCSlowSpaceChargeCorrection, 2);
65};
66
95
96class TPCFastTransform : public FlatObject
97{
98 public:
99 static GPUglobalconstexpr() float DEFLUMI = -1e6f; // default value to check if member was set
100 static GPUglobalconstexpr() float DEFIDC = -1e6f; // default value to check if member was set
101
103
105 TPCFastTransform();
106
108 TPCFastTransform(const TPCFastTransform&) = delete;
109
111 TPCFastTransform& operator=(const TPCFastTransform&) = delete;
112
113 inline void destroy()
114 {
115 mCorrection.destroy();
116 FlatObject::destroy();
117 }
118
120#if !defined(GPUCA_GPUCODE)
121 ~TPCFastTransform()
122 {
123 delete mCorrectionSlow;
124 }
125#else
126 ~TPCFastTransform() = default;
127#endif
128
130
132
134 static constexpr size_t getClassAlignmentBytes() { return TPCFastSpaceChargeCorrection::getClassAlignmentBytes(); }
135
137 static constexpr size_t getBufferAlignmentBytes() { return TPCFastSpaceChargeCorrection::getBufferAlignmentBytes(); }
138
140
141 void cloneFromObject(const TPCFastTransform& obj, char* newFlatBufferPtr);
142
144
145 using FlatObject::releaseInternalBuffer;
146 void moveBufferTo(char* newBufferPtr);
147
149
150 void setActualBufferAddress(char* actualFlatBufferPtr);
151 void setFutureBufferAddress(char* futureFlatBufferPtr);
152
154
156 void startConstruction(const TPCFastSpaceChargeCorrection& correction);
157
162 void setCalibration(int64_t timeStamp, float t0, float vDrift);
163
165 void setLumi(float l) { mLumi = l; }
166 void setLumiError(float e) { mLumiError = e; }
167 void setLumiScaleFactor(float s) { mLumiScaleFactor = s; }
168 void setIDC(float l) { mIDC = l; }
169 void setIDCError(float e) { mIDCError = e; }
170 void setCTP2IDCFallBackThreshold(float v) { mCTP2IDCFallBackThreshold = v; }
172 void setTimeStamp(int64_t v) { mTimeStamp = v; }
173
175 GPUdi() const TPCFastSpaceChargeCorrection& getCorrection() const { return mCorrection; }
176
178 TPCFastSpaceChargeCorrection& getCorrection() { return mCorrection; }
179
181 void finishConstruction();
182
188 GPUd() void Transform(int32_t sector, int32_t row, float pad, float time, float& x, float& y, float& z, float vertexTime = 0, const TPCFastTransform* ref = nullptr, const TPCFastTransform* ref2 = nullptr, float scale = 0.f, float scale2 = 0.f, int32_t scaleMode = 0) const;
189 GPUd() void TransformXYZ(int32_t sector, int32_t row, float& x, float& y, float& z, const TPCFastTransform* ref = nullptr, const TPCFastTransform* ref2 = nullptr, float scale = 0.f, float scale2 = 0.f, int32_t scaleMode = 0) const;
190
192 GPUd() void TransformInTimeFrame(int32_t sector, int32_t row, float pad, float time, float& x, float& y, float& z, float maxTimeBin) const;
193 GPUd() void TransformInTimeFrame(int32_t sector, float time, float& z, float maxTimeBin) const;
194
196 GPUd() void InverseTransformInTimeFrame(int32_t sector, int32_t row, float /*x*/, float y, float z, float& pad, float& time, float maxTimeBin) const;
197 GPUd() float InverseTransformInTimeFrame(int32_t sector, float z, float maxTimeBin) const;
198
200 GPUd() void InverseTransformYZtoX(int32_t sector, int32_t row, float y, float z, float& x, const TPCFastTransform* ref = nullptr, const TPCFastTransform* ref2 = nullptr, float scale = 0.f, float scale2 = 0.f, int32_t scaleMode = 0) const;
201
203 GPUd() void InverseTransformYZtoNominalYZ(int32_t sector, int32_t row, float y, float z, float& ny, float& nz, const TPCFastTransform* ref = nullptr, const TPCFastTransform* ref2 = nullptr, float scale = 0.f, float scale2 = 0.f, int32_t scaleMode = 0) const;
204
206 GPUd() void InverseTransformXYZtoNominalXYZ(int32_t sector, int32_t row, float x, float y, float z, float& nx, float& ny, float& nz, const TPCFastTransform* ref = nullptr, const TPCFastTransform* ref2 = nullptr, float scale = 0.f, float scale2 = 0.f, int32_t scaleMode = 0) const;
207
209 GPUd() void TransformIdeal(int32_t sector, int32_t row, float pad, float time, float& x, float& y, float& z, float vertexTime) const;
210 GPUd() void TransformIdealZ(int32_t sector, float time, float& z, float vertexTime) const;
211
212 GPUd() void convPadTimeToLocal(int32_t sector, int32_t row, float pad, float time, float& y, float& z, float vertexTime) const;
213 GPUd() void convPadTimeToLocalInTimeFrame(int32_t sector, int32_t row, float pad, float time, float& y, float& z, float maxTimeBin) const;
214
215 GPUd() void convLocalToPadTime(int32_t sector, int32_t row, float y, float z, float& pad, float& time, float vertexTime) const;
216 GPUd() void convLocalToPadTimeInTimeFrame(int32_t sector, int32_t row, float y, float z, float& pad, float& time, float maxTimeBin) const;
217
218 GPUd() float convTimeToZinTimeFrame(int32_t sector, float time, float maxTimeBin) const;
219 GPUd() float convZtoTimeInTimeFrame(int32_t sector, float z, float maxTimeBin) const;
220 GPUd() float convDeltaTimeToDeltaZinTimeFrame(int32_t sector, float deltaTime) const;
221 GPUd() float convDeltaZtoDeltaTimeInTimeFrame(int32_t sector, float deltaZ) const;
222 GPUd() float convDeltaZtoDeltaTimeInTimeFrameAbs(float deltaZ) const;
223 GPUd() float convZOffsetToVertexTime(int32_t sector, float zOffset, float maxTimeBin) const;
224 GPUd() float convVertexTimeToZOffset(int32_t sector, float vertexTime, float maxTimeBin) const;
225
226 void setApplyCorrectionOn() { mApplyCorrection = 1; }
227 void setApplyCorrectionOff() { mApplyCorrection = 0; }
228 bool isCorrectionApplied() { return mApplyCorrection; }
229
231
233 GPUdi() const TPCFastTransformGeo& getGeometry() const { return mCorrection.getGeometry(); }
234
236 GPUdi() int64_t getTimeStamp() const { return mTimeStamp; }
237
239 GPUdi() float getVDrift() const { return mVdrift; }
240
242 GPUdi() float getT0() const { return mT0; }
243
245 GPUdi() float getLumi() const { return mLumi; }
246
247 GPUdi() float isLumiSet() const { return mLumi != DEFLUMI; }
248
250 GPUdi() float getLumiError() const { return mLumiError; }
251
253 GPUd() float getIDC() const;
254
255 GPUdi() bool isIDCSet() const { return mIDC != DEFIDC; }
256
258 GPUdi() float getIDCError() const { return mIDCError; }
259
260 GPUdi() float getCTP2IDCFallBackThreshold() const { return mCTP2IDCFallBackThreshold; }
261
263 GPUdi() float getLumiScaleFactor() const { return mLumiScaleFactor; }
264
266 GPUd() float getMaxDriftTime(int32_t sector, int32_t row, float pad) const;
267
269 GPUd() float getMaxDriftTime(int32_t sector, int32_t row) const;
270
272 GPUd() float getMaxDriftTime(int32_t sector) const;
273
274#if !defined(GPUCA_GPUCODE) && !defined(GPUCA_STANDALONE)
275
276 int32_t writeToFile(std::string outFName = "", std::string name = "");
277
278 void rectifyAfterReadingFromFile();
279
280 static TPCFastTransform* loadFromFile(std::string inpFName = "", std::string name = "");
281
283 void setSlowTPCSCCorrection(TFile& inpf);
284
286 const auto& getCorrectionSlow() const { return *mCorrectionSlow; }
287
288#endif // !GPUCA_GPUCODE
289
291 void print() const;
292
293 GPUd() float convDriftLengthToTime(float driftLength, float vertexTime) const;
294
295 private:
297 enum ConstructionExtraState : uint32_t {
298 CalibrationIsSet = 0x4
299 };
300
302
304
306
307 int64_t mTimeStamp;
308
314 TPCFastSpaceChargeCorrection mCorrection;
315
316 bool mApplyCorrection; // flag for applying correction
317
324 float mT0;
325 float mVdrift;
326
327 float mLumi;
328 float mLumiError;
329 float mLumiScaleFactor;
330
331 float mIDC;
332 float mIDCError;
333 float mCTP2IDCFallBackThreshold;
334
336 TPCSlowSpaceChargeCorrection* mCorrectionSlow{nullptr};
337
338 GPUd() void TransformLocal(int32_t sector, int32_t row, float& x, float& y, float& z, const TPCFastTransform* ref, const TPCFastTransform* ref2, float scale, float scale2, int32_t scaleMode) const;
339
340 ClassDefNV(TPCFastTransform, 5);
341};
342
343// =======================================================================
344// Inline implementations of some methods
345// =======================================================================
346
347// ----------------------------------------------------------------------
348
349GPUdi() void TPCFastTransform::convPadTimeToLocal(int32_t sector, int32_t row, float pad, float time, float& y, float& z, float vertexTime) const
350{
351 float l = (time - mT0 - vertexTime) * mVdrift; // drift length [cm]
352 getGeometry().convPadDriftLengthToLocal(sector, row, pad, l, y, z);
353}
354
355GPUdi() void TPCFastTransform::convPadTimeToLocalInTimeFrame(int32_t sector, int32_t row, float pad, float time, float& y, float& z, float maxTimeBin) const
356{
357 float l = getGeometry().getTPCzLength() + (time - mT0 - maxTimeBin) * mVdrift; // drift length [cm]
358 getGeometry().convPadDriftLengthToLocal(sector, row, pad, l, y, z);
359}
360
361// ----------------------------------------------------------------------
362
363GPUdi() float TPCFastTransform::convZOffsetToVertexTime(int32_t sector, float zOffset, float maxTimeBin) const
364{
365 if (sector < getGeometry().getNumberOfSectorsA()) {
366 return maxTimeBin - (getGeometry().getTPCzLength() + zOffset) / mVdrift;
367 } else {
368 return maxTimeBin - (getGeometry().getTPCzLength() - zOffset) / mVdrift;
369 }
370}
371
372GPUdi() float TPCFastTransform::convVertexTimeToZOffset(int32_t sector, float vertexTime, float maxTimeBin) const
373{
374 if (sector < getGeometry().getNumberOfSectorsA()) {
375 return (maxTimeBin - vertexTime) * mVdrift - getGeometry().getTPCzLength();
376 } else {
377 return -((maxTimeBin - vertexTime) * mVdrift - getGeometry().getTPCzLength());
378 }
379}
380
381GPUdi() float TPCFastTransform::convDriftLengthToTime(float driftLength, float vertexTime) const
382{
383 return (mT0 + vertexTime + driftLength / mVdrift);
384}
385
386// ----------------------------------------------------------------------
387
388GPUdi() void TPCFastTransform::convLocalToPadTime(int32_t sector, int32_t row, float y, float z, float& pad, float& time, float vertexTime) const
389{
390 float l;
391 getGeometry().convLocalToPadDriftLength(sector, row, y, z, pad, l);
392 time = convDriftLengthToTime(l, vertexTime);
393}
394
395GPUdi() void TPCFastTransform::convLocalToPadTimeInTimeFrame(int32_t sector, int32_t row, float y, float z, float& pad, float& time, float maxTimeBin) const
396{
397 float l;
398 getGeometry().convLocalToPadDriftLength(sector, row, y, z, pad, l);
399 time = convDriftLengthToTime(l, maxTimeBin);
400}
401
402// ----------------------------------------------------------------------
403
404GPUdi() void TPCFastTransform::TransformLocal(int32_t sector, int32_t row, float& x, float& y, float& z, const TPCFastTransform* ref, const TPCFastTransform* ref2, float scale, float scale2, int32_t scaleMode) const
405{
406 GPUCA_RTC_SPECIAL_CODE(ref2 = nullptr; scale2 = 0.f;);
407
408 if (!mApplyCorrection) {
409 return;
410 }
411
412 float dx = 0.f, dy = 0.f, dz = 0.f;
413
414 if ((scale >= 0.f) || (scaleMode == 1) || (scaleMode == 2)) {
415#ifndef GPUCA_GPUCODE
416 if (mCorrectionSlow) {
417 float gx, gy, gz;
418 getGeometry().convLocalToGlobal(sector, x, y, z, gx, gy, gz);
419 float gdxC, gdyC, gdzC;
420 mCorrectionSlow->getCorrections(gx, gy, gz, sector, gdxC, gdyC, gdzC);
421 getGeometry().convGlobalToLocal(sector, gdxC, gdyC, gdzC, dx, dy, dz);
422 } else
423#endif // GPUCA_GPUCODE
424 {
425 mCorrection.getCorrectionLocal(sector, row, y, z, dx, dy, dz);
426 if (ref) {
427 if ((scale > 0.f) && (scaleMode == 0)) { // scaling was requested
428 float dx1, dy1, dz1;
429 ref->mCorrection.getCorrectionLocal(sector, row, y, z, dx1, dy1, dz1);
430 dx = (dx - dx1) * scale + dx1;
431 dy = (dy - dy1) * scale + dy1;
432 dz = (dz - dz1) * scale + dz1;
433 } else if ((scale != 0.f) && ((scaleMode == 1) || (scaleMode == 2))) {
434 float dx1, dy1, dz1;
435 ref->mCorrection.getCorrectionLocal(sector, row, y, z, dx1, dy1, dz1);
436 dx = dx1 * scale + dx;
437 dy = dy1 * scale + dy;
438 dz = dz1 * scale + dz;
439 }
440 }
441 if (ref2 && (scale2 != 0)) {
442 float dx1, dy1, dz1;
443 ref2->mCorrection.getCorrectionLocal(sector, row, y, z, dx1, dy1, dz1);
444 dx = dx1 * scale2 + dx;
445 dy = dy1 * scale2 + dy;
446 dz = dz1 * scale2 + dz;
447 }
448 }
449 }
450
451 GPUCA_DEBUG_STREAMER_CHECK(if (o2::utils::DebugStreamer::checkStream(o2::utils::StreamFlags::streamFastTransform)) {
452 float lx = x, ly = y, lz = z;
453
454 float gx, gy, gz;
455 getGeometry().convLocalToGlobal(sector, lx, ly, lz, gx, gy, gz);
456
457 float lxT = lx + dx;
458 float lyT = ly + dy;
459 float lzT = lz + dz;
460
461 float invYZtoXScaled;
462 InverseTransformYZtoX(sector, row, lyT, lzT, invYZtoXScaled, ref, ref2, scale, scale2, scaleMode);
463
464 float invYZtoX;
465 InverseTransformYZtoX(sector, row, lyT, lzT, invYZtoX);
466
467 float YZtoNominalY;
468 float YZtoNominalZ;
469 InverseTransformYZtoNominalYZ(sector, row, lyT, lzT, YZtoNominalY, YZtoNominalZ);
470
471 float YZtoNominalYScaled;
472 float YZtoNominalZScaled;
473 InverseTransformYZtoNominalYZ(sector, row, lyT, lzT, YZtoNominalYScaled, YZtoNominalZScaled, ref, ref2, scale, scale2, scaleMode);
474
475 float dxRef = 0.f, dyRef = 0.f, dzRef = 0.f;
476 if (ref) {
477 ref->mCorrection.getCorrectionLocal(sector, row, y, z, dxRef, dyRef, dzRef);
478 }
479
480 float dxRef2 = 0.f, dyRef2 = 0.f, dzRef2 = 0.f;
481 if (ref2) {
482 ref2->mCorrection.getCorrectionLocal(sector, row, y, z, dxRef2, dyRef2, dzRef2);
483 }
484
485 float dxOrig, dyOrig, dzOrig;
486 mCorrection.getCorrectionLocal(sector, row, y, z, dxOrig, dyOrig, dzOrig);
487
488 o2::utils::DebugStreamer::instance()->getStreamer("debug_fasttransform", "UPDATE") << o2::utils::DebugStreamer::instance()->getUniqueTreeName("tree_Transform").data()
489 // corrections in x, u, v
490 << "dxOrig=" << dxOrig
491 << "dyOrig=" << dyOrig
492 << "dzOrig=" << dzOrig
493 << "dxRef=" << dxRef
494 << "dyRef=" << dyRef
495 << "dzRef=" << dzRef
496 << "dxRef2=" << dxRef2
497 << "dyRef2=" << dyRef2
498 << "dzRef2=" << dzRef2
499 << "dx=" << dx
500 << "dy=" << dy
501 << "dz=" << dz
502 << "row=" << row
503 << "sector=" << sector
504 << "scale=" << scale
505 << "scale2=" << scale2
506 // original local coordinates
507 << "ly=" << ly
508 << "lz=" << lz
509 << "lx=" << lx
510 // corrected local coordinated
511 << "lxT=" << lxT
512 << "lyT=" << lyT
513 << "lzT=" << lzT
514 // global uncorrected coordinates
515 << "gx=" << gx
516 << "gy=" << gy
517 << "gz=" << gz
518 // some transformations which are applied
519 << "invYZtoX=" << invYZtoX
520 << "invYZtoXScaled=" << invYZtoXScaled
521 << "YZtoNominalY=" << YZtoNominalY
522 << "YZtoNominalYScaled=" << YZtoNominalYScaled
523 << "YZtoNominalZ=" << YZtoNominalZ
524 << "YZtoNominalZScaled=" << YZtoNominalZScaled
525 << "scaleMode=" << scaleMode
526 << "\n";
527 })
528
529 x += dx;
530 y += dy;
531 z += dz;
532}
533
534GPUdi() void TPCFastTransform::TransformXYZ(int32_t sector, int32_t row, float& x, float& y, float& z, const TPCFastTransform* ref, const TPCFastTransform* ref2, float scale, float scale2, int32_t scaleMode) const
535{
536
537 TransformLocal(sector, row, x, y, z, ref, ref2, scale, scale2, scaleMode);
538}
539
540GPUdi() void TPCFastTransform::Transform(int32_t sector, int32_t row, float pad, float time, float& x, float& y, float& z, float vertexTime, const TPCFastTransform* ref, const TPCFastTransform* ref2, float scale, float scale2, int32_t scaleMode) const
541{
547
548 const TPCFastTransformGeo::RowInfo& rowInfo = getGeometry().getRowInfo(row);
549
550 x = rowInfo.x;
551 convPadTimeToLocal(sector, row, pad, time, y, z, vertexTime);
552 TransformLocal(sector, row, x, y, z, ref, ref2, scale, scale2, scaleMode);
553}
554
555GPUdi() void TPCFastTransform::TransformInTimeFrame(int32_t sector, float time, float& z, float maxTimeBin) const
556{
557 float l = (time - mT0 - maxTimeBin) * mVdrift; // drift length cm
558 z = getGeometry().convDriftLengthToZ(sector, l);
559}
560
561GPUdi() void TPCFastTransform::TransformInTimeFrame(int32_t sector, int32_t row, float pad, float time, float& x, float& y, float& z, float maxTimeBin) const
562{
568
569 const TPCFastTransformGeo::RowInfo& rowInfo = getGeometry().getRowInfo(row);
570 x = rowInfo.x;
571 convPadTimeToLocalInTimeFrame(sector, row, pad, time, y, z, maxTimeBin);
572}
573
574GPUdi() void TPCFastTransform::InverseTransformInTimeFrame(int32_t sector, int32_t row, float /*x*/, float y, float z, float& pad, float& time, float maxTimeBin) const
575{
577 convLocalToPadTimeInTimeFrame(sector, row, y, z, pad, time, maxTimeBin);
578}
579
580GPUdi() float TPCFastTransform::InverseTransformInTimeFrame(int32_t sector, float z, float maxTimeBin) const
581{
582 float pad, time;
583 InverseTransformInTimeFrame(sector, 0, 0, 0, z, pad, time, maxTimeBin);
584 return time;
585}
586
587GPUdi() void TPCFastTransform::TransformIdealZ(int32_t sector, float time, float& z, float vertexTime) const
588{
595
596 float l = (time - mT0 - vertexTime) * mVdrift; // drift length cm
597 z = getGeometry().convDriftLengthToZ(sector, l);
598}
599
600GPUdi() void TPCFastTransform::TransformIdeal(int32_t sector, int32_t row, float pad, float time, float& x, float& y, float& z, float vertexTime) const
601{
608
609 x = getGeometry().getRowInfo(row).x;
610 float driftLength = (time - mT0 - vertexTime) * mVdrift; // drift length cm
611 getGeometry().convPadDriftLengthToLocal(sector, row, pad, driftLength, y, z);
612}
613
614GPUdi() float TPCFastTransform::convTimeToZinTimeFrame(int32_t sector, float time, float maxTimeBin) const
615{
622
623 float v = (time - mT0 - maxTimeBin) * mVdrift; // drift length cm
624 float z = (sector < getGeometry().getNumberOfSectorsA()) ? -v : v;
625 return z;
626}
627
628GPUdi() float TPCFastTransform::convZtoTimeInTimeFrame(int32_t sector, float z, float maxTimeBin) const
629{
631 float v = (sector < getGeometry().getNumberOfSectorsA()) ? -z : z;
632 return mT0 + maxTimeBin + v / mVdrift;
633}
634
635GPUdi() float TPCFastTransform::convDeltaTimeToDeltaZinTimeFrame(int32_t sector, float deltaTime) const
636{
637 float deltaZ = deltaTime * mVdrift;
638 return sector < getGeometry().getNumberOfSectorsA() ? -deltaZ : deltaZ;
639}
640
641GPUdi() float TPCFastTransform::convDeltaZtoDeltaTimeInTimeFrameAbs(float deltaZ) const
642{
643 return deltaZ / mVdrift;
644}
645
646GPUdi() float TPCFastTransform::convDeltaZtoDeltaTimeInTimeFrame(int32_t sector, float deltaZ) const
647{
648 float deltaT = deltaZ / mVdrift;
649 return sector < getGeometry().getNumberOfSectorsA() ? -deltaT : deltaT;
650}
651
652GPUdi() float TPCFastTransform::getMaxDriftTime(int32_t sector, int32_t row, float pad) const
653{
655 return convDriftLengthToTime(getGeometry().getTPCzLength(), 0.f);
656}
657
658GPUdi() float TPCFastTransform::getMaxDriftTime(int32_t sector, int32_t row) const
659{
661 return convDriftLengthToTime(getGeometry().getTPCzLength(), 0.f);
662}
663
664GPUdi() float TPCFastTransform::getMaxDriftTime(int32_t sector) const
665{
667 return convDriftLengthToTime(getGeometry().getTPCzLength(), 0.f);
668}
669
670GPUdi() void TPCFastTransform::InverseTransformYZtoX(int32_t sector, int32_t row, float realY, float realZ, float& realX, const TPCFastTransform* ref, const TPCFastTransform* ref2, float scale, float scale2, int32_t scaleMode) const
671{
672 GPUCA_RTC_SPECIAL_CODE(ref2 = nullptr; scale2 = 0.f;);
674
675 float dx = 0.f;
676
677 if ((scale >= 0.f) || (scaleMode == 1) || (scaleMode == 2)) {
678 dx = mCorrection.getCorrectionXatRealYZ(sector, row, realY, realZ);
679 if (ref) { // scaling was requested
680 if (scaleMode == 0 && scale > 0.f) {
681 float dxref = ref->mCorrection.getCorrectionXatRealYZ(sector, row, realY, realZ);
682 dx = (dx - dxref) * scale + dxref;
683 } else if ((scale != 0) && ((scaleMode == 1) || (scaleMode == 2))) {
684 float dxref = ref->mCorrection.getCorrectionXatRealYZ(sector, row, realY, realZ);
685 dx = dxref * scale + dx;
686 }
687 }
688 if (ref2 && (scale2 != 0)) {
689 float dxref = ref2->mCorrection.getCorrectionXatRealYZ(sector, row, realY, realZ);
690 dx = dxref * scale2 + dx;
691 }
692 }
693
694 realX = mCorrection.getGeometry().getRowInfo(row).x + dx;
695
696 GPUCA_DEBUG_STREAMER_CHECK(if (o2::utils::DebugStreamer::checkStream(o2::utils::StreamFlags::streamFastTransform)) {
697 o2::utils::DebugStreamer::instance()->getStreamer("debug_fasttransform", "UPDATE") << o2::utils::DebugStreamer::instance()->getUniqueTreeName("tree_InverseTransformYZtoX").data()
698 << "sector=" << sector
699 << "row=" << row
700 << "scale=" << scale
701 << "y=" << realY
702 << "z=" << realZ
703 << "x=" << realX
704 << "\n";
705 })
706}
707
708GPUdi() void TPCFastTransform::InverseTransformYZtoNominalYZ(int32_t sector, int32_t row, float realY, float realZ, float& measuredY, float& measuredZ, const TPCFastTransform* ref, const TPCFastTransform* ref2, float scale, float scale2, int32_t scaleMode) const
709{
711
712 GPUCA_RTC_SPECIAL_CODE(ref2 = nullptr; scale2 = 0.f;);
713
714 float dy = 0;
715 float dz = 0;
716
717 if ((scale >= 0.f) || (scaleMode == 1) || (scaleMode == 2)) {
718 mCorrection.getCorrectionYZatRealYZ(sector, row, realY, realZ, dy, dz);
719
720 if (ref) { // scaling was requested
721 if (scaleMode == 0 && scale > 0.f) {
722 float dy1, dz1;
723 ref->mCorrection.getCorrectionYZatRealYZ(sector, row, realY, realZ, dy1, dz1);
724 dy = (dy - dy1) * scale + dy1;
725 dz = (dz - dz1) * scale + dz1;
726 } else if ((scale != 0) && ((scaleMode == 1) || (scaleMode == 2))) {
727 float dy1, dz1;
728 ref->mCorrection.getCorrectionYZatRealYZ(sector, row, realY, realZ, dy1, dz1);
729 dy = dy1 * scale + dy;
730 dz = dz1 * scale + dz;
731 }
732 if (ref2 && (scale2 != 0)) {
733 float dy1, dz1;
734 ref2->mCorrection.getCorrectionYZatRealYZ(sector, row, realY, realZ, dy1, dz1);
735 dy = dy1 * scale2 + dy;
736 dz = dz1 * scale2 + dz;
737 }
738 }
739 }
740
741 measuredY = realY - dy;
742 measuredZ = realZ - dz;
743
744 GPUCA_DEBUG_STREAMER_CHECK(if (o2::utils::DebugStreamer::checkStream(o2::utils::StreamFlags::streamFastTransform)) {
745 o2::utils::DebugStreamer::instance()->getStreamer("debug_fasttransform", "UPDATE") << o2::utils::DebugStreamer::instance()->getUniqueTreeName("tree_InverseTransformYZtoNominalYZ").data()
746 << "sector=" << sector
747 << "row=" << row
748 << "scale=" << scale
749 << "real y=" << realY
750 << "real z=" << realZ
751 << "measured y=" << measuredY
752 << "measured z=" << measuredZ
753 << "\n";
754 })
755}
756
757GPUdi() void TPCFastTransform::InverseTransformXYZtoNominalXYZ(int32_t sector, int32_t row, float x, float y, float z, float& nx, float& ny, float& nz, const TPCFastTransform* ref, const TPCFastTransform* ref2, float scale, float scale2, int32_t scaleMode) const
758{
760 int32_t row2 = row + 1;
761 if (row2 >= getGeometry().getNumberOfRows()) {
762 row2 = row - 1;
763 }
764 float nx1, ny1, nz1; // nominal coordinates for row
765 float nx2, ny2, nz2; // nominal coordinates for row2
766 nx1 = getGeometry().getRowInfo(row).x;
767 nx2 = getGeometry().getRowInfo(row2).x;
768 InverseTransformYZtoNominalYZ(sector, row, y, z, ny1, nz1, ref, ref2, scale, scale2, scaleMode);
769 InverseTransformYZtoNominalYZ(sector, row2, y, z, ny2, nz2, ref, ref2, scale, scale2, scaleMode);
770 float c1 = (nx2 - nx) / (nx2 - nx1);
771 float c2 = (nx - nx1) / (nx2 - nx1);
772 nx = x;
773 ny = (ny1 * c1 + ny2 * c2);
774 nz = (nz1 * c1 + nz2 * c2);
775}
776
777} // namespace o2::gpu
778
779#endif
FlatObject.h
Definition of FlatObject class.
GPUCA_DEBUG_STREAMER_CHECK
#define GPUCA_DEBUG_STREAMER_CHECK(...)
Definition GPUCommonDef.h:75
GPUCA_RTC_SPECIAL_CODE
#define GPUCA_RTC_SPECIAL_CODE(...)
Definition GPUCommonDef.h:79
sector
configStandalone Light OpenGL display display_light Apply smoothing to points Enable Z buffer Highlight links Highlight seeds Highlight sector tracks Highlight final tracks Show individual sector
Definition GPUSettingsListDisplay.h:44
s
Timeframe settings Average distance in cm of events merged into Randomize distance around average distance of merged events Simulate correct bunch interactions instead of placing only the average number of events A value[n] sets TFlen for[n] collisions in Number of bunches per trainsort Number of bunch trains Instantaneous interaction rate Do not simulate border Total number of collisions to be placed in the interior of all time Do not select random but walk over array of events in stride steps h Event generator settings h configStandalone Use CPU for GPU Force usage of the specified GPU device no CPU fallback Continue processing after an error s
Definition GPUSettingsListStandalone.h:55
e
Timeframe settings Average distance in cm of events merged into Randomize distance around average distance of merged events Simulate correct bunch interactions instead of placing only the average number of events A value[n] sets TFlen for[n] collisions in Number of bunches per trainsort Number of bunch trains Instantaneous interaction rate Do not simulate border Total number of collisions to be placed in the interior of all time Do not select random but walk over array of events in stride steps h Event generator settings h configStandalone Use CPU for GPU Force usage of the specified GPU device no CPU fallback Continue processing after an error First event to Number of events to Number of iterations to Number of iterations to Number of initial iterations excluded from const char e
Definition GPUSettingsListStandalone.h:60
int32_t
configStandalone Reconstruction rec_tpc multiplier for the combined cluster track error during track following area in cm for the search of for z only used if decrease removal by when above this local occupancy rowx Shift of TPC revert cluster flag edge bit distance to edge exceeds this ZOffset for secondary multiple of radius max P of electron K BB bands crossing max P of electron p BB bands crossing max P of electron d BB bands crossing max P of electron t BB bands crossing min accepted dEdxMaxTPC of the track Additive extra cluster error for Y2 if edge flag set Additive extra cluster error for Y2 if single set Additive extra cluster error for Y2 if or single set Additive extra cluster error for Z2 if or single set Extra errors to add to track when crossing depending on addErrorsCECrossing Inner portion of y range in sector that is not used in searching for extrapolated track candidates factor2YS for track merging factor2K for track merging Margin in cm of Y position when rejecting edge clusters based on uncorrected track Y Max fraction of shared clusters for tracklet Except pad from cluster finding if total number of charges in a fragment is above this Except pad from cluster finding if number of consecutive charges in a fragment is above this Threshold where a timebin is considered disabling the noisy pad check for that pad Treat signal above this minimum as saturated Smoothing factor for the exponential Highly Ionising Particle tail filter Number of timebins to use for the averaging Add additional custom track errors when crossing relative as fraction of Do N fit passes in final fit of< 0:reject at max -n hits") AddOptionRTC(rejectIFCLowRadiusCluster, uint8_t, 1, "", 0, "Reject clusters that get the IFC mask error during refit") AddOptionRTC(dEdxTruncLow, uint8_t, 2, "", 0, "Low truncation threshold, fraction of 128") AddOptionRTC(dEdxTruncHigh, uint8_t, 77, "", 0, "High truncation threshold, fraction of 128") AddOptionRTC(extrapolationTracking, int8_t, 1, "", 0, "Enable Extrapolation Tracking(prolong tracks to adjacent sectors to find short segments)") AddOptionRTC(disableRefitAttachment, uint8_t, 0, "", 0, "Bitmask to disable certain attachment steps during refit(1:attachment, 2:propagation, 4:loop following, 8:mirroring)") AddOptionRTC(rejectionStrategy, uint8_t, o2::gpu::GPUSettings::RejectionStrategyA, "", 0, "Enable rejection of TPC clusters for compression(0=no, 1=strategy A, 2=strategy B)") AddOptionRTC(mergeLoopersAfterburner, uint8_t, 1, "", 0, "Run afterburner for additional looper merging") AddOptionRTC(compressionTypeMask, uint8_t, o2::gpu::GPUSettings::CompressionFull, "", 0, "TPC Compression mode bits(1=truncate charge/width LSB, 2=differences, 4=track-model)") AddOptionRTC(compressionSortOrder, uint8_t, o2::gpu::GPUSettings::SortTime, "", 0, "Sort order of TPC compression(0=time, 1=pad, 2=Z-time-pad, 3=Z-pad-time, 4=no sorting(use incoming order))") AddOptionRTC(sigBitsCharge, uint8_t, 4, "", 0, "Number of significant bits for TPC cluster charge in compression mode 1") AddOptionRTC(sigBitsWidth, uint8_t, 3, "", 0, "Number of significant bits for TPC cluster width in compression mode 1") AddOptionRTC(dropLoopers, uint8_t, 0, "", 0, "Drop looping tracks starting from second loop") AddOptionRTC(mergerInterpolateErrors, uint8_t, 1, "", 0, "Use interpolation instead of extrapolation for chi2 based cluster rejection") AddOptionRTC(mergerInterpolateRejectAlsoOnCurrentPosition, uint8_t, 1, "", 0, "When using mergerInterpolateErrors, reject based on chi2 twice computed with interpolated and current track position starting from NDF > mergerNonInterpolateRejectMinNDF Merge tracks accross the central electrode Enable PID response Add sub threshold clusters in TPC dEdx computation OR mask of TPC flags that will reject the cluster in alternative dEdx Reject edge clusters based on uncorrected track Y during track fit Increase decrease removal by when on this number of pads from the edge h configStandalone Reconstruction rec_trd Max chi2 for TRD tracklets to be matched to a track Chi2 cut for strict matching mode Number of sigmas for ITS TPC track time error estimate Addition to search road around track prolongation along Y in cm Optimal tracklet rphi resolution in maximum chi2 per attached tracklet for TRD tracks take into account the full likelihood of the slope including exponential tails Tracks with less attached tracklets are discarded after the tracking Post trigger Pile up integration time in BCs fits all iterations but the last one in projections Use material lookup table for TPC refit Apply workarounds to avoid cyclic merge should not be needed tpc dyn configStandalone Processing proc_rtc Replace constant variables static by constexpr expressions Compile RTC in deterministic with NO_FAST_MATH flags and GPUCA_DETERMINISTIC_MODE define Use RTC to optimize GPU code h configStandalone Processing proc_rtctech Use a file lock to serialize access to the cache folder int32_t
Definition GPUSettingsList.h:235
TPCFastSpaceChargeCorrection.h
Definition of TPCFastSpaceChargeCorrection class.
TPCFastTransformGeo.h
Definition of TPCFastTransformGeo class.
int64_t
o2::gpu::FlatObject::releaseInternalBuffer
char * releaseInternalBuffer()
_____________ Methods for making the data buffer external __________________________
Definition FlatObject.h:538
o2::gpu::FlatObject::destroy
void destroy()
_______________ Utilities _______________________________________________
Definition FlatObject.h:361
o2::gpu::FlatObject::startConstruction
void startConstruction()
_____________ Construction _________
Definition FlatObject.h:354
o2::gpu::TPCFastSpaceChargeCorrection::getBufferAlignmentBytes
static constexpr size_t getBufferAlignmentBytes()
_____________ FlatObject functionality, see FlatObject class for description ____________
Definition FlatObject.h:195
o2::gpu::TPCFastSpaceChargeCorrection::getClassAlignmentBytes
static constexpr size_t getClassAlignmentBytes()
GPUCA_GPUCODE.
Definition FlatObject.h:192
o2::gpu::TPCFastTransform::setActualBufferAddress
void setActualBufferAddress(char *actualFlatBufferPtr)
Moving the class with its external buffer to another location.
Definition TPCFastTransform.cxx:74
o2::gpu::TPCFastTransform::const
int32_t float float float float float float maxTimeBin const
Definition TPCFastTransform.h:192
o2::gpu::TPCFastTransform::GPUdi
GPUdi() float getVDrift() const
Return mVDrift in cm / time bin.
Definition TPCFastTransform.h:239
o2::gpu::TPCFastTransform::GPUdi
GPUdi() const TPCFastTransformGeo &getGeometry() const
_______________ Utilities _______________________________________________
Definition TPCFastTransform.h:233
o2::gpu::TPCFastTransform::scaleMode
int32_t float float float float float float const TPCFastTransform const TPCFastTransform float float int32_t scaleMode
Definition TPCFastTransform.h:188
o2::gpu::TPCFastTransform::setLumi
void setLumi(float l)
Set Lumi info.
Definition TPCFastTransform.h:165
o2::gpu::TPCFastTransform::GPUdi
GPUdi() float getLumi() const
Return map lumi.
Definition TPCFastTransform.h:245
o2::gpu::TPCFastTransform::finishConstruction
void finishConstruction()
Finishes initialization: puts everything to the flat buffer, releases temporary memory.
Definition TPCFastTransform.cxx:129
o2::gpu::TPCFastTransform::moveBufferTo
void moveBufferTo(char *newBufferPtr)
Definition TPCFastTransform.cxx:67
o2::gpu::TPCFastTransform::GPUd
GPUd() float getIDC() const
Return map lumi.
o2::gpu::TPCFastTransform::setTimeStamp
void setTimeStamp(int64_t v)
Sets the time stamp of the current calibaration.
Definition TPCFastTransform.h:172
o2::gpu::TPCFastTransform::cloneFromObject
void cloneFromObject(const TPCFastTransform &obj, char *newFlatBufferPtr)
Construction interface.
Definition TPCFastTransform.cxx:43
o2::gpu::TPCFastTransform::getBufferAlignmentBytes
static constexpr size_t getBufferAlignmentBytes()
Gives minimal alignment in bytes required for the flat buffer.
Definition TPCFastTransform.h:137
o2::gpu::TPCFastTransform::writeToFile
int32_t writeToFile(std::string outFName="", std::string name="")
Definition TPCFastTransform.cxx:159
o2::gpu::TPCFastTransform::setCTP2IDCFallBackThreshold
void setCTP2IDCFallBackThreshold(float v)
Definition TPCFastTransform.h:170
o2::gpu::TPCFastTransform::GPUdi
GPUdi() float getCTP2IDCFallBackThreshold() const
Definition TPCFastTransform.h:260
o2::gpu::TPCFastTransform::z
int32_t float float float float float & z
Definition TPCFastTransform.h:188
o2::gpu::TPCFastTransform::vertexTime
int32_t float float float float float float vertexTime
Definition TPCFastTransform.h:188
o2::gpu::TPCFastTransform::scale
int32_t float float float float float float const TPCFastTransform const TPCFastTransform float scale
Definition TPCFastTransform.h:188
o2::gpu::TPCFastTransform::setSlowTPCSCCorrection
void setSlowTPCSCCorrection(TFile &inpf)
setting the reference corrections
Definition TPCFastTransform.cxx:235
o2::gpu::TPCFastTransform::GPUdi
GPUdi() const TPCFastSpaceChargeCorrection &getCorrection() const
Gives a reference for external initialization of TPC corrections.
Definition TPCFastTransform.h:175
o2::gpu::TPCFastTransform::getClassAlignmentBytes
static constexpr size_t getClassAlignmentBytes()
_____________ FlatObject functionality, see FlatObject class for description ____________
Definition TPCFastTransform.h:134
o2::gpu::TPCFastTransform::GPUglobalconstexpr
static GPUglobalconstexpr() float DEFLUMI
o2::gpu::TPCFastTransform::getCorrection
TPCFastSpaceChargeCorrection & getCorrection()
Gives a reference for external initialization of TPC corrections.
Definition TPCFastTransform.h:178
o2::gpu::TPCFastTransform::y
int32_t float float float float & y
Definition TPCFastTransform.h:188
o2::gpu::TPCFastTransform::nz
int32_t float float float float & nz
Definition TPCFastTransform.h:203
o2::gpu::TPCFastTransform::setCalibration
void setCalibration(int64_t timeStamp, float t0, float vDrift)
Definition TPCFastTransform.cxx:116
o2::gpu::TPCFastTransform::GPUdi
GPUdi() float getIDCError() const
Return map lumi error.
Definition TPCFastTransform.h:258
o2::gpu::TPCFastTransform::TPCFastTransform
TPCFastTransform()
_____________ Constructors / destructors __________________________
Definition TPCFastTransform.cxx:37
o2::gpu::TPCFastTransform::loadFromFile
static TPCFastTransform * loadFromFile(std::string inpFName="", std::string name="")
Definition TPCFastTransform.cxx:193
o2::gpu::TPCFastTransform::GPUdi
GPUdi() float isLumiSet() const
Definition TPCFastTransform.h:247
o2::gpu::TPCFastTransform::x
int32_t float float float & x
Definition TPCFastTransform.h:188
o2::gpu::TPCFastTransform::setFutureBufferAddress
void setFutureBufferAddress(char *futureFlatBufferPtr)
Definition TPCFastTransform.cxx:81
o2::gpu::TPCFastTransform::scale2
int32_t float float float float float float const TPCFastTransform const TPCFastTransform float float scale2
Definition TPCFastTransform.h:188
o2::gpu::TPCFastTransform::ref2
int32_t float float float float float float const TPCFastTransform const TPCFastTransform * ref2
Definition TPCFastTransform.h:188
o2::gpu::TPCFastTransform::print
void print() const
Print method.
Definition TPCFastTransform.cxx:141
o2::gpu::TPCFastTransform::ny
int32_t float float float & ny
Definition TPCFastTransform.h:203
o2::gpu::TPCFastTransform::GPUdi
GPUdi() float getT0() const
Return T0 in time bin units.
Definition TPCFastTransform.h:242
o2::gpu::TPCFastTransform::GPUdi
GPUdi() int64_t getTimeStamp() const
Gives the time stamp of the current calibaration parameters.
Definition TPCFastTransform.h:236
o2::gpu::TPCFastTransform::getCorrectionSlow
const auto & getCorrectionSlow() const
Definition TPCFastTransform.h:286
o2::gpu::TPCFastTransform::GPUdi
GPUdi() float getLumiScaleFactor() const
Return map user defined lumi scale factor.
Definition TPCFastTransform.h:263
o2::gpu::TPCFastTransform::GPUd
GPUd() void Transform(int32_t sector
o2::gpu::TPCFastTransform::GPUdi
GPUdi() float getLumiError() const
Return map lumi error.
Definition TPCFastTransform.h:250
o2::gpu::TPCFastTransform::nx
int32_t float float float float & nx
Definition TPCFastTransform.h:206
x
GLint GLenum GLint x
Definition glcorearb.h:403
v
const GLdouble * v
Definition glcorearb.h:832
name
GLuint const GLchar * name
Definition glcorearb.h:781
f
GLdouble f
Definition glcorearb.h:310
y
GLint y
Definition glcorearb.h:270
void
typedef void(APIENTRYP PFNGLCULLFACEPROC)(GLenum mode)
t0
GLuint GLfloat GLfloat GLfloat GLfloat GLfloat GLfloat GLfloat t0
Definition glcorearb.h:5034
ref
GLint ref
Definition glcorearb.h:291
z
GLdouble GLdouble GLdouble z
Definition glcorearb.h:843
o2::gpu::GPUdi
GPUdi() o2
Definition TrackTRD.h:39
o2::its::math_utils::dx1
const float dx1
Definition MathUtils.h:56
o2::its::math_utils::dy1
const float dy1
Definition MathUtils.h:56
o2::tpc
Global TPC definitions and constants.
Definition TrackGeometry.h:655
o2::utils::streamFastTransform
@ streamFastTransform
stream tpc fast transform
Definition DebugStreamer.h:37
o2::gpu::TPCSlowSpaceChargeCorrection
simple struct to hold the space charge object which can be used for CPU reconstruction only
Definition TPCFastTransform.h:42
o2::gpu::TPCSlowSpaceChargeCorrection::ClassDefNV
ClassDefNV(TPCSlowSpaceChargeCorrection, 2)
o2::gpu::TPCSlowSpaceChargeCorrection::getCorrections
void getCorrections(const float gx, const float gy, const float gz, const int32_t sector, float &gdxC, float &gdyC, float &gdzC) const
getting the corrections for global coordinates
Definition TPCFastTransform.cxx:229
o2::gpu::TPCSlowSpaceChargeCorrection::mCorr
o2::tpc::SpaceCharge< float > * mCorr
reference space charge corrections
Definition TPCFastTransform.h:51