Skip to content

Write tracked V0s, cascades and 3-bodies in collision order #15838

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

Merged
sawenzel merged 2 commits into from
Sep 24, 2026
+10 −8
Merged

Write tracked V0s, cascades and 3-bodies in collision order #15838

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
29cf278
Write tracked V0s, cascades and 3-bodies in collision order
sawenzel Sep 24, 2026
d526624
Keep strange tracks in decay order also with one vertexer thread
sawenzel Sep 24, 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
14 changes: 8 additions & 6 deletions Detectors/AOD/src/AODProducerWorkflowSpec.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1527,7 +1527,7 @@ void AODProducerWorkflowDPL::prepareStrangenessTracking(const o2::globaltracking
std::exclusive_scan(mVertexStrLUT.begin(), mVertexStrLUT.end(), mVertexStrLUT.begin(), 0);

// sort by collision ID
std::sort(mCollisionStrTrk.begin(), mCollisionStrTrk.end(), [](const auto& a, const auto& b) { return a.first < b.first; });
std::stable_sort(mCollisionStrTrk.begin(), mCollisionStrTrk.end(), [](const auto& a, const auto& b) { return a.first < b.first; });
mStrTrkIndices.clear();
mStrTrkIndices.resize(mCollisionStrTrk.size(), -1);
}
Expand All @@ -1536,7 +1536,6 @@ template <typename V0C, typename CC, typename D3BC>
void AODProducerWorkflowDPL::fillStrangenessTrackingTables(const o2::globaltracking::RecoContainer& recoData, V0C& v0Curs, CC& cascCurs, D3BC& d3BodyCurs)
{
int itsTableIdx = -1;
int sTrkID = 0;
int nV0 = 0;
int nCasc = 0;
int nD3Body = 0;
Expand All @@ -1555,7 +1554,10 @@ void AODProducerWorkflowDPL::fillStrangenessTrackingTables(const o2::globaltrack
cascCurs.reserve(nCasc);
d3BodyCurs.reserve(nD3Body);

for (const auto& sTrk : recoData.getStrangeTracks()) {
// Write the rows grouped by collision, the order analyses slice these tables in
auto sTracks = recoData.getStrangeTracks();
for (const auto& collStrTrk : mCollisionStrTrk) {
const auto& sTrk = sTracks[collStrTrk.second];
auto ITSIndex = GIndex{sTrk.mITSRef, GIndex::ITS};
auto item = mGIDToTableID.find(ITSIndex);
if (item != mGIDToTableID.end()) {
Expand All @@ -1565,7 +1567,7 @@ void AODProducerWorkflowDPL::fillStrangenessTrackingTables(const o2::globaltrack
continue;
}
if (sTrk.mPartType == dataformats::kStrkV0) {
v0Curs(mStrTrkIndices[sTrkID++],
v0Curs(mStrTrkIndices[collStrTrk.second],
itsTableIdx,
sTrk.mDecayRef,
sTrk.mDecayVtx[0],
Expand All @@ -1577,7 +1579,7 @@ void AODProducerWorkflowDPL::fillStrangenessTrackingTables(const o2::globaltrack
sTrk.mTopoChi2,
sTrk.getAverageClusterSize());
} else if (sTrk.mPartType == dataformats::kStrkCascade) {
cascCurs(mStrTrkIndices[sTrkID++],
cascCurs(mStrTrkIndices[collStrTrk.second],
itsTableIdx,
sTrk.mDecayRef,
sTrk.mDecayVtx[0],
Expand All @@ -1589,7 +1591,7 @@ void AODProducerWorkflowDPL::fillStrangenessTrackingTables(const o2::globaltrack
sTrk.mTopoChi2,
sTrk.getAverageClusterSize());
} else {
d3BodyCurs(mStrTrkIndices[sTrkID++],
d3BodyCurs(mStrTrkIndices[collStrTrk.second],
itsTableIdx,
sTrk.mDecayRef,
sTrk.mDecayVtx[0],
Expand Down
4 changes: 2 additions & 2 deletions Detectors/Vertexing/src/SVertexer.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -225,8 +225,8 @@ void SVertexer::produceOutput(o2::framework::ProcessingContext& pc)

std::vector<int> sortIdx(strTracksTmp.size());
std::iota(sortIdx.begin(), sortIdx.end(), 0);
// if mNTreads > 1 we need to sort tracks, clus and MCLabs by their mDecayRef
if (mNThreads > 1 && mNStrangeTracks > 1) {
// sort tracks, clus and MCLabs by their mDecayRef, also with one thread, so that they follow the vertex order
if (mNStrangeTracks > 1) {
std::sort(sortIdx.begin(), sortIdx.end(), [&strTracksTmp](int i1, int i2) { return strTracksTmp[i1].mDecayRef < strTracksTmp[i2].mDecayRef; });
}

Expand Down
Loading