Implement savefig.directory for gtk3 gui

mspacek · mspacek · commit a9cb8cefd8c4 · 2013-02-01T14:44:32.000-08:00
diff --git a/lib/matplotlib/backends/backend_gtk3.py b/lib/matplotlib/backends/backend_gtk3.py
@@ -537,6 +537,7 @@ def get_filechooser(self):
         fc = FileChooserDialog(
             title='Save the figure',
             parent=self.win,
+            path=os.path.expanduser(rcParams.get('savefig.directory', '')),
             filetypes=self.canvas.get_supported_filetypes(),
             default_filetype=self.canvas.get_default_filetype())
         fc.set_current_name(self.canvas.get_default_filename())
@@ -545,6 +546,13 @@ def get_filechooser(self):
     def save_figure(self, *args):
         fname, format = self.get_filechooser().get_filename_from_user()
         if fname:
+            startpath = os.path.expanduser(rcParams.get('savefig.directory', ''))
+            if startpath == '':
+                # explicitly missing key or empty str signals to use cwd
+                rcParams['savefig.directory'] = startpath
+            else:
+                # save dir for next time
+                rcParams['savefig.directory'] = os.path.dirname(str(fname))
             try:
                 self.canvas.print_figure(fname, format=format)
             except Exception, e:
