Skip to content

Commit b740a31

Browse files
authored
[PWGDQ] fill event information in runMCGen (#16101)
1 parent f2c4a8e commit b740a31

1 file changed

Lines changed: 8 additions & 4 deletions

File tree

PWGDQ/Tasks/dqEfficiency_withAssoc_direct.cxx

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2017,7 +2017,7 @@ struct AnalysisSameEventPairing {
20172017
PresliceUnsorted<aod::McParticles> perReducedMcEvent = aod::mcparticle::mcCollisionId;
20182018

20192019
// template <int TPairType, typename TEventsMC>
2020-
template <int TPairType, typename TEvents, typename TEventsMC>
2020+
template <int TPairType, uint32_t TEventFillMap, typename TEvents, typename TEventsMC>
20212021
void runMCGen(TEvents const& events, TEventsMC const& mcEvents, McParticles const& mcTracks)
20222022
{
20232023
cout << "AnalysisSameEventPairing::runMCGen() called" << endl;
@@ -2053,7 +2053,11 @@ struct AnalysisSameEventPairing {
20532053
eFromJpsiMcParticleIndices.clear();
20542054

20552055
auto mcCollisionGlobalIndex = event.mcCollisionId();
2056-
// auto mcEvent = mcEvents.rawIteratorAt(mcCollisionGlobalIndex);
2056+
auto mcEvent = mcEvents.rawIteratorAt(mcCollisionGlobalIndex);
2057+
2058+
// fill event information
2059+
VarManager::FillEvent<TEventFillMap>(event);
2060+
VarManager::FillEvent<VarManager::ObjTypes::CollisionMC>(mcEvent);
20572061

20582062
auto groupedMCTracks = mcTracks.sliceBy(perReducedMcEvent, mcCollisionGlobalIndex);
20592063
groupedMCTracks.bindInternalIndicesTo(&mcTracks);
@@ -2161,7 +2165,7 @@ struct AnalysisSameEventPairing {
21612165
{
21622166
cout << "AnalysisSameEventPairing::processBarrelOnly() called" << endl;
21632167
runSameEventPairing<true, VarManager::kDecayToEE, gkEventFillMapWithMults, gkTrackFillMapWithCov>(events, bcs, trackAssocsPerCollision, barrelAssocs, barrelTracks, mcEvents, mcTracks);
2164-
runMCGen<VarManager::kDecayToEE>(events, mcEvents, mcTracks);
2168+
runMCGen<VarManager::kDecayToEE, gkEventFillMapWithMults>(events, mcEvents, mcTracks);
21652169
cout << "AnalysisSameEventPairing::processBarrelOnly() completed" << endl;
21662170
}
21672171

@@ -2171,7 +2175,7 @@ struct AnalysisSameEventPairing {
21712175
{
21722176
cout << "AnalysisSameEventPairing::processBarrelPbPbOnly() called" << endl;
21732177
runSameEventPairing<true, VarManager::kDecayToEE, gkEventFillMapWithCentAndMults, gkTrackFillMapWithCov>(events, bcs, trackAssocsPerCollision, barrelAssocs, barrelTracks, mcEvents, mcTracks);
2174-
runMCGen<VarManager::kDecayToEE>(events, mcEvents, mcTracks);
2178+
runMCGen<VarManager::kDecayToEE, gkEventFillMapWithCentAndMults>(events, mcEvents, mcTracks);
21752179
cout << "AnalysisSameEventPairing::processBarrelPbPbOnly() completed" << endl;
21762180
}
21772181

0 commit comments

Comments
 (0)