From 17d0aff2144414ff0aba957c4845caa091f6ccb9 Mon Sep 17 00:00:00 2001
From: gac-x03da <gac-x03da@x03da-cons-02.psi.ch>
Date: Tue, 15 Aug 2023 12:19:12 +0200
Subject: [PATCH] Script execution

---
 script/local.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/script/local.py b/script/local.py
index 12ac92a6..766d07ca 100644
--- a/script/local.py
+++ b/script/local.py
@@ -35,7 +35,12 @@ sinp = SimulatedInput()
 def integrate_image(vertical = True):
     data = Scienta.dataArray.read()
     #Integrate and plot
-    (width,height) = Scienta.getImageSize().tolist()
+    dims = Scienta.getImageSize().tolist()
+    if len(dims)==2:
+        (width,height) = dims
+    else:
+        (width,height, images) = dims    
+
     integration = []
     if vertical:
         for i in range(width):