Skip to content

[PWGDQ] Fix table reservation problem #17230

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
JifengDeng wants to merge 2 commits into
base: master
Choose a base branch
from
+18 −3
Draft

[PWGDQ] Fix table reservation problem #17230

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ebfff6d
Fix the reserve size problem in asymmetric pairing of tableReader_wit…
Jul 28, 2026
1c7a30f
Please consider the following formatting changes
alibuild Jul 28, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 18 additions & 3 deletions PWGDQ/Tasks/tableReader_withAssoc.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3472,7 +3472,7 @@ struct AnalysisAsymmetricPairing {

// Template function to run same event pairing with asymmetric pairs (e.g. kaon-pion)
template <bool TTwoProngFitter, int TPairType, uint32_t TEventFillMap, uint32_t TTrackFillMap, typename TEvents, typename TTrackAssocs, typename TTracks>
void runAsymmetricPairing(TEvents const& events, Preslice<TTrackAssocs>& preslice, TTrackAssocs const& assocs, TTracks const& /*tracks*/)
void runAsymmetricPairing(TEvents const& events, Preslice<TTrackAssocs>& preslice, TTrackAssocs const& /*assocs*/, TTracks const& /*tracks*/)
{
fPairCount.clear();

Expand All @@ -3485,8 +3485,23 @@ struct AnalysisAsymmetricPairing {

int sign1 = 0;
int sign2 = 0;
ditrackList.reserve(assocs.size());
ditrackExtraList.reserve(assocs.size());

// Reserve capacity for the output tables
int64_t reserveSize = 0;
for (auto const& event : events) {
if (!event.isEventSelected_bit(0)) {
continue;
}
if (fConfigRemoveCollSplittingCandidates.value && event.isEventSelected_bit(2)) {
continue;
}
auto groupedLegAAssocs = legACandidateAssocs.sliceBy(preslice, event.globalIndex());
auto groupedLegBAssocs = legBCandidateAssocs.sliceBy(preslice, event.globalIndex());
reserveSize += static_cast<int64_t>(groupedLegAAssocs.size()) *
static_cast<int64_t>(groupedLegBAssocs.size());
}
ditrackList.reserve(reserveSize);
ditrackExtraList.reserve(reserveSize);

constexpr bool trackHasCov = ((TTrackFillMap & VarManager::ObjTypes::TrackCov) > 0 || (TTrackFillMap & VarManager::ObjTypes::ReducedTrackBarrelCov) > 0);

Expand Down