Add Pdg to SourceType mapping

Author: jpxrk

PWGHF/HFL/Tasks/taskSingleElectron.cxx

@@ -320,25 +320,24 @@ struct HfTaskSingleElectron {
       if (mctrack.size()) {
         auto const& grmothersIdsVec = mctrack.front().mothersIds();
         if (grmothersIdsVec.empty()) {
-          if (motherPdg == kPi0) {
-            return Pi0;
-          } else if (motherPdg == kEta) {
-            return Eta;
-          } else if (motherPdg == kOmega) {
-            return Omega;
-          } else if (motherPdg == kPhi) {
-            return Phi;
-          } else if (motherPdg == kEtaPrime) {
-            return EtaPrime;
-          } else if (motherPdg == kRho770_0) {
-            return Rho0;
-          } else if (motherPdg == kKPlus) {
-            return Ke3;
-          } else if (motherPdg == kK0Long) {
-            return K0l;
-          } else {
-            return Else;
-          }
+
+		  static const std::map<int, SourceType> PdgToSource = {
+			{kPi0, Pi0},
+			{kEta, Eta},
+			{kOmega, Omega},
+			{kPhi, Phi},
+			{kEtaPrime, EtaPrime},
+			{kRho770_0, Rho0},
+			{kKPlus, Ke3},
+			{kK0Long, K0l}
+		  };
+
+		  auto it = PdgToSource.find(motherPdg);
+		  if (it != PdgToSource.end()) {
+			return it->second;
+		  }
+		  return Else;
+
         } else {
           if (motherPdg == kPi0) {
             grmotherPt = mctrack.front().pt();