[PWGLF] fixed strange hadron selection in generated mc

Author: alcaliva

PWGLF/Tasks/Strangeness/strangenessInJets.cxx

@@ -1202,6 +1202,13 @@ struct StrangenessInJets {
       // Loop over all MC particles and select physical primaries within acceptance
       for (const auto& particle : mcParticlesPerColl) {
 
+        // Store properties of strange hadrons
+        int pdgAbs = std::abs(particle.pdgCode());
+        if (particle.isPhysicalPrimary() && (pdgAbs == kK0Short || pdgAbs == kLambda0 || pdgAbs == kXiMinus || pdgAbs == kOmegaMinus)) {
+          pdg.emplace_back(particle.pdgCode());
+          strHadronMomentum.emplace_back(particle.px(), particle.py(), particle.pz());
+        }
+
         // Select physical primary particles or HF decay products
         if (!isPhysicalPrimaryOrFromHF(particle, mcParticles))
           continue;
@@ -1216,13 +1223,6 @@ struct StrangenessInJets {
         fastjet::PseudoJet fourMomentum(particle.px(), particle.py(), particle.pz(), energy);
         fourMomentum.set_user_index(particle.pdgCode());
         fjParticles.emplace_back(fourMomentum);
-
-        // Store properties of strange hadrons
-        int pdgAbs = std::abs(particle.pdgCode());
-        if (pdgAbs == kK0Short || pdgAbs == kLambda0 || pdgAbs == kXiMinus || pdgAbs == kOmegaMinus) {
-          pdg.emplace_back(particle.pdgCode());
-          strHadronMomentum.emplace_back(particle.px(), particle.py(), particle.pz());
-        }
       }
 
       // Skip events with no particles