[PWGDQ] Change the reserve size in tableReader.cxx

PWGDQ/Tasks/tableReader.cxx

@@ -350,6 +350,8 @@
 
   int fCurrentRun; // needed to detect if the run changed and trigger update of calibrations etc.
 
+  int64_t reserveSize = 0;
+
   void init(o2::framework::InitContext& context)
   {
     if (context.mOptions.get<bool>("processDummy")) {
@@ -376,7 +378,7 @@
 
       // set one histogram directory for each defined track cut
       TString histDirNames = "TrackBarrel_BeforeCuts;";
       for (auto& cut : fTrackCuts) {
         histDirNames += Form("TrackBarrel_%s;", cut.GetName());
       }
 
@@ -416,12 +418,13 @@
       fCurrentRun = event.runNumber();
     }
 
-    trackSel.reserve(tracks.size());
+    reserveSize += tracks.size();
+    trackSel.reserve(reserveSize);
     uint32_t filterMap = 0;
     bool prefilterSelected = false;
     int iCut = 0;
 
     for (auto& track : tracks) {
       filterMap = 0;
       prefilterSelected = false;
       VarManager::FillTrack<TTrackFillMap>(track);
@@ -477,6 +480,8 @@
 
   Filter filterEventSelected = aod::dqanalysisflags::isEventSelected == 1;
 
+  int64_t reserveSize = 0;
+
   void init(o2::framework::InitContext& context)
   {
     if (context.mOptions.get<bool>("processDummy")) {
@@ -500,7 +505,7 @@
 
       // set one histogram directory for each defined track cut
       TString histDirNames = "TrackMuon_BeforeCuts;";
       for (auto& cut : fMuonCuts) {
         histDirNames += Form("TrackMuon_%s;", cut.GetName());
       }
 
@@ -516,14 +521,15 @@
     VarManager::ResetValues(0, VarManager::kNMuonTrackVariables);
     VarManager::FillEvent<TEventFillMap>(event);
 
-    muonSel.reserve(muons.size());
+    reserveSize += muons.size();
+    muonSel.reserve(reserveSize);
     uint32_t filterMap = 0;
     int iCut = 0;
 
     // First loop to get muon multiplicity for single muon cumulants
     if constexpr (static_cast<bool>(TEventFillMap & VarManager::ObjTypes::ReducedEventQvector)) {
       int multMuon = 0;
       for (auto& muon : muons) {
         filterMap = 0;
         VarManager::FillTrack<TMuonFillMap>(muon);
 
@@ -540,7 +546,7 @@
       VarManager::fgValues[VarManager::kMultSingleMuons] = multMuon;
     }
 
     for (auto& muon : muons) {
       filterMap = 0;
       VarManager::FillTrack<TMuonFillMap>(muon);
       if (fConfigQA) { // TODO: make this compile time
@@ -630,7 +636,7 @@
   template <int TPairType, uint32_t TTrackFillMap, typename TTracks1, typename TTracks2>
   void runPrefilterPairing(TTracks1 const& tracks1, TTracks2 const& tracks2)
   {
     for (auto& [track1, track2] : o2::soa::combinations(o2::soa::CombinationsFullIndexPolicy(tracks1, tracks2))) {
       if (track1.sign() * track2.sign() > 0) {
         continue;
       }
@@ -660,7 +666,7 @@
       fCurrentRun = events.begin().runNumber();
     }
 
     for (auto& event : events) {
       if (event.isEventSelected()) {
         auto groupedPrefilterCandidates = filteredTracks.sliceBy(perCollision, event.globalIndex());
         auto groupedBarrelCandidates = barrelTracksSelected->sliceByCached(aod::reducedtrack::reducedeventId, event.globalIndex(), cache);
@@ -669,7 +675,7 @@
     } // end loop events
 
     // Fill Prefilter bits for all tracks to have something joinable to MyBarrelTracksSelected
     for (auto& track : tracks) {
       prefilter(static_cast<int>(fPrefiltermap[track.globalIndex()]));
     }
   }
@@ -823,8 +829,8 @@
 
     uint32_t twoTrackFilter = 0;
     uint32_t mult_dimuons = 0;
     for (auto& track1 : tracks1) {
       for (auto& track2 : tracks2) {
         if constexpr (TPairType == VarManager::kDecayToMuMu) {
           twoTrackFilter = static_cast<uint32_t>(track1.isMuonSelected()) & static_cast<uint32_t>(track2.isMuonSelected()) & fTwoMuonFilterMask;
         }
@@ -1103,6 +1109,8 @@
   std::vector<std::vector<TString>> fTrackMuonHistNames;
   std::vector<AnalysisCompositeCut> fPairCuts;
 
+  int64_t reserveSize = 0;
+
   void init(o2::framework::InitContext& context)
   {
     if (context.mOptions.get<bool>("processDummy")) {
@@ -1372,20 +1380,26 @@
     uint32_t twoTrackFilter = 0;
     uint32_t dileptonFilterMap = 0;
     uint32_t dileptonMcDecision = 0; // placeholder, copy of the dqEfficiency.cxx one
-    dielectronList.reserve(1);
-    dimuonList.reserve(1);
-    dielectronExtraList.reserve(1);
-    dielectronInfoList.reserve(1);
-    dimuonExtraList.reserve(1);
-    dileptonInfoList.reserve(1);
-    dileptonFlowList.reserve(1);
+
+    if constexpr (TPairType != VarManager::kElectronMuon) {
+      // tracks1 = tracks2
+      // since emu does not require any table in this task, reserveSize is updated for same particle only now
+      reserveSize += tracks1.size() * (tracks1.size() - 1) / 2;
+    }
+    dielectronList.reserve(reserveSize);
+    dimuonList.reserve(reserveSize);
+    dielectronExtraList.reserve(reserveSize);
+    dielectronInfoList.reserve(reserveSize);
+    dimuonExtraList.reserve(reserveSize);
+    dileptonInfoList.reserve(reserveSize);
+    dileptonFlowList.reserve(reserveSize);
     if (fConfigFlatTables.value) {
-      dielectronAllList.reserve(1);
-      dielectronMlList.reserve(1);
-      dimuonAllList.reserve(1);
+      dielectronAllList.reserve(reserveSize);
+      dielectronMlList.reserve(reserveSize);
+      dimuonAllList.reserve(reserveSize);
     }
     if (useMiniTree.fConfigMiniTree) {
-      dileptonMiniTree.reserve(1);
+      dileptonMiniTree.reserve(reserveSize);
     }
 
     bool isSelectedBDT = false;