[PWGCF] Fixed bug and added some QA histos (#9797)

Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: cnkoster

PWGCF/Flow/Tasks/flowSP.cxx

@@ -162,6 +162,16 @@ struct FlowSP {
     nEventSelections
   };
 
+  enum TrackSelections {
+    trackSel_FilteredTracks,
+    trackSel_NCls,
+    trackSel_FshCls,
+    trackSel_TPCBoundary,
+    trackSel_ZeroCharge,
+    trackSel_ParticleWeights,
+    nTrackSelections
+  };
+
   enum ChargeType {
     kInclusive,
     kPositive,
@@ -190,6 +200,8 @@ struct FlowSP {
     AxisSpec t0cAxis = {70, 0, 70000, "N_{ch} (T0C)"};
     AxisSpec t0aAxis = {200, 0, 200, "N_{ch}"};
     AxisSpec multpvAxis = {4000, 0, 4000, "N_{ch} (PV)"};
+    AxisSpec shclAxis = {200, 0, 1, "Fraction shared cl. TPC"};
+    AxisSpec clAxis = {160, 0, 160, "Number of cl. TPC"};
 
     int ptbins = ptbinning.size() - 1;
 
@@ -315,8 +327,10 @@ struct FlowSP {
       registry.add<TH1>("incl/QA/hPhiCorrected", "", kTH1D, {axisPhi});
       registry.add<TH1>("incl/QA/hEta", "", kTH1D, {axisEta});
       registry.add<TH3>("incl/QA/hPhi_Eta_vz", "", kTH3D, {axisPhi, axisEta, axisVz});
-      registry.add<TH1>("incl/QA/hDCAxy", "", kTH1D, {axisDCAxy});
-      registry.add<TH1>("incl/QA/hDCAz", "", kTH1D, {axisDCAz});
+      registry.add<TH2>("incl/QA/hDCAxy_pt", "", kTH2D, {axisPt, axisDCAxy});
+      registry.add<TH2>("incl/QA/hDCAz_pt", "", kTH2D, {axisPt, axisDCAz});
+      registry.add("incl/QA/hSharedClusters_pt", "", {HistType::kTH2D, {axisPt, shclAxis}});
+      registry.add("incl/QA/hCrossedRows_pt", "", {HistType::kTH2D, {axisPt, clAxis}});
 
       registry.addClone("incl/", "pos/");
       registry.addClone("incl/", "neg/");
@@ -340,6 +354,14 @@ struct FlowSP {
     registry.get<TH1>(HIST("hEventCount"))->GetXaxis()->SetBinLabel(evSel_kIsGoodITSLayersAll + 1, "kkIsGoodITSLayersAll");
     registry.get<TH1>(HIST("hEventCount"))->GetXaxis()->SetBinLabel(evSel_isSelectedZDC + 1, "isSelected");
 
+    registry.add("hTrackCount", "Number of Tracks; Cut; #Tracks Passed Cut", {HistType::kTH1D, {{nTrackSelections, 0, nTrackSelections}}});
+    registry.get<TH1>(HIST("hTrackCount"))->GetXaxis()->SetBinLabel(trackSel_FilteredTracks + 1, "Filtered Track");
+    registry.get<TH1>(HIST("hTrackCount"))->GetXaxis()->SetBinLabel(trackSel_NCls + 1, "nClusters TPC");
+    registry.get<TH1>(HIST("hTrackCount"))->GetXaxis()->SetBinLabel(trackSel_FshCls + 1, "Frac. sh. Cls TPC");
+    registry.get<TH1>(HIST("hTrackCount"))->GetXaxis()->SetBinLabel(trackSel_TPCBoundary + 1, "TPC Boundary");
+    registry.get<TH1>(HIST("hTrackCount"))->GetXaxis()->SetBinLabel(trackSel_ZeroCharge + 1, "Only charged");
+    registry.get<TH1>(HIST("hTrackCount"))->GetXaxis()->SetBinLabel(trackSel_ParticleWeights + 1, "Apply weights");
+
     if (cfgUseAdditionalEventCut) {
       fMultPVCutLow = new TF1("fMultPVCutLow", "[0]+[1]*x+[2]*x*x+[3]*x*x*x+[4]*x*x*x*x - 3.5*([5]+[6]*x+[7]*x*x+[8]*x*x*x+[9]*x*x*x*x)", 0, 100);
       fMultPVCutLow->SetParameters(3257.29, -121.848, 1.98492, -0.0172128, 6.47528e-05, 154.756, -1.86072, -0.0274713, 0.000633499, -3.37757e-06);
@@ -535,9 +557,11 @@ struct FlowSP {
 
     if (track.tpcNClsFound() < cfgNcls)
       return false;
+    registry.fill(HIST("hTrackCount"), trackSel_NCls);
 
-    if (track.tpcFractionSharedCls() < cfgFshcls)
+    if (track.tpcFractionSharedCls() > cfgFshcls)
       return false;
+    registry.fill(HIST("hTrackCount"), trackSel_FshCls);
 
     double phimodn = track.phi();
     if (field < 0) // for negative polarity field
@@ -556,6 +580,7 @@ struct FlowSP {
         return false; // reject track
     }
     registry.fill(HIST("QA/after/pt_phi"), track.pt(), phimodn);
+    registry.fill(HIST("hTrackCount"), trackSel_TPCBoundary);
     return true;
   }
 
@@ -679,8 +704,10 @@ struct FlowSP {
     registry.fill(HIST(Charge[ct]) + HIST("QA/hPhiCorrected"), track.phi(), wacc * weff);
     registry.fill(HIST(Charge[ct]) + HIST("QA/hEta"), track.eta());
     registry.fill(HIST(Charge[ct]) + HIST("QA/hPhi_Eta_vz"), track.phi(), track.eta(), vz);
-    registry.fill(HIST(Charge[ct]) + HIST("QA/hDCAxy"), track.dcaXY());
-    registry.fill(HIST(Charge[ct]) + HIST("QA/hDCAz"), track.dcaZ());
+    registry.fill(HIST(Charge[ct]) + HIST("QA/hDCAxy_pt"), track.pt(), track.dcaXY());
+    registry.fill(HIST(Charge[ct]) + HIST("QA/hDCAz_pt"), track.pt(), track.dcaZ());
+    registry.fill(HIST(Charge[ct]) + HIST("QA/hSharedClusters_pt"), track.pt(), track.tpcFractionSharedCls());
+    registry.fill(HIST(Charge[ct]) + HIST("QA/hCrossedRows_pt"), track.pt(), track.tpcNClsFound());
   }
 
   void processData(UsedCollisions::iterator const& collision, aod::BCsWithTimestamps const&, UsedTracks const& tracks)
@@ -785,6 +812,7 @@ struct FlowSP {
 
       for (const auto& track : tracks) {
         registry.fill(HIST("QA/before/hPt_inclusive"), track.pt());
+        registry.fill(HIST("hTrackCount"), trackSel_FilteredTracks);
 
         float weff = 1, wacc = 1;
         float weffP = 1, waccP = 1;
@@ -795,6 +823,7 @@ struct FlowSP {
 
         if (track.sign() == 0.0)
           return;
+        registry.fill(HIST("hTrackCount"), trackSel_ZeroCharge);
         bool pos = (track.sign() > 0) ? true : false;
 
         // Fill NUA weights
@@ -816,6 +845,8 @@ struct FlowSP {
         if (!pos && !setCurrentParticleWeights(kNegative, weffN, waccN, track.phi(), track.eta(), track.pt(), vtxz))
           return;
 
+        registry.fill(HIST("hTrackCount"), trackSel_ParticleWeights);
+
         registry.fill(HIST("QA/after/hPt_inclusive"), track.pt(), wacc * weff);
 
         // // constrain angle to 0 -> [0,0+2pi]