From 218d8c0e450969b9c7062d66e7763d0c7ea786c3 Mon Sep 17 00:00:00 2001
From: Marshall Scott <mbscott@anl.gov>
Date: Fri, 7 May 2021 15:56:41 -0400
Subject: [PATCH] Added count function

---
 .../scripts/emcal_barrel_pions_electrons_analysis.cxx     | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/benchmarks/barrel_ecal/scripts/emcal_barrel_pions_electrons_analysis.cxx b/benchmarks/barrel_ecal/scripts/emcal_barrel_pions_electrons_analysis.cxx
index ae12c88b..7bd5275e 100644
--- a/benchmarks/barrel_ecal/scripts/emcal_barrel_pions_electrons_analysis.cxx
+++ b/benchmarks/barrel_ecal/scripts/emcal_barrel_pions_electrons_analysis.cxx
@@ -61,10 +61,12 @@ void emcal_barrel_pions_electrons_analysis(const char* input_fname = "sim_output
   auto Esim_ele = [](const std::vector<dd4pod::CalorimeterHitData>& evt, std::vector<dd4pod::Geant4ParticleData> const& input) {
     std::vector<double> result;
     auto total_edep = 0.0;
+    int count = 0;
     if (input[2].pdgID == 11)// Electron
     { 
-      for (const auto& i: evt)
+      for (const auto& i: evt && count > 1)
         total_edep += i.energyDeposit;
+        count++;
     }
     result.push_back(total_edep);
   return result;
@@ -74,10 +76,12 @@ void emcal_barrel_pions_electrons_analysis(const char* input_fname = "sim_output
   auto Esim_pi = [](const std::vector<dd4pod::CalorimeterHitData>& evt, std::vector<dd4pod::Geant4ParticleData> const& input) {
     std::vector<double> result;
     auto total_edep = 0.0;
+    int count = 0;
     if (input[2].pdgID == -211)// Negative pion
     { 
-      for (const auto& i: evt)
+      for (const auto& i: evt && count < 1)
         total_edep += i.energyDeposit;
+        count++;
     }
     result.push_back(total_edep);
   return result;
-- 
GitLab