U

Author: njacazio

DPG/Tasks/AOTTrack/PID/qaPIDTPCSignal.cxx

@@ -38,7 +38,15 @@ struct tpcPidQaSignal {
   Configurable<int> trdSelection{"trdSelection", 0, "Flag for the TRD selection: -1 no TRD, 0 no selection, 1 TRD"};
   Configurable<float> fractionOfEvents{"fractionOfEvents", 0.1f, "Downsampling factor for the events for derived data"};
   Configurable<float> minP{"minP", 0.01, "Minimum momentum in range"};
+  Configurable<float> maxEta{"maxEta", 0.8, "Maximum eta in range"};
+  Configurable<float> maxITSChi2{"maxITSChi2", 36, "Maximum chi2 in range"};
+  Configurable<float> maxTPCChi2{"maxTPCChi2", 4, "Maximum chi2 in range"};
   Configurable<float> maxP{"maxP", 20, "Maximum momentum in range"};
+  Configurable<int> minITSCls{"minITSCls", 6, "Minimum number of ITS clusters"};
+  Configurable<int> pidInTracking{"pidInTracking", -1, "PID in tracking"};
+  Configurable<int> minTPCCls{"minTPCCls", 0, "Minimum number of TPC clusters"};
+  Configurable<int> minTPCClsFindable{"minTPCClsFindable", 0, "Minimum number of TPC findable clusters"};
+  Configurable<float> minCrossedRowsOverFindableCls{"minCrossedRowsOverFindableCls", 0.8f, "Minimum number of TPC found/findable clusters"};
   ConfigurableAxis dEdxBins{"dEdxBins", {5000, 0.f, 5000.f}, "Binning in dE/dx"};
   Configurable<float> minTPCNcls{"minTPCNcls", 0.f, "Minimum number or TPC Clusters for tracks"};
   Configurable<float> minNClsCrossedRows{"minNClsCrossedRows", 70.f, "Minimum number or TPC crossed rows for tracks"};
@@ -63,6 +71,25 @@ struct tpcPidQaSignal {
     h->GetXaxis()->SetBinLabel(2, "Passed ev. sel.");
     h->GetXaxis()->SetBinLabel(3, "Passed vtx Z");
 
+    // Event properties
+    h = histos.add<TH1>("trksel", "", kTH1D, {{10, 0.5, 10.5, "Trk. Sel."}});
+    h->GetXaxis()->SetBinLabel(1, "Tracks read");
+    h->GetXaxis()->SetBinLabel(2, Form("Has > %i ITS clusters", minITSCls.value));
+    h->GetXaxis()->SetBinLabel(3, Form("Has > %f TPC clusters findable", minTPCClsFindable.value));
+    h->GetXaxis()->SetBinLabel(4, Form("Has > %f TPC clusters found", minTPCNcls.value));
+    h->GetXaxis()->SetBinLabel(5, Form("Has > %f Found/Findable Ratio", minCrossedRowsOverFindableCls.value));
+    h->GetXaxis()->SetBinLabel(6, Form("Has > %f Xrows", minNClsCrossedRows.value));
+    h->GetXaxis()->SetBinLabel(7, "All PID in trk");
+    if (pidInTracking == -1) {
+      h->GetXaxis()->SetBinLabel(7, Form("PID in trk %i", pidInTracking.value));
+    }
+    h->GetXaxis()->SetBinLabel(8, "no TRD sel.");
+    if (trdSelection == -1) {
+      h->GetXaxis()->SetBinLabel(8, "has no TRD");
+    } else if (trdSelection == 1) {
+      h->GetXaxis()->SetBinLabel(8, "has TRD");
+    }
+
     histos.add("event/vertexz", "", kTH1D, {vtxZAxis});
     hdedx = histos.add<TH3>("event/tpcsignal", "", kTH3D, {pAxis, dedxAxis, chargeAxis});
     LOG(info) << "QA PID TPC histograms:";
@@ -72,43 +99,54 @@ struct tpcPidQaSignal {
   template <typename T>
   bool isTrackSelected(const T& track)
   {
-    if (t.itsNCls() < 6) {
+    histos.fill(HIST("trksel"), 1);
+    if (track.itsNCls() < minITSCls) {
       return false;
     }
-    if (t.tpcNClsFindable() <= 0) {
+    histos.fill(HIST("trksel"), 2);
+    if (track.tpcNClsFindable() <= minTPCClsFindable) {
       return false;
     }
-    if (t.tpcNClsFound() < minTPCNcls) {
+    histos.fill(HIST("trksel"), 3);
+    if (track.tpcNClsFound() < minTPCNcls) {
       return false;
     }
-    if (t.tpcCrossedRowsOverFindableCls() < 0.8f) {
+    histos.fill(HIST("trksel"), 4);
+    if (track.tpcCrossedRowsOverFindableCls() < minCrossedRowsOverFindableCls) {
       return false;
     }
-    if (t.tpcNClsCrossedRows() < minNClsCrossedRows) {
+    histos.fill(HIST("trksel"), 5);
+    if (track.tpcNClsCrossedRows() < minNClsCrossedRows) {
       return false;
     }
+    histos.fill(HIST("trksel"), 6);
+    if (pidInTracking != -1 && track.pidForTracking() != pidInTracking) {
+      return false;
+    }
+    histos.fill(HIST("trksel"), 7);
     switch (trdSelection) {
       case 0:
         break;
       case -1:
-        if (t.hasTRD()) {
+        if (track.hasTRD()) {
           return false;
         }
         break;
       case 1:
-        if (!t.hasTRD()) {
+        if (!track.hasTRD()) {
           return false;
         }
         break;
       default:
         LOG(fatal) << "Invalid TRD selection";
     }
+    histos.fill(HIST("trksel"), 8);
     return true;
   }
 
-  Filter trackFilterEta = (nabs(aod::track::eta) < 0.8f);
-  Filter trackFilterITS = (aod::track::itsChi2NCl < 36.f);
-  Filter trackFilterTPC = ((aod::track::tpcChi2NCl < 4.f));
+  Filter trackFilterEta = (nabs(aod::track::eta) < maxEta);
+  Filter trackFilterITS = (aod::track::itsChi2NCl < maxITSChi2);
+  Filter trackFilterTPC = ((aod::track::tpcChi2NCl < maxTPCChi2));
   using TrackCandidates = soa::Join<aod::TracksIU, aod::TracksExtra>;
   void processEvSel(soa::Join<aod::Collisions, aod::EvSels>::iterator const& collision,
                     soa::Filtered<TrackCandidates> const& tracks,
@@ -166,23 +204,21 @@ struct tpcPidQaSignal {
   }
   PROCESS_SWITCH(tpcPidQaSignal, processEvSel, "Process with event selection", false);
 
-  void processNoEvSel(aod::Collision const& collision,
-                      soa::Filtered<TrackCandidates> const& tracks)
+  void processNoEvSel(soa::Filtered<TrackCandidates> const& tracks, aod::Collisions const& collisions)
   {
     if (fractionOfEvents < 1.f && (static_cast<float>(rand_r(&randomSeed)) / static_cast<float>(RAND_MAX)) > fractionOfEvents) { // Skip events that are not sampled
       return;
     }
 
-    histos.fill(HIST("event/evsel"), 1);
-    histos.fill(HIST("event/evsel"), 2);
-
-    if (abs(collision.posZ()) > 10.f) {
-      return;
-    }
-    histos.fill(HIST("event/evsel"), 3);
-    histos.fill(HIST("event/vertexz"), collision.posZ());
+    histos.fill(HIST("event/evsel"), 1, collisions.size());
 
     for (const auto& t : tracks) {
+      if (!t.has_collision()) {
+        continue;
+      }
+      if (abs(t.collision().posZ()) > 10.f) {
+        continue;
+      }
       if (!isTrackSelected(t)) {
         continue;
       }