From 8842f7765e14f82c51f3075dba9e7449456dc505 Mon Sep 17 00:00:00 2001
From: Danilo Ferreira de Lima <danilo.enoque.ferreira.de.lima@xfel.de>
Date: Wed, 18 Jan 2023 16:22:40 +0100
Subject: [PATCH] Added function to retrieve low res range.
---
pes_to_spec/model.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py
index e65f029..818cdc3 100644
--- a/pes_to_spec/model.py
+++ b/pes_to_spec/model.py
@@ -540,6 +540,10 @@ class Model(TransformerMixin, BaseEstimator):
"""Get x-axis of high-resolution data."""
return self.y_model.named_steps["smoothen"].energy
+ def get_low_resolution_range(self) -> Tuple[int, int]:
+ """Get bin number with first and last relevant bins in the low-resolution spectrum."""
+ return self.x_model.named_steps['select'].tof_start, (self.x_model.named_steps['select'].tof_start + self.y_model.named_steps['select'].delta_tof)
+
def debug_peak_finding(self, low_res_data: Dict[str, np.ndarray], filename: str):
"""
Produce image to understand if the peak finding step worked well.
--
GitLab