[PWGCF] calculate eff(eta,pT) in MC (#9685)

Author: SwatiSaha-1997

PWGCF/EbyEFluctuations/Tasks/netprotonCumulantsMc.cxx

@@ -180,6 +180,8 @@ struct NetprotonCumulantsMc {
     // Variable bin width axis
     std::vector<double> ptBinning = {0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0, 1.1, 1.2, 1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2.0, 2.1, 2.2, 2.3, 2.4, 2.5, 2.6, 2.7, 2.8, 2.9, 3.0, 3.1, 3.2, 3.3, 3.4, 3.5, 3.6, 3.7, 3.8, 3.9, 4.0};
     AxisSpec ptAxis = {ptBinning, "#it{p}_{T} (GeV/#it{c})"};
+    std::vector<double> etaBinning = {-0.8, -0.7, -0.6, -0.5, -0.4, -0.3, -0.2, -0.1, 0.0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8};
+    AxisSpec etaAxis = {etaBinning, "#it{#eta}"};
     std::vector<double> centBining = {0, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80, 85, 90};
     AxisSpec centAxis = {centBining, "Multiplicity percentile from FT0M (%)"};
     AxisSpec netprotonAxis = {41, -20.5, 20.5, "net-proton number"};
@@ -220,6 +222,10 @@ struct NetprotonCumulantsMc {
     histos.add("hCorrProfileTotalProton", "Eff. Corrected total proton number vs. centrality", kTProfile, {centAxis});
     histos.add("hCorrProfileProton", "Eff. Corrected proton number vs. centrality", kTProfile, {centAxis});
     histos.add("hCorrProfileAntiproton", "Eff. Corrected antiproton number vs. centrality", kTProfile, {centAxis});
+    histos.add("hrec2DEtaVsPtProton", "2D hist of Reconstructed Proton y: eta vs. x: pT", kTH2F, {ptAxis, etaAxis});
+    histos.add("hrec2DEtaVsPtAntiproton", "2D hist of Reconstructed Anti-proton y: eta vs. x: pT", kTH2F, {ptAxis, etaAxis});
+    histos.add("hgen2DEtaVsPtProton", "2D hist of Generated Proton y: eta vs. x: pT", kTH2F, {ptAxis, etaAxis});
+    histos.add("hgen2DEtaVsPtAntiproton", "2D hist of Generated Anti-proton y: eta vs. x: pT", kTH2F, {ptAxis, etaAxis});
 
     if (cfgIsCalculateCentral) {
       // uncorrected
@@ -926,6 +932,7 @@ struct NetprotonCumulantsMc {
             if (mcParticle.pdgCode() == 2212) {
               histos.fill(HIST("hgenPtProton"), mcParticle.pt()); //! hist for p gen
               histos.fill(HIST("hgenPtDistProtonVsCentrality"), mcParticle.pt(), cent);
+              histos.fill(HIST("hgen2DEtaVsPtProton"), mcParticle.pt(), mcParticle.eta());
               histos.fill(HIST("hgenEtaProton"), mcParticle.eta());
               histos.fill(HIST("hgenPhiProton"), mcParticle.phi());
               if (mcParticle.pt() < cfgCutPtUpper)
@@ -934,6 +941,7 @@ struct NetprotonCumulantsMc {
             if (mcParticle.pdgCode() == -2212) {
               histos.fill(HIST("hgenPtAntiproton"), mcParticle.pt()); //! hist for anti-p gen
               histos.fill(HIST("hgenPtDistAntiprotonVsCentrality"), mcParticle.pt(), cent);
+              histos.fill(HIST("hgen2DEtaVsPtAntiproton"), mcParticle.pt(), mcParticle.eta());
               histos.fill(HIST("hgenEtaAntiproton"), mcParticle.eta());
               histos.fill(HIST("hgenPhiAntiproton"), mcParticle.phi());
               if (mcParticle.pt() < cfgCutPtUpper)
@@ -1011,6 +1019,7 @@ struct NetprotonCumulantsMc {
             histos.fill(HIST("hrecPartPtProton"), particle.pt()); //! hist for p rec
             histos.fill(HIST("hrecPtProton"), track.pt());        //! hist for p rec
             histos.fill(HIST("hrecPtDistProtonVsCentrality"), particle.pt(), cent);
+            histos.fill(HIST("hrec2DEtaVsPtProton"), particle.pt(), particle.eta());
             histos.fill(HIST("hrecEtaProton"), particle.eta());
             histos.fill(HIST("hrecPhiProton"), particle.phi());
             histos.fill(HIST("hrecDcaXYProton"), track.dcaXY());
@@ -1025,6 +1034,7 @@ struct NetprotonCumulantsMc {
             histos.fill(HIST("hrecPartPtAntiproton"), particle.pt()); //! hist for anti-p rec
             histos.fill(HIST("hrecPtAntiproton"), track.pt());        //! hist for anti-p rec
             histos.fill(HIST("hrecPtDistAntiprotonVsCentrality"), particle.pt(), cent);
+            histos.fill(HIST("hrec2DEtaVsPtAntiproton"), particle.pt(), particle.eta());
             histos.fill(HIST("hrecEtaAntiproton"), particle.eta());
             histos.fill(HIST("hrecPhiAntiproton"), particle.phi());
             histos.fill(HIST("hrecDcaXYAntiproton"), track.dcaXY());