GPUTRDRecoParam.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef O2_GPU_TRD_RECOPARAM_H
#define O2_GPU_TRD_RECOPARAM_H
 
#include "GPUCommonDef.h"
#include "GPUCommonRtypes.h"
#include "GPUCommonArray.h"
#include "GPUCommonMath.h"
 
namespace o2
{
namespace gpu
{
struct GPUSettingsRec;
 
class GPUTRDRecoParam
{
 public:
  GPUTRDRecoParam() = default;
  GPUTRDRecoParam(const GPUTRDRecoParam&) = default;
  ~GPUTRDRecoParam() = default;
 
  void init(float bz, const GPUSettingsRec* rec = nullptr);
 
#if !defined(GPUCA_GPUCODE_DEVICE)
  GPUd() void recalcTrkltCov(const float tilt, const float snp, const float rowSize, std::array<float, 3>& cov) const
  {
    recalcTrkltCov(tilt, snp, rowSize, cov.data());
  }
#endif
  GPUd() void recalcTrkltCov(const float tilt, const float snp, const float rowSize, float* cov) const;
 
  GPUd() float getRPhiRes(float snp) const { return (mRPhiA2 + mRPhiC2 * (snp - mLorentzAngle) * (snp - mLorentzAngle)); }
  GPUd() float getDyRes(float snp) const { return mDyA2 + mDyC2 * (snp - mLorentzAngle) * (snp - mLorentzAngle); }           // a^2 + c^2 * (snp - b)^2
  GPUd() float convertAngleToDy(float snp) const { return 3.f * snp / CAMath::Sqrt(1 - snp * snp); }                         // when calibrated, sin(phi) = (dy / xDrift) / sqrt(1+(dy/xDrift)^2) works well
  GPUd() float getCorrYDy(float snp) const { return mCorrYDyA + mCorrYDyC * (snp - mLorentzAngle) * (snp - mLorentzAngle); } // a + c * (snp - b)^2
 
  GPUd() float getZCorrCoeffNRC() const { return mZCorrCoefNRC; }
 
 private:
  // tracklet error parameterization depends on the magnetic field
  float mLorentzAngle{0.f};
  // rphi
  float mRPhiA2{1.f}; 
  float mRPhiC2{0.f}; 
  // angle
  float mDyA2{1.225e-3f}; 
  float mDyC2{0.f};       
  // correlation coefficient between y residual and dy residual
  float mCorrYDyA{0.f};
  float mCorrYDyC{0.f};
 
  float mZCorrCoefNRC{1.4f}; 
 
  ClassDefNV(GPUTRDRecoParam, 3);
};
 
} // namespace gpu
} // namespace o2
 
#endif // O2_GPU_TRD_RECOPARAM_H