ADD: Read, write and view vtkImageData funcations

tatiana
1 parent bee27cf2
Showing 1 changed file with 26 additions and 0 deletions Show diff stats
invesalius/data/imagedata_utils.py
@@ -124,3 +124,29 @@ def BuildEditedImage(imagedata, points):
         imagedata.Update()
     return imagedata
+
+def Export(imagedata, filename):
+    writer = vtk.vtkXMLImageDataWriter()
+    writer.SetFileName(filename)
+    writer.SetDataModeToAscii()
+    #writer.SetDataModeToBinary()
+    writer.SetInput(imagedata)
+    writer.Write()
+
+def Read(filename):
+    reader = vtk.vtkXMLImageDataReader()
+    reader.SetFileName(filename)
+    #reader.WholeSlicesOn()
+    reader.Update()
+    return reader.GetOutput()
+
+def View(imagedata):
+    viewer = vtk.vtkImageViewer()
+    viewer.SetInput(imagedata)
+    viewer.SetZSlice(10)
+    viewer.SetColorWindow(200)
+    viewer.SetColorLevel(100)
+    viewer.Render()
+    
+    import time
+    time.sleep(10)
	@@ -124,3 +124,29 @@ def BuildEditedImage(imagedata, points):		@@ -124,3 +124,29 @@ def BuildEditedImage(imagedata, points):
124	imagedata.Update()	124	imagedata.Update()
125		125
126	return imagedata	126	return imagedata
		127	+
		128	+def Export(imagedata, filename):
		129	+ writer = vtk.vtkXMLImageDataWriter()
		130	+ writer.SetFileName(filename)
		131	+ writer.SetDataModeToAscii()
		132	+ #writer.SetDataModeToBinary()
		133	+ writer.SetInput(imagedata)
		134	+ writer.Write()
		135	+
		136	+def Read(filename):
		137	+ reader = vtk.vtkXMLImageDataReader()
		138	+ reader.SetFileName(filename)
		139	+ #reader.WholeSlicesOn()
		140	+ reader.Update()
		141	+ return reader.GetOutput()
		142	+
		143	+def View(imagedata):
		144	+ viewer = vtk.vtkImageViewer()
		145	+ viewer.SetInput(imagedata)
		146	+ viewer.SetZSlice(10)
		147	+ viewer.SetColorWindow(200)
		148	+ viewer.SetColorLevel(100)
		149	+ viewer.Render()
		150	+
		151	+ import time
		152	+ time.sleep(10)