lfhcal: scale event count with momentum

eic · Nov 13, 2024 · aa1e19b · aa1e19b
1 parent bd1cd14
commit aa1e19b
Showing 1 changed file with 8 additions and 1 deletion.
diff --git 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 \