ITS: fix cluster label access for non-staggered

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

Author: f3sch

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

@@ -150,7 +150,7 @@ struct TimeFrame {
 
   const TrackingFrameInfo& getClusterTrackingFrameInfo(int layerId, const Cluster& cl) const;
   gsl::span<const MCCompLabel> getClusterLabels(int layerId, const Cluster& cl) const { return getClusterLabels(layerId, cl.clusterId); }
-  gsl::span<const MCCompLabel> getClusterLabels(int layerId, const int clId) const { return mClusterLabels[layerId]->getLabels(mClusterExternalIndices[layerId][clId]); }
+  gsl::span<const MCCompLabel> getClusterLabels(int layerId, const int clId) const { return mClusterLabels[((mIsStaggered) ? layerId : 0)]->getLabels(mClusterExternalIndices[layerId][clId]); }
   int getClusterExternalIndex(int layerId, const int clId) const { return mClusterExternalIndices[layerId][clId]; }
   int getClusterSize(int layer, int clusterId) const { return mClusterSize[layer][clusterId]; }
   void setClusterSize(int layer, bounded_vector<uint8_t>& v) { mClusterSize[layer] = std::move(v); }
@@ -194,6 +194,9 @@ struct TimeFrame {
   unsigned long getArtefactsMemory() const;
   void printArtefactsMemory() const;
 
+  /// staggering
+  void setIsStaggered(bool b) noexcept { mIsStaggered = b; }
+
   /// ROF cuts
   int getROFCutClusterMult() const { return mCutClusterMult; };
   int getROFCutVertexMult() const { return mCutVertexMult; };
@@ -316,6 +319,8 @@ struct TimeFrame {
   ROFVertexLookupTableN mROFVertexLookupTable;
   ROFVertexLookupTableN::View mROFVertexLookupTableView;
 
+  bool mIsStaggered{false};
+
   std::shared_ptr<BoundedMemoryResource> mMemoryPool;
 };
 

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

@@ -72,6 +72,7 @@ void ITSTrackingInterface::initialise()
   }
   mVertexer->setNThreads(vertConf.nThreads, mTaskArena);
   mTracker->setNThreads(trackConf.nThreads, mTaskArena);
+  mTimeFrame->setIsStaggered(mDoStaggering);
 
   // prepare data filter
   for (int iLayer = 0; iLayer < ((mDoStaggering) ? NLayers : 1); ++iLayer) {