[PWGUD] Refactor of personal task (#8709)

Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: rolavick

PWGUD/Core/UPCTauCentralBarrelHelperRL.h

@@ -28,7 +28,23 @@ enum MyParticle {
   P_MUON = 1,
   P_PION = 2,
   P_KAON = 3,
-  P_PROTON = 4
+  P_PROTON = 4,
+  P_ENUM_COUNTER = 5
+};
+
+enum MyTauChannel {
+  CH_EE = 0,
+  CH_MUMU = 1,
+  CH_EMU = 2,
+  CH_PIPI = 3,
+  CH_EPI = 4,
+  CH_MUPI = 5,
+  CH_FOURPI = 6,
+  CH_ETHREEPI = 7,
+  CH_MUTHREEPI = 8,
+  CH_SIXPI = 9,
+  CH_EMUPI = 10,
+  CH_ENUM_COUNTER = 11
 };
 
 void printLargeMessage(std::string info)
@@ -106,18 +122,18 @@ int testPIDhypothesis(T trackPIDinfo, float maxNsigmaTPC = 5.0, float maxNsigmaT
 }
 
 template <typename T>
-int trackPDG(T trackPIDinfo)
+int trackPDG(T trackPIDinfo, float maxNsigmaTPC = 5.0, float maxNsigmaTOF = 5.0, bool useTOF = true, bool useTOFsigmaAfterTPC = true, float nSigmaShift = 0., bool isMC = false)
 // using testPIDhypothesis, reads enumMyParticle and return pdg value
 {
-  if (testPIDhypothesis(trackPIDinfo) == P_ELECTRON) {
+  if (testPIDhypothesis(trackPIDinfo, maxNsigmaTPC, maxNsigmaTOF, useTOF, useTOFsigmaAfterTPC, nSigmaShift, isMC) == P_ELECTRON) {
     return 11;
-  } else if (testPIDhypothesis(trackPIDinfo) == P_MUON) {
+  } else if (testPIDhypothesis(trackPIDinfo, maxNsigmaTPC, maxNsigmaTOF, useTOF, useTOFsigmaAfterTPC, nSigmaShift, isMC) == P_MUON) {
     return 13;
-  } else if (testPIDhypothesis(trackPIDinfo) == P_PION) {
+  } else if (testPIDhypothesis(trackPIDinfo, maxNsigmaTPC, maxNsigmaTOF, useTOF, useTOFsigmaAfterTPC, nSigmaShift, isMC) == P_PION) {
     return 211;
-  } else if (testPIDhypothesis(trackPIDinfo) == P_KAON) {
+  } else if (testPIDhypothesis(trackPIDinfo, maxNsigmaTPC, maxNsigmaTOF, useTOF, useTOFsigmaAfterTPC, nSigmaShift, isMC) == P_KAON) {
     return 321;
-  } else if (testPIDhypothesis(trackPIDinfo) == P_PROTON) {
+  } else if (testPIDhypothesis(trackPIDinfo, maxNsigmaTPC, maxNsigmaTOF, useTOF, useTOFsigmaAfterTPC, nSigmaShift, isMC) == P_PROTON) {
     return 2212;
   } else {
     printDebugMessage("Something is wrong with track PDG selector");
@@ -198,6 +214,32 @@ double calculateAcoplanarity(double phi_trk1, double phi_trk2)
     return (o2::constants::math::TwoPI - aco);
 }
 
+template <typename Ps>
+int countPhysicalPrimary(Ps particles)
+// Function to loop over particles associated to a mcCollision and return total of physical primary particles
+{
+  int nTotal = 0;
+  for (auto& particle : particles) {
+    if (!particle.isPhysicalPrimary())
+      continue;
+    nTotal++;
+  }
+  return nTotal;
+}
+
+template <typename Ps>
+int countParticlesWithoutMother(Ps particles)
+// Function to loop over particles associated to a mcCollision and return total of particles without mothers (hopely alternative to isPhysicalPrimary)
+{
+  int nTotal = 0;
+  for (auto& particle : particles) {
+    if (particle.has_mothers())
+      continue;
+    nTotal++;
+  }
+  return nTotal;
+}
+
 template <typename T>
 float getAvgITSClSize(T const& track)
 {