Optional TRD/TOF tracks request for PV used for interpolation

Author: shahor02

Detectors/TPC/calibration/SpacePoints/include/SpacePoints/SpacePointsCalibConfParam.h

@@ -38,6 +38,7 @@ struct SpacePointsCalibConfParam : public o2::conf::ConfigurableParamHelper<Spac
   float maxITSChi2 = 20.f;         ///< cut on ITS reduced chi2
   float maxTRDChi2 = 10.f;         ///< cut on TRD reduced chi2
   float minPtNoOuterPoint = 0.8f;  ///< minimum pt for ITS-TPC tracks to be considered for extrapolation
+  int minTOFTRDPVContributors = 0; ///< min contributors from TRD or TOF (fast detectors) to consider tracks of this PV
   bool ignoreNonPVContrib = true;  ///< flag if tracks which did not contribute to the PV should be ignored or not
   bool enableTrackDownsampling = false; ///< flag if track sampling shall be enabled or not
   float tsalisThreshold = 1.f;          ///< in case the sampling functions returns a value smaller than this the track is discarded (1. means no downsampling)

Detectors/TPC/calibration/SpacePoints/src/TrackInterpolation.cxx

@@ -149,12 +149,30 @@ void TrackInterpolation::prepareInputTrackSample(const o2::globaltracking::RecoC
   auto vtxRefs = mRecoCont->getPrimaryVertexMatchedTrackRefs(); // references from vertex to these track IDs
   int nv = vtxRefs.size() - 1;
   GTrackID::mask_t allowedSources = GTrackID::getSourcesMask("ITS-TPC,ITS-TPC-TRD,ITS-TPC-TOF,ITS-TPC-TRD-TOF");
+  constexpr std::array<int, 3> SrcFast = {int(GTrackID::ITSTPCTRD), int(GTrackID::ITSTPCTOF), int(GTrackID::ITSTPCTRDTOF)};
 
   for (int iv = 0; iv < nv; iv++) {
     LOGP(debug, "processing PV {} of {}", iv, nv);
 
     const auto& vtref = vtxRefs[iv];
     auto pv = pvvec[iv];
+    if (mParams->minTOFTRDPVContributors > 0) { // we want only PVs constrained by fast detectors
+      int nfound = 0;
+      bool usePV = false;
+      for (uint32_t is = 0; is < SrcFast.size() && !usePV; is++) {
+        int src = SrcFast[is], idMin = vtref.getFirstEntryOfSource(src), idMax = idMin + vtref.getEntriesOfSource(src);
+        for (int i = idMin; i < idMax; i++) {
+          if (trackIndex[i].isPVContributor() && (++nfound == mParams->minTOFTRDPVContributors)) {
+            usePV = true;
+            break;
+          }
+        }
+      }
+      if (!usePV) {
+        continue;
+      }
+    }
+
     for (int is = GTrackID::NSources; is >= 0; is--) {
       if (!allowedSources[is]) {
         continue;