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

从 matplotlib工具栏源码探析一（禁用工具栏、默认工具栏和工具栏管理器三种模式的差异） 一文可知 matplotlib 内置实现了多个工具项的实现，而默认工具栏中的工具项只是其中的一部分，有没有方法直接管理工具栏，添加、删除内置工具项？

matplotlib内置的工具项

由源码可知， matplotlib.backend_tools.default_tools 变量为字典类型，实例化了基于 matplotlib.backend_tools.ToolBase 类定义的内置工具项。

源码

​
    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 plt
    import matplotlib as mpl
    from pprint import pprint
    
    plt.rcParams['toolbar'] = 'toolmanager'
    fig = plt.gcf()
    pprint(mpl.backend_tools.default_tools)

输出

{'allnav': <class 'matplotlib.backend_tools.ToolEnableAllNavigation'>, 'back': <class 'matplotlib.backend_tools.ToolBack'>, 'copy': 'ToolCopyToClipboard', 'cursor': 'ToolSetCursor', 'forward': <class 'matplotlib.backend_tools.ToolForward'>, 'fullscreen': <class 'matplotlib.backend_tools.ToolFullScreen'>, 'grid': <class 'matplotlib.backend_tools.ToolGrid'>, 'grid_minor': <class 'matplotlib.backend_tools.ToolMinorGrid'>, 'help': 'ToolHelp', 'home': <class 'matplotlib.backend_tools.ToolHome'>, 'nav': <class 'matplotlib.backend_tools.ToolEnableNavigation'>, 'pan': <class 'matplotlib.backend_tools.ToolPan'>, 'position': <class 'matplotlib.backend_tools.ToolCursorPosition'>, 'quit': <class 'matplotlib.backend_tools.ToolQuit'>, 'quit_all': <class 'matplotlib.backend_tools.ToolQuitAll'>, 'rubberband': 'ToolRubberband', 'save': 'ToolSaveFigure', 'subplots': 'ToolConfigureSubplots', 'viewpos': <class 'matplotlib.backend_tools.ToolViewsPositions'>, 'xscale': <class 'matplotlib.backend_tools.ToolXScale'>, 'yscale': <class 'matplotlib.backend_tools.ToolYScale'>, 'zoom': <class 'matplotlib.backend_tools.ToolZoom'>}

使用工具栏管理器管理内置工具项

由源码可知默认工具栏模式 toolbar2 模式没有提供添加、删除工具项的接口。因此，管理工具栏需要使用工具栏管理器模式 toolmanager ，与该模式相关的重要定义有：

• matplotlib.backend_bases.ToolContainerBase(toolmanager) 类：工具栏容器的基类，定义了工具栏编辑的方法。构造函数参数为 toolmanager ，表示工具栏容器容纳的工具栏。

• matplotlib.backend_managers.ToolManager(figure=None) 类：管理用户触发工具栏工具项按钮而产生的动作。 matplotlib.backend_tools.ToolBase 类：所有工具栏工具项的基类，所有工具项均由 matplotlib.backend_managers.ToolManager 实例化。

• matplotlib.backend_tools.default_tools 变量：字典类型，实例化基于 matplotlib.backend_tools.ToolBase 类定义的内置工具项。

• matplotlib.backend_tools.default_toolbar_tools 变量：嵌套列表，以类似格式 [[分组1, [工具1, 工具2 ...]], [分组2, [...]]] 定义工具栏布局。

• matplotlib.backend_tools.add_tools_to_container 函数：设置 toolbarmanager 模式默认工具栏。

使用系统函数实现添加工具项

根据源码可知， matplotlib.backend_tools.add_tools_to_container 函数可以设置 toolbarmanager 模式默认工具栏。

案例

案例说明：为工具栏添加 全屏切换 工具项。

​
    import matplotlib.pyplot as plt
    import matplotlib as mpl
    
    plt.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()

案例解析： add_tools_to_container 函数有两个参数 container 和 tools ，由源码可知 container 参数的值应为 fig.canvas.manager.toolbar ， tools 参数按照 [[分组1, [工具1, 工具2 ...]], [分组2, [...]]] 格式取值。

使用工具栏管理器实现添加、删除内置工具项

根据源码可知：

添加内置工具项有两种方法

• toolbar 对象可以通过 add_tool 方法添加内置工具项，参数为 name 和 tool ， name 为工具项的名称， tool 为添加的工具项对应的类或者字符串。

• toolbar 对象可以通过 add_toolitem 方法添加内置工具项，参数为 name 、 group 、 position 、 image_file 、 description 和 toggle ， name 为工具项的名称， group 为工具项所在组， position 为工具项在组中的位置，取值为列表索引，一般取 -1 即在所在组末尾追加，设置为 0 即在所在组的首位， image_file 为工具项图像，值为字符串， description 为工具项描述， toggle 为是否为切换式工具项，布尔值。

• 删除内置工具项有两种方法

• toolbar 对象可以通过 remove_toolitem 方法删除内置工具项，参数为 name ，即工具项的名称。

• toolmanager 对象可以通过 remove_tool 方法删除内置工具项，参数为 name ，即工具项的名称。

案例

案例说明：删除 向前 工具项，添加 全屏切换 工具项。

​
    import matplotlib.pyplot as plt
    import matplotlib as mpl
    
    plt.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()

总结

通过工具栏管理器添加、删除内置工具项的方法很多种，需要注意调用对象、方法、参数，阅读下面的 matplotlib 源码可能会有所启发。

相关源码

matplotlib.backends.backend_qt5 模块

​
    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

matplotlib.backend_tools 模块

​
    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)

到此这篇关于python matplotlib工具栏源码探析二之添加、删除内置工具项的案例的文章就介绍到这了,更多相关python matplotlib内置工具项内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！