GPUTPCTrackLinearisation.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 GPUTPCTRACKLINEARISATION_H
#define GPUTPCTRACKLINEARISATION_H
 
#include "GPUTPCTrackParam.h"
 
namespace o2::gpu
{
class GPUTPCTrackLinearisation
{
 public:
  GPUTPCTrackLinearisation() : mSinPhi(0), mCosPhi(1), mDzDs(0), mQPt(0) {}
  GPUTPCTrackLinearisation(float SinPhi1, float CosPhi1, float DzDs1, float QPt1) : mSinPhi(SinPhi1), mCosPhi(CosPhi1), mDzDs(DzDs1), mQPt(QPt1) {}
 
  GPUd() GPUTPCTrackLinearisation(const GPUTPCTrackParam& t);
 
  GPUd() void Set(float SinPhi1, float CosPhi1, float DzDs1, float QPt1);
 
  GPUd() float SinPhi() const { return mSinPhi; }
  GPUd() float CosPhi() const { return mCosPhi; }
  GPUd() float DzDs() const { return mDzDs; }
  GPUd() float QPt() const { return mQPt; }
 
  GPUd() float GetSinPhi() const { return mSinPhi; }
  GPUd() float GetCosPhi() const { return mCosPhi; }
  GPUd() float GetDzDs() const { return mDzDs; }
  GPUd() float GetQPt() const { return mQPt; }
 
  GPUd() void SetSinPhi(float v) { mSinPhi = v; }
  GPUd() void SetCosPhi(float v) { mCosPhi = v; }
  GPUd() void SetDzDs(float v) { mDzDs = v; }
  GPUd() void SetQPt(float v) { mQPt = v; }
 
 private:
  float mSinPhi; // SinPhi
  float mCosPhi; // CosPhi
  float mDzDs;   // DzDs
  float mQPt;    // QPt
};
 
GPUdi() GPUTPCTrackLinearisation::GPUTPCTrackLinearisation(const GPUTPCTrackParam& GPUrestrict() t) : mSinPhi(t.SinPhi()), mCosPhi(0), mDzDs(t.DzDs()), mQPt(t.QPt())
{
  if (mSinPhi > GPUCA_MAX_SIN_PHI) {
    mSinPhi = GPUCA_MAX_SIN_PHI;
  } else if (mSinPhi < -GPUCA_MAX_SIN_PHI) {
    mSinPhi = -GPUCA_MAX_SIN_PHI;
  }
  mCosPhi = CAMath::Sqrt(1 - mSinPhi * mSinPhi);
  if (t.SignCosPhi() < 0) {
    mCosPhi = -mCosPhi;
  }
}
 
GPUdi() void GPUTPCTrackLinearisation::Set(float SinPhi1, float CosPhi1, float DzDs1, float QPt1)
{
  SetSinPhi(SinPhi1);
  SetCosPhi(CosPhi1);
  SetDzDs(DzDs1);
  SetQPt(QPt1);
}
} // namespace o2::gpu
 
#endif // GPUTPCTRACKLINEARISATION_H