provide downsampling

Author: lubynets

DPG/Tasks/TPC/treeCreatorPidTpcDiagnostics.cxx

@@ -41,6 +41,7 @@
 #include <Framework/runDataProcessing.h>
 #include <ReconstructionDataFormats/PID.h>
 
+#include <TRandom3.h>
 #include <TString.h>
 
 #include <boost/preprocessor/cat.hpp>
@@ -89,7 +90,7 @@ struct TreeCreatorPidTpcDiagnostics {
   Configurable<bool> treatLimitedAcceptanceAsBad{"treatLimitedAcceptanceAsBad", false, "reject all events where the detectors relevant for the specified Runlist are flagged as LimitedAcceptance"};
   Configurable<bool> requireGoodRct{"requireGoodRct", false, "require good detector flag in run condtion table"};
 
-  struct ParticleWiseCuts : ConfigurableGroup {
+  struct ParticleWiseCfgs : ConfigurableGroup {
 
 #define DECLARE_PARTICLE_WISE_CONFIGURABLES(Unused1, ParticleNameLong, Unused2)                                                                                                                                                                            \
   Configurable<float> cutTpcInnerParameterMin##ParticleNameLong{"cutTpcInnerParameterMin" #ParticleNameLong, 0.f, "Lower-value cut on tpcInnerParam for " #ParticleNameLong};    /* o2-linter: disable=name/configurable (Configurable defined in macro)*/ \
@@ -99,27 +100,28 @@ struct TreeCreatorPidTpcDiagnostics {
   Configurable<float> cutDeDxExpectedMax##ParticleNameLong{"cutDeDxExpectedMax" #ParticleNameLong, 1e9f, "Upper-value cut on expected dE/dx for " #ParticleNameLong};            /* o2-linter: disable=name/configurable (Configurable defined in macro)*/ \
   Configurable<float> cutDeDxDiffMin##ParticleNameLong{"cutDeDxDiffMin" #ParticleNameLong, -1e9f, "Lower-value cut on real - expected dE/dx difference for " #ParticleNameLong}; /* o2-linter: disable=name/configurable (Configurable defined in macro)*/ \
   Configurable<float> cutDeDxDiffMax##ParticleNameLong{"cutDeDxDiffMax" #ParticleNameLong, 1e9f, "Upper-value cut on real - expected dE/dx difference for " #ParticleNameLong};  /* o2-linter: disable=name/configurable (Configurable defined in macro)*/ \
-  Configurable<float> cutNSigmaTpcAbs##ParticleNameLong{"cutNSigmaTpcAbs" #ParticleNameLong, 999.f, "Cut on absolute value of nSigmaTpc for " #ParticleNameLong};                // o2-linter: disable=name/configurable (Configurable defined in macro)
+  Configurable<float> cutNSigmaTpcAbs##ParticleNameLong{"cutNSigmaTpcAbs" #ParticleNameLong, 999.f, "Cut on absolute value of nSigmaTpc for " #ParticleNameLong};                /* o2-linter: disable=name/configurable (Configurable defined in macro)*/ \
+  Configurable<float> downsampleFactor##ParticleNameLong{"downsampleFactor" #ParticleNameLong, 1.f, "Downsampling factor for " #ParticleNameLong};                               // o2-linter: disable=name/configurable (Configurable defined in macro)
 
     DO_FOR_ALL_PARTICLES(DECLARE_PARTICLE_WISE_CONFIGURABLES, _)
 #undef DECLARE_PARTICLE_WISE_CONFIGURABLES
-  } particleWiseCuts;
+  } particleWiseCfgs;
 
-#define PARTICLE_WISE_CUT_NAMES \
-  (cutTpcInnerParameterMin)(cutTpcInnerParameterMax)(cutExpSigmaMax)(cutDeDxExpectedMin)(cutDeDxExpectedMax)(cutDeDxDiffMin)(cutDeDxDiffMax)(cutNSigmaTpcAbs)
+#define PARTICLE_WISE_CFG_NAMES \
+  (cutTpcInnerParameterMin)(cutTpcInnerParameterMax)(cutExpSigmaMax)(cutDeDxExpectedMin)(cutDeDxExpectedMax)(cutDeDxDiffMin)(cutDeDxDiffMax)(cutNSigmaTpcAbs)(downsampleFactor)
 
-#define MAKE_PARTICLE_CUT_POINTER(Unused, LongName, CutName) \
-  &particleWiseCuts.BOOST_PP_CAT(CutName, LongName),
+#define MAKE_PARTICLE_CUT_POINTER(Unused, ParticleNameLong, CfgName) \
+  &particleWiseCfgs.BOOST_PP_CAT(CfgName, ParticleNameLong),
 
-#define PACK_CONFIGURABLES_TO_ARRAY(Unused1, Unused2, CutName)    \
-  const std::array<Configurable<float>*, PID::Alpha + 1> CutName{ \
-    DO_FOR_ALL_PARTICLES(MAKE_PARTICLE_CUT_POINTER, CutName)};
+#define PACK_CONFIGURABLES_TO_ARRAY(Unused1, Unused2, CfgName)    \
+  const std::array<Configurable<float>*, PID::Alpha + 1> CfgName{ \
+    DO_FOR_ALL_PARTICLES(MAKE_PARTICLE_CUT_POINTER, CfgName)};
 
-  BOOST_PP_SEQ_FOR_EACH(PACK_CONFIGURABLES_TO_ARRAY, _, PARTICLE_WISE_CUT_NAMES)
+  BOOST_PP_SEQ_FOR_EACH(PACK_CONFIGURABLES_TO_ARRAY, _, PARTICLE_WISE_CFG_NAMES)
 
 #undef PACK_CONFIGURABLES_TO_ARRAY
 #undef MAKE_PARTICLE_CUT_POINTER
-#undef PARTICLE_WISE_CUT_NAMES
+#undef PARTICLE_WISE_CFG_NAMES
 
   HistogramRegistry registry{"registry", {}};
 
@@ -129,6 +131,8 @@ struct TreeCreatorPidTpcDiagnostics {
 
   o2::aod::rctsel::RCTFlagsChecker rctChecker;
 
+  TRandom3 mRnd{};
+
   using CollisionsExtra = soa::Join<aod::Collisions, aod::Mults, aod::EvSels>;
   using TrackCandidates = soa::Join<aod::Tracks, aod::TracksExtra, aod::TrackSelection>;
 
@@ -224,6 +228,10 @@ struct TreeCreatorPidTpcDiagnostics {
       const auto tracksFromCollision = tracks.sliceBy(perCollisionTracks, static_cast<int>(collision.globalIndex()));
 
       for (const auto& track : tracksFromCollision) {
+        if (*downsampleFactor.at(ParticleId) < 1.f && mRnd.Uniform(1) > *downsampleFactor.at(ParticleId)) {
+          continue;
+        }
+
         bool isGoodTrack = isTrackSelected(track, trackSelection);
         isGoodTrack &= (!requireGlobalTrack || track.isGlobalTrack());
         isGoodTrack &= (!requireIts || track.hasITS());