TrackLTIntegral.h

// 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.

/// \file TrackLTIntegral.h
/// \brief Track Length and TOF integral
/// \author ruben.shahoyan@cern.ch

#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 q2p2);
  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