Handling orientation

Thiago Franco de Moraes
1 parent afdb3ce1
Showing 1 changed file with 13 additions and 2 deletions Show diff stats
invesalius/data/imagedata_utils.py
@@ -681,11 +681,22 @@ def dcmmf2memmap(dcm_file, orientation):
  
     d = numpy_support.vtk_to_numpy(o.GetPointData().GetScalars())
     d.shape = z, y, x
-    matrix[:] = d
-    matrix.flush()
+    if orientation == 'CORONAL':
+        matrix.shape = y, z, x
+        for n in xrange(z):
+            matrix[:, n, :] = d[n]
+    elif orientation == 'SAGITTAL':
+        matrix.shape = x, z, y
+        for n in xrange(z):
+            matrix[:, :, n] = d[n]
+    else:
+        matrix[:] = d
  
+    matrix.flush()
     scalar_range = matrix.min(), matrix.max()
  
+    print "ORIENTATION", orientation
+
     return matrix, spacing, scalar_range, temp_file
...	...	@@ -681,11 +681,22 @@ def dcmmf2memmap(dcm_file, orientation):
681	681
682	682	d = numpy_support.vtk_to_numpy(o.GetPointData().GetScalars())
683	683	d.shape = z, y, x
684		- matrix[:] = d
685		- matrix.flush()
	684	+ if orientation == 'CORONAL':
	685	+ matrix.shape = y, z, x
	686	+ for n in xrange(z):
	687	+ matrix[:, n, :] = d[n]
	688	+ elif orientation == 'SAGITTAL':
	689	+ matrix.shape = x, z, y
	690	+ for n in xrange(z):
	691	+ matrix[:, :, n] = d[n]
	692	+ else:
	693	+ matrix[:] = d
686	694
	695	+ matrix.flush()
687	696	scalar_range = matrix.min(), matrix.max()
688	697
	698	+ print "ORIENTATION", orientation
	699	+
689	700	return matrix, spacing, scalar_range, temp_file
690	701
691	702
...	...