[PWGUD] : Fixing the runtime overhead

PWGUD/Tasks/upcPhotonuclearAnalysisJMG.cxx

@@ -1,4 +1,4 @@
 // Copyright 2019-2020 CERN and copyright holders of ALICE O2.
 // See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
 // All rights not expressly granted are reserved.
 //
@@ -11,7 +11,7 @@
 ///
 /// \brief
 /// \author Josué Martínez García, josuem@cern.ch
 /// \file   upcPhotonuclearAnalysisJMG.cxx
 
 #include <vector>
 
@@ -47,9 +47,9 @@
 DECLARE_SOA_COLUMN(PtSideA, ptSideA, std::vector<float>);
 DECLARE_SOA_COLUMN(RapSideA, rapSideA, std::vector<float>);
 DECLARE_SOA_COLUMN(PhiSideA, phiSideA, std::vector<float>);
 DECLARE_SOA_COLUMN(TPCSignalSideA, tpcSignalSideA, std::vector<float>);
 DECLARE_SOA_COLUMN(TOFSignalSideA, tofSignalSideA, std::vector<float>);
 DECLARE_SOA_COLUMN(TPCNSigmaPiSideA, tpcNSigmaPiSideA, std::vector<float>);
 DECLARE_SOA_COLUMN(TOFNSigmaPiSideA, tofNSigmaPiSideA, std::vector<float>);
 DECLARE_SOA_COLUMN(TPCNSigmaKaSideA, tpcNSigmaKaSideA, std::vector<float>);
 DECLARE_SOA_COLUMN(TOFNSigmaKaSideA, tofNSigmaKaSideA, std::vector<float>);
@@ -100,7 +100,8 @@
   HistogramRegistry histos{"histos", {}, OutputObjHandlingPolicy::AnalysisObject};
 
   // Declare configurables on events/collisions
-  Configurable<int> nEvenstMixed{"nEvenstMixed", 3, {"Events to be Mixed"}};
+  Configurable<int> nEventsMixed{"nEventsMixed", 3, {"Events to be Mixed"}};
+  Configurable<int> factorEventsMixed{"factorEventsMixed", 100, {"factorEventsMixed to events mixed"}};
   Configurable<float> myZVtxCut{"myZVtxCut", 10., {"My collision cut"}};
   Configurable<float> myTimeZNACut{"myTimeZNACut", 2., {"My collision cut"}};
   Configurable<float> myTimeZNCCut{"myTimeZNCCut", 2., {"My collision cut"}};
@@ -170,7 +171,7 @@
     const AxisSpec axisPt{402, -0.05, 20.05};
     const AxisSpec axisP{402, -10.05, 10.05};
     const AxisSpec axisTPCSignal{802, -0.05, 400.05};
     const AxisSpec axisPhi{64, -2 * PI, 2 * PI};
     const AxisSpec axisEta{50, -1.2, 1.2};
     const AxisSpec axisNch{201, -0.5, 200.5};
     const AxisSpec axisZNEnergy{1002, -0.5, 500.5};
@@ -181,7 +182,7 @@
     const AxisSpec axisTPCNClsCrossedRowsMin{100, -0.05, 2.05};
 
     histos.add("yields", "multiplicity vs pT vs eta", {HistType::kTH3F, {{100, 0, 100, "multiplicity"}, {40, 0, 20, "p_{T}"}, {100, -2, 2, "#eta"}}});
     histos.add("etaphi", "multiplicity vs eta vs phi", {HistType::kTH3F, {{100, 0, 100, "multiplicity"}, {100, -2, 2, "#eta"}, {200, 0, 2 * PI, "#varphi"}}});
 
     const int maxMixBin = axisMultiplicity->size() * axisVertex->size();
     histos.add("eventcount", "bin", {HistType::kTH1F, {{maxMixBin + 2, -2.5, -0.5 + maxMixBin, "bin"}}});
@@ -275,23 +276,18 @@
     mixedGapSideC.setObject(new CorrelationContainer("mixedEventGapSideC", "mixedEventGapSideC", corrAxis, effAxis, {}));
   }
 
-  std::vector<double> vtxBinsEdges{VARIABLE_WIDTH, -10.0f, -5.0f, 0.0f, 5.0f, 10.0f};
+  std::vector<double> vtxBinsEdges{VARIABLE_WIDTH, -10.0f, -7.0f, -5.0f, -2.5f, 0.0f, 2.5f, 5.0f, 7.0f, 10.0f};
   std::vector<double> gapSideBinsEdges{VARIABLE_WIDTH, -0.5, 0.5, 1.5};
 
   SliceCache cache;
+  int countGapA = 0;
+  int countGapC = 0;
 
   // Binning only on PosZ without multiplicity
   // using BinningType = ColumnBinningPolicy<aod::collision::PosZ>;
   using BinningType = ColumnBinningPolicy<aod::collision::PosZ, aod::udcollision::GapSide>;
-  BinningType bindingOnVtx{{vtxBinsEdges, {gapSideBinsEdges}}, true};
-  SameKindPair<FullSGUDCollision, FullUDTracks, BinningType> pairs{bindingOnVtx, nEvenstMixed, -1, &cache};
-
-  // ColumnBinningPolicy<aod::collision::PosZ, aod::udcollision::TotalFT0AmplitudeC> bindingOnVtx{{vtxBinsEdges, multBinsEdges}, true};
-
-  // SameKindPair<soa::Join<aod::UDCollisions, aod::UDCollisionsSels, aod::SGCollisions, aod::UDZdcsReduced>,
-  //              FullUDTracks,
-  //              ColumnBinningPolicy<aod::collision::PosZ, aod::udcollision::TotalFT0AmplitudeC>>
-  //              pair{bindingOnVtx, 5, -1, &cache};
+  BinningType bindingOnVtx{{vtxBinsEdges, gapSideBinsEdges}, true};
+  SameKindPair<FullSGUDCollision, FullUDTracks, BinningType> pairs{bindingOnVtx, nEventsMixed, -1, &cache};
 
   template <typename CSG>
   bool isCollisionCutSG(CSG const& collision, int SideGap)
@@ -412,11 +408,11 @@
           continue;
         }
         float deltaPhi = phi(track1.px(), track1.py()) - phi(track2.px(), track2.py());
         if (deltaPhi > 1.5f * PI) {
           deltaPhi -= TwoPI;
         }
         if (deltaPhi < -PIHalf) {
           deltaPhi += TwoPI;
         }
         target->getPairHist()->Fill(CorrelationContainer::kCFStepReconstructed, eta(track1.px(), track1.py(), track1.pz()) - eta(track2.px(), track2.py(), track2.pz()), track2.pt(), track1.pt(), multiplicity, deltaPhi, posZ, 1.0);
       }
@@ -584,7 +580,8 @@
         if (fillCollisionUD(sameGapSideC, multiplicity) == false) {
           return;
         }
-        // LOGF(debug, "Filling sameGapSideC events");
+        histos.fill(HIST("eventcount"), -1);
+        // LOGF(info, "Filling sameGapSideC events");
         fillCorrelationsUD(sameGapSideC, reconstructedTracks, reconstructedTracks, multiplicity, reconstructedCollision.posZ());
         break;
       default:
@@ -601,39 +598,57 @@
     // int sgSide = reconstructedCollision.gapSide();
     // int sgSide = 0;
 
+    int maxCountGapA = 0;
+    int maxCountGapC = 0;
+
+    if (auto histEventCount = histos.get<TH1>(HIST("eventcount"))) {
+      int binA = histEventCount->GetXaxis()->FindBin(-2); // Gap A
+      int binC = histEventCount->GetXaxis()->FindBin(-1); // Gap C
+
+      maxCountGapA = histEventCount->GetBinContent(binA) * factorEventsMixed;
+      maxCountGapC = histEventCount->GetBinContent(binC) * factorEventsMixed;
+    }
+
     for (const auto& [collision1, tracks1, collision2, tracks2] : pairs) {
       if (collision1.size() == 0 || collision2.size() == 0) {
-        // LOGF(debug, "One or both collisions are empty.");
+        // LOGF(info, "One or both collisions are empty.");
         continue;
       }
 
+      if (countGapA >= maxCountGapA && countGapC >= maxCountGapC) {
+        break;
+      }
       float multiplicity = 0;
       if (collision1.gapSide() == 0 && collision2.gapSide() == 0) { // gap on side A
         if (isCollisionCutSG(collision1, 0) == false && isCollisionCutSG(collision2, 0) == false) {
           continue;
         }
-        // LOGF(debug, "In the pairs loop, gap side A");
+        // std::cout << "Counts for Gap A: " << countGapA << " Maximum Count for Gap A " << maxCountGapA << std::endl;
+        ++countGapA;
+        // LOGF(info, "In the pairs loop, gap side A");
         multiplicity = tracks1.size();
         if (fillCollisionUD(mixedGapSideA, multiplicity) == false) {
           return;
         }
         // histos.fill(HIST("eventcount"), bindingOnVtx.getBin({collision1.posZ()}));
         histos.fill(HIST("eventcount"), bindingOnVtx.getBin({collision1.posZ(), collision1.gapSide()}));
         fillCorrelationsUD(mixedGapSideA, tracks1, tracks2, multiplicity, collision1.posZ());
-        // LOGF(debug, "Filling mixedGapSideA events, Gap for side A");
+        // LOGF(info, "Filling mixedGapSideA events, Gap for side A");
       }
 
       if (collision1.gapSide() == 1 && collision2.gapSide() == 1) { // gap on side C
         if (isCollisionCutSG(collision1, 1) == false && isCollisionCutSG(collision2, 1) == false) {
           continue;
         }
-        // LOGF(debug, "In the pairs loop, gap side C");
+        // std::cout << "Counts for Gap C: " << countGapC << " Maximum Count for Gap C" << maxCountGapC << std::endl;
+        ++countGapC;
+        // LOGF(info, "In the pairs loop, gap side C");
         multiplicity = tracks1.size();
         if (fillCollisionUD(mixedGapSideC, multiplicity) == false) {
           return;
         }
         fillCorrelationsUD(mixedGapSideC, tracks1, tracks2, multiplicity, collision1.posZ());
-        // LOGF(debug, "Filling mixedGapSideC events, Gap for side C");
+        // LOGF(info, "Filling mixedGapSideC events, Gap for side C");
       } else {
         continue;
       }