自学教程：python matplotlib工具栏源码探析二之添加、删除内置工具项的案例

default_tools = {'home': ToolHome, 'back': ToolBack, 'forward': ToolForward,     'zoom': ToolZoom, 'pan': ToolPan,     'subplots': 'ToolConfigureSubplots',     'save': 'ToolSaveFigure',     'grid': ToolGrid,     'grid_minor': ToolMinorGrid,     'fullscreen': ToolFullScreen,     'quit': ToolQuit,     'quit_all': ToolQuitAll,     'allnav': _ToolEnableAllNavigation,     'nav': _ToolEnableNavigation,     'xscale': ToolXScale,     'yscale': ToolYScale,     'position': ToolCursorPosition,     _views_positions: ToolViewsPositions,     'cursor': 'ToolSetCursor',     'rubberband': 'ToolRubberband',     'help': 'ToolHelp',     'copy': 'ToolCopyToClipboard',     }

import matplotlib.pyplot as pltimport matplotlib as mplfrom pprint import pprintplt.rcParams['toolbar'] = 'toolmanager'fig = plt.gcf()pprint(mpl.backend_tools.default_tools)

import matplotlib.pyplot as pltimport matplotlib as mplplt.rcParams['toolbar'] = 'toolmanager'fig = plt.gcf()# 通过mpl.backend_tools.add_tools_to_container函数添加工具项mpl.backend_tools.add_tools_to_container(fig.canvas.manager.toolbar, tools=[['foo', [ 'fullscreen']]])plt.show()

import matplotlib.pyplot as pltimport matplotlib as mplplt.rcParams['toolbar'] = 'toolmanager'fig = plt.gcf()fig.canvas.manager.toolmanager.remove_tool('forward')fig.canvas.manager.toolbar.remove_toolitem('back')fig.canvas.manager.toolbar.add_tool('quit', 'foo')fig.canvas.manager.toolbar.add_toolitem('fullscreen', 'foo', -1,'fullscreen','fullscreen',False) plt.show()

class FigureManagerQT(FigureManagerBase): self.toolbar = self._get_toolbar(self.canvas, self.window) if self.toolmanager:  backend_tools.add_tools_to_manager(self.toolmanager)  if self.toolbar:   backend_tools.add_tools_to_container(self.toolbar) if self.toolbar:  self.window.addToolBar(self.toolbar)  tbs_height = self.toolbar.sizeHint().height() else:  tbs_height = 0

 def _get_toolbar(self, canvas, parent):  # must be inited after the window, drawingArea and figure  # attrs are set  if matplotlib.rcParams['toolbar'] == 'toolbar2':   toolbar = NavigationToolbar2QT(canvas, parent, True)  elif matplotlib.rcParams['toolbar'] == 'toolmanager':   toolbar = ToolbarQt(self.toolmanager, self.window)  else:   toolbar = None  return toolbar

 class ToolbarQt(ToolContainerBase, QtWidgets.QToolBar): def __init__(self, toolmanager, parent):  ToolContainerBase.__init__(self, toolmanager)  QtWidgets.QToolBar.__init__(self, parent)  self.setAllowedAreas(   QtCore.Qt.TopToolBarArea | QtCore.Qt.BottomToolBarArea)  message_label = QtWidgets.QLabel("")  message_label.setAlignment(   QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)  message_label.setSizePolicy(   QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Expanding,         QtWidgets.QSizePolicy.Ignored))  self._message_action = self.addWidget(message_label)  self._toolitems = {}  self._groups = {} def add_toolitem(   self, name, group, position, image_file, description, toggle):  button = QtWidgets.QToolButton(self)  if image_file:   button.setIcon(NavigationToolbar2QT._icon(self, image_file))  button.setText(name)  if description:   button.setToolTip(description)  def handler():   self.trigger_tool(name)  if toggle:   button.setCheckable(True)   button.toggled.connect(handler)  else:   button.clicked.connect(handler)  self._toolitems.setdefault(name, [])  self._add_to_group(group, name, button, position)  self._toolitems[name].append((button, handler)) def _add_to_group(self, group, name, button, position):  gr = self._groups.get(group, [])  if not gr:   sep = self.insertSeparator(self._message_action)   gr.append(sep)  before = gr[position]  widget = self.insertWidget(before, button)  gr.insert(position, widget)  self._groups[group] = gr def toggle_toolitem(self, name, toggled):  if name not in self._toolitems:   return  for button, handler in self._toolitems[name]:   button.toggled.disconnect(handler)   button.setChecked(toggled)   button.toggled.connect(handler) def remove_toolitem(self, name):  for button, handler in self._toolitems[name]:   button.setParent(None)  del self._toolitems[name] def set_message(self, s):  self.widgetForAction(self._message_action).setText(s

def add_tools_to_container(container, tools=default_toolbar_tools): """ Add multiple tools to the container. Parameters ---------- container : Container  `backend_bases.ToolContainerBase` object that will get the tools added. tools : list, optional  List in the form ``[[group1, [tool1, tool2 ...]], [group2, [...]]]``  where the tools ``[tool1, tool2, ...]`` will display in group1.  See `add_tool` for details. """ for group, grouptools in tools:  for position, tool in enumerate(grouptools):   container.add_tool(tool, group, position)

def add_tools_to_manager(toolmanager, tools=default_tools): """ Add multiple tools to a `.ToolManager`. Parameters ---------- toolmanager : `.backend_managers.ToolManager`  Manager to which the tools are added. tools : {str: class_like}, optional  The tools to add in a {name: tool} dict, see `add_tool` for more  info. """ for name, tool in tools.items():  toolmanager.add_tool(name, tool)