From d234558a85d4f70b3475b062c2361de89f124a38 Mon Sep 17 00:00:00 2001
From: David Hammer <dhammer@mailbox.org>
Date: Thu, 2 Jun 2022 15:38:58 +0200
Subject: [PATCH] JF: also override shape of gain map (hotfixed at SPB)

---
 src/calng/JungfrauCorrection.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/calng/JungfrauCorrection.py b/src/calng/JungfrauCorrection.py
index 553f86db..5aa8bb5c 100644
--- a/src/calng/JungfrauCorrection.py
+++ b/src/calng/JungfrauCorrection.py
@@ -346,8 +346,11 @@ class JungfrauCorrection(BaseCorrection):
         if len(cell_table.shape) == 0:
             cell_table = cell_table[np.newaxis]
         try:
+            gain_map = data_hash.get("data.gain")
+            if self.unsafe_get("dataFormat.overrideInputAxisOrder"):
+                gain_map.shape = self.input_data_shape
             self.kernel_runner.load_data(
-                image_data, data_hash.get("data.gain"), cell_table
+                image_data, gain_map, cell_table
             )
         except ValueError as e:
             self.log_status_warn(f"Failed to load data: {e}")
-- 
GitLab