From 624a339ef3e88132e79a376b374ac014aca86731 Mon Sep 17 00:00:00 2001
From: Alberto Caliva <alberto.caliva@cern.ch>
Date: Sat, 20 Apr 2024 08:32:30 +0200
Subject: [PATCH] antip feed-down correction from mc

---
 PWGLF/Tasks/Nuspex/nuclei_in_jets.cxx | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/PWGLF/Tasks/Nuspex/nuclei_in_jets.cxx b/PWGLF/Tasks/Nuspex/nuclei_in_jets.cxx
index 262f445126f..3d3d9aab197 100644
--- a/PWGLF/Tasks/Nuspex/nuclei_in_jets.cxx
+++ b/PWGLF/Tasks/Nuspex/nuclei_in_jets.cxx
@@ -93,8 +93,8 @@ struct nuclei_in_jets {
   Configurable<float> min_pt{"min_pt", 0.2f, "minimum pt of the tracks"};
   Configurable<float> min_eta{"min_eta", -0.8f, "minimum eta"};
   Configurable<float> max_eta{"max_eta", +0.8f, "maximum eta"};
-  Configurable<float> min_y{"min_y", -5.0f, "minimum y"};
-  Configurable<float> max_y{"max_y", +5.0f, "maximum y"};
+  Configurable<float> min_y{"min_y", -0.5f, "minimum y"};
+  Configurable<float> max_y{"max_y", +0.5f, "maximum y"};
   Configurable<float> max_dcaxy{"max_dcaxy", 0.1f, "Maximum DCAxy"};
   Configurable<float> max_dcaz{"max_dcaz", 0.1f, "Maximum DCAz"};
   Configurable<float> min_nsigmaTPC{"min_nsigmaTPC", -3.0f, "Minimum nsigma TPC"};
@@ -194,6 +194,10 @@ struct nuclei_in_jets {
     registryMC.add("antiproton_dca_prim", "antiproton_dca_prim", HistType::kTH2F, {{100, 0.0, 5.0, "#it{p}_{T} (GeV/#it{c})"}, {200, -0.5, 0.5, "DCA_{xy} (cm)"}});
     registryMC.add("antiproton_dca_sec", "antiproton_dca_sec", HistType::kTH2F, {{100, 0.0, 5.0, "#it{p}_{T} (GeV/#it{c})"}, {200, -0.5, 0.5, "DCA_{xy} (cm)"}});
 
+    // Fraction of Primary Antiprotons from MC
+    registryMC.add("antiproton_prim", "antiproton_prim", HistType::kTH1F, {{100, 0.0, 5.0, "#it{p}_{T} (GeV/#it{c})"}});
+    registryMC.add("antiproton_all", "antiproton_all", HistType::kTH1F, {{100, 0.0, 5.0, "#it{p}_{T} (GeV/#it{c})"}});
+
     // Histograms for reweighting
     registryMC.add("antiproton_eta_pt_pythia", "antiproton_eta_pt_pythia", HistType::kTH2F, {{100, 0.0, 5.0, "#it{p}_{T} (GeV/#it{c})"}, {80, -0.8, 0.8, "#eta"}});
     registryMC.add("antideuteron_eta_pt", "antideuteron_eta_pt", HistType::kTH2F, {{100, 0.0, 5.0, "#it{p}_{T} (GeV/#it{c})"}, {80, -0.8, 0.8, "#eta"}});
@@ -895,7 +899,7 @@ struct nuclei_in_jets {
     }
   }
 
-  void processRec(SimCollisions const& collisions, MCTracks const& mcTracks, aod::McCollisions const& /*mcCollisions*/, const aod::McParticles& /*mcParticles*/)
+  void processRec(SimCollisions const& collisions, MCTracks const& mcTracks, aod::McCollisions const& mcCollisions, const aod::McParticles& mcParticles)
   {
 
     for (const auto& collision : collisions) {
@@ -960,15 +964,23 @@ struct nuclei_in_jets {
         if (particle.pdgCode() == -2212 && (!particle.isPhysicalPrimary()) && TMath::Abs(track.dcaZ()) < max_dcaz)
           registryMC.fill(HIST("antiproton_dca_sec"), pt, track.dcaXY());
 
-        if (!particle.isPhysicalPrimary())
-          continue;
-
         // DCA Cuts
         if (TMath::Abs(track.dcaXY()) > max_dcaxy)
           continue;
         if (TMath::Abs(track.dcaZ()) > max_dcaz)
           continue;
 
+        // Fraction of Primary Antiprotons
+        if (particle.pdgCode() == -2212) {
+          registryMC.fill(HIST("antiproton_all"), pt);
+          if (particle.isPhysicalPrimary()) {
+            registryMC.fill(HIST("antiproton_prim"), pt);
+          }
+        }
+
+        if (!particle.isPhysicalPrimary())
+          continue;
+
         // Antiproton
         if (particle.pdgCode() == -2212) {
           if (pt < 1.0 && nsigmaTPCPr > min_nsigmaTPC && nsigmaTPCPr < max_nsigmaTPC) {