re-generate legend: fixes

anykraus · anykraus · commit 07930c02dd18 · 2014-04-22T16:44:15.000+02:00
Fixed crash if there was no previous legend.
Fixed ncol inheritance.
diff --git a/lib/matplotlib/backends/qt4_editor/figureoptions.py b/lib/matplotlib/backends/qt4_editor/figureoptions.py
@@ -127,10 +127,13 @@ def apply_callback(data):
         
         # re-generate legend, if checkbox is checked. Stefan Kraus/tacaswell 2014-04-22
         if generate_legend:
-            old_legend = axes.get_legend()
-            new_legend = axes.legend()
-            new_legend._ncol = old_legend._ncol
-            new_legend.draggable(old_legend._draggable is not None)
+            if axes.legend_ is not None:
+                old_legend = axes.get_legend()
+                new_legend = axes.legend(ncol = old_legend._ncol)
+                new_legend.draggable(old_legend._draggable is not None)
+            else:
+                new_legend = axes.legend()
+                new_legend.draggable(True)
 
         # Redraw
         figure = axes.get_figure()
