incapsulate fill RecSig THns into function

Author: lubynets

PWGHF/D2H/Tasks/taskLc.cxx

@@ -396,14 +396,15 @@ struct HfTaskLc {
           double massLc(-1);
           double outputBkg(-1), outputPrompt(-1), outputFD(-1);
           const float properLifetime = hfHelper.ctLc(candidate) * CtToProperLifetimePs;
-          if ((candidate.isSelLcToPKPi() >= selectionFlagLc) && pdgCodeProng0 == kProton) {
-            massLc = hfHelper.invMassLcToPKPi(candidate);
+
+          auto fillTHnRecSig = [&](bool isPKPi) {
+            massLc = isPKPi ? hfHelper.invMassLcToPKPi(candidate) : hfHelper.invMassLcToPiKP(candidate);
 
             if constexpr (fillMl) {
               if (candidate.mlProbLcToPKPi().size() == NumberOfMlClasses) {
-                outputBkg = candidate.mlProbLcToPKPi()[MlClassBackground]; /// bkg score
-                outputPrompt = candidate.mlProbLcToPKPi()[MlClassPrompt];  /// prompt score
-                outputFD = candidate.mlProbLcToPKPi()[MlClassNonPrompt];   /// non-prompt score
+                outputBkg = isPKPi ? candidate.mlProbLcToPKPi()[MlClassBackground] : candidate.mlProbLcToPiKP()[MlClassBackground]; /// bkg score
+                outputPrompt = isPKPi ? candidate.mlProbLcToPKPi()[MlClassPrompt] : candidate.mlProbLcToPiKP()[MlClassPrompt];      /// prompt score
+                outputFD = isPKPi ? candidate.mlProbLcToPKPi()[MlClassNonPrompt] : candidate.mlProbLcToPiKP()[MlClassNonPrompt];    /// non-prompt score
               }
               /// Fill the ML outputScores and variables of candidate
               std::vector<double> valuesToFill{massLc, pt, cent, outputBkg, outputPrompt, outputFD, static_cast<double>(numPvContributors), ptRecB, static_cast<double>(originType)};
@@ -424,35 +425,13 @@ struct HfTaskLc {
               }
               registry.get<THnSparse>(HIST("hnLcVars"))->Fill(valuesToFill.data());
             }
+          };
+
+          if ((candidate.isSelLcToPKPi() >= selectionFlagLc) && pdgCodeProng0 == kProton) {
+            fillTHnRecSig(true);
           }
           if ((candidate.isSelLcToPiKP() >= selectionFlagLc) && pdgCodeProng0 == kPiPlus) {
-            massLc = hfHelper.invMassLcToPiKP(candidate);
-
-            if constexpr (fillMl) {
-              if (candidate.mlProbLcToPiKP().size() == NumberOfMlClasses) {
-                outputBkg = candidate.mlProbLcToPiKP()[MlClassBackground]; /// bkg score
-                outputPrompt = candidate.mlProbLcToPiKP()[MlClassPrompt];  /// prompt score
-                outputFD = candidate.mlProbLcToPiKP()[MlClassNonPrompt];   /// non-prompt score
-              }
-              /// Fill the ML outputScores and variables of candidate (todo: add multiplicity)
-              std::vector<double> valuesToFill{massLc, pt, cent, outputBkg, outputPrompt, outputFD, static_cast<double>(numPvContributors), ptRecB, static_cast<double>(originType)};
-              if (storeOccupancy && occEstimator != o2::hf_occupancy::OccupancyEstimator::None) {
-                valuesToFill.push_back(occ);
-              }
-              if (storeProperLifetime) {
-                valuesToFill.push_back(properLifetime);
-              }
-              registry.get<THnSparse>(HIST("hnLcVarsWithBdt"))->Fill(valuesToFill.data());
-            } else {
-              std::vector<double> valuesToFill{massLc, pt, cent, ptProng0, ptProng1, ptProng2, chi2PCA, decayLength, cpa, static_cast<double>(numPvContributors), ptRecB, static_cast<double>(originType)};
-              if (storeOccupancy && occEstimator != o2::hf_occupancy::OccupancyEstimator::None) {
-                valuesToFill.push_back(occ);
-              }
-              if (storeProperLifetime) {
-                valuesToFill.push_back(properLifetime);
-              }
-              registry.get<THnSparse>(HIST("hnLcVars"))->Fill(valuesToFill.data());
-            }
+            fillTHnRecSig(false);
           }
         }
       }