From 3ccfbc98b9e822a238e60d893170ff0d083f75ed Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Sat, 7 Dec 2024 09:17:27 +0100
Subject: [PATCH] Adjust RoiSet focus relative to 3d patch focus

---
 model_server/base/roiset.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/model_server/base/roiset.py b/model_server/base/roiset.py
index 0a8775e6..41edb804 100644
--- a/model_server/base/roiset.py
+++ b/model_server/base/roiset.py
@@ -1017,7 +1017,10 @@ class RoiSet(object):
 
             if pad_to and expanded:
                 patch = pad(patch, pad_to)
-            return {'patch': patch, 'zif': zif}
+            return {
+                'patch': patch,
+                'zif': (roi.z0 + zif) if hasattr(roi, 'z0') else roi.zi,
+            }
 
         df_processed_patches = self._df.apply(lambda r: _make_patch(r), axis=1, result_type='expand')
         if update_focus_zi:
-- 
GitLab