Added whats new and validation for figure.savefig dpi change

patchen · patchen · commit 1f25f15eb534 · 2015-04-03T20:20:39.000-04:00
diff --git a/doc/users/whats_new/updated_figure.rst b/doc/users/whats_new/updated_figure.rst
@@ -0,0 +1,9 @@
+Updated Figure.savefig()
+------------------------
+
+Added support to save the figure with the same dpi as the figure on the screen using dpi='figure'
+
+Example:
+	f = plt.figure(dpi=25)				# dpi set to 25
+	S = plt.scatter([1,2,3],[4,5,6])
+	f.savefig('output.png', dpi='figure')	# output savefig dpi set to 25 (same as figure)
diff --git a/lib/matplotlib/rcsetup.py b/lib/matplotlib/rcsetup.py
@@ -117,6 +117,15 @@ def validate_float_or_None(s):
     except ValueError:
         raise ValueError('Could not convert "%s" to float' % s)
 
+def validate_dpi(s):
+    """confirm s is string 'figure' or convert s to float or raise"""
+    if s == 'figure':
+        return s
+    try:
+        return float(s)
+    except ValueError:
+        raise ValueError('"%s" is not string "figure" or'
+            ' could not convert "%s" to float' % (s, s))
 
 def validate_int(s):
     """convert s to int or raise"""
@@ -749,7 +758,7 @@ def __call__(self, s):
                                                      closedmax=False)],
 
     ## Saving figure's properties
-    'savefig.dpi':         [100, validate_float],   # DPI
+    'savefig.dpi':         [100, validate_dpi],   # DPI
     'savefig.facecolor':   ['w', validate_color],  # facecolor; white
     'savefig.edgecolor':   ['w', validate_color],  # edgecolor; white
     'savefig.frameon':     [True, validate_bool],
