diff --git a/PWGEM/PhotonMeson/TableProducer/associateMCinfoPhoton.cxx b/PWGEM/PhotonMeson/TableProducer/associateMCinfoPhoton.cxx index e89f5c765e6..7248b009e04 100644 --- a/PWGEM/PhotonMeson/TableProducer/associateMCinfoPhoton.cxx +++ b/PWGEM/PhotonMeson/TableProducer/associateMCinfoPhoton.cxx @@ -57,7 +57,7 @@ using namespace o2::soa; using namespace o2::aod::pwgem::photonmeson::utils::mcutil; using namespace o2::constants::physics; -using MyCollisionsMC = soa::Join; +using MyCollisionsMC = soa::Join; using TracksMC = soa::Join; using FwdTracksMC = soa::Join; using MyEMCClusters = soa::Join; @@ -95,6 +95,7 @@ struct AssociateMCInfoPhoton { Configurable maxY{"maxY", 0.9f, "max |rapidity| for BinnedGenPts table"}; Configurable doStoreAllDaughters{"doStoreAllDaughters", false, "flag to enable storing of all photon, pi0, eta, eta' and omega daughters. This will increase the dervied data size!"}; Configurable useIsSelected{"useIsSelected", true, "flag to enable the usage of isSelected from our EM event selection. Disabling this will increase the derived data size!"}; + Configurable useIsEOI{"useIsEOI", true, "flag to enable the usage of the EOI table"}; HistogramRegistry registry{"EMMCEvent"}; @@ -172,12 +173,16 @@ struct AssociateMCInfoPhoton { template bool passesCollisionGate(TCollision const& collisionIter) { + // event selection same as in createEMEventPhoton if (!collisionIter.has_mcCollision()) { return false; } if (useIsSelected.value && !collisionIter.isSelected()) { return false; } + if (useIsEOI.value && !collisionIter.isEoI()) { + return false; + } return true; }