ENH: Update slicer.util.updateVolumeFromArray to support 1D NumPy array

jcfr · Aug 1, 2023 · 780f35c · 780f35c
1 parent b8e8faf
commit 780f35c
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/Base/Python/slicer/util.py b/Base/Python/slicer/util.py
@@ -2245,7 +2245,16 @@ def updateVolumeFromArray(volumeNode, narray):
     """
 
     vshape = tuple(reversed(narray.shape))
-    if len(vshape) == 2:
+    if len(vshape) == 1:
+        # Line of pixels
+        vcomponents = 1
+        # Put the slice into a single-slice 3D volume
+        import numpy as np
+        narray3d = np.zeros([1, 1, narray.shape[0]])
+        narray3d[0, 0, :] = narray
+        narray = narray3d
+        vshape = tuple(reversed(narray.shape))
+    elif len(vshape) == 2:
         # Scalar 2D volume
         vcomponents = 1
         # Put the slice into a single-slice 3D volume