Add other species

Author: vkucera

PWGHF/TableProducer/pidCreator.cxx

@@ -28,6 +28,14 @@ using namespace o2::framework;
 struct HfPidCreator {
   Produces<aod::TracksPidFullElS> tracksPidFullElS;
   Produces<aod::TracksPidTinyElS> tracksPidTinyElS;
+  Produces<aod::TracksPidFullMuS> tracksPidFullMuS;
+  Produces<aod::TracksPidTinyMuS> tracksPidTinyMuS;
+  Produces<aod::TracksPidFullPiS> tracksPidFullPiS;
+  Produces<aod::TracksPidTinyPiS> tracksPidTinyPiS;
+  Produces<aod::TracksPidFullKaS> tracksPidFullKaS;
+  Produces<aod::TracksPidTinyKaS> tracksPidTinyKaS;
+  Produces<aod::TracksPidFullPrS> tracksPidFullPrS;
+  Produces<aod::TracksPidTinyPrS> tracksPidTinyPrS;
 
   static constexpr float defaultNSigmaTolerance = .1f;
   static constexpr float defaultNSigma = -999.f + defaultNSigmaTolerance; // -999.f is the default value set in TPCPIDResponse.h and PIDTOF.h
@@ -48,6 +56,14 @@ struct HfPidCreator {
   {
     checkTableSwitch(initContext, "TracksPidFullElS", doprocessFullEl);
     checkTableSwitch(initContext, "TracksPidTinyElS", doprocessTinyEl);
+    checkTableSwitch(initContext, "TracksPidFullMuS", doprocessFullMu);
+    checkTableSwitch(initContext, "TracksPidTinyMuS", doprocessTinyMu);
+    checkTableSwitch(initContext, "TracksPidFullPiS", doprocessFullPi);
+    checkTableSwitch(initContext, "TracksPidTinyPiS", doprocessTinyPi);
+    checkTableSwitch(initContext, "TracksPidFullKaS", doprocessFullKa);
+    checkTableSwitch(initContext, "TracksPidTinyKaS", doprocessTinyKa);
+    checkTableSwitch(initContext, "TracksPidFullPrS", doprocessFullPr);
+    checkTableSwitch(initContext, "TracksPidTinyPrS", doprocessTinyPr);
   }
 
   /// Function to combine TPC and TOF NSigma
@@ -77,6 +93,7 @@ struct HfPidCreator {
   void processDummy(aod::Collisions const&) {}
   PROCESS_SWITCH(HfPidCreator, processDummy, "Process nothing", true);
 
+  // Macro for declaring process functions per species
   #define PROCESS_PID(_Species_) \
   void processFull##_Species_(aod::TracksPid##_Species_ const& tracks) \
   { \
@@ -94,7 +111,12 @@ struct HfPidCreator {
   } \
   PROCESS_SWITCH(HfPidCreator, processTiny##_Species_, "Process tiny "#_Species_, false);
 
+  // Declare process functions for all species.
   PROCESS_PID(El)
+  PROCESS_PID(Mu)
+  PROCESS_PID(Pi)
+  PROCESS_PID(Ka)
+  PROCESS_PID(Pr)
 
   #undef PROCESS_PID
 };