Saving screenshots with vtk6

Thiago Franco de Moraes
1 parent 48a76302
Showing 2 changed files with 7 additions and 5 deletions Show diff stats
invesalius/data/viewer_slice.py
invesalius/data/viewer_volume.py
@@ -753,6 +753,7 @@ class Viewer(wx.Panel):
                 image = vtk.vtkRenderLargeImage()
                 image.SetInput(ren)
                 image.SetMagnification(1)
+                image.Update()
                 image = image.GetOutput()
@@ -769,8 +770,8 @@ class Viewer(wx.Panel):
                 elif (filetype == const.FILETYPE_TIF):
                     writer = vtk.vtkTIFFWriter()
                     filename = "%s.tif"%filename.strip(".tif")
-                
-                writer.SetInput(image)
+
+                writer.SetInputData(image)
                 writer.SetFileName(filename)
                 writer.Write()
@@ -318,6 +318,7 @@ class Viewer(wx.Panel):
                 image = vtk.vtkRenderLargeImage()
                 image.SetInput(self.ren)
                 image.SetMagnification(1)
+                image.Update()
                 image = image.GetOutput()
@@ -333,12 +334,12 @@ class Viewer(wx.Panel):
                 elif (filetype == const.FILETYPE_TIF):
                     writer = vtk.vtkTIFFWriter()
                     filename = "%s.tif"%filename.strip(".tif")
-                
-                writer.SetInput(image)
+
+                writer.SetInputData(image)
                 writer.SetFileName(filename)
                 writer.Write()
         Publisher.sendMessage('End busy cursor')
- 
+
     def OnCloseProject(self, pubsub_evt):
         if self.raycasting_volume:
             self.raycasting_volume = False