From aa1e19b957aeeb595b042096777fe470cf8ec8ba Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
Date: Tue, 12 Nov 2024 01:18:10 -0500
Subject: [PATCH] lfhcal: scale event count with momentum

---
 benchmarks/lfhcal/Snakefile | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/benchmarks/lfhcal/Snakefile b/benchmarks/lfhcal/Snakefile
index 459392c0..cef068e5 100644
--- a/benchmarks/lfhcal/Snakefile
+++ b/benchmarks/lfhcal/Snakefile
@@ -1,3 +1,10 @@
+def get_n_events(wildcards):
+    energy = float(wildcards.ENERGY.replace("GeV", "").replace("MeV", "e-3"))
+    n_events = 1000
+    n_events = int(n_events // (energy ** 0.5))
+    return n_events
+
+
 rule lfhcal_sim:
     input:
         steering_file="EPIC/EVGEN/SINGLE/{PARTICLE}/{ENERGY}/{PHASE_SPACE}/{PARTICLE}_{ENERGY}_{PHASE_SPACE}.steer",
@@ -12,7 +19,7 @@ rule lfhcal_sim:
         PHASE_SPACE="3to50deg",
         INDEX="\d{4}",
     params:
-        N_EVENTS=1000
+        N_EVENTS=get_n_events,
     shell:
         """
 ddsim \
-- 
GitLab