fix: correcting mc truth filling for the secondary contamination estimation

Author: Zuzanna Chochulska

PWGCF/FemtoUniverse/Core/FemtoUniverseEfficiencyCorrection.h

@@ -50,7 +50,6 @@ struct EffCorConfigurableGroup : framework::ConfigurableGroup {
   framework::Configurable<std::string> confEffCorCCDBPath{"confEffCorCCDBPath", "", "[Efficiency Correction] CCDB path to histograms"};
   framework::Configurable<std::vector<std::string>> confEffCorCCDBTimestamps{"confEffCorCCDBTimestamps", {}, "[Efficiency Correction] Timestamps of histograms in CCDB (0 can be used as a placeholder, e.g. when running subwagons)"};
   framework::Configurable<std::string> confEffCorVariables{"confEffCorVariables", "pt", "[Efficiency Correction] Variables for efficiency correction histogram dimensions (available: 'pt'; 'pt,eta'; 'pt,mult'; 'pt,eta,mult')"};
-  framework::Configurable<bool> confEffCorSetMultToConst{"confEffCorSetMultToConst", false, "[Efficiency Correction] Multiplicity for the histograms set to the constant value"};
 };
 
 class EfficiencyCorrection
@@ -63,7 +62,6 @@ class EfficiencyCorrection
   auto init(framework::HistogramRegistry* registry, std::vector<framework::AxisSpec> axisSpecs) -> void
   {
     shouldFillHistograms = config->confEffCorFillHist;
-    shouldSetMultToConst = config->confEffCorSetMultToConst;
 
     histRegistry = registry;
     if (shouldFillHistograms) {
@@ -128,7 +126,7 @@ class EfficiencyCorrection
     histRegistry->fill(HIST(histDirectory) + HIST("/") + HIST(histSuffix[N - 1]) + HIST("/hMCTruth"),
                        particle.pt(),
                        particle.eta(),
-                       shouldSetMultToConst ? 100 : particle.template fdCollision_as<CollisionType>().multV0M());
+                       particle.template fdCollision_as<CollisionType>().multV0M());
   }
 
   template <uint8_t N, typename CollisionType = aod::FdCollisions>
@@ -279,7 +277,6 @@ class EfficiencyCorrection
 
   bool shouldApplyCorrection{false};
   bool shouldFillHistograms{false};
-  bool shouldSetMultToConst{false};
 
   o2::ccdb::BasicCCDBManager& ccdb{o2::ccdb::BasicCCDBManager::instance()};
   std::array<TH1*, 2> hLoaded{nullptr, nullptr};

PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackPhi.cxx

@@ -117,11 +117,6 @@ struct FemtoUniversePairTaskTrackPhi {
                                                    (aod::femtouniverseparticle::pt > ConfTrackPtLow) &&
                                                    (aod::femtouniverseparticle::pt < ConfTrackPtHigh);
 
-  // Partition<soa::Join<FilteredFemtoFullParticles, aod::FDMCLabels>> partsTrackMCTruth =
-  //                                                  aod::femtouniverseparticle::partType == static_cast<uint8_t>(aod::femtouniverseparticle::ParticleType::kMCTruthTrack) &&
-  //                                                  aod::femtouniverseparticle::pt < ConfTrackPtHigh &&
-  //                                                  aod::femtouniverseparticle::pt > ConfTrackPtLow;
-
   /// Particle 2 --- PHI MESON
   Configurable<float> ConfPhiPtLow{"ConfPhiPtLow", 0.8, "Lower limit of the Phi pT."};
   Configurable<float> ConfPhiPtHigh{"ConfPhiPtHigh", 4.0, "Higher limit of the Phi pT."};
@@ -141,11 +136,6 @@ struct FemtoUniversePairTaskTrackPhi {
                                                  (aod::femtouniverseparticle::tempFitVar > confInvMassLowLimitPhi) &&
                                                  (aod::femtouniverseparticle::tempFitVar < confInvMassUpLimitPhi);
 
-  // Partition<soa::Join<FilteredFemtoFullParticles, aod::FDMCLabels>> partsPhiMCTruth =
-  //                                                aod::femtouniverseparticle::partType == static_cast<uint8_t>(aod::femtouniverseparticle::ParticleType::kMCTruthTrack) &&
-  //                                                aod::femtouniverseparticle::pt < ConfPhiPtHigh &&
-  //                                                aod::femtouniverseparticle::pt > ConfPhiPtLow;
-
   /// Partitions  for Phi daughters kPhiChild
   Partition<FilteredFemtoFullParticles> partsPhiDaugh = (aod::femtouniverseparticle::partType == uint8_t(aod::femtouniverseparticle::ParticleType::kPhiChild));
   Partition<soa::Join<FilteredFemtoFullParticles, aod::FDMCLabels>> partsPhiDaughMC = (aod::femtouniverseparticle::partType == uint8_t(aod::femtouniverseparticle::ParticleType::kPhiChild));
@@ -523,13 +513,6 @@ struct FemtoUniversePairTaskTrackPhi {
       if constexpr (isMC) {
         // reco
         effCorrection.fillRecoHist<ParticleNo::ONE, FilteredFDCollisions>(phicandidate, 333);
-        // truth
-        auto mcPartId1 = phicandidate.fdMCParticleId();
-        auto const& mcpart1 = mcParts.iteratorAt(mcPartId1);
-        if (mcpart1.pdgMCTruth() != 333) {
-          continue;
-        }
-        effCorrection.fillTruthHist<ParticleNo::ONE, FilteredFDCollisions>(phicandidate);
       }
     }
 
@@ -570,14 +553,6 @@ struct FemtoUniversePairTaskTrackPhi {
 
       if constexpr (isMC) {
         effCorrection.fillRecoHist<ParticleNo::TWO, FilteredFDCollisions>(track, ConfTrackPDGCode);
-
-        // truth
-        auto mcPartId2 = track.fdMCParticleId();
-        auto const& mcpart2 = mcParts.iteratorAt(mcPartId2);
-        if (mcpart2.pdgMCTruth() != ConfTrackPDGCode) {
-          continue;
-        }
-        effCorrection.fillTruthHist<ParticleNo::TWO, FilteredFDCollisions>(track);
       }
     }
 
@@ -738,6 +713,10 @@ struct FemtoUniversePairTaskTrackPhi {
         continue;
       }
 
+      if (pdgCode == ConfTrackPDGCode) {
+        effCorrection.fillTruthHist<ParticleNo::TWO, FilteredFDCollisions>(part);
+      }
+
       // charge +
       if (pdgParticle->Charge() > 0.0) {
         registryMCtruth.fill(HIST("MCtruthAllPositivePt"), part.pt());
@@ -755,6 +734,7 @@ struct FemtoUniversePairTaskTrackPhi {
       if (pdgCode == 333) {
         registryMCtruth.fill(HIST("MCtruthPhi"), part.pt(), part.eta());
         registryMCtruth.fill(HIST("MCtruthPhiPt"), part.pt());
+        effCorrection.fillTruthHist<ParticleNo::ONE, FilteredFDCollisions>(part);
         continue;
       }