refactor(EyeVolumeVoxelAnnotation): simplify code for plotting

MedVisBonn · Sep 15, 2022 · 355e7f6 · 355e7f6
1 parent 0339fb6
commit 355e7f6
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/eyepy/core/eyevolume.py b/eyepy/core/eyevolume.py
@@ -207,12 +207,9 @@ def plot(
         if vmax is None:
             vmax = max([enface_projection.max(), vmin])
 
-        visible = np.zeros(enface_projection[region].shape)
-        visible[
-            np.logical_and(
-                vmin <= enface_projection[region], enface_projection[region] <= vmax
-            )
-        ] = 1
+        enface_crop = enface_projection[region]
+        visible = np.zeros(enface_crop.shape)
+        visible[np.logical_and(vmin <= enface_crop, enface_crop <= vmax)] = 1
 
         if cbar:
             divider = make_axes_locatable(ax)
@@ -223,8 +220,8 @@ def plot(
             )
 
         ax.imshow(
-            enface_projection[region],
-            alpha=visible[region] * alpha,
+            enface_crop,
+            alpha=visible * alpha,
             cmap=cmap,
             vmin=vmin,
             vmax=vmax,