[DPG] Update qaEfficiency (#6653)

- To Add the Tertiary  tracks from the secondary weak decay

Author: upasanasharma31

DPG/Tasks/AOTTrack/qaEfficiency.cxx

@@ -107,6 +107,12 @@ std::array<std::shared_ptr<TH1>, nParticles> hPtTrkItsTpcMat;
 std::array<std::shared_ptr<TH1>, nParticles> hPtItsTpcTofMat;
 std::array<std::shared_ptr<TH1>, nParticles> hPtGeneratedMat;
 
+// Pt for tertiaries from secondary weak decay
+std::array<std::shared_ptr<TH1>, nParticles> hPtItsTpcTer;
+std::array<std::shared_ptr<TH1>, nParticles> hPtTrkItsTpcTer;
+std::array<std::shared_ptr<TH1>, nParticles> hPtItsTpcTofTer;
+std::array<std::shared_ptr<TH1>, nParticles> hPtGeneratedTer;
+
 // P
 std::array<std::shared_ptr<TH1>, nParticles> hPItsTpc;
 std::array<std::shared_ptr<TH1>, nParticles> hPTrkItsTpc;
@@ -325,6 +331,12 @@ struct QaEfficiency {
     hPtItsTpcTofStr[histogramIndex] = histos.add<TH1>(Form("MC/pdg%i/pt/str/its_tpc_tof", PDGs[histogramIndex]), "ITS-TPC-TOF tracks (from weak decays) " + tagPt, kTH1D, {axisPt});
     hPtGeneratedStr[histogramIndex] = histos.add<TH1>(Form("MC/pdg%i/pt/str/generated", PDGs[histogramIndex]), "Generated (from weak decays) " + tagPt, kTH1D, {axisPt});
 
+    // Ter
+    hPtItsTpcTer[histogramIndex] = histos.add<TH1>(Form("MC/pdg%i/pt/ter/its_tpc", PDGs[histogramIndex]), "ITS-TPC tracks (from secondary weak decays) " + tagPt, kTH1D, {axisPt});
+    hPtTrkItsTpcTer[histogramIndex] = histos.add<TH1>(Form("MC/pdg%i/pt/ter/trk/its_tpc", PDGs[histogramIndex]), "ITS-TPC tracks (reco from secondary weak decays) " + tagPt, kTH1D, {axisPt});
+    hPtItsTpcTofTer[histogramIndex] = histos.add<TH1>(Form("MC/pdg%i/pt/ter/its_tpc_tof", PDGs[histogramIndex]), "ITS-TPC-TOF tracks (from secondary weak decays) " + tagPt, kTH1D, {axisPt});
+    hPtGeneratedTer[histogramIndex] = histos.add<TH1>(Form("MC/pdg%i/pt/ter/generated", PDGs[histogramIndex]), "Generated (from secondary weak decays) " + tagPt, kTH1D, {axisPt});
+
     // Mat
     hPtItsTpcMat[histogramIndex] = histos.add<TH1>(Form("MC/pdg%i/pt/mat/its_tpc", PDGs[histogramIndex]), "ITS-TPC tracks (from material)" + tagPt, kTH1D, {axisPt});
     hPtTrkItsTpcMat[histogramIndex] = histos.add<TH1>(Form("MC/pdg%i/pt/mat/trk/its_tpc", PDGs[histogramIndex]), "ITS-TPC tracks (reco from material) " + tagPt, kTH1D, {axisPt});
@@ -437,17 +449,19 @@ struct QaEfficiency {
     makeEfficiency("ITS-TPC_vsPt_Prm_Trk", hPtTrkItsTpcPrm[histogramIndex]);
     makeEfficiency("ITS-TPC-TOF_vsPt_Prm", hPtItsTpcTofPrm[histogramIndex]);
     makeEfficiency("ITS-TPC-TOF_vsPt_Prm_Trk", hPtTrkItsTpcTofPrm[histogramIndex]);
-
     makeEfficiency("ITS-TPC_vsPt_Prm_RecoEv", hPtItsTpcPrm[histogramIndex]);
 
     makeEfficiency("ITS-TPC_vsPt_Str", hPtItsTpcStr[histogramIndex]);
     makeEfficiency("ITS-TPC_vsPt_Str_Trk", hPtTrkItsTpcStr[histogramIndex]);
     makeEfficiency("ITS-TPC-TOF_vsPt_Str", hPtItsTpcTofStr[histogramIndex]);
-
     makeEfficiency("ITS-TPC_vsPt_Mat", hPtItsTpcMat[histogramIndex]);
     makeEfficiency("ITS-TPC_vsPt_Mat_Trk", hPtTrkItsTpcMat[histogramIndex]);
     makeEfficiency("ITS-TPC-TOF_vsPt_Mat", hPtItsTpcTofMat[histogramIndex]);
 
+    makeEfficiency("ITS-TPC_vsPt_Ter", hPtItsTpcTer[histogramIndex]);
+    makeEfficiency("ITS-TPC_vsPt_Ter_Trk", hPtTrkItsTpcTer[histogramIndex]);
+    makeEfficiency("ITS-TPC-TOF_vsPt_Ter", hPtItsTpcTofTer[histogramIndex]);
+
     makeEfficiency("ITS-TPC_vsP", hPItsTpc[histogramIndex]);
     makeEfficiency("ITS-TPC_vsP_Trk", hPTrkItsTpc[histogramIndex]);
     makeEfficiency("ITS-TPC-TOF_vsP", hPItsTpcTof[histogramIndex]);
@@ -909,7 +923,24 @@ struct QaEfficiency {
     }
     return mcParticle.isPhysicalPrimary();
   }
-
+  bool isFinal(const o2::aod::McParticles::iterator& mcParticle)
+  {
+    // Example conditions to determine if a particle is final (tertiary)
+    // Here, we assume that final state particles are those not originating from primary vertex
+    // and not further decaying into other particles
+    // Check if the particle has no daughters
+    if (!mcParticle.has_daughters()) {
+
+      // Check if the particle is not a primary particle
+      if (!mcParticle.isPhysicalPrimary()) {
+        // Check if the particle is produced in a secondary decay
+        if (mcParticle.getProcess() == 4) {
+          return true; // Consider it as a tertiary particle
+        }
+      }
+    }
+    return false; // Otherwise, not considered a tertiary particle
+  }
   template <int pdgSign, o2::track::PID::ID id>
   void fillMCTrackHistograms(const TrackCandidatesMC::iterator& track, const bool doMakeHistograms)
   {
@@ -927,7 +958,6 @@ struct QaEfficiency {
         return;
       }
     }
-
     constexpr int histogramIndex = id + pdgSign * nSpecies;
     LOG(debug) << "fillMCTrackHistograms for pdgSign '" << pdgSign << "' and id '" << static_cast<int>(id) << "' " << particleName(pdgSign, id) << " with index " << histogramIndex;
     const o2::aod::McParticles::iterator& mcParticle = track.mcParticle();
@@ -1036,6 +1066,15 @@ struct QaEfficiency {
           hPtItsTpcTofStr[histogramIndex]->Fill(mcParticle.pt());
         }
       }
+      if (isFinal(mcParticle)) {
+        if (passedITS && passedTPC && motherIsAccepted) {
+          hPtItsTpcTer[histogramIndex]->Fill(mcParticle.pt());
+          hPtTrkItsTpcTer[histogramIndex]->Fill(track.pt());
+          if (passedTOF) {
+            hPtItsTpcTofTer[histogramIndex]->Fill(mcParticle.pt());
+          }
+        }
+      }
     } else { // Material
       if (passedITS && passedTPC) {
         hPtItsTpcMat[histogramIndex]->Fill(mcParticle.pt());
@@ -1107,12 +1146,14 @@ struct QaEfficiency {
         }
         if (motherIsAccepted) {
           hPtGeneratedStr[histogramIndex]->Fill(mcParticle.pt());
+          if (isFinal(mcParticle)) {
+            hPtGeneratedTer[histogramIndex]->Fill(mcParticle.pt());
+          }
         }
       } else { // Material
         hPtGeneratedMat[histogramIndex]->Fill(mcParticle.pt());
       }
     }
-
     hEtaGenerated[histogramIndex]->Fill(mcParticle.eta());
     hYGenerated[histogramIndex]->Fill(mcParticle.y());
     hPhiGenerated[histogramIndex]->Fill(mcParticle.phi());
@@ -1191,6 +1232,10 @@ struct QaEfficiency {
     doFillEfficiency("ITS-TPC_vsPt_Mat_Trk", hPtTrkItsTpcMat[histogramIndex], hPtGeneratedMat[histogramIndex]);
     doFillEfficiency("ITS-TPC-TOF_vsPt_Mat", hPtItsTpcTofMat[histogramIndex], hPtGeneratedMat[histogramIndex]);
 
+    doFillEfficiency("ITS-TPC_vsPt_Ter", hPtItsTpcTer[histogramIndex], hPtGeneratedTer[histogramIndex]);
+    doFillEfficiency("ITS-TPC_vsPt_Ter_Trk", hPtTrkItsTpcTer[histogramIndex], hPtGeneratedTer[histogramIndex]);
+    doFillEfficiency("ITS-TPC-TOF_vsPt_Ter", hPtItsTpcTofTer[histogramIndex], hPtGeneratedTer[histogramIndex]);
+
     doFillEfficiency("ITS-TPC_vsP", hPItsTpc[histogramIndex], hPGenerated[histogramIndex]);
     doFillEfficiency("ITS-TPC_vsP_Trk", hPTrkItsTpc[histogramIndex], hPGenerated[histogramIndex]);
     doFillEfficiency("ITS-TPC-TOF_vsP", hPItsTpcTof[histogramIndex], hPGenerated[histogramIndex]);