# HG changeset patch
# User greg
# Date 1509732817 14400
# Node ID 2b914b6a0c8384e50308b6f65030af5ba031824d
# Parent  85b0aa4ab75f5a589a09cc106181083fff7e97e5
Uploaded

diff -r 85b0aa4ab75f -r 2b914b6a0c83 dmri.py
--- a/dmri.py	Fri Nov 03 14:13:32 2017 -0400
+++ b/dmri.py	Fri Nov 03 14:13:37 2017 -0400
@@ -4,12 +4,15 @@
 import nibabel
 import shutil
 
+from dipy.core.gradients import gradient_table
 from dipy.data import fetch_sherbrooke_3shell
+from dipy.io import read_bvals_bvecs
 from matplotlib import pyplot
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--input', dest='input', help='Input dataset')
-parser.add_argument('--output', dest='output', help='Output dataset')
+parser.add_argument('--output_gtab', dest='output_gtab', help='Output gradient table dataset')
+parser.add_argument('--output_png', dest='output_png', help='Output dataset')
 
 args = parser.parse_args()
 
@@ -33,4 +36,9 @@
 pyplot.subplot(1, 2, 2).set_axis_off()
 pyplot.imshow(data[:, :, axial_middle, 10].T, cmap='gray', origin='lower')
 pyplot.savefig('data.png', bbox_inches='tight')
-shutil.move('data.png', args.output)
+shutil.move('data.png', args.output_png)
+# Load the b-values and b-vectors.
+bvals, bvecs = read_bvals_bvecs(fbval, fbvec)
+gtab = gradient_table(bvals, bvecs)
+with open(args.output_gtab) as fh:
+    fh.write(gtab.info)
\ No newline at end of file