From 70105d28adf3b064ed89e02a41e3e2f1cac4d0fa Mon Sep 17 00:00:00 2001
From: wyzula-jan <jan.wyzula@psi.ch>
Date: Tue, 16 Dec 2025 00:33:12 +0100
Subject: [PATCH] wip transpose preview signal fix

---
 ophyd_devices/utils/bec_signals.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/ophyd_devices/utils/bec_signals.py b/ophyd_devices/utils/bec_signals.py
index 8a6639f..e8ef2ec 100644
--- a/ophyd_devices/utils/bec_signals.py
+++ b/ophyd_devices/utils/bec_signals.py
@@ -583,10 +583,18 @@ class PreviewSignal(BECMessageSignal):
         if self.ndim == 1:
             return value
 
+        if not isinstance(value, np.ndarray):
+            value = np.asarray(value)
+
         if self.num_rotation_90:
             value = np.rot90(value, k=self.num_rotation_90, axes=(0, 1))
         if self.transpose:
-            value = np.transpose(value)
+            if value.ndim == 2:
+                value = value.T
+            elif value.ndim == 3:
+                value = np.transpose(value, (1, 0, 2))
+            else:
+                value = np.transpose(value)
 
         return value