From 056daaa9c3bc03fb218c6a9ac61ae1d38903a9c9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <alan.munoz@ed.ac.uk>
Date: Wed, 28 Sep 2022 00:48:29 +0100
Subject: [PATCH] refactor(post): use signal.lineage

---
 src/postprocessor/core/processor.py | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/src/postprocessor/core/processor.py b/src/postprocessor/core/processor.py
index 5f28dbba..915139db 100644
--- a/src/postprocessor/core/processor.py
+++ b/src/postprocessor/core/processor.py
@@ -310,23 +310,15 @@ class PostProcessor(ProcessABC):
                 parameters = self.parameters_classfun[process].default()
 
             if isinstance(parameters, LineageProcessParameters):
-                with h5py.File(self._filename, "r") as f:
-                    trap_mo_da = f[parameters.lineage_location]
-                    lineage = np.array(
-                        (
-                            trap_mo_da["trap"],
-                            trap_mo_da["mother_label"],
-                            trap_mo_da["daughter_label"],
-                        )
-                    ).T
+                lineage = self._signal.lineage(
+                    # self.parameters.lineage_location
+                )
                 loaded_process = self.classfun[process](parameters)
                 loaded_process.load_lineage(lineage)
             else:
                 loaded_process = self.classfun[process](parameters)
 
             for dataset in datasets:
-                # print("Processing", process, "for", dataset)
-
                 if isinstance(dataset, list):  # multisignal process
                     signal = [self._signal[d] for d in dataset]
                 elif isinstance(dataset, str):
-- 
GitLab