Skip to content
Snippets Groups Projects
Commit 056daaa9 authored by Alán Muñoz's avatar Alán Muñoz
Browse files

refactor(post): use signal.lineage

parent 792fb895
No related branches found
No related tags found
No related merge requests found
...@@ -310,23 +310,15 @@ class PostProcessor(ProcessABC): ...@@ -310,23 +310,15 @@ class PostProcessor(ProcessABC):
parameters = self.parameters_classfun[process].default() parameters = self.parameters_classfun[process].default()
if isinstance(parameters, LineageProcessParameters): if isinstance(parameters, LineageProcessParameters):
with h5py.File(self._filename, "r") as f: lineage = self._signal.lineage(
trap_mo_da = f[parameters.lineage_location] # self.parameters.lineage_location
lineage = np.array( )
(
trap_mo_da["trap"],
trap_mo_da["mother_label"],
trap_mo_da["daughter_label"],
)
).T
loaded_process = self.classfun[process](parameters) loaded_process = self.classfun[process](parameters)
loaded_process.load_lineage(lineage) loaded_process.load_lineage(lineage)
else: else:
loaded_process = self.classfun[process](parameters) loaded_process = self.classfun[process](parameters)
for dataset in datasets: for dataset in datasets:
# print("Processing", process, "for", dataset)
if isinstance(dataset, list): # multisignal process if isinstance(dataset, list): # multisignal process
signal = [self._signal[d] for d in dataset] signal = [self._signal[d] for d in dataset]
elif isinstance(dataset, str): elif isinstance(dataset, str):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment