update reserve system

Author: zjxiongOvO

PWGDQ/Tasks/dqEfficiency_withAssoc.cxx

@@ -2011,9 +2011,21 @@ struct AnalysisSameEventPairing {
     for (auto& event : events) {
       if (event.isEventSelected_bit(0)) {
         auto groupedAssocs = assocs.sliceBy(preslice, event.globalIndex());
-        reserveSize += (groupedAssocs.size() * (groupedAssocs.size() - 1)) / 2; // n choose 2 combinations
+        size_t nGood = 0;
+        for (auto const& t : groupedAssocs) {
+          if constexpr (TPairType == VarManager::kDecayToEE) {
+            if (t.isBarrelSelected_raw()) {
+              nGood++;
+            }
+          } else if constexpr (TPairType == VarManager::kDecayToMuMu) {
+            if (t.isMuonSelected_raw()) {
+              nGood++;
+            }
+          }
+        }
+        reserveSize += nGood * (nGood - 1) / 2;
       }
-    }
+    } 
 
     dielectronList.reserve(reserveSize);
     dimuonList.reserve(reserveSize);

PWGDQ/Tasks/dqEfficiency_withAssoc_direct.cxx

@@ -1675,9 +1675,22 @@ struct AnalysisSameEventPairing {
     for (auto& event : events) {
       if (event.isEventSelected_bit(0)) {
         auto groupedAssocs = assocs.sliceBy(preslice, event.globalIndex());
-        reserveSize += (groupedAssocs.size() * (groupedAssocs.size() - 1)) / 2; // n choose 2 combinations
+        size_t nGood = 0;
+        for (auto const& t : groupedAssocs) {
+          if constexpr (TPairType == VarManager::kDecayToEE) {
+            if (t.isBarrelSelected_raw()) {
+              nGood++;
+            }
+          }
+          /*else if constexpr (TPairType == VarManager::kDecayToMuMu) {
+            if (t.isMuonSelected_raw()) {
+              nGood++;
+            }
+          }*/
+        }
+        reserveSize += nGood * (nGood - 1) / 2;
       }
-    }
+    } 
 
     dielectronList.reserve(reserveSize);
     dielectronsExtraList.reserve(reserveSize);

PWGDQ/Tasks/tableReader_withAssoc.cxx

@@ -1849,7 +1849,6 @@ struct AnalysisSameEventPairing {
         reserveSize += nGood * (nGood - 1) / 2;
       }
     }
-    LOG(info) << "Reserving capacity for " << reserveSize << " pairs in the output tables";
 
     dielectronList.reserve(reserveSize);
     dimuonList.reserve(reserveSize);

PWGDQ/Tasks/tableReader_withAssoc_direct.cxx

@@ -1642,9 +1642,17 @@ struct AnalysisSameEventPairing {
     for (auto& event : events) {
       if (event.isEventSelected_bit(0)) {
         auto groupedAssocs = assocs.sliceBy(preslice, event.globalIndex());
-        reserveSize += (groupedAssocs.size() * (groupedAssocs.size() - 1)) / 2; // n choose 2 combinations
+        size_t nGood = 0;
+        for (auto const& t : groupedAssocs) {
+          if constexpr (TPairType == VarManager::kDecayToEE) {
+            if (t.isBarrelSelected_raw()) {
+              nGood++;
+            }
+          }
+        }
+        reserveSize += nGood * (nGood - 1) / 2;
       }
-    }
+    } 
 
     dielectronList.reserve(reserveSize);
     dielectronsExtraList.reserve(reserveSize);