Fix PID assignment in Xic0 and Omegac0 candidate creator

Author: mtorresc15

PWGHF/TableProducer/candidateCreatorXic0Omegac0.cxx

@@ -97,7 +97,7 @@ enum McMatchFlag : uint8_t {
 };
 
 // Convert the KFParticle PDG code to the O2 track PID enum needed by getTrackParCovFromKFP()
-o2::track::PID getTrackPIDFromPDG(const int pdg)
+o2::track::PID::ID getTrackPIDFromPDG(const int pdg)
 {
   switch (std::abs(pdg)) {
     case 211:
@@ -106,13 +106,15 @@ o2::track::PID getTrackPIDFromPDG(const int pdg)
       return o2::track::PID::Kaon;
     case 2212:
       return o2::track::PID::Proton;
+    case 3122:
+      return o2::track::PID::Lambda;
     case 3312:
       return o2::track::PID::XiMinus;
     case 3334:
       return o2::track::PID::OmegaMinus;
     default:
       LOGF(fatal, "Unsupported PDG code %d in getTrackPIDFromPDG()", pdg);
-      return static_cast<o2::track::PID>(-1);
+      return o2::track::PID::Pion; 
   }
 }
 
@@ -979,15 +981,15 @@ struct HfCandidateCreatorXic0Omegac0 {
 
       omegaDauChargedTrackParCov = getTrackParCovFromKFP(kfBachKaonToOmega, o2::track::PID::Kaon, bachCharge); // Cascade bach kaon
       omegaDauChargedTrackParCov.setAbsCharge(1);
-      auto pidCasc = getTrackPIDFromPDG(kfOmegaToOmegaC.GetPDG());
+      o2::track::PID::ID pidCasc = getTrackPIDFromPDG(kfOmegaToOmegaC.GetPDG());
       o2::track::TrackParCov trackCasc = getTrackParCovFromKFP(kfOmegaToOmegaC, pidCasc, bachCharge);
       trackCasc.setAbsCharge(1);
 
-      auto pidV0Dau0 = getTrackPIDFromPDG(kfPos.GetPDG());
+      o2::track::PID::ID pidV0Dau0 = getTrackPIDFromPDG(kfPos.GetPDG());
       trackParCovV0Dau0 = getTrackParCovFromKFP(kfPos, pidV0Dau0, +1); // V0 postive daughter
       trackParCovV0Dau0.setAbsCharge(1);
 
-      auto pidV0Dau1 = getTrackPIDFromPDG(kfNeg.GetPDG());
+      o2::track::PID::ID pidV0Dau1 = getTrackPIDFromPDG(kfNeg.GetPDG());
       trackParCovV0Dau1 = getTrackParCovFromKFP(kfNeg, pidV0Dau1, -1); // V0 negative daughter
       trackParCovV0Dau1.setAbsCharge(1);
 
@@ -1477,15 +1479,15 @@ struct HfCandidateCreatorXic0Omegac0 {
       xiDauChargedTrackParCov = getTrackParCovFromKFP(kfBachPionToXi, o2::track::PID::Pion, bachCharge); // Cascade bach pion
       xiDauChargedTrackParCov.setAbsCharge(1);
 
-      auto pidCasc = getTrackPIDFromPDG(kfXiToXiC.GetPDG());
+      o2::track::PID::ID pidCasc = getTrackPIDFromPDG(kfXiToXiC.GetPDG());
       o2::track::TrackParCov trackCasc = getTrackParCovFromKFP(kfXiToXiC, pidCasc, bachCharge);
       trackCasc.setAbsCharge(1);
 
-      auto pidV0Dau0 = getTrackPIDFromPDG(kfPos.GetPDG());
+      o2::track::PID::ID pidV0Dau0 = getTrackPIDFromPDG(kfPos.GetPDG());
       trackParCovV0Dau0 = getTrackParCovFromKFP(kfPos, pidV0Dau0, +1); // V0 postive daughter
       trackParCovV0Dau0.setAbsCharge(1);
 
-      auto pidV0Dau1 = getTrackPIDFromPDG(kfNeg.GetPDG());
+      o2::track::PID::ID pidV0Dau1 = getTrackPIDFromPDG(kfNeg.GetPDG());
       trackParCovV0Dau1 = getTrackParCovFromKFP(kfNeg, pidV0Dau1, -1); // V0 negative daughter
       trackParCovV0Dau1.setAbsCharge(1);