[PWGCF] FemtoUniverse: Add function to fill only MCTruthTracks

Author: prchakra

PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx

@@ -323,12 +323,9 @@ struct FemtoUniverseProducerTask {
 
   // D0/D0bar mesons
   struct : o2::framework::ConfigurableGroup {
-    Configurable<float> trackD0CandEtaMax{"trackD0CandEtaMax", 0.8, "max. track/D0 cand. pseudorapidity"};
-    Configurable<double> yD0CandGenMax{"yD0CandGenMax", 0.5, "max. gen. D0 cand. rapidity"};
-    Configurable<double> yD0CandMax{"yD0CandMax", 0.8, "max. D0 cand. rapidity"};
+    Configurable<float> confD0D0barCandEtaCut{"confD0D0barCandEtaCut", 0.8, "max. cand. pseudorapidity"};
     Configurable<float> trackD0pTGenMin{"trackD0pTGenMin", 0.0, "MC Truth, min. pT for tracks and D0/D0bar cand."};
     Configurable<float> trackD0pTGenMax{"trackD0pTGenMax", 24.0, "MC Truth, max. pT for tracks and D0/D0bar cand."};
-    Configurable<bool> useYCutD0Cand{"useYCutD0Cand", true, "True - apply cut on y of D0 cand./false - apply cut on eta"};
     Configurable<bool> storeD0D0barDoubleMassHypo{"storeD0D0barDoubleMassHypo", false, "Store D0/D0bar cand. which pass selection criteria for both, D0 and D0bar"};
     Configurable<std::vector<int>> classMlD0D0bar{"classMlD0D0bar", {0, 1, 2}, "Indexes of ML scores to be stored. Three indexes max."};
   } ConfD0Selection;
@@ -1139,6 +1136,9 @@ struct FemtoUniverseProducerTask {
   {
     const auto occupancy = col.trackOccupancyInTimeRange();
     outputCollExtra(irrate, occupancy);
+    if (occupancy > ConfGeneral.confTPCOccupancyMax) {
+      std::cout << "occupancy" << occupancy << std::endl;
+    }
   }
 
   template <bool isMC, typename TrackType>
@@ -1487,11 +1487,7 @@ struct FemtoUniverseProducerTask {
         continue;
       }
 
-      if (ConfD0Selection.useYCutD0Cand && std::abs(hfHelper.yD0(hfCand)) > ConfD0Selection.yD0CandMax) {
-        continue;
-      }
-
-      if (!(ConfD0Selection.useYCutD0Cand) && std::abs(hfCand.eta()) > ConfD0Selection.trackD0CandEtaMax) {
+      if (std::abs(hfCand.eta()) > ConfD0Selection.confD0D0barCandEtaCut) {
         continue;
       }
 
@@ -1610,11 +1606,7 @@ struct FemtoUniverseProducerTask {
         continue;
       }
 
-      if (ConfD0Selection.useYCutD0Cand && std::abs(hfHelper.yD0(hfCand)) > ConfD0Selection.yD0CandMax) {
-        continue;
-      }
-
-      if (!(ConfD0Selection.useYCutD0Cand) && std::abs(hfCand.eta()) > ConfD0Selection.trackD0CandEtaMax) {
+      if (std::abs(hfCand.eta()) > ConfD0Selection.confD0D0barCandEtaCut) {
         continue;
       }
 
@@ -1740,11 +1732,7 @@ struct FemtoUniverseProducerTask {
         continue;
       }
 
-      if (ConfD0Selection.useYCutD0Cand && std::abs(hfHelper.yD0(hfCand)) > ConfD0Selection.yD0CandMax) {
-        continue;
-      }
-
-      if (!(ConfD0Selection.useYCutD0Cand) && std::abs(hfCand.eta()) > ConfD0Selection.trackD0CandEtaMax) {
+      if (std::abs(hfCand.eta()) > ConfD0Selection.confD0D0barCandEtaCut) {
         continue;
       }
 
@@ -2133,7 +2121,8 @@ struct FemtoUniverseProducerTask {
     std::vector<int> tmpIDtrack;
 
     for (const auto& particle : mcParts) {
-
+      if (particle.eta() < -ConfFilterCuts.confEtaFilterCut || particle.eta() > ConfFilterCuts.confEtaFilterCut)
+        continue;
       if (particle.pt() < ConfD0Selection.trackD0pTGenMin || particle.pt() > ConfD0Selection.trackD0pTGenMax)
         continue;
 
@@ -2166,18 +2155,9 @@ struct FemtoUniverseProducerTask {
         /// check if we have D0(bar) → π± K∓
         continue;
       }
-
       if (std::abs(particle.pdgCode()) == Pdg::kD0) {
-        if (std::abs(particle.y()) > ConfD0Selection.yD0CandGenMax) {
-          continue;
-        } else {
-          origin = RecoDecay::getCharmHadronOrigin(mcParts, particle);
-          mcGenFlag = particle.flagMcMatchGen();
-        }
-      } else {
-        if (std::abs(particle.eta()) > ConfD0Selection.trackD0CandEtaMax) {
-          continue;
-        }
+        origin = RecoDecay::getCharmHadronOrigin(mcParts, particle);
+        mcGenFlag = particle.flagMcMatchGen();
       }
 
       outputParts(outputCollision.lastIndex(),