Optionally store refs on ITS parts of all PV contributors

To allow accessing them w/o going throught all kinds of tracks

Author: shahor02

DataFormats/Reconstruction/include/ReconstructionDataFormats/VtxTrackRef.h

@@ -80,12 +80,17 @@ class VtxTrackRef : public RangeReference<int, int>
   // set the last +1 element index and finalize all references
   void setEnd(int end);
 
+  const RangeReference<int, int>& getITSGloContributors() const { return mITSGloContributors; }
+  RangeReference<int, int>& getITSGloContributors() { return mITSGloContributors; }
+
  private:
   using RangeReference<int, int>::RangeReference;
   int mVtxID = -1; // vertex ID. The reference for unassigned tracks will have it negative!
   std::array<int, VtxTrackIndex::Source::NSources - 1> mFirstEntrySource{0};
 
-  ClassDefNV(VtxTrackRef, 2);
+  RangeReference<int, int> mITSGloContributors; // optionally filled to keep ITS parts of all contributors (including ITS-only ones)
+
+  ClassDefNV(VtxTrackRef, 3);
 };
 
 std::ostream& operator<<(std::ostream& os, const o2::dataformats::VtxTrackRef& v);

DataFormats/Reconstruction/src/VtxTrackRef.cxx

@@ -31,6 +31,9 @@ std::string VtxTrackRef::asString(bool skipEmpty) const
       str += fmt::format(", N{:s} : {:d}", VtxTrackIndex::getSourceName(i), getEntriesOfSource(i));
     }
   }
+  if (mITSGloContributors.getEntries()) {
+    str += fmt::format("| ITScontributors: {} from {}", mITSGloContributors.getEntries(), mITSGloContributors.getFirstEntry());
+  }
   return str;
 }
 

Detectors/GlobalTrackingWorkflow/src/VertexTrackMatcherSpec.cxx

@@ -98,6 +98,7 @@ void VertexTrackMatcherSpec::updateTimeDependentParams(ProcessingContext& pc)
     mMatcher.setITSROFrameLengthMUS(o2::base::GRPGeomHelper::instance().getGRPECS()->isDetContinuousReadOut(o2::detectors::DetID::ITS) ? alpParamsITS.roFrameLengthInBC * o2::constants::lhc::LHCBunchSpacingMUS : alpParamsITS.roFrameLengthTrig * 1.e-3);
     const auto& alpParamsMFT = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::MFT>::Instance();
     mMatcher.setMFTROFrameLengthMUS(o2::base::GRPGeomHelper::instance().getGRPECS()->isDetContinuousReadOut(o2::detectors::DetID::MFT) ? alpParamsMFT.roFrameLengthInBC * o2::constants::lhc::LHCBunchSpacingMUS : alpParamsMFT.roFrameLengthTrig * 1.e-3);
+    mMatcher.init();
     LOGP(info, "VertexTrackMatcher ITSROFrameLengthMUS:{} MFTROFrameLengthMUS:{}", mMatcher.getITSROFrameLengthMUS(), mMatcher.getMFTROFrameLengthMUS());
   }
   // we may have other params which need to be queried regularly

Detectors/Vertexing/include/DetectorsVertexing/PVertexerParams.h

@@ -116,7 +116,7 @@ struct PVertexerParams : public o2::conf::ConfigurableParamHelper<PVertexerParam
   float nSigmaTimeTrack = 4.;       ///< define track time bracket as +- this number of sigmas of its time resolution
   float timeMarginTrackTime = 0.5;  ///< additive marginal error in \mus to track time bracket
   float timeMarginVertexTime = 0.0; ///< additive marginal error to \mus vertex time bracket
-
+  bool fillITSGloContributors = false; ///< fill separate entry for ITS contributors
   O2ParamDef(PVertexerParams, "pvertexer");
 };
 

Detectors/Vertexing/include/DetectorsVertexing/VertexTrackMatcher.h

@@ -51,6 +51,7 @@ class VertexTrackMatcher
     int origID = -1;     ///< vertex origin id
   };
 
+  void init();
   void process(const o2::globaltracking::RecoContainer& recoData,
                std::vector<VTIndex>& trackIndex, // Global ID's for associated tracks
                std::vector<VRef>& vtxRefs);      // references on these tracks
@@ -69,6 +70,7 @@ class VertexTrackMatcher
 
  private:
   void extractTracks(const o2::globaltracking::RecoContainer& data, const std::unordered_map<GIndex, bool>& vcont);
+  std::vector<int> mITSGloSources;
   std::vector<TrackTBracket> mTBrackets;
   float mITSROFrameLengthMUS = 0;       ///< ITS RO frame in mus
   float mMFTROFrameLengthMUS = 0;       ///< MFT RO frame in mus

Detectors/Vertexing/src/VertexTrackMatcher.cxx

@@ -21,6 +21,19 @@
 
 using namespace o2::vertexing;
 
+void VertexTrackMatcher::init()
+{
+  const auto& PVParams = o2::vertexing::PVertexerParams::Instance();
+  mITSGloSources.clear();
+  if (PVParams.fillITSGloContributors) { // collect sources of global tracks with ITS contributor
+    for (int i = 0; i < GIndex::NSources; i++) {
+      if (GIndex::includesDet(o2::detectors::DetID::ITS, GIndex::getSourceDetectorsMask(i)) && i != GIndex::ITSAB) {
+        mITSGloSources.push_back(i);
+      }
+    }
+  }
+}
+
 void VertexTrackMatcher::process(const o2::globaltracking::RecoContainer& recoData,
                                  std::vector<VTIndex>& trackIndex,
                                  std::vector<VRef>& vtxRefs)
@@ -123,6 +136,22 @@ void VertexTrackMatcher::process(const o2::globaltracking::RecoContainer& recoDa
       vr.setFirstEntryOfSource(oldSrc, trackIndex.size());
     }
     vr.setEnd(trackIndex.size());
+
+    if (PVParams.fillITSGloContributors) {
+      auto& ITSGloContributors = vr.getITSGloContributors();
+      ITSGloContributors.setFirstEntry(trackIndex.size());
+      for (auto srcITS : mITSGloSources) {
+        const int fst = vr.getFirstEntryOfSource(srcITS), lst = fst + vr.getEntriesOfSource(srcITS);
+        for (int ii = fst; ii < lst; ii++) {
+          auto vid = trackIndex[ii];
+          if (vid.isPVContributor()) {
+            trackIndex.push_back(ii == GIndex::ITS ? vid : recoData.getITSContributorGID(vid));
+          }
+        }
+      }
+      ITSGloContributors.setEntries(trackIndex.size() - ITSGloContributors.getFirstEntry());
+    }
+
     if (logVertices) {
       LOG(info) << vr;
     }