.. _user_interfaces-toolmanager:

user_interfaces example code: toolmanager.py
============================================

[`source code <toolmanager.py>`_]

::

    '''This example demonstrates how to:
    * Modify the Toolbar
    * Create tools
    * Add tools
    * Remove tools
    Using `matplotlib.backend_managers.ToolManager`
    '''
    
    
    from __future__ import print_function
    import matplotlib
    matplotlib.use('GTK3Cairo')
    matplotlib.rcParams['toolbar'] = 'toolmanager'
    import matplotlib.pyplot as plt
    from matplotlib.backend_tools import ToolBase, ToolToggleBase
    from gi.repository import Gtk, Gdk
    
    
    class ListTools(ToolBase):
        '''List all the tools controlled by the `ToolManager`'''
        # keyboard shortcut
        default_keymap = 'm'
        description = 'List Tools'
    
        def trigger(self, *args, **kwargs):
            print('_' * 80)
            print("{0:12} {1:45} {2}".format('Name (id)',
                                             'Tool description',
                                             'Keymap'))
            print('-' * 80)
            tools = self.toolmanager.tools
            for name in sorted(tools.keys()):
                if not tools[name].description:
                    continue
                keys = ', '.join(sorted(self.toolmanager.get_tool_keymap(name)))
                print("{0:12} {1:45} {2}".format(name,
                                                 tools[name].description,
                                                 keys))
            print('_' * 80)
            print("Active Toggle tools")
            print("{0:12} {1:45}".format("Group", "Active"))
            print('-' * 80)
            for group, active in self.toolmanager.active_toggle.items():
                print("{0:12} {1:45}".format(group, active))
    
    
    class GroupHideTool(ToolToggleBase):
        '''Hide lines with a given gid'''
        default_keymap = 'G'
        description = 'Hide by gid'
    
        def __init__(self, *args, **kwargs):
            self.gid = kwargs.pop('gid')
            ToolToggleBase.__init__(self, *args, **kwargs)
    
        def enable(self, *args):
            self.set_lines_visibility(False)
    
        def disable(self, *args):
            self.set_lines_visibility(True)
    
        def set_lines_visibility(self, state):
            gr_lines = []
            for ax in self.figure.get_axes():
                for line in ax.get_lines():
                    if line.get_gid() == self.gid:
                        line.set_visible(state)
            self.figure.canvas.draw()
    
    
    fig = plt.figure()
    plt.plot([1, 2, 3], gid='mygroup')
    plt.plot([2, 3, 4], gid='unknown')
    plt.plot([3, 2, 1], gid='mygroup')
    
    # Add the custom tools that we created
    fig.canvas.manager.toolmanager.add_tool('List', ListTools)
    fig.canvas.manager.toolmanager.add_tool('Hide', GroupHideTool, gid='mygroup')
    
    
    # Add an existing tool to new group `foo`.
    # It can be added as many times as we want
    fig.canvas.manager.toolbar.add_tool('zoom', 'foo')
    
    # Remove the forward button
    fig.canvas.manager.toolmanager.remove_tool('forward')
    
    # To add a custom tool to the toolbar at specific location inside
    # the navigation group
    fig.canvas.manager.toolbar.add_tool('Hide', 'navigation', 1)
    
    plt.show()
    

Keywords: python, matplotlib, pylab, example, codex (see :ref:`how-to-search-examples`)