Merge pull request #4909 from jrevans/issue06

FIX: Qt figure options preserve units
matplotlib · Sep 13, 2015 · b9bc7d1 · b9bc7d1
2 parents f8ceb85 + e5940b7
commit b9bc7d1
Showing 1 changed file with 14 additions and 0 deletions.
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):