diff --git a/src/calng/JungfrauCorrection.py b/src/calng/JungfrauCorrection.py index 33a1b00619243f318c8f2b9521fa6132f558fd15..8dc6210a8e4d0173510e05606b70b01e042f9fd5 100644 --- a/src/calng/JungfrauCorrection.py +++ b/src/calng/JungfrauCorrection.py @@ -419,20 +419,21 @@ class JungfrauCorrection(BaseCorrection): ) def _load_constant_to_runner(self, constant, constant_data): - if constant is JungfrauConstants.Offset10Hz: + if constant_data.shape[0] == self.get("dataFormat.pixelsX"): + constant_data = np.transpose(constant_data, (2, 1, 0, 3)) + else: constant_data = np.transpose(constant_data, (2, 0, 1, 3)) + if constant is JungfrauConstants.Offset10Hz: self.kernel_runner.offset_map_gpu.set(constant_data.astype(np.float32)) if not self.get("corrections.offset.available"): self.set("corrections.offset.available", True) elif constant is JungfrauConstants.RelativeGain10Hz: - constant_data = np.transpose(constant_data, (2, 1, 0, 3)) self.kernel_runner.rel_gain_map_gpu.set(constant_data.astype(np.float32)) if not self.get("corrections.relGain.available"): self.set("corrections.relGain.available", True) elif constant in ( JungfrauConstants.BadPixelsDark10Hz, JungfrauConstants.BadPixelsFF10Hz ): - constant_data = np.transpose(constant_data, (2, 0, 1, 3)) self.kernel_runner.bad_pixel_map_gpu |= cupy.asarray(constant_data) if not self.get("corrections.badPixels.available"): self.set("corrections.badPixels.available", True)