adding an additional Mult histo (counting events with >1 particle candidate)

Author: glromane

PWGCF/Femto3D/Tasks/femto3dPairTask.cxx

@@ -15,29 +15,26 @@
 
 #include "PWGCF/Femto3D/Core/femto3dPairTask.h"
 
-#include "PWGCF/Femto3D/DataModel/PIDutils.h"
 #include "PWGCF/Femto3D/DataModel/singletrackselector.h"
 
-#include <Framework/ASoA.h>
-#include <Framework/AnalysisDataModel.h>
-#include <Framework/AnalysisTask.h>
-#include <Framework/Configurable.h>
-#include <Framework/Expressions.h>
-#include <Framework/HistogramRegistry.h>
-#include <Framework/HistogramSpec.h>
-#include <Framework/InitContext.h>
-#include <Framework/OutputObjHeader.h>
-#include <Framework/runDataProcessing.h>
-
-#include <TH1.h>
-#include <TH2.h>
-#include <TH3.h>
-#include <TString.h>
-#include <TVector3.h>
+#include "Common/DataModel/Multiplicity.h"
 
+#include "Framework/ASoA.h"
+#include "Framework/AnalysisDataModel.h"
+#include "Framework/AnalysisTask.h"
+#include "Framework/DataTypes.h"
+#include "Framework/Expressions.h"
+#include "Framework/HistogramRegistry.h"
+#include "Framework/StaticFor.h"
+#include "Framework/runDataProcessing.h"
+#include "MathUtils/Utils.h"
+
+#include "TLorentzVector.h"
+#include <TH1F.h>
+#include <TParameter.h>
+
+#include <algorithm> // std::random_shuffle
 #include <chrono>
-#include <cmath>
-#include <cstdint>
 #include <ctime>
 #include <map>
 #include <memory>
@@ -165,6 +162,7 @@ struct FemtoCorrelations {
   std::shared_ptr<TH2> TOFhisto_second;
 
   std::vector<std::shared_ptr<TH1>> MultHistos;
+  std::vector<std::shared_ptr<TH1>> MultHistos_pair;
   std::vector<std::vector<std::shared_ptr<TH1>>> kThistos;
   std::vector<std::vector<std::shared_ptr<TH1>>> mThistos; // test
   std::vector<std::vector<std::shared_ptr<TH1>>> SEhistos_1D;
@@ -210,6 +208,11 @@ struct FemtoCorrelations {
       auto hMult = registry.add<TH1>(Form("Cent%i/TPCMult_cent%i", i, i), Form("TPCMult_cent%i", i), kTH1F, {{5001, -0.5, 5000.5, "Mult."}});
       MultHistos.push_back(std::move(hMult));
 
+      if (IsIdentical) {
+        auto hMult_pair = registry.add<TH1>(Form("Cent%i/TPCMult_pair_cond_cent%i", i, i), Form("TPCMult_pair_cond_cent%i", i), kTH1F, {{5001, -0.5, 5000.5, "Mult."}});
+        MultHistos_pair.push_back(std::move(hMult_pair));
+      }
+
       for (unsigned int j = 0; j < _kTbins.value.size() - 1; j++) {
         auto hSE_1D = registry.add<TH1>(Form("Cent%i/SE_1D_cent%i_kT%i", i, i, j), Form("SE_1D_cent%i_kT%i", i, j), kTH1F, {{CFkStarBinning, "k* (GeV/c)"}});
         auto hME_1D = registry.add<TH1>(Form("Cent%i/ME_1D_cent%i_kT%i", i, i, j), Form("ME_1D_cent%i_kT%i", i, j), kTH1F, {{CFkStarBinning, "k* (GeV/c)"}});
@@ -513,6 +516,10 @@ struct FemtoCorrelations {
           unsigned int centBin = std::floor((i->first).second);
           MultHistos[centBin]->Fill(col1->mult());
 
+          if (selectedtracks_1[col1->index()].size() > 1) {
+            MultHistos_pair[centBin]->Fill(col1->mult());
+          }
+
           mixTracks(selectedtracks_1[col1->index()], centBin); // mixing SE identical
 
           for (unsigned int indx2 = indx1 + 1; indx2 < EvPerBin; indx2++) { // nested loop for all the combinations of collisions in a chosen mult/vertex bin