Feat: initial mc support in producer

Author: ariedel-cern

PWGCF/Femto/Core/cascadeBuilder.h

@@ -379,8 +379,8 @@ class CascadeBuilder
     LOG(info) << "Initialization done...";
   }
 
-  template <modes::System system, typename T1, typename T2, typename T3, typename T4, typename T5, typename T6, typename T7, typename T8, typename T9>
-  void fillCascades(T1 const& col, T2& collisionBuilder, T3& collisionProducts, T4& trackProducts, T5& cascadeProducts, T6 const& fullCascades, T7 const& fullTracks, T8& trackBuilder, T9& indexMap)
+  template <modes::System system, typename T1, typename T2, typename T3, typename T4, typename T5, typename T6, typename T7, typename T8>
+  void fillCascades(T1 const& col, T2& collisionBuilder, T3& collisionProducts, T4& trackProducts, T5& cascadeProducts, T6 const& fullCascades, T7 const& fullTracks, T8& trackBuilder)
   {
     if (!mFillAnyTable) {
       return;
@@ -404,9 +404,9 @@ class CascadeBuilder
 
       collisionBuilder.template fillCollision<system>(collisionProducts, col);
 
-      bachelorIndex = trackBuilder.template getDaughterIndex<modes::Track::kCascadeBachelor>(bachelor, trackProducts, collisionProducts, indexMap);
-      posDaughterIndex = trackBuilder.template getDaughterIndex<modes::Track::kV0Daughter>(posDaughter, trackProducts, collisionProducts, indexMap);
-      negDaughterIndex = trackBuilder.template getDaughterIndex<modes::Track::kV0Daughter>(negDaughter, trackProducts, collisionProducts, indexMap);
+      bachelorIndex = trackBuilder.template getDaughterIndex<modes::Track::kCascadeBachelor>(bachelor, trackProducts, collisionProducts);
+      posDaughterIndex = trackBuilder.template getDaughterIndex<modes::Track::kV0Daughter>(posDaughter, trackProducts, collisionProducts);
+      negDaughterIndex = trackBuilder.template getDaughterIndex<modes::Track::kV0Daughter>(negDaughter, trackProducts, collisionProducts);
 
       fillCascade(collisionProducts, cascadeProducts, cascade, col, bachelorIndex, posDaughterIndex, negDaughterIndex);
     }

PWGCF/Femto/Core/collisionBuilder.h

@@ -446,7 +446,7 @@ class CollisionBuilder
       return;
     }
 
-    if (mCollisionAleadyFilled) {
+    if (mCollisionAlreadyFilled) {
       return;
     }
 
@@ -488,17 +488,28 @@ class CollisionBuilder
       }
     }
 
-    mCollisionAleadyFilled = true;
+    mCollisionAlreadyFilled = true;
   }
 
-  void reset()
+  template <modes::System system, typename T1, typename T2, typename T3, typename T4, typename T5>
+  void fillMcCollision(T1& collisionProducts, T2 const& col, T3 const& mcCols, T4& mcProducts, T5& mcBuilder)
+  {
+    if (mCollisionAlreadyFilled) {
+      return;
+    }
+    this->template fillCollision<system>(collisionProducts, col);
+    mcBuilder.template fillMcCollisionWithLabel<system>(mcProducts, col, mcCols);
+  }
+
+  void
+    reset()
   {
-    mCollisionAleadyFilled = false;
+    mCollisionAlreadyFilled = false;
   }
 
  private:
   CollisionSelection<HistName> mCollisionSelection;
-  bool mCollisionAleadyFilled = false;
+  bool mCollisionAlreadyFilled = false;
   Zorro mZorro;
   bool mUseTrigger = false;
   int mRunNumber = -1;

PWGCF/Femto/Core/dataTypes.h

@@ -49,6 +49,9 @@ using TwoTrackResonanceType = uint16_t;
 using CascadeMaskType = uint16_t;
 using CascadeType = uint16_t;
 
+// datatype for origin of mc particle
+using McOriginType = uint8_t;
+
 // datatype for particles
 using ParticleType = uint16_t;
 

PWGCF/Femto/Core/kinkBuilder.h

@@ -396,8 +396,8 @@ class KinkBuilder
     LOG(info) << "Initialization done...";
   }
 
-  template <modes::System system, typename T1, typename T2, typename T3, typename T4, typename T5, typename T6, typename T7, typename T8, typename T9>
-  void fillKinks(T1 const& col, T2& collisionBuilder, T3& collisionProducts, T4& trackProducts, T5& kinkProducts, T6 const& kinks, T7 const& tracks, T8& trackBuilder, T9& indexMap)
+  template <modes::System system, typename T1, typename T2, typename T3, typename T4, typename T5, typename T6, typename T7, typename T8>
+  void fillKinks(T1 const& col, T2& collisionBuilder, T3& collisionProducts, T4& trackProducts, T5& kinkProducts, T6 const& kinks, T7 const& tracks, T8& trackBuilder)
   {
     if (!mFillAnyTable) {
       return;
@@ -422,7 +422,7 @@ class KinkBuilder
 
       auto daughter = kink.template trackDaug_as<T7>();
       collisionBuilder.template fillCollision<system>(collisionProducts, col);
-      daughterIndex = trackBuilder.template getDaughterIndex<modes::Track::kKinkDaughter>(daughter, trackProducts, collisionProducts, indexMap);
+      daughterIndex = trackBuilder.template getDaughterIndex<modes::Track::kKinkDaughter>(daughter, trackProducts, collisionProducts);
       if constexpr (modes::isEqual(kinkType, modes::Kink::kSigma)) {
         fillSigma(collisionProducts, kinkProducts, kink, daughterIndex);
       }