Merge pull request #4909 from jrevans/issue06

tacaswell · tacaswell · commit b9bc7d16dc5c · 2015-09-12T21:00:38.000-04:00
FIX: Qt figure options preserve units
diff --git a/lib/matplotlib/backends/qt_editor/figureoptions.py b/lib/matplotlib/backends/qt_editor/figureoptions.py
@@ -63,6 +63,12 @@ def figure_edit(axes, parent=None):
                ('(Re-)Generate automatic legend', False),
                ]
 
+    # Save the unit data
+    xconverter = axes.xaxis.converter
+    yconverter = axes.yaxis.converter
+    xunits = axes.xaxis.get_units()
+    yunits = axes.yaxis.get_units()
+
     if has_curve:
         # Get / Curves
         linedict = {}
@@ -122,6 +128,14 @@ def apply_callback(data):
         axes.set_ylim(ymin, ymax)
         axes.set_ylabel(ylabel)
 
+        # Restore the unit data
+        axes.xaxis.converter = xconverter
+        axes.yaxis.converter = yconverter
+        axes.xaxis.set_units(xunits)
+        axes.yaxis.set_units(yunits)
+        axes.xaxis._update_axisinfo()
+        axes.yaxis._update_axisinfo()
+
         if has_curve:
             # Set / Curves
             for index, curve in enumerate(curves):
