From fc40a36f45a0ff66356f94bd4932b219f7aaaf63 Mon Sep 17 00:00:00 2001
From: rettigl <l.rettig@web.de>
Date: Fri, 3 Nov 2023 23:40:58 +0100
Subject: [PATCH] add timed dataframe to new processor functions

---
 sed/core/processor.py | 40 ++++++++++++++++++++++++++++++----------
 1 file changed, 30 insertions(+), 10 deletions(-)

diff --git a/sed/core/processor.py b/sed/core/processor.py
index bd96f7de..15efc245 100644
--- a/sed/core/processor.py
+++ b/sed/core/processor.py
@@ -1208,11 +1208,7 @@ def save_energy_calibration(
                 "Energy calibration parameters not found, need to generate parameters first!",
             ) from exc
 
-        config = {
-            "energy": {
-                "calibration": calibration,
-            },
-        }
+        config = {"energy": {"calibration": calibration}}
         if isinstance(self.ec.offset, dict):
             config["energy"]["offset"] = self.ec.offset
         save_config(config, filename, overwrite)
@@ -1289,12 +1285,12 @@ def add_energy_offset(
             ValueError: If the energy column is not in the dataframe.
         """
         energy_column = self._config["dataframe"]["energy_column"]
-        if energy_column not in self._dataframe.columns:
-            raise ValueError(
-                f"Energy column {energy_column} not found in dataframe! "
-                "Run `append energy axis` first.",
-            )
         if self.dataframe is not None:
+            if energy_column not in self._dataframe.columns:
+                raise ValueError(
+                    f"Energy column {energy_column} not found in dataframe! "
+                    "Run `append energy axis` first.",
+                )
             df, metadata = self.ec.add_offsets(
                 df=self._dataframe,
                 constant=constant,
@@ -1304,6 +1300,17 @@ def add_energy_offset(
                 reductions=reductions,
                 preserve_mean=preserve_mean,
             )
+            if self._timed_dataframe is not None:
+                if energy_column in self._timed_dataframe.columns:
+                    self._timed_dataframe, _ = self.ec.add_offsets(
+                        df=self._timed_dataframe,
+                        constant=constant,
+                        columns=columns,
+                        energy_column=energy_column,
+                        signs=signs,
+                        reductions=reductions,
+                        preserve_mean=preserve_mean,
+                    )
             self._attributes.add(
                 metadata,
                 "add_energy_offset",
@@ -1336,6 +1343,12 @@ def append_tof_ns_axis(
                 df=self._dataframe,
                 **kwargs,
             )
+            if self._timed_dataframe is not None:
+                if self._config["dataframe"]["tof_column"] in self._timed_dataframe.columns:
+                    self._timed_dataframe, _ = self.ec.append_tof_ns_axis(
+                        df=self._timed_dataframe,
+                        **kwargs,
+                    )
             self._attributes.add(
                 metadata,
                 "tof_ns_conversion",
@@ -1357,6 +1370,13 @@ def align_dld_sectors(self, sector_delays: np.ndarray = None, **kwargs):
                 sector_delays=sector_delays,
                 **kwargs,
             )
+            if self._timed_dataframe is not None:
+                if self._config["dataframe"]["tof_column"] in self._timed_dataframe.columns:
+                    self._timed_dataframe, _ = self.ec.align_dld_sectors(
+                        df=self._timed_dataframe,
+                        sector_delays=sector_delays,
+                        **kwargs,
+                    )
             self._attributes.add(
                 metadata,
                 "dld_sector_alignment",