From cb82cd89df148722a8b5ff4e48de2273a84daea6 Mon Sep 17 00:00:00 2001 From: David Hammer <dhammer@mailbox.org> Date: Wed, 15 Jan 2025 13:25:47 +0100 Subject: [PATCH] Reorder things in some addons' init --- src/calng/base_correction.py | 1 - src/calng/correction_addons/base_addon.py | 2 +- src/calng/correction_addons/litpixel_counter.py | 10 +++++----- src/calng/correction_addons/peakfinder9.py | 2 +- 4 files changed, 7 insertions(+), 8 deletions(-) diff --git a/src/calng/base_correction.py b/src/calng/base_correction.py index 0271fea3..e62a59cf 100644 --- a/src/calng/base_correction.py +++ b/src/calng/base_correction.py @@ -588,7 +588,6 @@ class BaseCorrection(PythonDevice): ) # update device schema in one fell swoop self.updateSchema(schema_override) - print(self._enabled_addons) self._geometry = None def _on_geometry(geometry): diff --git a/src/calng/correction_addons/base_addon.py b/src/calng/correction_addons/base_addon.py index 307e3b14..1ab99090 100644 --- a/src/calng/correction_addons/base_addon.py +++ b/src/calng/correction_addons/base_addon.py @@ -17,9 +17,9 @@ class BaseCorrectionAddon: def __init__(self, device, prefix, config): """Will get node from extend_device_schema; node paths won't include prefix""" - self.reconfigure(config) self._device = device self._prefix = prefix + self.reconfigure(config) def post_correction( self, train_id, processed_data, cell_table, pulse_table, output_hash diff --git a/src/calng/correction_addons/litpixel_counter.py b/src/calng/correction_addons/litpixel_counter.py index bd2b2c03..90c4eaba 100644 --- a/src/calng/correction_addons/litpixel_counter.py +++ b/src/calng/correction_addons/litpixel_counter.py @@ -22,16 +22,16 @@ def factors(n): class LitPixelCounter(BaseCorrectionAddon): def __init__(self, device, prefix, config): - # note: this includes reconfigure - super().__init__(device, prefix, config) - global cupy - import cupy - self._ss_size = device.kernel_runner.num_pixels_ss self._ss_factors = factors(self._ss_size) self._fs_size = device.kernel_runner.num_pixels_fs self._fs_factors = factors(self._fs_size) + # note: this includes reconfigure + super().__init__(device, prefix, config) + global cupy + import cupy + # these properties may be changed to the factors # of module dimensions due to protection device.set(f"{prefix}.ssAsicSize", self._ss_asic_size) diff --git a/src/calng/correction_addons/peakfinder9.py b/src/calng/correction_addons/peakfinder9.py index 1647350c..4a9d199f 100644 --- a/src/calng/correction_addons/peakfinder9.py +++ b/src/calng/correction_addons/peakfinder9.py @@ -170,11 +170,11 @@ class Peakfinder9(BaseCorrectionAddon): pass def __init__(self, device, prefix, config): + self._config = config super().__init__(device, prefix, config) global cupy import cupy - self._config = config _src_dir = pathlib.Path(__file__).absolute().parent.parent with (_src_dir / "kernels" / "peakfinder9_gpu.cu").open("r") as fd: -- GitLab