[PWGHF] Keep kaon origin classification in charm producer

Author: FDUEnrich

PWGCF/FemtoDream/Core/femtoDreamUtils.h

@@ -18,12 +18,9 @@
 
 #include "PWGCF/DataModel/FemtoDerived.h"
 
-#include "Common/Core/RecoDecay.h"
-
 #include <CommonConstants/PhysicsConstants.h>
 #include <Framework/Logger.h>
 
-#include <TMCProcess.h>
 #include <TPDGCode.h>
 
 #include <cstdint>
@@ -105,28 +102,6 @@ inline float getMass(int pdgCode)
   return mass;
 }
 
-/// Classify true kaons from the correct collision for the DCA templates.
-template <typename T>
-inline int getKaonOrigin(T const& particles, typename T::iterator const& particle)
-{
-  using namespace o2::aod::femtodreamMCparticle;
-  constexpr int GenFromTransport = -1;
-  if (particle.getProcess() == TMCProcess::kPHInhelastic && particle.getGenStatusCode() == GenFromTransport) {
-    return kMaterial;
-  }
-  // HF decay products can be physical primaries. Check their ancestry first.
-  if (RecoDecay::getParticleOrigin(particles, particle) != RecoDecay::OriginType::None) {
-    return kSecondaryFromHF;
-  }
-  if (particle.isPhysicalPrimary()) {
-    return kPrimary;
-  }
-  if (particle.getProcess() == TMCProcess::kPDecay && particle.getGenStatusCode() == GenFromTransport && particle.has_mothers()) {
-    return kSecondaryFromLF;
-  }
-  return kElse;
-}
-
 inline int checkDaughterType(o2::aod::femtodreamparticle::ParticleType partType, int motherPDG, int daughterPDG = 0)
 {
   int partOrigin = 0;

PWGHF/HFC/TableProducer/producerCharmHadronsTrackFemtoDream.cxx

@@ -465,6 +465,28 @@ struct HfProducerCharmHadronsTrackFemtoDream {
                             -999., -999., -999.);
   }
 
+  /// Classify true kaons from the correct collision for the DCA templates.
+  template <typename T>
+  static int getKaonOrigin(T const& particles, typename T::iterator const& particle)
+  {
+    using namespace o2::aod::femtodreamMCparticle;
+    constexpr int GenFromTransport = -1;
+    if (particle.getProcess() == TMCProcess::kPHInhelastic && particle.getGenStatusCode() == GenFromTransport) {
+      return kMaterial;
+    }
+    // HF decay products can be physical primaries. Check their ancestry first.
+    if (RecoDecay::getParticleOrigin(particles, particle) != RecoDecay::OriginType::None) {
+      return kSecondaryFromHF;
+    }
+    if (particle.isPhysicalPrimary()) {
+      return kPrimary;
+    }
+    if (particle.getProcess() == TMCProcess::kPDecay && particle.getGenStatusCode() == GenFromTransport && particle.has_mothers()) {
+      return kSecondaryFromLF;
+    }
+    return kElse;
+  }
+
   template <typename CollisionType, typename ParticleType>
   void fillMcParticle(CollisionType const& col, ParticleType const& particle, o2::aod::femtodreamparticle::ParticleType fdparttype, aod::McParticles const& mcParticles)
   {