Added collision selection from jetderiveddatautilities for MC collisions too.

Author: christianreckziegel

PWGJE/Tasks/hfFragmentationFunction.cxx

@@ -25,15 +25,7 @@
 #include "PWGJE/Core/JetUtilities.h"
 #include "PWGJE/DataModel/Jet.h"
 #include "PWGJE/DataModel/JetReducedData.h"
-<<<<<<< HEAD
-=======
 
-<<<<<<< HEAD
-#include "PWGHF/Core/DecayChannels.h"
->>>>>>> 8fcbef4d9 (Fix: clang-formatting done manually)
-
-=======
->>>>>>> ee38d6bf6 (Fix: formatting with automatic git-clang-format feature)
 #include "Common/Core/RecoDecay.h"
 
 #include "Framework/ASoA.h"
@@ -329,7 +321,7 @@ struct HfFragmentationFunction {
 
       registry.fill(HIST("h_collision_counter"), 0.0);
       // skip collisions outside of |z| < vertexZCut
-      if (std::abs(mccollision.posZ()) > vertexZCut) {
+      if (!jetderiveddatautilities::selectCollision(mccollision, eventSelectionBits) || !(std::abs(mccollision.posZ()) < vertexZCut)) {
         continue;
       }
       registry.fill(HIST("h_collision_counter"), 1.0);
@@ -420,7 +412,7 @@ struct HfFragmentationFunction {
     for (const auto& mccollision : mccollisions) {
       registry.fill(HIST("h_collision_counter"), 0.0);
       // skip collisions outside of |z| < vertexZCut
-      if (std::abs(mccollision.posZ()) > vertexZCut) {
+      if (!jetderiveddatautilities::selectCollision(mccollision, eventSelectionBits) || !(std::abs(mccollision.posZ()) < vertexZCut)) {
         continue;
       }
       registry.fill(HIST("h_collision_counter"), 1.0);
@@ -462,28 +454,6 @@ struct HfFragmentationFunction {
             }
             int selectedAs = 0;
 
-            if (mcdcand.flagMcMatchRec() == decayChannel) { // matched to HF on truth level
-              matchedFrom = 1;
-            } else if (mcdcand.flagMcMatchRec() == -decayChannel) { // matched to HFbar on truth level
-              matchedFrom = -1;
-            }
-            // bitwise AND operation: Checks whether BIT(i) is set, regardless of other bits
-            if (mcdcand.candidateSelFlag() & BIT(0)) { // CandidateSelFlag == BIT(0) -> selected as HF
-              selectedAs = 1;
-            } else if (mcdcand.candidateSelFlag() & BIT(1)) { // CandidateSelFlag == BIT(1) -> selected as HFbar
-              selectedAs = -1;
-            }
-
-            // reflection information for storage: HF = +1, HFbar = -1, neither = 0
-            int matchedFrom = 0;
-            int decayChannel = 0;
-            if (jethfutilities::isD0Table<TCandidatesMCD>()) {
-              decayChannel = o2::hf_decay::hf_cand_2prong::DecayChannelMain::D0ToPiK;
-            } else if (jethfutilities::isLcTable<TCandidatesMCD>()) {
-              decayChannel = o2::hf_decay::hf_cand_3prong::DecayChannelMain::LcToPKPi;
-            }
-            int selectedAs = 0;
-
             if (mcdcand.flagMcMatchRec() == decayChannel) { // matched to HF on truth level
               matchedFrom = 1;
             } else if (mcdcand.flagMcMatchRec() == -decayChannel) { // matched to HFbar on truth level