TrackLTIntegral.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 ALICEO2_TRACK_LTINTEGRAL_H_
#define ALICEO2_TRACK_LTINTEGRAL_H_
 
#include "GPUCommonRtypes.h"
#include "GPUCommonDef.h"
#include "ReconstructionDataFormats/PID.h"
 
namespace o2
{
namespace track
{
 
class TrackLTIntegral
{
 public:
  static constexpr float NeglectTime = -1.; // if 1st mT slot contains this, don't fill time
 
  GPUdDefault() TrackLTIntegral() = default;
  GPUdDefault() TrackLTIntegral(const TrackLTIntegral& stc) = default;
  GPUdDefault() ~TrackLTIntegral() = default;
 
  GPUd() static constexpr int getNTOFs() { return o2::track::PID::NIDs; }
 
  GPUd() float getL() const { return mL; }
  GPUd() float getX2X0() const { return mX2X0; }
  GPUd() float getXRho() const { return mXRho; }
  GPUd() float getTOF(int id) const { return mT[id]; }
 
  GPUd() void clear()
  {
    mL = 0.f;
    mX2X0 = 0.f;
    mXRho = 0.f;
    for (int i = getNTOFs(); i--;) {
      mT[i] = 0.f;
    }
  }
 
  GPUd() void clearFast()
  {
    mL = 0.f;
    mX2X0 = 0.f;
    mXRho = 0.f;
    if (!isTimeNotNeeded()) {
      for (int i = getNTOFs(); i--;) {
        mT[i] = 0.f;
      }
    }
  }
 
  GPUd() void addStep(float dL, float p2Inv);
  GPUd() void addX2X0(float d) { mX2X0 += d; }
  GPUd() void addXRho(float d) { mXRho += d; }
 
  GPUd() void setL(float l) { mL = l; }
  GPUd() void setX2X0(float x) { mX2X0 = x; }
  GPUd() void setXRho(float x) { mXRho = x; }
  GPUd() void setTOF(float t, int id) { mT[id] = t; }
 
  GPUd() void setTimeNotNeeded() { mT[0] = NeglectTime; }
  GPUd() bool isTimeNotNeeded() const { return mT[0] == NeglectTime; }
 
  GPUd() void print() const;
 
 private:
  float mL = 0.;                         // length in cm
  float mX2X0 = 0.;                      // integrated X/X0
  float mXRho = 0.;                      // average X*rho
  float mT[o2::track::PID::NIDs] = {0.}; // TOF in ps
 
  ClassDefNV(TrackLTIntegral, 2);
};
}; // namespace track
}; // namespace o2
 
#endif