[PWGCF] flow: add supicious Runs removal, DCAz pt dependent cut, DCAxy ppPass3 cut … (AliceO2Group#8253)

Author: Luzhiyongg

PWGCF/Flow/Tasks/FlowRunbyRun.cxx

@@ -17,6 +17,9 @@
 #include <cmath>
 #include <vector>
 #include <map>
+#include <string>
+#include <memory>
+#include <utility>
 #include "Framework/runDataProcessing.h"
 #include "Framework/AnalysisTask.h"
 #include "Framework/ASoAHelpers.h"
@@ -106,8 +109,7 @@ struct FlowRunbyRun {
     ccdb->setCreatedNotAfter(nolaterthan.value);
 
     // Add output histograms to the registry
-    std::vector<int> temp = cfgRunNumbers;
-    RunNumbers = temp;
+    RunNumbers = cfgRunNumbers;
     for (auto& runNumber : RunNumbers) {
       CreateOutputObjectsForRun(runNumber);
     }

PWGCF/Flow/Tasks/FlowTask.cxx

@@ -16,6 +16,9 @@
 #include <DataFormatsParameters/GRPMagField.h>
 #include <cmath>
 #include <vector>
+#include <unordered_map>
+#include <string>
+#include <memory>
 #include "Framework/runDataProcessing.h"
 #include "Framework/AnalysisTask.h"
 #include "Framework/ASoAHelpers.h"
@@ -59,6 +62,8 @@ struct FlowTask {
   O2_DEFINE_CONFIGURABLE(cfgCutChi2prTPCcls, float, 2.5f, "max chi2 per TPC clusters")
   O2_DEFINE_CONFIGURABLE(cfgCutTPCclu, float, 70.0f, "minimum TPC clusters")
   O2_DEFINE_CONFIGURABLE(cfgCutDCAz, float, 2.0f, "max DCA to vertex z")
+  O2_DEFINE_CONFIGURABLE(cfgCutDCAxyppPass3Enabled, bool, false, "switch of ppPass3 DCAxy pt dependent cut")
+  O2_DEFINE_CONFIGURABLE(cfgCutDCAzPtDepEnabled, bool, false, "switch of DCAz pt dependent cut")
   O2_DEFINE_CONFIGURABLE(cfgTrkSelSwitch, bool, false, "switch for self-defined track selection")
   O2_DEFINE_CONFIGURABLE(cfgTrkSelRun3ITSMatch, bool, false, "GlobalTrackRun3ITSMatching::Run3ITSall7Layers selection")
   O2_DEFINE_CONFIGURABLE(cfgRejectionTPCsectorOverlap, bool, true, "rejection for TPC sector overlap")
@@ -84,6 +89,7 @@ struct FlowTask {
   O2_DEFINE_CONFIGURABLE(cfgUseSmallMemory, bool, false, "Use small memory mode")
   Configurable<std::vector<std::string>> cfgUserDefineGFWCorr{"cfgUserDefineGFWCorr", std::vector<std::string>{"refN02 {2} refP02 {-2}", "refN12 {2} refP12 {-2}"}, "User defined GFW CorrelatorConfig"};
   Configurable<std::vector<std::string>> cfgUserDefineGFWName{"cfgUserDefineGFWName", std::vector<std::string>{"Ch02Gap22", "Ch12Gap22"}, "User defined GFW Name"};
+  Configurable<std::vector<int>> cfgRunRemoveList{"cfgRunRemoveList", std::vector<int>{-1}, "excluded run numbers"};
 
   ConfigurableAxis axisVertex{"axisVertex", {40, -20, 20}, "vertex axis for histograms"};
   ConfigurableAxis axisPhi{"axisPhi", {60, 0.0, constants::math::TwoPI}, "phi axis for histograms"};
@@ -168,7 +174,7 @@ struct FlowTask {
     registry.add("hEventCount", "Number of Event;; Count", {HistType::kTH1D, {{5, 0, 5}}});
     registry.get<TH1>(HIST("hEventCount"))->GetXaxis()->SetBinLabel(1, "Filtered event");
     registry.get<TH1>(HIST("hEventCount"))->GetXaxis()->SetBinLabel(2, "after sel8");
-    registry.get<TH1>(HIST("hEventCount"))->GetXaxis()->SetBinLabel(3, "after strict Pile-up cut");
+    registry.get<TH1>(HIST("hEventCount"))->GetXaxis()->SetBinLabel(3, "after supicious Runs removal");
     registry.get<TH1>(HIST("hEventCount"))->GetXaxis()->SetBinLabel(4, "after additional event cut");
     registry.get<TH1>(HIST("hEventCount"))->GetXaxis()->SetBinLabel(5, "after correction loads");
     registry.add("hVtxZ", "Vexter Z distribution", {HistType::kTH1D, {axisVertex}});
@@ -310,6 +316,8 @@ struct FlowTask {
     fGFW->AddRegion("refP10", 0.5, 0.8, 1, 1);
     fGFW->AddRegion("refN12", -0.8, -0.6, 1, 1);
     fGFW->AddRegion("refP12", 0.6, 0.8, 1, 1);
+    fGFW->AddRegion("refN14", -0.8, -0.7, 1, 1);
+    fGFW->AddRegion("refP14", 0.7, 0.8, 1, 1);
     fGFW->AddRegion("refN", -0.8, -0.4, 1, 1);
     fGFW->AddRegion("refP", 0.4, 0.8, 1, 1);
     fGFW->AddRegion("refM", -0.4, 0.4, 1, 1);
@@ -398,6 +406,8 @@ struct FlowTask {
       myTrackSel = getGlobalTrackSelectionRun3ITSMatch(TrackSelection::GlobalTrackRun3ITSMatching::Run3ITSibAny, TrackSelection::GlobalTrackRun3DCAxyCut::Default);
     }
     myTrackSel.SetMinNClustersTPC(cfgCutTPCclu);
+    if (cfgCutDCAxyppPass3Enabled)
+      myTrackSel.SetMaxDcaXYPtDep([](float pt) { return 0.004f + 0.013f / pt; }); // Tuned on the LHC22f anchored MC LHC23d1d on primary pions. 7 Sigmas of the resolution
   }
 
   template <char... chars>
@@ -582,6 +592,9 @@ struct FlowTask {
   template <typename TTrack>
   bool trackSelected(TTrack track)
   {
+    if (cfgCutDCAzPtDepEnabled && (track.dcaZ() > (0.004f + 0.013f / track.pt())))
+      return false;
+
     if (cfgTrkSelSwitch) {
       return myTrackSel.IsSelected(track);
     } else {
@@ -633,6 +646,15 @@ struct FlowTask {
       return;
     if (tracks.size() < 1)
       return;
+    registry.fill(HIST("hEventCount"), 1.5);
+    auto bc = collision.bc_as<aod::BCsWithTimestamps>();
+    int currentRunNumber = bc.runNumber();
+    for (auto& ExcludedRun : cfgRunRemoveList.value) {
+      if (currentRunNumber == ExcludedRun) {
+        return;
+      }
+    }
+    registry.fill(HIST("hEventCount"), 2.5);
     if (!cfgUseSmallMemory) {
       registry.fill(HIST("BeforeCut_globalTracks_centT0C"), collision.centFT0C(), tracks.size());
       registry.fill(HIST("BeforeCut_PVTracks_centT0C"), collision.centFT0C(), collision.multNTracksPV());
@@ -642,9 +664,6 @@ struct FlowTask {
       registry.fill(HIST("BeforeCut_multV0A_multT0A"), collision.multFT0A(), collision.multFV0A());
       registry.fill(HIST("BeforeCut_multT0C_centT0C"), collision.centFT0C(), collision.multFT0C());
     }
-    registry.fill(HIST("hEventCount"), 1.5);
-    // place holder for pile-up rejection
-    registry.fill(HIST("hEventCount"), 2.5);
     const auto cent = collision.centFT0C();
     if (cfgUseAdditionalEventCut && !eventSelected(collision, tracks.size(), cent))
       return;
@@ -655,7 +674,6 @@ struct FlowTask {
     registry.fill(HIST("hMult"), tracks.size());
     registry.fill(HIST("hCent"), collision.centFT0C());
     fGFW->Clear();
-    auto bc = collision.bc_as<aod::BCsWithTimestamps>();
     if (cfgGetInteractionRate) {
       initHadronicRate(bc);
       double hadronicRate = mRateFetcher.fetch(ccdb.service, bc.timestamp(), mRunNumber, "ZNC hadronic") * 1.e-3; //