Merge pull request #14719 from anntzer/qtexclusive

timhoffm · web-flow · commit 2d56d4256530 · 2019-07-17T20:39:20.000+02:00
Make Qt navtoolbar more robust against removal of either pan or zoom.
diff --git a/lib/matplotlib/backends/backend_qt5.py b/lib/matplotlib/backends/backend_qt5.py
@@ -685,6 +685,8 @@ def _icon(self, name, color=None):
 
     def _init_toolbar(self):
         self.basedir = str(cbook._get_data_path('images'))
+        # Ensure that zoom and pan are mutually exclusive.
+        self._button_group = QtWidgets.QButtonGroup()
 
         background_color = self.palette().color(self.backgroundRole())
         foreground_color = self.palette().color(self.foregroundRole())
@@ -700,6 +702,7 @@ def _init_toolbar(self):
                 self._actions[callback] = a
                 if callback in ['zoom', 'pan']:
                     a.setCheckable(True)
+                    self._button_group.addButton(self.widgetForAction(a))
                 if tooltip_text is not None:
                     a.setToolTip(tooltip_text)
                 if text == 'Subplots':
@@ -774,18 +777,13 @@ def edit_parameters(self):
             ax = axes[titles.index(item)]
         figureoptions.figure_edit(ax, self)
 
-    def _update_buttons_checked(self):
-        # sync button checkstates to match active mode
-        self._actions['pan'].setChecked(self._active == 'PAN')
-        self._actions['zoom'].setChecked(self._active == 'ZOOM')
-
     def pan(self, *args):
         super().pan(*args)
-        self._update_buttons_checked()
+        self._actions['pan'].setChecked(self._active == 'PAN')
 
     def zoom(self, *args):
         super().zoom(*args)
-        self._update_buttons_checked()
+        self._actions['zoom'].setChecked(self._active == 'ZOOM')
 
     def set_message(self, s):
         self.message.emit(s)
