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	124	imagedata.Update()
125	125
126	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)
...	...