From 1872bed0c78de561b3ed321fb609da6a21b02e27 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Le=20Guyader?= <loic.le.guyader@xfel.eu>
Date: Mon, 2 Aug 2021 13:19:09 +0200
Subject: [PATCH] Display roi rectangles in inspect_rois
---
src/toolbox_scs/routines/boz.py | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
diff --git a/src/toolbox_scs/routines/boz.py b/src/toolbox_scs/routines/boz.py
index f118fb8..5e47f03 100644
--- a/src/toolbox_scs/routines/boz.py
+++ b/src/toolbox_scs/routines/boz.py
@@ -451,6 +451,23 @@ def inspect_rois(data_mean, rois, threshold=None, allrois=False):
vmax=np.percentile(data_mean[:, :256], 99))
main_ax.set_aspect('equal')
+ from matplotlib.patches import Rectangle
+ roi = rois['n']
+ main_ax.add_patch(Rectangle((roi['xl'], 128-roi['yh']),
+ roi['xh'] - roi['xl'],
+ roi['yh'] - roi['yl'],
+ alpha=0.3, color='b'))
+ roi = rois['0']
+ main_ax.add_patch(Rectangle((roi['xl'], 128-roi['yh']),
+ roi['xh'] - roi['xl'],
+ roi['yh'] - roi['yl'],
+ alpha=0.3, color='g'))
+ roi = rois['p']
+ main_ax.add_patch(Rectangle((roi['xl'], 128-roi['yh']),
+ roi['xh'] - roi['xl'],
+ roi['yh'] - roi['yl'],
+ alpha=0.3, color='r'))
+
x.plot(Xs, pX)
x.invert_yaxis()
if threshold is not None:
--
GitLab