Update QAHistTask.cxx (#4003)

Added Ka & Pi.

Author: RafRaf11

PWGLF/Tasks/QAHistTask.cxx

@@ -53,12 +53,16 @@ struct QAHistTask {
   void init(o2::framework::InitContext&)
   {
 
-    if ((process_proton == true && (process_deuteron == true || process_triton == true || process_He3 == true || process_He4 == true)) || (process_deuteron == true && (process_triton == true || process_He3 == true || process_He4 == true)) || (process_triton == true && (process_He3 == true || process_He4 == true)) || (process_He3 == true && process_He4 == true)) {
+    if ((process_pion == true && (process_kaon == true || process_proton == true || process_deuteron == true || process_triton == true || process_He3 == true || process_He4 == true)) || (process_kaon == true && (process_proton == true || process_deuteron == true || process_triton == true || process_He3 == true || process_He4 == true)) || (process_proton == true && (process_deuteron == true || process_triton == true || process_He3 == true || process_He4 == true)) || (process_deuteron == true && (process_triton == true || process_He3 == true || process_He4 == true)) || (process_triton == true && (process_He3 == true || process_He4 == true)) || (process_He3 == true && process_He4 == true)) {
       LOG(fatal) << "++++++++ Can't enable more than one species at a time, use subwagons for that purpose. ++++++++";
     }
 
     std::string species;
 
+    if (process_pion)
+      species = "pi";
+    if (process_kaon)
+      species = "ka";
     if (process_proton)
       species = "p";
     if (process_deuteron)
@@ -198,6 +202,8 @@ struct QAHistTask {
   }
 
   // Configurables
+  Configurable<bool> process_pion{"process_pion", false, "0: disabled, 1: enabled"};
+  Configurable<bool> process_kaon{"process_kaon", false, "0: disabled, 1: enabled"};
   Configurable<bool> process_proton{"process_proton", false, "0: disabled, 1: enabled"};
   Configurable<bool> process_deuteron{"process_deuteron", false, "0: disabled, 1: enabled"};
   Configurable<bool> process_triton{"process_triton", false, "0: disabled, 1: enabled"};
@@ -255,6 +261,10 @@ struct QAHistTask {
 
       float nSigmaSpecies = 999.0;
 
+      if (process_pion)
+        nSigmaSpecies = track.tpcNSigmaPi();
+      if (process_kaon)
+        nSigmaSpecies = track.tpcNSigmaKa();
       if (process_proton)
         nSigmaSpecies = track.tpcNSigmaPr();
       if (process_deuteron)
@@ -360,7 +370,7 @@ struct QAHistTask {
         QA_reg.fill(HIST("histTofSignalData"), track.tpcInnerParam() * track.sign(), track.beta());
       }
 
-      // fill QA histograms (proton)
+      // fill QA histograms
       if (TMath::Abs(nSigmaSpecies) < nsigmacut) {
         if (track.sign() > 0) {
           QA_species_pos.fill(HIST("histDcaVsPtData"), track.pt(), track.dcaXY());
@@ -483,11 +493,11 @@ struct QAHistTask {
 
   using EventCandidatesDataCent = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs>>;
 
-  using TrackCandidatesData = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::pidTPCLfFullPr, aod::pidTOFFullPr, aod::pidTPCLfFullDe, aod::pidTOFFullDe, aod::pidTPCLfFullTr, aod::pidTOFFullTr, aod::pidTPCLfFullHe, aod::pidTOFFullHe, aod::pidTPCLfFullAl, aod::pidTOFFullAl, aod::TrackSelection, aod::TrackSelectionExtension, aod::TOFSignal, aod::pidTOFmass, aod::pidTOFbeta>>;
+  using TrackCandidatesData = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::pidTPCLfFullPi, aod::pidTOFFullPi, aod::pidTPCLfFullKa, aod::pidTOFFullKa, aod::pidTPCLfFullPr, aod::pidTOFFullPr, aod::pidTPCLfFullDe, aod::pidTOFFullDe, aod::pidTPCLfFullTr, aod::pidTOFFullTr, aod::pidTPCLfFullHe, aod::pidTOFFullHe, aod::pidTPCLfFullAl, aod::pidTOFFullAl, aod::TrackSelection, aod::TrackSelectionExtension, aod::TOFSignal, aod::pidTOFmass, aod::pidTOFbeta>>;
 
   using EventCandidatesMC = soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::CentFT0Cs>>;
 
-  using TrackCandidatesMC = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::pidTPCLfFullPr, aod::pidTOFFullPr, aod::pidTPCLfFullDe, aod::pidTOFFullDe, aod::pidTPCLfFullTr, aod::pidTOFFullTr, aod::pidTPCLfFullHe, aod::pidTOFFullHe, aod::pidTPCLfFullAl, aod::pidTOFFullAl, aod::TrackSelection, aod::TrackSelectionExtension, aod::TOFSignal, aod::pidTOFmass, aod::pidTOFbeta>>;
+  using TrackCandidatesMC = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::pidTPCLfFullPi, aod::pidTOFFullPi, aod::pidTPCLfFullKa, aod::pidTOFFullKa, aod::pidTPCLfFullPr, aod::pidTOFFullPr, aod::pidTPCLfFullDe, aod::pidTOFFullDe, aod::pidTPCLfFullTr, aod::pidTOFFullTr, aod::pidTPCLfFullHe, aod::pidTOFFullHe, aod::pidTPCLfFullAl, aod::pidTOFFullAl, aod::TrackSelection, aod::TrackSelectionExtension, aod::TOFSignal, aod::pidTOFmass, aod::pidTOFbeta>>;
 
   void processData(EventCandidatesData::iterator const& event, TrackCandidatesData const& tracks)
   {