From 2ed4d85b8327220778882842bfb8d120a96b9856 Mon Sep 17 00:00:00 2001
From: bloomdt-uw <bloomdt@uw.edu>
Date: Tue, 6 Oct 2020 15:39:23 -0700
Subject: [PATCH] Moved show_anatomical_slices to viz.utils

---
 AFQ/viz/utils.py                        | 28 ++++++++++++++++++++
 examples/plot_callosal_tract_profile.py | 34 +------------------------
 2 files changed, 29 insertions(+), 33 deletions(-)

diff --git a/AFQ/viz/utils.py b/AFQ/viz/utils.py
index 3a3ee96a2..3c7e4ecd5 100644
--- a/AFQ/viz/utils.py
+++ b/AFQ/viz/utils.py
@@ -1335,3 +1335,31 @@ def visualize_gif_inline(fname, use_s3fs=False):
         fs.get(fname_remote, fname)
 
     display.display(display.Image(fname))
+
+
+def show_anatomical_slices(img_data, title):
+    """
+    display anatomical slices from midpoint
+
+    based on:
+    https://nipy.org/nibabel/coordinate_systems.html
+    """
+
+    axial_slice = img_data[:, :, int(img_data.shape[2] / 2)]
+    coronal_slice = img_data[:, int(img_data.shape[1] / 2), :]
+    sagittal_slice = img_data[int(img_data.shape[0] / 2), :, :]
+
+    fig = plt.figure(constrained_layout=False)
+    gs = fig.add_gridspec(nrows=3, ncols=2, wspace=0.01, hspace=0.01)
+    ax1 = fig.add_subplot(gs[:-1, :])
+    ax1.imshow(axial_slice.T, cmap="gray", origin="lower")
+    ax1.axis('off')
+    ax2 = fig.add_subplot(gs[2, 0])
+    ax2.imshow(coronal_slice.T, cmap="gray", origin="lower")
+    ax2.axis('off')
+    ax3 = fig.add_subplot(gs[2, 1])
+    ax3.imshow(sagittal_slice.T, cmap="gray", origin="lower")
+    ax3.axis('off')
+
+    plt.suptitle(title)
+    plt.show()
diff --git a/examples/plot_callosal_tract_profile.py b/examples/plot_callosal_tract_profile.py
index 3fcdf6b72..94264ce49 100644
--- a/examples/plot_callosal_tract_profile.py
+++ b/examples/plot_callosal_tract_profile.py
@@ -39,6 +39,7 @@
 import AFQ.models.dti as dti
 import AFQ.segmentation as seg
 from AFQ.utils.volume import patch_up_roi, density_map
+from AFQ.viz.utils import show_anatomical_slices
 from AFQ.viz.plotly_backend import visualize_bundles, visualize_volume
 
 import logging
@@ -56,38 +57,6 @@
 # Target directory for this example's output files
 working_dir = "./callosal_tract_profile"
 
-##########################################################################
-# Helper functions:
-# -----------------
-
-
-def show_anatomical_slices(img_data, title):
-    """
-    display anatomical slices from midpoint
-
-    based on:
-    https://nipy.org/nibabel/coordinate_systems.html
-    """
-
-    axial_slice = img_data[:, :, int(img_data.shape[2] / 2)]
-    coronal_slice = img_data[:, int(img_data.shape[1] / 2), :]
-    sagittal_slice = img_data[int(img_data.shape[0] / 2), :, :]
-
-    fig = plt.figure(constrained_layout=False)
-    gs = fig.add_gridspec(nrows=3, ncols=2, wspace=0.01, hspace=0.01)
-    ax1 = fig.add_subplot(gs[:-1, :])
-    ax1.imshow(axial_slice.T, cmap="gray", origin="lower")
-    ax1.axis('off')
-    ax2 = fig.add_subplot(gs[2, 0])
-    ax2.imshow(coronal_slice.T, cmap="gray", origin="lower")
-    ax2.axis('off')
-    ax3 = fig.add_subplot(gs[2, 1])
-    ax3.imshow(sagittal_slice.T, cmap="gray", origin="lower")
-    ax3.axis('off')
-
-    plt.suptitle(title)
-    plt.show()
-
 
 ##########################################################################
 # Get example data:
@@ -420,7 +389,6 @@ def show_anatomical_slices(img_data, title):
 fiber_groups = segmentation.fiber_groups
 
 for bundle in bundles:
-    
     tractogram = StatefulTractogram(fiber_groups[bundle]['sl'].streamlines,
                                     img,
                                     Space.VOX)