Merge branch 'AliceO2Group:master' into master

Author: YazhenLin

PWGCF/EbyEFluctuations/Tasks/netprotcumulants.cxx

@@ -53,6 +53,41 @@ using namespace o2;
 using namespace o2::framework;
 using namespace o2::framework::expressions;
 
+namespace o2::aod
+{
+
+// ===== Event columns =====
+DECLARE_SOA_COLUMN(EventId, eventId, int);
+DECLARE_SOA_COLUMN(MyCent, myCent, float);
+DECLARE_SOA_COLUMN(Vz, vz, float);
+DECLARE_SOA_COLUMN(Ntrk, ntrk, int);
+DECLARE_SOA_COLUMN(Np, np, int);
+DECLARE_SOA_COLUMN(Npbar, npbar, int);
+
+// ===== Track columns =====
+DECLARE_SOA_COLUMN(Pt, pt, float);
+DECLARE_SOA_COLUMN(Eta, eta, float);
+DECLARE_SOA_COLUMN(Pid, pid, int);
+DECLARE_SOA_COLUMN(Eff, eff, float);
+
+// ===== Tables =====
+DECLARE_SOA_TABLE(EventTable, "AOD", "MYEVENT",
+                  EventId,
+                  MyCent,
+                  Vz,
+                  Ntrk,
+                  Np,
+                  Npbar);
+
+DECLARE_SOA_TABLE(MYTrackTable, "AOD", "MYTRACK",
+                  EventId,
+                  Pt,
+                  Eta,
+                  Pid,
+                  Eff);
+
+} // namespace o2::aod
+
 struct NetProtCumulants {
   // events
   Configurable<float> cfgCutVertex{"cfgCutVertex", 10.0f, "Accepted z-vertex range"};
@@ -110,6 +145,26 @@ struct NetProtCumulants {
 
   HistogramRegistry histos{"Histos", {}, OutputObjHandlingPolicy::AnalysisObject};
 
+  Produces<aod::EventTable> eventTable;
+  Produces<aod::MYTrackTable> trackTable;
+
+  int eventCounter = 0;
+  int eventId = 0;
+
+  // event variables
+  int Ntrk_event = 0;
+  int Np_event = 0;
+  int Npbar_event = 0;
+
+  float cent_event = 0;
+  float vz_event = 0;
+
+  // track variables
+  float pt_track = 0;
+  float eta_track = 0;
+  int pid_track = 0;
+  float eff_track = 0;
+
   TRandom3* fRndm = new TRandom3(0);
 
   // Eff histograms 2d: eff(pT, eta)
@@ -855,6 +910,7 @@ struct NetProtCumulants {
     if (cfgUsePtDepDCAz) {
       fPtDepDCAz = new TF1("ptDepDCAz", cfgDCAzFunc->c_str(), 0.001, 10.0);
     }
+
   } // end init()
 
   template <typename T>
@@ -2134,6 +2190,18 @@ struct NetProtCumulants {
 
   void processDataRec(AodCollisions::iterator const& coll, aod::BCsWithTimestamps const&, AodTracks const& inputTracks)
   {
+
+    // reset per event
+    Ntrk_event = 0;
+    Np_event = 0;
+    Npbar_event = 0;
+
+    // assign event ID
+    eventId = eventCounter++;
+
+    // store event info
+    cent_event = coll.centFT0C();
+    vz_event = coll.posZ();
     if (!coll.sel8()) {
       return;
     }
@@ -2263,6 +2331,7 @@ struct NetProtCumulants {
       if (cfgIfMandatoryTOF && !track.hasTOF()) {
         continue;
       }
+      Ntrk_event++;
 
       bool trackSelected = false;
       if (cfgPIDchoice == 0)
@@ -2293,11 +2362,18 @@ struct NetProtCumulants {
 
           if (track.pt() < cfgCutPtUpper) {
             nProt = nProt + 1.0;
+            Np_event++;
             float pEff = getEfficiency(track); // get efficiency of track
             if (pEff != 0) {
               for (int i = 1; i < 7; i++) {
                 powerEffProt[i] += std::pow(1.0 / pEff, i);
               }
+              pt_track = track.pt();
+              eta_track = track.eta();
+              pid_track = +1;
+              eff_track = pEff;
+
+              trackTable(eventId, pt_track, eta_track, pid_track, eff_track);
             }
           }
         }
@@ -2311,18 +2387,27 @@ struct NetProtCumulants {
           histos.fill(HIST("hrecDcaZAntiproton"), track.dcaZ());
           if (track.pt() < cfgCutPtUpper) {
             nAntiprot = nAntiprot + 1.0;
+            Npbar_event++;
             float pEff = getEfficiency(track); // get efficiency of track
             if (pEff != 0) {
               for (int i = 1; i < 7; i++) {
                 powerEffAntiprot[i] += std::pow(1.0 / pEff, i);
               }
+              pt_track = track.pt();
+              eta_track = track.eta();
+              pid_track = -1;
+              eff_track = pEff;
+
+              trackTable(eventId, pt_track, eta_track, pid_track, eff_track);
             }
           }
         }
 
       } //! checking PID
     } //! end track loop
 
+    eventTable(eventId, cent_event, vz_event, Ntrk_event, Np_event, Npbar_event);
+
     float netProt = nProt - nAntiprot;
     float nTracks = nProt + nAntiprot;
     histos.fill(HIST("hNTracksVsCent"), cent, nTracks);

PWGCF/EbyEFluctuations/Tasks/v0ptHadPiKaProt.cxx

@@ -100,7 +100,8 @@ enum PairHist {
 
   // mixing qa
   kSeNpart1VsNpart2,                         // number of unique particles 1 vs unique number of particles 2 in each same event
-  kMeMixingWindow,                           // mixing window size
+  kMeMixingWindowRaw,                        // mixing window size
+  kMeMixingWindowEffective,                  // mixing window size, counting event pairs with particle pairs
   kMeNpart1VsNpart2,                         // number of unique particles 1 vs number of unique particles 2 in each mixed event
   kMeVtz1VsMult1VsCent1VsVtz2VsMult2VsCent2, // correlation of event properties in each mixing bin
 
@@ -233,7 +234,8 @@ constexpr std::array<histmanager::HistInfo<PairHist>, kPairHistogramLast>
       {kTrueMultVsMult, o2::framework::HistType::kTH2F, "hTrueMultVsMult", "Multiplicity_{True} vs Multiplicity; Multiplicity_{True} ;  Multiplicity"},
       {kTrueCentVsCent, o2::framework::HistType::kTH2F, "hTrueCentVsCent", "Centrality_{True} vs Centrality; Centrality_{True} (%); Centrality (%)"},
       {kSeNpart1VsNpart2, o2::framework::HistType::kTH2F, "hSeNpart1VsNpart2", "# unique particle 1 vs # unique particle 2 in each same event; # partilce 1; # particle 2"},
-      {kMeMixingWindow, o2::framework::HistType::kTH1F, "hMeMixingWindow", "Mixing Window; Mixing Windown ; Entries"},
+      {kMeMixingWindowRaw, o2::framework::HistType::kTH1F, "hMeMixingWindowRaw", "Raw Mixing Window; Raw Mixing Window ; Entries"},
+      {kMeMixingWindowEffective, o2::framework::HistType::kTH1F, "hMeMixingWindowEffective", "Effective Mixing Window; Effective Mixing Windown ; Entries"},
       {kMeNpart1VsNpart2, o2::framework::HistType::kTH2F, "hMeNpart1VsNpart2", "# unique particle 1 vs # unique partilce 2 in each mixing bin; # partilce 1; # particle 2"},
       {kMeVtz1VsMult1VsCent1VsVtz2VsMult2VsCent2, o2::framework::HistType::kTHnSparseF, "hVtz1VsMult1VsCent1VsVtz2VsMult2VsCent2", "Mixing bins; V_{z,1} (cm); multiplicity_{1}; centrality_{1} (%); V_{z,2} (cm); multiplicity_{2}; centrality_{2} (%)"},
     }};
@@ -276,7 +278,8 @@ constexpr std::array<histmanager::HistInfo<PairHist>, kPairHistogramLast>
     {kKstarVsMtVsMinvVsPt1VsPt2VsMultVsCent, {confAnalysis.kstar, confAnalysis.mt, confAnalysis.massInv, confAnalysis.pt1, confAnalysis.pt2, confAnalysis.multiplicity, confAnalysis.centrality}},                           \
     {kDalitz, {confAnalysis.kstar, confAnalysis.dalitzMtot, confAnalysis.dalitzM12, confAnalysis.dalitzM13}},                                                                                                                \
     {kSeNpart1VsNpart2, {confMixing.particleBinning, confMixing.particleBinning}},                                                                                                                                           \
-    {kMeMixingWindow, {confMixing.particleBinning}},                                                                                                                                                                         \
+    {kMeMixingWindowRaw, {confMixing.particleBinning}},                                                                                                                                                                      \
+    {kMeMixingWindowEffective, {confMixing.particleBinning}},                                                                                                                                                                \
     {kMeNpart1VsNpart2, {confMixing.particleBinning, confMixing.particleBinning}},                                                                                                                                           \
     {kMeVtz1VsMult1VsCent1VsVtz2VsMult2VsCent2, {confMixing.vtxBins, confMixing.multBins, confMixing.centBins, confMixing.vtxBins, confMixing.multBins, confMixing.centBins}},
 
@@ -599,7 +602,9 @@ class PairHistManager
   void resetTrackedParticlesPerEvent()
   {
     mParticles1PerEvent.clear();
+    mParticles1PerEvent.reserve(100);
     mParticles2PerEvent.clear();
+    mParticles2PerEvent.reserve(100);
   }
 
   void fillMixingQaSe()
@@ -616,10 +621,11 @@ class PairHistManager
     }
   }
 
-  void fillMixingQaMePerMixingBin(int windowSize)
+  void fillMixingQaMePerMixingBin(int windowSizeRaw, int windowSizeEffective)
   {
-    if (mEventMixingQa) {
-      mHistogramRegistry->fill(HIST(prefix) + HIST(MixingQaDir) + HIST(getHistName(kMeMixingWindow, HistTable)), windowSize);
+    if (mPairCorrelationQa) {
+      mHistogramRegistry->fill(HIST(prefix) + HIST(MixingQaDir) + HIST(getHistName(kMeMixingWindowRaw, HistTable)), windowSizeRaw);
+      mHistogramRegistry->fill(HIST(prefix) + HIST(MixingQaDir) + HIST(getHistName(kMeMixingWindowEffective, HistTable)), windowSizeEffective);
     }
   }
 
@@ -732,7 +738,8 @@ class PairHistManager
   {
     std::string mcDir = std::string(prefix) + std::string(MixingQaDir);
     if (mPairCorrelationQa) {
-      mHistogramRegistry->add(mcDir + getHistNameV2(kMeMixingWindow, HistTable), getHistDesc(kMeMixingWindow, HistTable), getHistType(kMeMixingWindow, HistTable), {Specs.at(kMeMixingWindow)});
+      mHistogramRegistry->add(mcDir + getHistNameV2(kMeMixingWindowRaw, HistTable), getHistDesc(kMeMixingWindowRaw, HistTable), getHistType(kMeMixingWindowRaw, HistTable), {Specs.at(kMeMixingWindowRaw)});
+      mHistogramRegistry->add(mcDir + getHistNameV2(kMeMixingWindowEffective, HistTable), getHistDesc(kMeMixingWindowEffective, HistTable), getHistType(kMeMixingWindowEffective, HistTable), {Specs.at(kMeMixingWindowEffective)});
       mHistogramRegistry->add(mcDir + getHistNameV2(kMeNpart1VsNpart2, HistTable), getHistDesc(kMeNpart1VsNpart2, HistTable), getHistType(kMeNpart1VsNpart2, HistTable), {Specs.at(kMeNpart1VsNpart2)});
     }
     if (mEventMixingQa) {