PWGHF: Add Perfect PID case for D0 PID performance (#142)

Author: skundu692

PWGHF/DataModel/HFCandidateSelectionTables.h

@@ -35,13 +35,15 @@ namespace hf_selcandidate_d0_ALICE3_Barrel
 {
 DECLARE_SOA_COLUMN(IsSelHFFlag, isSelHFFlag, int);                     //!
 DECLARE_SOA_COLUMN(IsSelD0NoPID, isSelD0NoPID, int);                   //!
+DECLARE_SOA_COLUMN(IsSelD0PerfectPID, isSelD0PerfectPID, int);         //!
 DECLARE_SOA_COLUMN(IsSelD0TOFPID, isSelD0TOFPID, int);                 //!
 DECLARE_SOA_COLUMN(IsSelD0RICHPID, isSelD0RICHPID, int);               //!
 DECLARE_SOA_COLUMN(IsSelD0TOFplusRICHPID, isSelD0TOFplusRICHPID, int); //!
 } // namespace hf_selcandidate_d0_ALICE3_Barrel
 DECLARE_SOA_TABLE(HFSelD0CandidateALICE3Barrel, "AOD", "HFSELD0CANDA3B", //!
                   hf_selcandidate_d0_ALICE3_Barrel::IsSelHFFlag,
                   hf_selcandidate_d0_ALICE3_Barrel::IsSelD0NoPID,
+                  hf_selcandidate_d0_ALICE3_Barrel::IsSelD0PerfectPID,
                   hf_selcandidate_d0_ALICE3_Barrel::IsSelD0TOFPID,
                   hf_selcandidate_d0_ALICE3_Barrel::IsSelD0RICHPID,
                   hf_selcandidate_d0_ALICE3_Barrel::IsSelD0TOFplusRICHPID);

PWGHF/TableProducer/HFD0CandidateSelectorALICE3Barrel.cxx

@@ -189,7 +189,7 @@ struct HFD0CandidateSelectorALICE3Barrel {
     return true;
   }
 
-  using Trks = soa::Join<aod::BigTracksPID, aod::Tracks, aod::RICHTracksIndex, aod::TracksExtra>;
+  using Trks = soa::Join<aod::BigTracksPID, aod::Tracks, aod::RICHTracksIndex, aod::McTrackLabels, aod::TracksExtra>;
   void process(aod::HfCandProng2 const& candidates, Trks const& barreltracks, const aod::McParticles& mcParticles, const aod::RICHs&, const aod::FRICHs&)
   {
 
@@ -198,19 +198,20 @@ struct HFD0CandidateSelectorALICE3Barrel {
       // selection flag
       int statusHFFlag = 0;
       int statusD0NoPID = 0;
+      int statusD0PerfectPID = 0;
       int statusD0TOFPID = 0;
       int statusD0RICHPID = 0;
       int statusD0TOFplusRICHPID = 0;
 
       if (!(candidate.hfflag() & 1 << DecayType::D0ToPiK)) {
-        hfSelD0CandidateALICE3Barrel(statusHFFlag, statusD0NoPID, statusD0TOFPID, statusD0RICHPID, statusD0TOFplusRICHPID);
+        hfSelD0CandidateALICE3Barrel(statusHFFlag, statusD0NoPID, statusD0PerfectPID, statusD0TOFPID, statusD0RICHPID, statusD0TOFplusRICHPID);
         continue;
       }
       statusHFFlag = 1;
 
       // conjugate-independent topological selection
       if (!selectionTopol(candidate)) {
-        hfSelD0CandidateALICE3Barrel(statusHFFlag, statusD0NoPID, statusD0TOFPID, statusD0RICHPID, statusD0TOFplusRICHPID);
+        hfSelD0CandidateALICE3Barrel(statusHFFlag, statusD0NoPID, statusD0PerfectPID, statusD0TOFPID, statusD0RICHPID, statusD0TOFplusRICHPID);
         continue;
       }
 
@@ -224,10 +225,15 @@ struct HFD0CandidateSelectorALICE3Barrel {
       bool topolD0bar = selectionTopolConjugate(candidate, trackNeg, trackPos);
 
       if (!topolD0 && !topolD0bar) {
-        hfSelD0CandidateALICE3Barrel(statusHFFlag, statusD0NoPID, statusD0TOFPID, statusD0RICHPID, statusD0TOFplusRICHPID);
+        hfSelD0CandidateALICE3Barrel(statusHFFlag, statusD0NoPID, statusD0PerfectPID, statusD0TOFPID, statusD0RICHPID, statusD0TOFplusRICHPID);
         continue;
       }
 
+      const auto mcParticlePositive = trackPos.mcParticle();
+      const auto mcParticleNegative = trackNeg.mcParticle();
+      int pdgPositive = mcParticlePositive.pdgCode();
+      int pdgNegative = mcParticleNegative.pdgCode();
+      //std::cout << "barrel= " << pdgPositive <<"\t"<< pdgNegative << "\n";
       float nsigmaTOFNegKaon = -5000.0;
       float nsigmaRICHNegKaon = -5000.0;
       float nsigmaTOFPosPion = -5000.0;
@@ -261,14 +267,16 @@ struct HFD0CandidateSelectorALICE3Barrel {
 
       if (topolD0) {
         statusD0NoPID = 1;
+        if (pdgPositive == 211 && pdgNegative == -321)
+          statusD0PerfectPID = 1;
         if ((std::abs(nsigmaTOFPosPion) < 3.0 && std::abs(nsigmaTOFNegKaon) < 3.0))
           statusD0TOFPID = 1;
         if ((std::abs(nsigmaRICHPosPion) < 3.0 && std::abs(nsigmaRICHNegKaon) < 3.0))
           statusD0RICHPID = 1;
         if (selectPionTOFplusRICH && selectKaonTOFplusRICH)
           statusD0TOFplusRICHPID = 1;
       }
-      hfSelD0CandidateALICE3Barrel(statusHFFlag, statusD0NoPID, statusD0TOFPID, statusD0RICHPID, statusD0TOFplusRICHPID);
+      hfSelD0CandidateALICE3Barrel(statusHFFlag, statusD0NoPID, statusD0PerfectPID, statusD0TOFPID, statusD0RICHPID, statusD0TOFplusRICHPID);
     }
   }
 };

PWGHF/Tasks/taskD0ALICE3Barrel.cxx

@@ -52,7 +52,10 @@ struct TaskD0ALICE3BarrelMC {
      {"hMassReflBkgD0RICHPID", "2-prong candidates (checked);#it{m}_{inv} (GeV/#it{c}^{2}); #it{p}_{T}; #it{y}", {HistType::kTH3F, {{120, 1.5848, 2.1848}, {150, 0., 30.}, {8, 0., 4.}}}},
      {"hMassSigBkgD0TOFplusRICHPID", "2-prong candidates (not checked);#it{m}_{inv} (GeV/#it{c}^{2}); #it{p}_{T}; #it{y}", {HistType::kTH3F, {{120, 1.5848, 2.1848}, {150, 0., 30.}, {8, 0, 4.0}}}},
      {"hMassSigD0TOFplusRICHPID", "2-prong candidates (matched);#it{m}_{inv} (GeV/#it{c}^{2}); #it{p}_{T}; #it{y}", {HistType::kTH3F, {{120, 1.5848, 2.1848}, {150, 0., 30.}, {8, 0., 4.}}}},
-     {"hMassBkgD0TOFplusRICHPID", "2-prong candidates (checked);#it{m}_{inv} (GeV/#it{c}^{2}); #it{p}_{T}; #it{y}", {HistType::kTH3F, {{120, 1.5848, 2.1848}, {150, 0., 30.}, {8, 0., 4.}}}}}};
+     {"hMassBkgD0TOFplusRICHPID", "2-prong candidates (checked);#it{m}_{inv} (GeV/#it{c}^{2}); #it{p}_{T}; #it{y}", {HistType::kTH3F, {{120, 1.5848, 2.1848}, {150, 0., 30.}, {8, 0., 4.}}}},
+     {"hMassSigBkgD0PerfectPID", "2-prong candidates (not checked);#it{m}_{inv} (GeV/#it{c}^{2}); #it{p}_{T}; #it{y}", {HistType::kTH3F, {{120, 1.5848, 2.1848}, {150, 0., 30.}, {8, 0, 4.0}}}},
+     {"hMassSigD0PerfectPID", "2-prong candidates (matched);#it{m}_{inv} (GeV/#it{c}^{2}); #it{p}_{T}; #it{y}", {HistType::kTH3F, {{120, 1.5848, 2.1848}, {150, 0., 30.}, {8, 0., 4.}}}},
+     {"hMassBkgD0PerfectPID", "2-prong candidates (checked);#it{m}_{inv} (GeV/#it{c}^{2}); #it{p}_{T}; #it{y}", {HistType::kTH3F, {{120, 1.5848, 2.1848}, {150, 0., 30.}, {8, 0., 4.}}}}}};
 
   Filter filterSelectCandidates = (aod::hf_selcandidate_d0_ALICE3_Barrel::isSelHFFlag >= 1);
 
@@ -109,6 +112,14 @@ struct TaskD0ALICE3BarrelMC {
           registry.fill(HIST("hMassBkgD0TOFplusRICHPID"), massD0, ptCandidate, rapidityCandidate);
         }
       }
+      if (candidate.isSelD0PerfectPID() >= 1) {
+        registry.fill(HIST("hMassSigBkgD0PerfectPID"), massD0, ptCandidate, rapidityCandidate);
+        if (candidate.flagMCMatchRec() == (1 << DecayType::D0ToPiK)) {
+          registry.fill(HIST("hMassSigD0PerfectPID"), massD0, ptCandidate, rapidityCandidate);
+        } else {
+          registry.fill(HIST("hMassBkgD0PerfectPID"), massD0, ptCandidate, rapidityCandidate);
+        }
+      }
     }
 
     for (auto& particle : particlesMC) {