X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=util%2Fview-mri-data.py;h=01bddbdcfc8d91aa24f6ae3156a5d57ccb985a69;hb=3a564c16360e72cf89d3569e6e35aad0fa042e90;hp=96fbfa57b23b0d07333f541ded8832c16b947dff;hpb=a4a270ea5c31d76a240c108a46d71ea37e578b17;p=spline3.git

diff --git a/util/view-mri-data.py b/util/view-mri-data.py
index 96fbfa5..01bddbd 100755
--- a/util/view-mri-data.py
+++ b/util/view-mri-data.py
@@ -1,4 +1,4 @@
-#!/usr/bin/python2
+#!/usr/bin/python3
 
 """
 Display volumetric data from the Stanford Volume Data Archive. The
@@ -11,7 +11,7 @@ import sys
 
 def usage():
     binary = sys.argv[0]
-    print "Usage:", binary, "<data file>"
+    print(f"Usage: {binary} <data file>")
 
 if len(sys.argv) < 2:
     usage()
@@ -33,18 +33,19 @@ def cube_root(n):
 import numpy as np
 data = np.fromfile(sys.argv[1], dtype='>u2')
 
-original_data_length = 7143424
-multiplier = cube_root(len(data) / original_data_length)
+original_data_length = 109*256*256
+multiplier = cube_root(len(data) // original_data_length)
 data.shape = (109*multiplier, 256*multiplier, 256*multiplier)
 data = data.T
 
 # Display the data
-mlab.figure(bgcolor=(0, 0, 0), size=(1000, 1000))
+mlab.figure( bgcolor=(0,0,0), size=(1000,1000) )
 
+# Our data is scalar (grayscale), but not equally spaced in all
+# directions.
 src = mlab.pipeline.scalar_field(data)
-
-# Our data is not equally spaced in all directions:
 src.spacing = [1, 1, 1.5]
 src.update_image_data = True
-v = mlab.pipeline.volume(src, vmax=2500, vmin=1400)
+
+mlab.pipeline.volume(src, vmin=1400, vmax=2500)
 mlab.show()