[PWGCF] FemtoUniverse: Adding histograms for more kT bins (AliceO2Gro…

…up#9125)
ddobrigk · Dec 25, 2024 · 31d5088 · 31d5088
1 parent 75e687d
commit 31d5088
Showing 1 changed file with 21 additions and 14 deletions.
diff --git a/PWGCF/FemtoUniverse/Core/FemtoUniversePairSHCentMultKt.h b/PWGCF/FemtoUniverse/Core/FemtoUniversePairSHCentMultKt.h
@@ -22,6 +22,8 @@
 #include <memory>
 #include "Framework/HistogramRegistry.h"
 
+using namespace o2::constants::physics;
+
 namespace o2::analysis::femto_universe
 {
 
@@ -198,6 +200,12 @@ class PairSHCentMultKt
       ktbinval = 2;
     } else if (ktval < ktBins[4]) {
       ktbinval = 3;
+    } else if (ktval < ktBins[5]) {
+      ktbinval = 4;
+    } else if (ktval < ktBins[6]) {
+      ktbinval = 5;
+    } else if (ktval < ktBins[7]) {
+      ktbinval = 6;
     } else {
       return;
     }
@@ -207,12 +215,10 @@ class PairSHCentMultKt
   /// Set the PDG codes of the two particles involved
   /// \param pdg1 PDG code of particle one
   /// \param pdg2 PDG code of particle two
-  void setPDGCodes(const int pdg1, const int pdg2)
+  void setPionPairMass()
   {
-    mMassOne = TDatabasePDG::Instance()->GetParticle(pdg1)->Mass();
-    mMassTwo = TDatabasePDG::Instance()->GetParticle(pdg2)->Mass();
-    mPDGOne = pdg1;
-    mPDGTwo = pdg2;
+    mMassOne = o2::constants::physics::MassPiPlus; // FIXME: Get from the PDG service of the common header
+    mMassTwo = o2::constants::physics::MassPiPlus; // FIXME: Get from the PDG service of the common header
   }
 
   /// To compute the bin value for cavariance matrix
@@ -242,6 +248,7 @@ class PairSHCentMultKt
     int fKtBin = ktval;
     std::vector<std::complex<double>> fYlmBuffer(kMaxJM);
     std::vector<double> f3d;
+    setPionPairMass();
     f3d = FemtoUniverseMath::newpairfunc(part1, mMassOne, part2, mMassTwo,
                                          isiden);
 
@@ -336,7 +343,7 @@ class PairSHCentMultKt
   void fillMultkTCov(uint8_t ChosenEventType, int MaxJM)
   {
     for (int multbinvalcov = 0;
-         multbinvalcov < static_cast<int>(centMultBins.size() - 2);
+         multbinvalcov < static_cast<int>(centMultBins.size() - 1);
          multbinvalcov++) {
       for (int ktbinvalcov = 0;
            ktbinvalcov < static_cast<int>(ktBins.size() - 1); ktbinvalcov++) {
@@ -346,22 +353,22 @@ class PairSHCentMultKt
   }
 
  private:
-  std::array<std::array<std::array<std::shared_ptr<TH1>, 10>, 4>, 4> fnumsreal{};
-  std::array<std::array<std::array<std::shared_ptr<TH1>, 10>, 4>, 4> fnumsimag{};
-  std::array<std::array<std::array<std::shared_ptr<TH1>, 10>, 4>, 4> fdensreal{};
-  std::array<std::array<std::array<std::shared_ptr<TH1>, 10>, 4>, 4> fdensimag{};
+  std::array<std::array<std::array<std::shared_ptr<TH1>, 10>, 7>, 4> fnumsreal{};
+  std::array<std::array<std::array<std::shared_ptr<TH1>, 10>, 7>, 4> fnumsimag{};
+  std::array<std::array<std::array<std::shared_ptr<TH1>, 10>, 7>, 4> fdensreal{};
+  std::array<std::array<std::array<std::shared_ptr<TH1>, 10>, 7>, 4> fdensimag{};
 
   TH1D* fbinctn[10][10];
   TH1D* fbinctd[10][10];
 
   static constexpr int kMaxL = 2;
   static constexpr int kMaxJM = (kMaxL + 1) * (kMaxL + 1);
 
-  std::array<std::array<std::array<float, (kMaxJM * kMaxJM * 4 * 100)>, 4>, 4> fcovmnum{}; ///< Covariance matrix for the numerator
-  std::array<std::array<std::array<float, (kMaxJM * kMaxJM * 4 * 100)>, 4>, 4> fcovmden{}; ///< Covariance matrix for the numerator
+  std::array<std::array<std::array<float, (kMaxJM * kMaxJM * 4 * 100)>, 7>, 4> fcovmnum{}; ///< Covariance matrix for the numerator
+  std::array<std::array<std::array<float, (kMaxJM * kMaxJM * 4 * 100)>, 7>, 4> fcovmden{}; ///< Covariance matrix for the numerator
 
-  std::array<std::array<std::shared_ptr<TH3>, 4>, 4> fcovnum{};
-  std::array<std::array<std::shared_ptr<TH3>, 4>, 4> fcovden{};
+  std::array<std::array<std::shared_ptr<TH3>, 7>, 4> fcovnum{};
+  std::array<std::array<std::shared_ptr<TH3>, 7>, 4> fcovden{};
 
  protected:
   HistogramRegistry* pairSHCentMultKtRegistry = nullptr;