Skip to content

Commit 3652e4a

Browse files
anntzeranntzer
committed
Interactive setting of savefig rcParams (Qt only).
An additional window pops after selecting the filename to allow setting of savefig-related rcParams. For simplicity and consistency, the "ps.usedistiller" rcParam is now normalized to `None` when not set.
1 parent ab98852 commit 3652e4a

File tree

3 files changed

+110
-19
lines changed

3 files changed

+110
-19
lines changed

‎lib/matplotlib/backends/backend_qt5.py

Lines changed: 102 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,8 @@
2727

2828
from .qt_compat import (QtCore, QtGui, QtWidgets, _getSaveFileName,
2929
__version__, is_pyqt5)
30-
from matplotlib.backends.qt_editor.formsubplottool import UiSubplotTool
30+
from .qt_editor import formlayout
31+
from .qt_editor.formsubplottool import UiSubplotTool
3132

3233
backend_version = __version__
3334

@@ -725,19 +726,19 @@ def save_figure(self, *args):
725726
startpath = matplotlib.rcParams.get('savefig.directory', '')
726727
startpath = os.path.expanduser(startpath)
727728
start = os.path.join(startpath, self.canvas.get_default_filename())
729+
728730
filters = []
729-
selectedFilter = None
731+
selected_filter = None
730732
for name, exts in sorted_filetypes:
731-
exts_list = " ".join(['*.%s' % ext for ext in exts])
732-
filter = '%s (%s)' % (name, exts_list)
733+
filters.append(
734+
'{} ({})'.format(name, ' '.join(map('*.{}'.format, exts))))
733735
if default_filetype in exts:
734-
selectedFilter = filter
735-
filters.append(filter)
736+
selected_filter = filters[-1]
736737
filters = ';;'.join(filters)
737738

738-
fname, filter = _getSaveFileName(self.parent,
739-
"Choose a filename to save to",
740-
start, filters, selectedFilter)
739+
fname, selected_filter = _getSaveFileName(
740+
self.parent, "Choose a filename to save to",
741+
start, filters, selected_filter)
741742
if fname:
742743
if startpath == '':
743744
# explicitly missing key or empty str signals to use cwd
@@ -746,6 +747,25 @@ def save_figure(self, *args):
746747
# save dir for next time
747748
savefig_dir = os.path.dirname(six.text_type(fname))
748749
matplotlib.rcParams['savefig.directory'] = savefig_dir
750+
options = _default_savefig_options
751+
try:
752+
options = (options
753+
+ [None]
754+
+ _extra_savefig_options[
755+
os.path.splitext(fname)[1].lower()])
756+
except KeyError:
757+
pass
758+
fedit_arg = []
759+
for option in options:
760+
if option is None:
761+
fedit_arg.append((None, None))
762+
else:
763+
fedit_arg.append(option.make_fedit_entry())
764+
fedit_res = formlayout.fedit(fedit_arg, "Options")
765+
if not fedit_res:
766+
return
767+
for option, res in zip(filter(None, options), fedit_res):
768+
option.setter(res)
749769
try:
750770
self.canvas.print_figure(six.text_type(fname))
751771
except Exception as e:
@@ -754,6 +774,79 @@ def save_figure(self, *args):
754774
QtWidgets.QMessageBox.Ok, QtWidgets.QMessageBox.NoButton)
755775

756776

777+
class _Option(object):
778+
779+
def __init__(self, name, rc_key=None, getter=None, setter=None):
780+
if rc_key and not (getter or setter):
781+
def make_fedit_entry():
782+
return (name, matplotlib.rcParams[rc_key])
783+
784+
def setter(val):
785+
matplotlib.rcParams[rc_key] = val
786+
787+
elif getter and setter and not rc_key:
788+
def make_fedit_entry():
789+
return (name, getter())
790+
791+
else:
792+
raise ValueError("Invalid entry")
793+
794+
self.name = name
795+
self.make_fedit_entry = make_fedit_entry
796+
self.setter = setter
797+
798+
799+
_default_savefig_options = [
800+
_Option("DPI", "savefig.dpi"),
801+
_Option("Face color", "savefig.facecolor"),
802+
_Option("Edge color", "savefig.edgecolor"),
803+
_Option("Tight bounding box",
804+
getter=lambda: matplotlib.rcParams["savefig.bbox"] == "tight",
805+
setter=lambda val: matplotlib.rcParams.__setitem__(
806+
"savefig.bbox", "tight" if val else "standard")),
807+
_Option("Tight bounding box padding", "savefig.pad_inches"),
808+
_Option("Transparent background", "savefig.transparent")]
809+
810+
811+
_extra_savefig_options = {
812+
".jpg": [
813+
_Option("JPEG quality", "savefig.jpeg_quality")],
814+
".jpeg": [
815+
_Option("JPEG quality", "savefig.jpeg_quality")],
816+
".pdf": [
817+
_Option("PDF compression", "pdf.compression"),
818+
_Option("PDF font type",
819+
getter=lambda:
820+
[str(matplotlib.rcParams["pdf.fonttype"]),
821+
"3", "42"],
822+
setter=lambda val:
823+
matplotlib.rcParams.__setitem__("pdf.fonttype", val))],
824+
".ps": [
825+
_Option("PS paper size",
826+
getter=lambda:
827+
[matplotlib.rcParams["ps.papersize"]]
828+
+ ["auto", "letter", "legal", "ledger"]
829+
+ ["A{}".format(i) for i in range(11)]
830+
+ ["B{}".format(i) for i in range(11)],
831+
setter=lambda val:
832+
matplotlib.rcParams.__setitem__("ps.papersize", val)),
833+
_Option("PS use AFM font", "ps.useafm"),
834+
_Option("PS distiller",
835+
getter=lambda:
836+
[str(matplotlib.rcParams["ps.usedistiller"])]
837+
+ ["None", "ghostscript", "xpdf"],
838+
setter=lambda val:
839+
matplotlib.rcParams.__setitem__("ps.usedistiller", val)),
840+
_Option("PS distiller resolution", "ps.distiller.res"),
841+
_Option("PS font type",
842+
getter=lambda:
843+
[str(matplotlib.rcParams["ps.fonttype"]),
844+
"3", "42"],
845+
setter=lambda val:
846+
matplotlib.rcParams.__setitem__("ps.fonttype", val))]
847+
}
848+
849+
757850
class SubplotToolQt(SubplotTool, UiSubplotTool):
758851
def __init__(self, targetfig, parent):
759852
UiSubplotTool.__init__(self, None)

‎lib/matplotlib/rcsetup.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -504,10 +504,8 @@ def update_savefig_format(value):
504504
def validate_ps_distiller(s):
505505
if isinstance(s, six.string_types):
506506
s = s.lower()
507-
if s in ('none', None):
507+
if s in ('none', None, 'false', False):
508508
return None
509-
elif s in ('false', False):
510-
return False
511509
elif s in ('ghostscript', 'xpdf'):
512510
return s
513511
else:
@@ -1269,7 +1267,7 @@ def validate_animation_writer_path(p):
12691267
'ps.papersize': ['letter', validate_ps_papersize],
12701268
'ps.useafm': [False, validate_bool], # Set PYTHONINSPECT
12711269
# use ghostscript or xpdf to distill ps output
1272-
'ps.usedistiller': [False, validate_ps_distiller],
1270+
'ps.usedistiller': [None, validate_ps_distiller],
12731271
'ps.distiller.res': [6000, validate_int], # dpi
12741272
'ps.fonttype': [3, validate_fonttype], # 3 (Type3) or 42 (Truetype)
12751273
# compression level from 0 to 9; 0 to disable

‎matplotlibrc.template

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -538,12 +538,12 @@ backend : $TEMPLATE_BACKEND
538538
# ps backend params
539539
#ps.papersize : letter # auto, letter, legal, ledger, A0-A10, B0-B10
540540
#ps.useafm : False # use of afm fonts, results in small files
541-
#ps.usedistiller : False # can be: None, ghostscript or xpdf
542-
# Experimental: may produce smaller files.
543-
# xpdf intended for production of publication quality files,
544-
# but requires ghostscript, xpdf and ps2eps
545-
#ps.distiller.res : 6000 # dpi
546-
#ps.fonttype : 3 # Output Type 3 (Type3) or Type 42 (TrueType)
541+
#ps.usedistiller : None # can be: None, ghostscript or xpdf
542+
# Experimental: may produce smaller files.
543+
# xpdf intended for production of publication quality files,
544+
# but requires ghostscript, xpdf and ps2eps
545+
#ps.distiller.res : 6000 # dpi
546+
#ps.fonttype : 3 # Output Type 3 (Type3) or Type 42 (TrueType)
547547

548548
# pdf backend params
549549
#pdf.compression : 6 # integer from 0 to 9

0 commit comments

Comments
 (0)