Re-allows VTK cells to export PDF

VisTrails · Apr 8, 2014 · 4ed7580 · 4ed7580
1 parent f0eb9c5
commit 4ed7580
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/vistrails/packages/vtk/vtkcell.py b/vistrails/packages/vtk/vtkcell.py
@@ -38,6 +38,7 @@
 # VTK/GUISupport/QVTK. Combine altogether to a single class: QVTKWidget
 ################################################################################
 import vtk
+import os
 from PyQt4 import QtCore, QtGui
 import sip
 from vistrails.core import system
@@ -983,7 +984,11 @@ def dumpToFile(self, filename):
         """dumpToFile() -> None
         Dumps itself as an image to a file, calling saveToPNG
         """
-        self.saveToPNG(filename)
+        ext = os.path.splitext(filename)[1].lower()
+        if ext == '.pdf':
+            self.saveToPDF(filename)
+        else:
+            self.saveToPNG(filename)
 
 class QVTKWidgetSaveCamera(QtGui.QAction):
     """

diff --git a/vistrails/packages/vtk/vtkviewcell.py b/vistrails/packages/vtk/vtkviewcell.py
@@ -38,6 +38,7 @@
 # VTK/GUISupport/QVTK. Combine altogether to a single class: QVTKViewWidget
 ################################################################################
 import vtk
+import os
 from PyQt4 import QtCore, QtGui
 import sip
 from vistrails.core import system
@@ -889,7 +890,11 @@ def dumpToFile(self, filename):
         """dumpToFile() -> None
         Dumps itself as an image to a file, calling saveToPNG
         """
-        self.saveToPNG(filename)
+        ext = os.path.splitext(filename)[1].lower()
+        if ext == '.pdf':
+            self.saveToPDF(filename)
+        else:
+            self.saveToPNG(filename)
 
 class QVTKViewWidgetSaveCamera(QtGui.QAction):
     """