From f69a77908101898a7de8a0c87361eab241009279 Mon Sep 17 00:00:00 2001
From: David Hammer <dhammer@mailbox.org>
Date: Tue, 18 Jan 2022 16:09:25 +0100
Subject: [PATCH] Fix map shape
---
src/calng/JungfrauCorrection.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/calng/JungfrauCorrection.py b/src/calng/JungfrauCorrection.py
index 028e2678..77cd6dcb 100644
--- a/src/calng/JungfrauCorrection.py
+++ b/src/calng/JungfrauCorrection.py
@@ -71,7 +71,7 @@ class JungfrauGpuRunner(base_gpu.BaseGpuRunner):
# TODO: avoid superclass creating cell table with wrong dtype first
self.cell_table_gpu = cupy.empty(self.memory_cells, dtype=cupy.uint8)
self.input_gain_map_gpu = cupy.empty(self.input_shape, dtype=cupy.uint8)
- self.map_shape = self.input_shape + (3,)
+ self.map_shape = (self.constant_memory_cells, self.pixels_y, self.pixels_x, 3)
# is jungfrau stuff gain mapped?
self.offset_map_gpu = cupy.zeros(self.map_shape, dtype=cupy.float32)
self.rel_gain_map_gpu = cupy.ones(self.map_shape, dtype=cupy.float32)
--
GitLab