[TrackLTIntegral] optionally update only L and mat.budget integral

shahor02 · shahor02 · commit 02e9daff230f · 2021-04-13T23:26:27.000+02:00
diff --git a/DataFormats/Reconstruction/include/ReconstructionDataFormats/TrackLTIntegral.h b/DataFormats/Reconstruction/include/ReconstructionDataFormats/TrackLTIntegral.h
@@ -27,6 +27,8 @@ 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;
@@ -35,32 +37,40 @@ class TrackLTIntegral
 
   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 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, 1);
+  ClassDefNV(TrackLTIntegral, 2);
 };
 }; // namespace track
 }; // namespace o2
diff --git a/DataFormats/Reconstruction/src/TrackLTIntegral.cxx b/DataFormats/Reconstruction/src/TrackLTIntegral.cxx
@@ -21,9 +21,13 @@ namespace track
 GPUd() void TrackLTIntegral::print() const
 {
 #ifndef GPUCA_GPUCODE_DEVICE
-  printf("L(cm): %6.2f, X2X0: %5.3f TOF(ps): ", getL(), getX2X0());
-  for (int i = 0; i < getNTOFs(); i++) {
-    printf(" %7.1f |", getTOF(i));
+  printf("L(cm): %6.2f, X2X0: %e XRho: %e TOF(ps): ", getL(), getX2X0(), getXRho());
+  if (isTimeNotNeeded()) {
+    printf(" Times not filled");
+  } else {
+    for (int i = 0; i < getNTOFs(); i++) {
+      printf(" %7.1f |", getTOF(i));
+    }
   }
   printf("\n");
 #endif
@@ -34,6 +38,9 @@ GPUd() void TrackLTIntegral::addStep(float dL, float p2Inv)
 {
   ///< add step in cm to integrals
   mL += dL;
+  if (isTimeNotNeeded()) {
+    return;
+  }
   const float dTns = dL * 1000.f / o2::constants::physics::LightSpeedCm2NS; // time change in ps for beta = 1 particle
   for (int id = 0; id < getNTOFs(); id++) {
     const float m2z = track::PID::getMass2Z(id);
