trk

Author: f3sch

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

@@ -66,6 +66,11 @@ struct LayerTiming {
   {
     return std::format("NROFsPerTF {} ROFLength {} ROFDelay {} ROFDelta {}", mNROFsTF, mROFLength, mROFDelay, mROFDelta);
   }
+
+  GPUh() void print() const
+  {
+    LOG(info) << asString();
+  }
 };
 
 // Base class for lookup to define layers
@@ -165,6 +170,7 @@ struct ROFOverlapTableView {
   {
     if (from == to) {
       LOGP(error, "No self-lookup supported");
+      return;
     }
 
     constexpr int w_index = 10;

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

@@ -62,6 +62,7 @@ class Tracker
   void adoptTimeFrame(TimeFrame<NLayers>& tf);
 
   void clustersToTracks(const LogFunc& info, const LogFunc& error);
+  const auto& getParameters() const { return mTrkParams; }
   void setParameters(const std::vector<TrackingParameters>& p) { mTrkParams = p; }
   void setMemoryPool(std::shared_ptr<BoundedMemoryResource> pool) { mMemoryPool = pool; }
   std::vector<TrackingParameters>& getParameters() { return mTrkParams; }

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

@@ -272,7 +272,7 @@ void TimeFrame<NLayers>::initialise(const int iteration, const TrackingParameter
       mPositionResolution[iLayer] = o2::gpu::CAMath::Sqrt(0.5f * (trkParam.SystErrorZ2[iLayer] + trkParam.SystErrorY2[iLayer]) + trkParam.LayerResolution[iLayer] * trkParam.LayerResolution[iLayer]);
     }
     for (unsigned int iLayer{0}; iLayer < std::min((int)mClusters.size(), maxLayers); ++iLayer) {
-      clearResizeBoundedArray(mIndexTables, getNrof(iLayer) * ((trkParam.ZBins * trkParam.PhiBins) + 1), getMaybeExternalHostResource(maxLayers != NLayers));
+      clearResizeBoundedVector(mIndexTables[iLayer], getNrof(iLayer) * ((trkParam.ZBins * trkParam.PhiBins) + 1), getMaybeExternalHostResource(maxLayers != NLayers));
     }
     clearResizeBoundedVector(mLines, getNrof(0), mMemoryPool.get());
     clearResizeBoundedVector(mTrackletClusters, getNrof(0), mMemoryPool.get());
@@ -398,14 +398,12 @@ void TimeFrame<NLayers>::setMemoryPool(std::shared_ptr<BoundedMemoryResource> po
 {
   mMemoryPool = std::move(pool);
 
-  auto initVector = [&]<typename T>(bounded_vector<T> & vec, bool useExternal = false)
-  {
+  auto initVector = [&]<typename T>(bounded_vector<T>& vec, bool useExternal = false) {
     std::pmr::memory_resource* mr = (useExternal) ? mExtMemoryPool.get() : mMemoryPool.get();
     deepVectorClear(vec, mr);
   };
 
-  auto initContainers = [&]<typename Container>(Container & container, bool useExternal = false)
-  {
+  auto initContainers = [&]<typename Container>(Container& container, bool useExternal = false) {
     for (auto& v : container) {
       initVector(v, useExternal);
     }

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

@@ -29,7 +29,6 @@
 #include "CommonDataFormat/IRFrame.h"
 #include "DetectorsBase/GRPGeomHelper.h"
 #include "ITStracking/BoundedAllocator.h"
-#include "DetectorsRaw/HBFUtils.h"
 #include "Framework/InputRecordWalker.h"
 #include "Framework/DataRefUtils.h"
 #include "Framework/DeviceSpec.h"
@@ -83,30 +82,35 @@ void ITSTrackingInterface::initialise()
       mFilter.emplace_back("itsmclabels", "ITS", "CLUSTERSMCTR", iLayer, Lifetime::Timeframe);
     }
   }
-
-  // prepare rof lookup table(s)
-  const int nOrbitsPerTF = raw::HBFUtils::Instance().getNOrbitsPerTF();
-  TimeFrameN::ROFOverlapTableN rofTable;
-  TimeFrameN::ROFVertexLookupTableN vtxTable;
-  TimeFrameN::ROFTimeSliceTableN sliceTable;
-  // can use DPL param since call to ccdb has been made
-  const auto& par = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::ITS>::Instance();
-  for (int iLayer = 0; iLayer < NLayers; ++iLayer) {
-    const int nROFsPerOrbit = o2::constants::lhc::LHCMaxBunches / par.getROFLengthInBC(iLayer);
-    rofTable.defineLayer(iLayer, nOrbitsPerTF * nROFsPerOrbit, par.getROFLengthInBC(iLayer), par.getROFDelayInBC(iLayer), trackParams[0].DeltaROF[iLayer]);
-    vtxTable.defineLayer(iLayer, nOrbitsPerTF * nROFsPerOrbit, par.getROFLengthInBC(iLayer), par.getROFDelayInBC(iLayer), trackParams[0].DeltaROF[iLayer]);
-    sliceTable.defineLayer(iLayer, nOrbitsPerTF * nROFsPerOrbit, par.getROFLengthInBC(iLayer), par.getROFDelayInBC(iLayer), trackParams[0].DeltaROF[iLayer]);
-  }
-  rofTable.init();
-  mTimeFrame->setROFOverlapTable(rofTable);
-  vtxTable.init();
-  mTimeFrame->setROFVertexLookupTable(vtxTable);
-  sliceTable.init(trackParams[0].NTimeSlices);
-  mTimeFrame->setROFTimeSliceTable(sliceTable);
 }
 
 void ITSTrackingInterface::run(framework::ProcessingContext& pc)
 {
+  if (static bool doneOnce{false}; !doneOnce) {
+    doneOnce = true;
+
+    // prepare rof lookup table(s)
+    // has to be done here to ensure we get the right number of HB per TF
+    const int nOrbitsPerTF = o2::base::GRPGeomHelper::getNHBFPerTF();
+    TimeFrameN::ROFOverlapTableN rofTable;
+    TimeFrameN::ROFVertexLookupTableN vtxTable;
+    TimeFrameN::ROFTimeSliceTableN sliceTable;
+    const auto& par = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::ITS>::Instance();
+    const auto& trackParams = mTracker->getParameters();
+    for (int iLayer = 0; iLayer < NLayers; ++iLayer) {
+      const int nROFsPerOrbit = o2::constants::lhc::LHCMaxBunches / par.getROFLengthInBC(iLayer);
+      rofTable.defineLayer(iLayer, nOrbitsPerTF * nROFsPerOrbit, par.getROFLengthInBC(iLayer), par.getROFDelayInBC(iLayer), trackParams[0].DeltaROF[iLayer]);
+      vtxTable.defineLayer(iLayer, nOrbitsPerTF * nROFsPerOrbit, par.getROFLengthInBC(iLayer), par.getROFDelayInBC(iLayer), trackParams[0].DeltaROF[iLayer]);
+      sliceTable.defineLayer(iLayer, nOrbitsPerTF * nROFsPerOrbit, par.getROFLengthInBC(iLayer), par.getROFDelayInBC(iLayer), trackParams[0].DeltaROF[iLayer]);
+    }
+    rofTable.init();
+    mTimeFrame->setROFOverlapTable(rofTable);
+    vtxTable.init();
+    mTimeFrame->setROFVertexLookupTable(vtxTable);
+    sliceTable.init(trackParams[0].NTimeSlices);
+    mTimeFrame->setROFTimeSliceTable(sliceTable);
+  }
+
   std::array<gsl::span<const itsmft::CompClusterExt>, NLayers> compClusters;
   std::array<gsl::span<const unsigned char>, NLayers> patterns;
   std::array<gsl::span<const itsmft::ROFRecord>, NLayers> rofsinput;