[PWGHF] add number of primary vertex contributors to data processing DPlus task

Author: fjonasALICE

PWGHF/D2H/Tasks/taskDplus.cxx

@@ -233,6 +233,9 @@ struct HfTaskDplus {
       if (storeIR) {
         axes.push_back(thnAxisIR);
       }
+      if (storePvContributors) {
+        axes.push_back(thnAxisPvContributors);
+      }
       if (doprocessDataWithMlWithUpc || doprocessDataWithUpc) {
         axes.push_back(thnAxisGapType);
         axes.push_back(thnAxisFT0A);
@@ -410,10 +413,16 @@ struct HfTaskDplus {
         }
       }
     } else { // Data
-      if (storeCentrality && storeOccupancy) {
+      if (storeCentrality && storeOccupancy && storePvContributors) {
+        registry.fill(HIST("hSparseMass"), HfHelper::invMassDplusToPiKPi(candidate), candidate.pt(), outputMl[0], outputMl[1], outputMl[2], centrality, occupancy, numPvContributors);
+      } else if (storeCentrality && storeOccupancy && !storePvContributors) {
         registry.fill(HIST("hSparseMass"), HfHelper::invMassDplusToPiKPi(candidate), candidate.pt(), outputMl[0], outputMl[1], outputMl[2], centrality, occupancy);
+      } else if (storeCentrality && !storeOccupancy && storePvContributors) {
+        registry.fill(HIST("hSparseMass"), HfHelper::invMassDplusToPiKPi(candidate), candidate.pt(), outputMl[0], outputMl[1], outputMl[2], centrality, numPvContributors);
       } else if (storeCentrality && !storeOccupancy) {
         registry.fill(HIST("hSparseMass"), HfHelper::invMassDplusToPiKPi(candidate), candidate.pt(), outputMl[0], outputMl[1], outputMl[2], centrality);
+      } else if (!storeCentrality && storeOccupancy && storePvContributors) {
+        registry.fill(HIST("hSparseMass"), HfHelper::invMassDplusToPiKPi(candidate), candidate.pt(), outputMl[0], outputMl[1], outputMl[2], occupancy, numPvContributors);
       } else if (!storeCentrality && storeOccupancy) {
         registry.fill(HIST("hSparseMass"), HfHelper::invMassDplusToPiKPi(candidate), candidate.pt(), outputMl[0], outputMl[1], outputMl[2], occupancy);
       } else if (!storeCentrality && !storeOccupancy && storePvContributors) {
@@ -559,7 +568,7 @@ struct HfTaskDplus {
           continue;
         }
 
-        if (storeCentrality || storeOccupancy) {
+        if (storeCentrality || storeOccupancy || storePvContributors) {
           auto collision = candidate.template collision_as<CollisionsCent>();
           if (storeCentrality && centEstimator != CentralityEstimator::None) {
             cent = getCentralityColl(collision, centEstimator);
@@ -771,7 +780,8 @@ struct HfTaskDplus {
         int const nAxesCent = storeCentrality ? 1 : 0; // centrality if storeCentrality
         int const nAxesOcc = storeOccupancy ? 1 : 0;   // occupancy if storeOccupancy
         int const nAxesIR = storeIR ? 1 : 0;           // IR if storeIR
-        int const nAxesTotal = NAxesBase + NAxesMl + nAxesCent + nAxesOcc + nAxesIR;
+        int const nAxesPv = storePvContributors ? 1 : 0;
+        int const nAxesTotal = NAxesBase + NAxesMl + nAxesCent + nAxesOcc + nAxesIR + nAxesPv;
 
         std::vector<double> valuesToFill;
         valuesToFill.reserve(nAxesTotal);
@@ -797,6 +807,9 @@ struct HfTaskDplus {
         if (storeIR) {
           valuesToFill.push_back(ir);
         }
+        if (storePvContributors) {
+          valuesToFill.push_back(static_cast<double>(collision.numContrib()));
+        }
         valuesToFill.push_back(static_cast<double>(gap));
         valuesToFill.push_back(static_cast<double>(fitInfo.ampFT0A));
         valuesToFill.push_back(static_cast<double>(fitInfo.ampFT0C));