The figure option dialog resets the axes and all unit information. This fixes

it so that unit information is not lost.

Author: James Evans

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 = {}
@@ -101,7 +107,7 @@ def figure_edit(axes, parent=None):
         has_curve = bool(curves)
 
     datalist = [(general, "Axes", "")]
-    if has_curve:
+    if has_curve and curves:
         datalist.append((curves, "Curves", ""))
 
     def apply_callback(data):
@@ -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):