From 46c7dd685ce1e460642ff2df999cf1c0b9807e36 Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
Date: Sat, 7 Dec 2024 12:38:58 -0500
Subject: [PATCH] backwards_ecal: put PLOT_TITLE to legend
---
benchmarks/backwards_ecal/Snakefile | 5 +++++
benchmarks/backwards_ecal/backwards_ecal.org | 4 ++--
2 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/benchmarks/backwards_ecal/Snakefile b/benchmarks/backwards_ecal/Snakefile
index 5d654ee9..2cccdb44 100644
--- a/benchmarks/backwards_ecal/Snakefile
+++ b/benchmarks/backwards_ecal/Snakefile
@@ -154,6 +154,11 @@ rule backwards_ecal:
directory("results/backwards_ecal/{CAMPAIGN}/")
shell:
"""
+if [[ "{wildcards.CAMPAIGN}" == "local" ]]; then
+export PLOT_TITLE="Benchmark simulation"
+else
+export PLOT_TITLE="\\textbf{{ePIC}} Simulation {wildcards.CAMPAIGN}"
+fi
env \
MATPLOTLIBRC={input.matplotlibrc} \
DETECTOR_CONFIG=""" + DETECTOR_CONFIG + """ \
diff --git a/benchmarks/backwards_ecal/backwards_ecal.org b/benchmarks/backwards_ecal/backwards_ecal.org
index 1d70e2c9..5ef59c82 100644
--- a/benchmarks/backwards_ecal/backwards_ecal.org
+++ b/benchmarks/backwards_ecal/backwards_ecal.org
@@ -127,7 +127,7 @@ fractions_below = {}
for ix, energy in enumerate(energies):
energy_value = float(energy.replace("GeV", "").replace("MeV", "e-3"))
- clf_label = "leading cluster"
+ clf_label = PLOT_TITLE
def clf(events):
return ak.drop_none(ak.max(events["EcalEndcapNClusters.energy"], axis=-1)) / energy_value
e_pred = clf(e_eval[energy])
@@ -242,7 +242,7 @@ rocs = {}
for ix, energy in enumerate(energies):
energy_value = float(energy.replace("GeV", "").replace("MeV", "e-3"))
- clf_label = "leading cluster"
+ clf_label = PLOT_TITLE
def clf(events):
return ak.drop_none(ak.max(events["EcalEndcapNClusters.energy"], axis=-1)) / energy_value
e_pred = clf(e_eval[energy])
--
GitLab