From b67947df4b25f23dae1c603883bf395533265a05 Mon Sep 17 00:00:00 2001
From: Daiki Sekihata <daiki.sekihata@cern.ch>
Date: Mon, 11 Aug 2025 11:03:51 +0200
Subject: [PATCH] PWGEM/Dilepton: add eoi in MC for consistency

---
 PWGEM/Dilepton/TableProducer/associateMCinfoDilepton.cxx | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/PWGEM/Dilepton/TableProducer/associateMCinfoDilepton.cxx b/PWGEM/Dilepton/TableProducer/associateMCinfoDilepton.cxx
index 52867a7609d..6bd004a0e40 100644
--- a/PWGEM/Dilepton/TableProducer/associateMCinfoDilepton.cxx
+++ b/PWGEM/Dilepton/TableProducer/associateMCinfoDilepton.cxx
@@ -42,7 +42,7 @@ struct AssociateMCInfoDilepton {
     kPCM = 0x4,
   };
 
-  using MyCollisionsMC = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels, aod::EMEvSels>;
+  using MyCollisionsMC = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels, aod::EMEvSels, aod::EMEoIs>;
   using TracksMC = soa::Join<aod::TracksIU, aod::McTrackLabels>;
   using FwdTracksMC = soa::Join<aod::FwdTracks, aod::McFwdTrackLabels>;
   using MFTTracksMC = soa::Join<aod::MFTTracks, aod::McMFTTrackLabels>;
@@ -172,6 +172,10 @@ struct AssociateMCInfoDilepton {
         continue;
       }
 
+      if (!collision.isEoI()) { // events with at least 1 lepton for data reduction.
+        continue;
+      }
+
       registry.fill(HIST("hEventCounter"), 2);
       auto mcCollision = collision.mcCollision();
       mceventlabels(fEventLabels.find(mcCollision.globalIndex())->second, collision.mcMask());