Commit dc7451d9488cd35d08c3f1469a18c091ccbdb6f3

Authored by tatiana
1 parent cdcc6888

ENH: Export imagedata binary or ascii mode

Showing 1 changed file with 5 additions and 3 deletions   Show diff stats
invesalius/data/imagedata_utils.py
@@ -155,11 +155,13 @@ def BuildEditedImage(imagedata, points): @@ -155,11 +155,13 @@ def BuildEditedImage(imagedata, points):
155 155
156 return imagedata 156 return imagedata
157 157
158 -def Export(imagedata, filename): 158 +def Export(imagedata, filename, bin=False):
159 writer = vtk.vtkXMLImageDataWriter() 159 writer = vtk.vtkXMLImageDataWriter()
160 writer.SetFileName(filename) 160 writer.SetFileName(filename)
161 - writer.SetDataModeToAscii()  
162 - #writer.SetDataModeToBinary() 161 + if bin:
  162 + writer.SetDataModeToBinary()
  163 + else:
  164 + writer.SetDataModeToAscii()
163 writer.SetInput(imagedata) 165 writer.SetInput(imagedata)
164 writer.Write() 166 writer.Write()
165 167