Factor out process functions for B+

Author: vkucera

PWGHF/TableProducer/derivedDataCreatorBplusToD0Pi.cxx

@@ -67,6 +67,7 @@ struct HfDerivedDataCreatorBplusToD0Pi {
 
   HfHelper hfHelper;
   SliceCache cache;
+  static constexpr double mass{o2::constants::physics::MassBPlus};
 
   using CollisionsWCentMult = soa::Join<aod::Collisions, aod::CentFV0As, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs, aod::PVMultZeqs>;
   using CollisionsWMcCentMult = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::CentFV0As, aod::CentFT0Ms, aod::CentFT0As, aod::CentFT0Cs, aod::PVMultZeqs>;
@@ -297,52 +298,6 @@ struct HfDerivedDataCreatorBplusToD0Pi {
     }
   }
 
-  template <typename CollisionType, typename ParticleType>
-  void preProcessMcCollisions(CollisionType const& mcCollisions,
-                              ParticleType const& mcParticles)
-  {
-    if (!confDerData.fillMcRCollId) {
-      return;
-    }
-    rowsCommon.hasMcParticles.clear();
-    // Fill MC collision flags
-    for (const auto& mcCollision : mcCollisions) {
-      auto thisMcCollId = mcCollision.globalIndex();
-      auto particlesThisMcColl = mcParticles.sliceBy(mcParticlesPerMcCollision, thisMcCollId);
-      LOGF(debug, "MC collision %d has %d MC particles (preprocess)", thisMcCollId, particlesThisMcColl.size());
-      rowsCommon.hasMcParticles[thisMcCollId] = (particlesThisMcColl.size() > 0);
-    }
-  }
-
-  template <typename CollisionType, typename ParticleType>
-  void processMcParticles(CollisionType const& mcCollisions,
-                          ParticleType const& mcParticles)
-  {
-    // Fill MC collision properties
-    auto sizeTableMcColl = mcCollisions.size();
-    rowsCommon.reserveTablesMcColl(sizeTableMcColl);
-    for (const auto& mcCollision : mcCollisions) {
-      auto thisMcCollId = mcCollision.globalIndex();
-      auto particlesThisMcColl = mcParticles.sliceBy(mcParticlesPerMcCollision, thisMcCollId);
-      auto sizeTablePart = particlesThisMcColl.size();
-      LOGF(debug, "MC collision %d has %d MC particles", thisMcCollId, sizeTablePart);
-      // Skip MC collisions without HF particles (and without HF candidates in matched reconstructed collisions if saving indices of reconstructed collisions matched to MC collisions)
-      LOGF(debug, "MC collision %d has %d saved derived rec. collisions", thisMcCollId, rowsCommon.matchedCollisions[thisMcCollId].size());
-      if (sizeTablePart == 0 && (!confDerData.fillMcRCollId || rowsCommon.matchedCollisions[thisMcCollId].empty())) {
-        LOGF(debug, "Skipping MC collision %d", thisMcCollId);
-        continue;
-      }
-      LOGF(debug, "Filling MC collision %d at derived index %d", thisMcCollId, rowsCommon.rowMcCollBase.lastIndex() + 1);
-      rowsCommon.fillTablesMcCollision(mcCollision);
-
-      // Fill MC particle properties
-      rowsCommon.reserveTablesParticles(sizeTablePart);
-      for (const auto& particle : particlesThisMcColl) {
-        rowsCommon.fillTablesParticle(particle, o2::constants::physics::MassBPlus);
-      }
-    }
-  }
-
   void processData(CollisionsWCentMult const& collisions,
                    SelectedCandidates const&,
                    THfCandDaughters const& candidatesDaughters,
@@ -361,9 +316,9 @@ struct HfDerivedDataCreatorBplusToD0Pi {
                     TracksWPid const& tracks,
                     aod::BCs const& bcs)
   {
-    preProcessMcCollisions(mcCollisions, mcParticles);
+    rowsCommon.preProcessMcCollisions(mcCollisions, mcParticlesPerMcCollision, mcParticles);
     processCandidates<false, true, false, true>(collisions, candidatesMcSig, candidatesDaughters, tracks, bcs);
-    processMcParticles(mcCollisions, mcParticles);
+    rowsCommon.processMcParticles(mcCollisions, mcParticlesPerMcCollision, mcParticles, mass);
   }
   PROCESS_SWITCH(HfDerivedDataCreatorBplusToD0Pi, processMcSig, "Process MC only for signals", false);
 
@@ -375,9 +330,9 @@ struct HfDerivedDataCreatorBplusToD0Pi {
                     TracksWPid const& tracks,
                     aod::BCs const& bcs)
   {
-    preProcessMcCollisions(mcCollisions, mcParticles);
+    rowsCommon.preProcessMcCollisions(mcCollisions, mcParticlesPerMcCollision, mcParticles);
     processCandidates<false, true, true, false>(collisions, candidatesMcBkg, candidatesDaughters, tracks, bcs);
-    processMcParticles(mcCollisions, mcParticles);
+    rowsCommon.processMcParticles(mcCollisions, mcParticlesPerMcCollision, mcParticles, mass);
   }
   PROCESS_SWITCH(HfDerivedDataCreatorBplusToD0Pi, processMcBkg, "Process MC only for background", false);
 
@@ -389,9 +344,9 @@ struct HfDerivedDataCreatorBplusToD0Pi {
                     TracksWPid const& tracks,
                     aod::BCs const& bcs)
   {
-    preProcessMcCollisions(mcCollisions, mcParticles);
+    rowsCommon.preProcessMcCollisions(mcCollisions, mcParticlesPerMcCollision, mcParticles);
     processCandidates<false, true, false, false>(collisions, candidatesMcAll, candidatesDaughters, tracks, bcs);
-    processMcParticles(mcCollisions, mcParticles);
+    rowsCommon.processMcParticles(mcCollisions, mcParticlesPerMcCollision, mcParticles, mass);
   }
   PROCESS_SWITCH(HfDerivedDataCreatorBplusToD0Pi, processMcAll, "Process MC", false);
 
@@ -415,9 +370,9 @@ struct HfDerivedDataCreatorBplusToD0Pi {
                       TracksWPid const& tracks,
                       aod::BCs const& bcs)
   {
-    preProcessMcCollisions(mcCollisions, mcParticles);
+    rowsCommon.preProcessMcCollisions(mcCollisions, mcParticlesPerMcCollision, mcParticles);
     processCandidates<true, true, false, true>(collisions, candidatesMcMlSig, candidatesDaughters, tracks, bcs);
-    processMcParticles(mcCollisions, mcParticles);
+    rowsCommon.processMcParticles(mcCollisions, mcParticlesPerMcCollision, mcParticles, mass);
   }
   PROCESS_SWITCH(HfDerivedDataCreatorBplusToD0Pi, processMcMlSig, "Process MC with ML only for signals", false);
 
@@ -429,9 +384,9 @@ struct HfDerivedDataCreatorBplusToD0Pi {
                       TracksWPid const& tracks,
                       aod::BCs const& bcs)
   {
-    preProcessMcCollisions(mcCollisions, mcParticles);
+    rowsCommon.preProcessMcCollisions(mcCollisions, mcParticlesPerMcCollision, mcParticles);
     processCandidates<true, true, true, false>(collisions, candidatesMcMlBkg, candidatesDaughters, tracks, bcs);
-    processMcParticles(mcCollisions, mcParticles);
+    rowsCommon.processMcParticles(mcCollisions, mcParticlesPerMcCollision, mcParticles, mass);
   }
   PROCESS_SWITCH(HfDerivedDataCreatorBplusToD0Pi, processMcMlBkg, "Process MC with ML only for background", false);
 
@@ -443,9 +398,9 @@ struct HfDerivedDataCreatorBplusToD0Pi {
                       TracksWPid const& tracks,
                       aod::BCs const& bcs)
   {
-    preProcessMcCollisions(mcCollisions, mcParticles);
+    rowsCommon.preProcessMcCollisions(mcCollisions, mcParticlesPerMcCollision, mcParticles);
     processCandidates<true, true, false, false>(collisions, candidatesMcMlAll, candidatesDaughters, tracks, bcs);
-    processMcParticles(mcCollisions, mcParticles);
+    rowsCommon.processMcParticles(mcCollisions, mcParticlesPerMcCollision, mcParticles, mass);
   }
   PROCESS_SWITCH(HfDerivedDataCreatorBplusToD0Pi, processMcMlAll, "Process MC with ML", false);
 };