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