[PWGCF] DptDpt - Tracking the fraction of shared TPC clusters (#9102)

Co-authored-by: Victor <victor@cern.ch>

Author: victor-gonzalez

PWGCF/Core/AnalysisConfigurableCuts.h

@@ -86,27 +86,28 @@ class TrackSelectionCfg
   ClassDefNV(TrackSelectionCfg, 1);
 };
 
-/// \brief Simple class for fine tuning a track selection object
-/// The tune is done after the actual track selection has ben performed
-/// Take into consideration that only more restrictive tunes are supported
-/// For more relaxed tunes the whole track selection object must be modified
+/// \brief Simple class for configuring the fine tuning a track selection object
+/// The tune should change the track selection objects and probably do further checks
+/// after the actual track selection has ben performed
 class TrackSelectionTuneCfg
 {
  public:
-  bool mUseIt = false;              ///< use this track selection tuning configuration
-  int mTPCclusters = 0;             ///< minimum number of TPC clusters
-  bool mUseTPCclusters = false;     ///< use or not the number of TPC clusters
-  int mTPCxRows = 70;               ///< minimum number of TPC crossed rows
-  bool mUseTPCxRows = false;        ///< use or not the number of TPC crossed rows
-  float mTPCXRoFClusters = 0.8;     ///< minimum value of the TPC ratio no of crossed rows over findable clusters
-  bool mUseTPCXRoFClusters = false; ///< use or not the TPC ration of no of crossed rows over findable clusters
-  float mDCAxy = 2.4;               ///< maximum DCA on xy plane
-  bool mUseDCAxy = false;           ///< use or not the maximum DCA on the xy plane
-  float mDCAz = 3.2;                ///< maximum DCA on z axis
-  bool mUseDCAz = false;            ///< use or not the maximum DCA on z asis
+  bool mUseIt = false;                        ///< use this track selection tuning configuration
+  int mTPCclusters = 0;                       ///< minimum number of TPC clusters
+  bool mUseTPCclusters = false;               ///< use or not the number of TPC clusters
+  int mTPCxRows = 70;                         ///< minimum number of TPC crossed rows
+  bool mUseTPCxRows = false;                  ///< use or not the number of TPC crossed rows
+  float mTPCXRoFClusters = 0.8;               ///< minimum value of the TPC ratio no of crossed rows over findable clusters
+  bool mUseTPCXRoFClusters = false;           ///< use or not the TPC ratio of no of crossed rows over findable clusters
+  float mFractionTpcSharedClusters = 0.4;     ///< the maximum fraction of TPC shared clusters
+  bool mUseFractionTpcSharedClusters = false; ///< use or not the fraction of TPC share clusters
+  float mDCAxy = 2.4;                         ///< maximum DCA on xy plane
+  bool mUseDCAxy = false;                     ///< use or not the maximum DCA on the xy plane
+  float mDCAz = 3.2;                          ///< maximum DCA on z axis
+  bool mUseDCAz = false;                      ///< use or not the maximum DCA on z asis
 
  private:
-  ClassDefNV(TrackSelectionTuneCfg, 1);
+  ClassDefNV(TrackSelectionTuneCfg, 2);
 };
 
 /// \brief Simple class to configure a selection based on PID

PWGCF/TableProducer/dptdptfilter.cxx

@@ -770,7 +770,7 @@ struct DptDptFilterTracks {
   Configurable<o2::analysis::CheckRangeCfg> cfgTraceDCAOutliers{"trackdcaoutliers", {false, 0.0, 0.0}, "Track the generator level DCAxy outliers: false/true, low dcaxy, up dcaxy. Default {false,0.0,0.0}"};
   Configurable<float> cfgTraceOutOfSpeciesParticles{"trackoutparticles", false, "Track the particles which are not e,mu,pi,K,p: false/true. Default false"};
   Configurable<int> cfgRecoIdMethod{"recoidmethod", 0, "Method for identifying reconstructed tracks: 0 No PID, 1 PID, 2 mcparticle, 3 mcparticle only primaries, 4 mcparticle only sec, 5 mcparicle only sec from decays, 6 mcparticle only sec from material. Default 0"};
-  Configurable<o2::analysis::TrackSelectionTuneCfg> cfgTuneTrackSelection{"tunetracksel", {}, "Track selection: {useit: true/false, tpccls-useit, tpcxrws-useit, tpcxrfc-useit, dcaxy-useit, dcaz-useit}. Default {false,0.70,false,0.8,false,2.4,false,3.2,false}"};
+  Configurable<o2::analysis::TrackSelectionTuneCfg> cfgTuneTrackSelection{"tunetracksel", {}, "Track selection: {useit: true/false, tpccls-useit, tpcxrws-useit, tpcxrfc-useit, tpcshcls-useit, dcaxy-useit, dcaz-useit}. Default {false,0.70,false,0.8,false,0.4,false,2.4,false,3.2,false}"};
   Configurable<o2::analysis::TrackSelectionPIDCfg> cfgPionPIDSelection{"pipidsel",
                                                                        {},
                                                                        "PID criteria for pions"};
@@ -835,7 +835,7 @@ struct DptDptFilterTracks {
 
     /* the track types and combinations */
     tracktype = cfgTrackType.value;
-    initializeTrackSelection(cfgTuneTrackSelection);
+    initializeTrackSelection(cfgTuneTrackSelection.value);
     traceDCAOutliers = cfgTraceDCAOutliers;
     traceOutOfSpeciesParticles = cfgTraceOutOfSpeciesParticles;
     recoIdMethod = cfgRecoIdMethod;

PWGCF/TableProducer/dptdptfilter.h

@@ -210,6 +210,7 @@ std::function<float(float)> maxDcaZPtDep{}; // max dca in z axis as function of
 
 std::vector<TrackSelection*> trackFilters = {};
 bool dca2Dcut = false;
+float sharedTpcClusters = 1.0; ///< max fraction of shared TPC clusters
 float maxDCAz = 1e6f;
 float maxDCAxy = 1e6f;
 
@@ -236,7 +237,7 @@ inline TList* getCCDBInput(auto& ccdb, const char* ccdbpath, const char* ccdbdat
   return lst;
 }
 
-inline void initializeTrackSelection(const TrackSelectionTuneCfg& tune)
+inline void initializeTrackSelection(TrackSelectionTuneCfg& tune)
 {
   switch (tracktype) {
     case 1: { /* Run2 global track */
@@ -341,12 +342,20 @@ inline void initializeTrackSelection(const TrackSelectionTuneCfg& tune)
         filter->SetMinNCrossedRowsOverFindableClustersTPC(tune.mTPCXRoFClusters);
       }
       if (tune.mUseDCAxy) {
+        /* DCAxy is tricky due to how the pT dependence is implemented */
+        filter->SetMaxDcaXYPtDep([&tune](float) { return tune.mDCAxy; });
         filter->SetMaxDcaXY(tune.mDCAxy);
       }
       if (tune.mUseDCAz) {
         filter->SetMaxDcaZ(tune.mDCAz);
       }
     }
+    if (tune.mUseDCAz) {
+      maxDcaZPtDep = [&tune](float) { return tune.mDCAz; };
+    }
+    if (tune.mUseFractionTpcSharedClusters) {
+      sharedTpcClusters = tune.mFractionTpcSharedClusters;
+    }
   }
 }
 
@@ -1128,6 +1137,10 @@ inline bool matchTrackType(TrackObject const& track)
         if (!checkDca2Dcut(track)) {
           return false;
         }
+        /* shared fraction of TPC clusters */
+        if (!(track.tpcFractionSharedCls() < sharedTpcClusters)) {
+          return false;
+        }
         return true;
       }
     }

PWGCF/Tasks/matchRecoGen.cxx

@@ -120,7 +120,7 @@ struct MatchRecoGen {
 
     /* the track types and combinations */
     tracktype = cfgTrackType.value;
-    initializeTrackSelection(cfgTuneTrackSelection);
+    initializeTrackSelection(cfgTuneTrackSelection.value);
     /* the centrality/multiplicity estimation */
     fCentMultEstimator = getCentMultEstimator(cfgCentMultEstimator);
     /* the trigger selection */