From 86d2de32a16d6e10ec875baf6aa5e8e5727b7a75 Mon Sep 17 00:00:00 2001
From: David Hammer <dhammer@mailbox.org>
Date: Wed, 26 Jan 2022 13:01:07 +0100
Subject: [PATCH] Tie burst mode more to memory cell count
---
src/calng/JungfrauCorrection.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/src/calng/JungfrauCorrection.py b/src/calng/JungfrauCorrection.py
index 03a8f27e..9209ffe7 100644
--- a/src/calng/JungfrauCorrection.py
+++ b/src/calng/JungfrauCorrection.py
@@ -44,9 +44,7 @@ class JungfrauGpuRunner(base_gpu.BaseGpuRunner):
input_data_dtype=cupy.uint16,
output_data_dtype=cupy.float32,
bad_pixel_mask_value=cupy.nan,
- burst_mode=False,
):
- self.burst_mode = burst_mode
self.input_shape = (memory_cells, pixels_y, pixels_x)
self.processed_shape = self.input_shape
super().__init__(
@@ -79,12 +77,16 @@ class JungfrauGpuRunner(base_gpu.BaseGpuRunner):
"input_data_dtype": utils.np_dtype_to_c_type(self.input_data_dtype),
"output_data_dtype": utils.np_dtype_to_c_type(self.output_data_dtype),
"corr_enum": utils.enum_to_c_template(CorrectionFlags),
- "burst_mode": self.memory_cells > 1,
+ "burst_mode": self.burst_mode,
}
)
self.source_module = cupy.RawModule(code=kernel_source)
self.correction_kernel = self.source_module.get_function("correct")
+ @property
+ def burst_mode(self):
+ return self.memory_cells > 1
+
def _get_raw_for_preview(self):
return self.input_data_gpu
@@ -305,7 +307,6 @@ class JungfrauCorrection(BaseCorrection):
self._kernel_runner_init_args = {
"bad_pixel_mask_value": self.bad_pixel_mask_value,
- "burst_mode": False, # TODO
}
def process_data(
--
GitLab