diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py index e65f02920ec22e614ad9d88f35907fd5d653106a..818cdc37b560e3eb39eff811e37afbf15a09a20e 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.