B
    d4                 @   s^   d dl mZmZmZ d dlmZ G dd dZG dd deZG dd dZ	G d	d
 d
Z
dS )    )_apicbookwidgetsNc               @   s   e Zd ZdZdddZdS )	ToolEventz)Event for tool manipulation (add/remove).Nc             C   s   || _ || _|| _|| _d S )N)namesendertooldata)selfr   r   r   r	    r   h/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/matplotlib/backend_managers.py__init__   s    zToolEvent.__init__)N)__name__
__module____qualname____doc__r   r   r   r   r   r      s   r   c                   s"   e Zd ZdZd fdd	Z  ZS )ToolTriggerEventz/Event to inform that a tool has been triggered.Nc                s   t  |||| || _d S )N)superr   canvasevent)r
   r   r   r   r   r	   )	__class__r   r   r      s    zToolTriggerEvent.__init__)NN)r   r   r   r   r   __classcell__r   r   )r   r   r      s   r   c               @   s   e Zd ZdZdd ZdS )ToolManagerMessageEventzs
    Event carrying messages from toolmanager.

    Messages usually get displayed to the user by the toolbar.
    c             C   s   || _ || _|| _d S )N)r   r   message)r
   r   r   r   r   r   r   r      s    z ToolManagerMessageEvent.__init__N)r   r   r   r   r   r   r   r   r   r      s   r   c               @   s   e Zd ZdZd/ddZedd Zedd Zejd	d Zd0ddZ	dd Z
dd Zd1ddZedd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd2d%d&Zd3d'd(Zd)d* Zed+d, Zd4d-d.ZdS )5ToolManagera  
    Manager for actions triggered by user interactions (key press, toolbar
    clicks, ...) on a Figure.

    Attributes
    ----------
    figure : `.Figure`
    keypresslock : `~matplotlib.widgets.LockDraw`
        `.LockDraw` object to know if the `canvas` key_press_event is locked.
    messagelock : `~matplotlib.widgets.LockDraw`
        `.LockDraw` object to know if the message is available to write.
    Nc             C   sJ   d | _ i | _i | _i | _t | _t | _	t | _
d | _| | d S )N)_key_press_handler_id_tools_keys_toggledr   ZCallbackRegistry
_callbacksr   ZLockDrawkeypresslockZmessagelock_figure
set_figure)r
   figurer   r   r   r   /   s    


zToolManager.__init__c             C   s   | j s
dS | j jS )z Canvas managed by FigureManager.N)r    canvas)r
   r   r   r   r#   ?   s    zToolManager.canvasc             C   s   | j S )zFigure that holds the canvas.)r    )r
   r   r   r   r"   F   s    zToolManager.figurec             C   s   |  | d S )N)r!   )r
   r"   r   r   r   r"   K   s    Tc             C   sR   | j r| j| j  || _|r0| jd| j| _ |rNx| j D ]
}||_q@W dS )z
        Bind the given figure to the tools.

        Parameters
        ----------
        figure : `.Figure`
        update_tools : bool, default: True
            Force tools to update figure.
        Zkey_press_eventN)	r   r#   Zmpl_disconnectr    Zmpl_connect
_key_pressr   valuesr"   )r
   r"   Zupdate_toolsr   r   r   r   r!   O   s    
zToolManager.set_figurec             C   s   | j ||S )a  
        Connect event with string *s* to *func*.

        Parameters
        ----------
        s : str
            The name of the event. The following events are recognized:

            - 'tool_message_event'
            - 'tool_removed_event'
            - 'tool_added_event'

            For every tool added a new event is created

            - 'tool_trigger_TOOLNAME', where TOOLNAME is the id of the tool.

        func : callable
            Callback function for the toolmanager event with signature::

                def func(event: ToolEvent) -> Any

        Returns
        -------
        cid
            The callback id for the connection. This can be used in
            `.toolmanager_disconnect`.
        )r   connect)r
   sfuncr   r   r   toolmanager_connectc   s    zToolManager.toolmanager_connectc             C   s   | j |S )z
        Disconnect callback id *cid*.

        Example usage::

            cid = toolmanager.toolmanager_connect('tool_trigger_zoom', onpress)
            #...later
            toolmanager.toolmanager_disconnect(cid)
        )r   Z
disconnect)r
   Zcidr   r   r   toolmanager_disconnect   s    
z"ToolManager.toolmanager_disconnectc             C   s.   |dkr| }d}t |||}| j|| dS )z!Emit a `ToolManagerMessageEvent`.NZtool_message_event)r   r   process)r
   r   r   r'   eventr   r   r   message_event   s
    zToolManager.message_eventc             C   s   | j S )zCurrently toggled tools.)r   )r
   r   r   r   active_toggle   s    zToolManager.active_togglec                s    fdd| j  D }|S )a  
        Return the keymap associated with the specified tool.

        Parameters
        ----------
        name : str
            Name of the Tool.

        Returns
        -------
        list of str
            List of keys associated with the tool.
        c                s   g | ]\}}| kr|qS r   r   ).0ki)r   r   r   
<listcomp>   s    z/ToolManager.get_tool_keymap.<locals>.<listcomp>)r   items)r
   r   keysr   )r   r   get_tool_keymap   s    zToolManager.get_tool_keymapc             C   s    x|  |D ]}| j|= qW d S )N)r5   r   )r
   r   r0   r   r   r   _remove_keys   s    zToolManager._remove_keysc          	   C   sz   || j krt| d| | t|tr2|g}xB|D ]:}|| jkrhtd| d| j|  d|  || j|< q8W dS )z
        Set the keymap to associate with the specified tool.

        Parameters
        ----------
        name : str
            Name of the Tool.
        key : str or list of str
            Keys to associate with the tool.
        z not in ToolszKey z changed from z to N)r   KeyErrorr6   
isinstancestrr   r   warn_external)r
   r   keyr0   r   r   r   update_keymap   s    




zToolManager.update_keymapc             C   s^   |  |}|  t|ddr*| |d | | d}t|| |}| j|| | j|= dS )z
        Remove tool named *name*.

        Parameters
        ----------
        name : str
            Name of the tool.
        toggledFZtoolmanagerZtool_removed_eventN)	get_tooldestroygetattrtrigger_toolr6   r   r   r+   r   )r
   r   r   r'   r,   r   r   r   remove_tool   s    


zToolManager.remove_toolc             O   s
  |  |}|stdt| || jkr<td | j| S |dkrn|tjkrntjdd|j	 d|j
 dd || |f||}|| j|< |jd	k	r| ||j t|tjr|jd	kr| jd	t  n| j|jd	 |jr| |d	d	d	 || j | | |S )
a  
        Add *tool* to `ToolManager`.

        If successful, adds a new event ``tool_trigger_{name}`` where
        ``{name}`` is the *name* of the tool; the event is fired every time the
        tool is triggered.

        Parameters
        ----------
        name : str
            Name of the tool, treated as the ID, has to be unique.
        tool : class_like, i.e. str or type
            Reference to find the class of the Tool to added.

        Notes
        -----
        args and kwargs get passed directly to the tools constructor.

        See Also
        --------
        matplotlib.backend_tools.ToolBase : The base class for tools.
        zImpossible to find class for %sz;A "Tool class" with the same name already exists, not addedcursorz3.5zOverriding ToolSetCursor with z was only necessary to provide the .set_cursor() method, which is deprecated since %(since)s and will be removed %(removal)s. Please report this to the z author.)r   N)_get_cls_to_instantiate
ValueErrorr9   r   r   r:   toolsZSetCursorBaseZwarn_deprecatedr   r   Zdefault_keymapr<   r8   ToolToggleBaseradio_groupr   
setdefaultsetr=   _handle_toggler!   r"   _tool_added_event)r
   r   r   argskwargsZtool_clsZtool_objr   r   r   add_tool   s,    







zToolManager.add_toolc             C   s"   d}t || |}| j|| d S )NZtool_added_event)r   r   r+   )r
   r   r'   r,   r   r   r   rL   "  s    zToolManager._tool_added_eventc             C   s   |j }|dkrH|j| jd kr2| jd |j n| jd |j dS | j| |jkr^d}n2| j| dkrt|j}n| | j| | || |j}|| j|< dS )a  
        Toggle tools, need to untoggle prior to using other Toggle tool.
        Called from trigger_tool.

        Parameters
        ----------
        tool : `.ToolBase`
        sender : object
            Object that wishes to trigger the tool.
        canvasevent : Event
            Original Canvas event or None.
        data : object
            Extra data to pass to the tool when triggering.
        N)rH   r   r   removeaddrA   )r
   r   r   r   r	   rH   r=   r   r   r   rK   '  s     zToolManager._handle_togglec             C   sZ   t |trF|t kr t | }n&d}t|t t |gd}t||d}t|rR|S d S d S )Nbackend_tools   F)r8   r9   globals
__import__localsr@   callable)r
   Zcallback_classmodZcurrent_moduler   r   r   rD   T  s    

z#ToolManager._get_cls_to_instantiatec             C   s\   |  |}|dkrdS |dkr"| }| |||| d| }t|||||}| j|| dS )a  
        Trigger a tool and emit the ``tool_trigger_{name}`` event.

        Parameters
        ----------
        name : str
            Name of the tool.
        sender : object
            Object that wishes to trigger the tool.
        canvasevent : Event
            Original Canvas event or None.
        data : object
            Extra data to pass to the tool when triggering.
        Nztool_trigger_%s)r>   _trigger_toolr   r   r+   )r
   r   r   r   r	   r   r'   r,   r   r   r   rA   e  s    
zToolManager.trigger_toolc             C   s8   |  |}t|tjr&| |||| |||| dS )zActually trigger a tool.N)r>   r8   rF   rG   rK   trigger)r
   r   r   r   r	   r   r   r   r   rY     s    
zToolManager._trigger_toolc             C   sF   |j d ks| j rd S | j|j d }|d kr4d S | j||d d S )N)r   )r;   r   lockedr   getrA   )r
   r,   r   r   r   r   r$     s    zToolManager._key_pressc             C   s   | j S )z,A dict mapping tool name -> controlled tool.)r   )r
   r   r   r   rF     s    zToolManager.toolsc             C   sH   t |tjr|j| jkr|S || jkr>|r:td|  dS | j| S )a  
        Return the tool object with the given name.

        For convenience, this passes tool objects through.

        Parameters
        ----------
        name : str or `.ToolBase`
            Name of the tool, or the tool itself.
        warn : bool, default: True
            Whether a warning should be emitted it no tool with the given name
            exists.

        Returns
        -------
        `.ToolBase` or None
            The tool or None if no tool with the given name exists.
        z"ToolManager does not control tool N)r8   rF   ZToolBaser   r   r   r:   )r
   r   warnr   r   r   r>     s    
zToolManager.get_tool)N)T)N)NNN)NNN)T)r   r   r   r   r   propertyr#   r"   setterr!   r)   r*   r-   r.   r5   r6   r<   rB   rO   rL   rK   rD   rA   rY   r$   rF   r>   r   r   r   r   r   !   s.   


	B-

	r   )Z
matplotlibr   r   r   Zmatplotlib.backend_toolsrR   rF   r   r   r   r   r   r   r   r   <module>   s
   	