[PWGHF] Converting the B+ derived data to always consider D0s with an ML score (#9312)

Author: nzardosh

PWGHF/TableProducer/derivedDataCreatorBplusToD0Pi.cxx

@@ -91,8 +91,7 @@ struct HfDerivedDataCreatorBplusToD0Pi {
   using SelectedCandidatesMcMl = soa::Filtered<soa::Join<aod::HfCandBplus, aod::HfCandBplusMcRec, aod::HfSelBplusToD0Pi, aod::HfMlBplusToD0Pi>>;
   using MatchedGenCandidatesMc = soa::Filtered<soa::Join<aod::McParticles, aod::HfCandBplusMcGen>>;
   using TypeMcCollisions = soa::Join<aod::McCollisions, aod::McCentFT0Ms>;
-  using THfCandDaughters = aod::HfCand2ProngWPid;
-  using THfCandDaughtersMl = soa::Join<THfCandDaughters, aod::HfMlD0>;
+  using THfCandDaughtersMl = soa::Join<aod::HfCand2ProngWPid, aod::HfMlD0>;
 
   Filter filterSelectCandidates = aod::hf_sel_candidate_bplus::isSelBplusToD0Pi >= 1;
   Filter filterMcGenMatching = nabs(aod::hf_cand_bplus::flagMcMatchGen) == static_cast<int8_t>(BIT(aod::hf_cand_bplus::DecayType::BplusToD0Pi));
@@ -300,14 +299,14 @@ struct HfDerivedDataCreatorBplusToD0Pi {
         float massBplusToD0Pi = hfHelper.invMassBplusToD0Pi(candidate);
         float mlScoreBplusToD0Pi{-1.f};
         std::vector<float> mlScoresD0;
-        bool isD0 = prongBachelor.sign() < 0; // D0 or D0bar
+        bool isD0 = prongBachelor.sign() < 0;
+        if (isD0) { // is D0
+          std::copy(prongCharm.mlProbD0().begin(), prongCharm.mlProbD0().end(), std::back_inserter(mlScoresD0));
+        } else { // is D0bar
+          std::copy(prongCharm.mlProbD0bar().begin(), prongCharm.mlProbD0bar().end(), std::back_inserter(mlScoresD0));
+        }
         if constexpr (isMl) {
           mlScoreBplusToD0Pi = candidate.mlProbBplusToD0Pi();
-          if (isD0) {
-            std::copy(prongCharm.mlProbD0().begin(), prongCharm.mlProbD0().end(), std::back_inserter(mlScoresD0));
-          } else {
-            std::copy(prongCharm.mlProbD0bar().begin(), prongCharm.mlProbD0bar().end(), std::back_inserter(mlScoresD0));
-          }
         }
         // flag = 0 for D0 pi-, flag = 1 for D0bar pi+
         fillTablesCandidate(candidate, prongCharm, prongBachelor, isD0 ? 0 : 1, massBplusToD0Pi, ct, y, flagMcRec, origin, mlScoreBplusToD0Pi, mlScoresD0);
@@ -317,7 +316,7 @@ struct HfDerivedDataCreatorBplusToD0Pi {
 
   void processData(CollisionsWCentMult const& collisions,
                    SelectedCandidates const&,
-                   THfCandDaughters const& candidatesDaughters,
+                   THfCandDaughtersMl const& candidatesDaughters,
                    TracksWPid const& tracks,
                    aod::BCs const& bcs)
   {
@@ -329,7 +328,7 @@ struct HfDerivedDataCreatorBplusToD0Pi {
                     SelectedCandidatesMc const&,
                     TypeMcCollisions const& mcCollisions,
                     MatchedGenCandidatesMc const& mcParticles,
-                    THfCandDaughters const& candidatesDaughters,
+                    THfCandDaughtersMl const& candidatesDaughters,
                     TracksWPid const& tracks,
                     aod::BCs const& bcs)
   {
@@ -343,7 +342,7 @@ struct HfDerivedDataCreatorBplusToD0Pi {
                     SelectedCandidatesMc const&,
                     TypeMcCollisions const& mcCollisions,
                     MatchedGenCandidatesMc const& mcParticles,
-                    THfCandDaughters const& candidatesDaughters,
+                    THfCandDaughtersMl const& candidatesDaughters,
                     TracksWPid const& tracks,
                     aod::BCs const& bcs)
   {
@@ -357,7 +356,7 @@ struct HfDerivedDataCreatorBplusToD0Pi {
                     SelectedCandidatesMc const&,
                     TypeMcCollisions const& mcCollisions,
                     MatchedGenCandidatesMc const& mcParticles,
-                    THfCandDaughters const& candidatesDaughters,
+                    THfCandDaughtersMl const& candidatesDaughters,
                     TracksWPid const& tracks,
                     aod::BCs const& bcs)
   {