ITS: add tracklet tuning

Signed-off-by: Felix Schlepper <felix.schlepper@cern.ch>

Author: f3sch

Detectors/ITSMFT/ITS/macros/tune/CheckTracklets.C

@@ -25,6 +25,7 @@ o2_add_library(ITStracking
                        src/TrackingConfigParam.cxx
                        src/Vertexer.cxx
                        src/VertexerTraits.cxx
+                       src/TuneExt.cxx
                PUBLIC_LINK_LIBRARIES
                        O2::GPUCommon
                        Microsoft.GSL::GSL
@@ -56,6 +57,7 @@ o2_target_root_dictionary(ITStracking
                                   include/ITStracking/Definitions.h
                                   include/ITStracking/FastMultEstConfig.h
                                   include/ITStracking/TrackingConfigParam.h
+                                  include/ITStracking/TuneExt.h
                           LINKDEF src/TrackingLinkDef.h)
 
 if(CUDA_ENABLED OR HIP_ENABLED)

Detectors/ITSMFT/ITS/tracking/CMakeLists.txt

@@ -85,11 +85,12 @@ class TrackerTraits
   std::shared_ptr<tbb::task_arena> mTaskArena;
 
  protected:
-  o2::gpu::GPUChainITS* mChain = nullptr;
-  TimeFrame<NLayers>* mTimeFrame;
-  std::vector<TrackingParameters> mTrkParams;
+  void createTrackletMC();
 
-  float mBz{-999.f};
+  o2::gpu::GPUChainITS* mChain{nullptr};
+  TimeFrame<NLayers>* mTimeFrame{nullptr};
+  std::vector<TrackingParameters> mTrkParams;
+  float mBz{constants::UnsetValue};
 };
 
 } // namespace its

Detectors/ITSMFT/ITS/tracking/include/ITStracking/TrackerTraits.h

@@ -0,0 +1,31 @@
+// Copyright 2019-2026 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.
+///
+/// Holder of auxiallary information used for tuning
+///
+
+#include "ITStracking/Constants.h"
+#include "GPUCommonRtypes.h"
+
+namespace o2::its
+{
+struct TrackletMC final {
+  float tgl{constants::UnsetValue}; // tanLambda
+  float phi{constants::UnsetValue}; // phi
+  bool ok{false};                   // truth
+  /// below only metrics valid if ok
+  bool prim{false};                 // primary
+  float dXY{constants::UnsetValue}; // transverse distance to event
+  float dZ{constants::UnsetValue};  // longitudinal distance to event
+  ClassDefNV(TrackletMC, 1);
+};
+
+} // namespace o2::its

Detectors/ITSMFT/ITS/tracking/include/ITStracking/TuneExt.h

@@ -223,26 +223,7 @@ void TrackerTraits<NLayers>::computeLayerTracklets(const int iteration, int iVer
 
     /// Create tracklets labels
     if (mTimeFrame->hasMCinformation() && mTrkParams[iteration].CreateArtefactLabels) {
-      tbb::parallel_for(0, static_cast<int>(topology.nTransitions), [&](const int transitionId) {
-        const auto& transition = topology.getTransition(transitionId);
-        for (auto& trk : mTimeFrame->getTracklets()[transitionId]) {
-          MCCompLabel label;
-          int currentId{mTimeFrame->getClusters()[transition.fromLayer][trk.firstClusterIndex].clusterId};
-          int nextId{mTimeFrame->getClusters()[transition.toLayer][trk.secondClusterIndex].clusterId};
-          for (const auto& lab1 : mTimeFrame->getClusterLabels(transition.fromLayer, currentId)) {
-            for (const auto& lab2 : mTimeFrame->getClusterLabels(transition.toLayer, nextId)) {
-              if (lab1 == lab2 && lab1.isValid()) {
-                label = lab1;
-                break;
-              }
-            }
-            if (label.isValid()) {
-              break;
-            }
-          }
-          mTimeFrame->getTrackletsLabel(transitionId).emplace_back(label);
-        }
-      });
+      createTrackletMC();
     }
   });
 }
@@ -907,6 +888,32 @@ void TrackerTraits<NLayers>::markTracks(int iteration)
   }
 }
 
+template <int NLayers>
+void TrackerTraits<NLayers>::createTrackletMC()
+{
+  const auto topology = mTimeFrame->getTrackingTopologyView();
+  tbb::parallel_for(0, static_cast<int>(topology.nTransitions), [&](const int transitionId) {
+    const auto& transition = topology.getTransition(transitionId);
+    for (auto& trk : mTimeFrame->getTracklets()[transitionId]) {
+      MCCompLabel label;
+      int currentId{mTimeFrame->getClusters()[transition.fromLayer][trk.firstClusterIndex].clusterId};
+      int nextId{mTimeFrame->getClusters()[transition.toLayer][trk.secondClusterIndex].clusterId};
+      for (const auto& lab1 : mTimeFrame->getClusterLabels(transition.fromLayer, currentId)) {
+        for (const auto& lab2 : mTimeFrame->getClusterLabels(transition.toLayer, nextId)) {
+          if (lab1 == lab2 && lab1.isValid()) {
+            label = lab1;
+            break;
+          }
+        }
+        if (label.isValid()) {
+          break;
+        }
+      }
+      mTimeFrame->getTrackletsLabel(transitionId).emplace_back(label);
+    }
+  });
+}
+
 template <int NLayers>
 void TrackerTraits<NLayers>::setBz(float bz)
 {

Detectors/ITSMFT/ITS/tracking/src/TrackerTraits.cxx

@@ -10,17 +10,81 @@
 // or submit itself to any jurisdiction.
 ///
 
+#include <format>
+#include <cmath>
+#include <limits>
+
 #include "ITStracking/TrackerTraitsMC.h"
+#include "ITStracking/TuneExt.h"
+#include "CommonUtils/TreeStreamRedirector.h"
+#include "Steer/MCKinematicsReader.h"
 
 #include "Framework/Logger.h"
 
 namespace o2::its
 {
 
+static o2::utils::TreeStreamRedirector* gDBG{nullptr};
+static o2::steer::MCKinematicsReader* gMCReader{nullptr};
+
 template <int NLayers>
 void TrackerTraitsMC<NLayers>::computeLayerTracklets(const int iteration, int iVertex)
 {
+  if (!gDBG) {
+    gDBG = new o2::utils::TreeStreamRedirector("its_tune.root");
+  }
+  if (!gMCReader) {
+    gMCReader = new o2::steer::MCKinematicsReader("collisioncontext.root");
+  }
+
+  // Create all tracklets we find in this iteration and dump them.
+  const std::string treeName = std::format("trklt_{}", iteration);
   TrackerTraits<NLayers>::computeLayerTracklets(iteration, iVertex);
+  this->createTrackletMC();
+  const auto topology = this->mTimeFrame->getTrackingTopologyView();
+  for (int transitionId{0}; transitionId < topology.nTransitions; ++transitionId) {
+    const auto& transition = topology.getTransition(transitionId);
+    for (int iTrklt{0}; iTrklt < this->mTimeFrame->getTracklets()[transitionId].size(); ++iTrklt) {
+      const auto& lbl = this->mTimeFrame->getTrackletsLabel(transitionId)[iTrklt];
+      const auto trklt = this->mTimeFrame->getTracklets()[transitionId][iTrklt];
+      TrackletMC trkltMC{
+        .tgl = trklt.tanLambda,
+        .phi = trklt.phi,
+        .ok = lbl.isValid(),
+      };
+      float dcaXY = std::numeric_limits<float>::max(), dcaZ = std::numeric_limits<float>::max();
+      if (lbl.isValid()) {
+        const auto& eve = gMCReader->getMCEventHeader(lbl.getSourceID(), lbl.getEventID());
+        const auto& firstCluster = this->mTimeFrame->getClusters()[transition.fromLayer][trklt.firstClusterIndex];
+        const auto& secondCluster = this->mTimeFrame->getClusters()[transition.toLayer][trklt.secondClusterIndex];
+        const float dx = secondCluster.xCoordinate - firstCluster.xCoordinate;
+        const float dy = secondCluster.yCoordinate - firstCluster.yCoordinate;
+        const float dz = secondCluster.zCoordinate - firstCluster.zCoordinate;
+        const float dxy2 = math_utils::hypot(dx, dy);
+        if (dxy2 > constants::Tolerance) {
+          const float t = ((eve.GetX() - firstCluster.xCoordinate) * dx + (eve.GetY() - firstCluster.yCoordinate) * dy) / dxy2;
+          const float xAtDCA = firstCluster.xCoordinate + t * dx;
+          const float yAtDCA = firstCluster.yCoordinate + t * dy;
+          const float zAtDCA = firstCluster.zCoordinate + t * dz;
+          const float curDCAx = xAtDCA - eve.GetX();
+          const float curDCAy = yAtDCA - eve.GetY();
+          dcaXY = math_utils::hypot(curDCAx, curDCAy);
+          dcaZ = zAtDCA - eve.GetZ();
+          trkltMC.dXY = dcaXY;
+          trkltMC.dZ = dcaZ;
+        }
+        const auto* mcTrk = gMCReader->getTrack(lbl);
+        if (mcTrk) {
+          trkltMC.prim = mcTrk->isPrimary();
+        }
+      }
+      (*gDBG) << treeName.c_str()
+              << "from=" << transition.fromLayer
+              << "to=" << transition.toLayer
+              << "trklt=" << trkltMC
+              << "\n";
+    }
+  }
 }
 
 template <int NLayers>
@@ -39,6 +103,10 @@ template <int NLayers>
 void TrackerTraitsMC<NLayers>::findRoads(const int iteration)
 {
   TrackerTraits<NLayers>::findRoads(iteration);
+
+  if (this->mTrkParams.size() - 1 == iteration) {
+    gDBG->Close();
+  }
 }
 
 template class TrackerTraitsMC<7>;

Detectors/ITSMFT/ITS/tracking/src/TrackerTraitsMC.cxx

@@ -18,6 +18,9 @@
 #pragma link C++ class o2::its::Tracklet + ;
 #pragma link C++ class std::vector < o2::its::Tracklet> + ;
 
+#pragma link C++ class o2::its::TrackletMC + ;
+#pragma link C++ class std::vector < o2::its::TrackletMC> + ;
+
 #pragma link C++ class o2::its::Cluster + ;
 #pragma link C++ class std::vector < o2::its::Cluster> + ;
 

Detectors/ITSMFT/ITS/tracking/src/TrackingLinkDef.h

@@ -0,0 +1,13 @@
+// Copyright 2019-2026 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.
+
+#include "ITStracking/TuneExt.h"
+ClassImp(o2::its::TrackletMC);