Skip to content

Commit 50b0984

Browse files
committed
test
1 parent c086f8f commit 50b0984

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

PWGDQ/Tasks/tableReader_withAssoc.cxx

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1834,12 +1834,16 @@ struct AnalysisSameEventPairing {
18341834
for (auto& event : events) {
18351835
if (event.isEventSelected_bit(0)) {
18361836
auto groupedAssocs = assocs.sliceBy(preslice, event.globalIndex());
1837-
reserveSize += (groupedAssocs.size() * (groupedAssocs.size() - 1)) / 2; // n choose 2 combinations
1837+
size_t nGood = 0;
1838+
for (auto const& t : groupedAssocs) {
1839+
if (t.isBarrelSelected_raw() && t.isBarrelSelectedPrefilter_raw()) {
1840+
nGood++;
1841+
}
1842+
}
1843+
reserveSize += nGood * (nGood - 1) / 2;
18381844
}
18391845
}
18401846
LOG(info) << "Reserving capacity for " << reserveSize << " pairs in the output tables";
1841-
LOG(info) << "number of Assocs: " << assocs.size() << ", number of Events: " << events.size();
1842-
reserveSize = 1;
18431847

18441848
dielectronList.reserve(reserveSize);
18451849
dimuonList.reserve(reserveSize);

0 commit comments

Comments
 (0)