in qt4 backend, define get_window_title and call it for default savefig filename

mspacek · mspacek · commit 395acc107cbe · 2012-07-05T14:25:05.000-07:00
revert prior changes to qt backend, since I can't test them
diff --git a/lib/matplotlib/backend_bases.py b/lib/matplotlib/backend_bases.py
@@ -2092,6 +2092,14 @@ def get_default_filetype(self):
         """
         return rcParams['savefig.format']
 
+    def get_window_title(self):
+        """
+        Get the title text of the window containing the figure.
+        Return None if there is no window (eg, a PS backend).
+        """
+        if hasattr(self, "manager"):
+            return self.manager.get_window_title()
+
     def set_window_title(self, title):
         """
         Set the title text of the window containing the figure.  Note that
@@ -2413,6 +2421,13 @@ def show_popup(self, msg):
         """
         pass
 
+    def get_window_title(self):
+        """
+        Get the title text of the window containing the figure.
+        Return None if there is no window (eg, a PS backend).
+        """
+        pass
+
     def set_window_title(self, title):
         """
         Set the title text of the window containing the figure.  Note that
diff --git a/lib/matplotlib/backends/backend_qt.py b/lib/matplotlib/backends/backend_qt.py
@@ -419,10 +419,8 @@ def save_figure(self, *args):
         sorted_filetypes = filetypes.items()
         sorted_filetypes.sort()
         default_filetype = self.canvas.get_default_filetype()
-        default_filename = str(self.canvas.window().windowTitle()) or 'image'
-        default_filename = default_filename.replace(' ', '_')
 
-        start = default_filename + '.' + default_filetype
+        start = 'image.' + default_filetype
         filters = []
         selectedFilter = None
         for name, exts in sorted_filetypes:
diff --git a/lib/matplotlib/backends/backend_qt4.py b/lib/matplotlib/backends/backend_qt4.py
@@ -481,6 +481,9 @@ def destroy( self, *args ):
         if DEBUG: print("destroy figure manager")
         self.window.close()
 
+    def get_window_title(self):
+        return str(self.window.windowTitle())
+
     def set_window_title(self, title):
         self.window.setWindowTitle(title)
 
@@ -614,7 +617,7 @@ def save_figure(self, *args):
         sorted_filetypes = filetypes.items()
         sorted_filetypes.sort()
         default_filetype = self.canvas.get_default_filetype()
-        default_filename = str(self.canvas.window().windowTitle()) or 'image'
+        default_filename = self.canvas.get_window_title() or 'image'
         default_filename = default_filename.replace(' ', '_')
 
         start = default_filename + '.' + default_filetype
