OwlCyberSecurity - MANAGER
Edit File: backend_managers.cpython-38.pyc
U ��]32 � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZ e � e �Z G dd� de �ZG dd� de�ZG dd� de �ZG d d � d e �ZdS ) � N)�validate_stringlistc @ s e Zd ZdZddd�ZdS )� ToolEventz)Event for tool manipulation (add/remove).Nc C s || _ || _|| _|| _d S �N)�name�sender�tool�data)�selfr r r r � r �=/usr/lib/python3/dist-packages/matplotlib/backend_managers.py�__init__ s zToolEvent.__init__)N��__name__� __module__� __qualname__�__doc__r r r r r r s r c @ s e Zd ZdZddd�ZdS )�ToolTriggerEventz/Event to inform that a tool has been triggered.Nc C s t �| ||||� || _d S r )r r � canvasevent)r r r r r r r r r r s zToolTriggerEvent.__init__)NNr r r r r r s r c @ s e Zd ZdZdd� ZdS )�ToolManagerMessageEventzs Event carrying messages from toolmanager. Messages usually get displayed to the user by the toolbar. c C s || _ || _|| _d S r )r r �message)r r r r r r r r ! s z ToolManagerMessageEvent.__init__Nr r r r r r s r c @ s� e Zd ZdZd/dd�Zedd� �Zedd� �Zejd d� �Zd0d d �Z d d� Z dd� Z d1dd�Z edd� �Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd2d%d&�Zd3d'd(�Zd)d*� Zed+d,� �Zd4d-d.�ZdS )5� ToolManagera� Manager for actions triggered by user interactions (key press, toolbar clicks, ...) on a Figure. Attributes ---------- figure : `Figure` keypresslock : `widgets.LockDraw` `LockDraw` object to know if the `canvas` key_press_event is locked messagelock : `widgets.LockDraw` `LockDraw` object to know if the message is available to write Nc C sT t �d� d | _i | _i | _i | _t�� | _t � � | _ t � � | _ d | _ | �|� d S )Nz�Treat the new Tool classes introduced in v1.5 as experimental for now, the API will likely change in version 2.1 and perhaps the rcParam as well)�_logZwarning�_key_press_handler_id�_tools�_keys�_toggled�cbookZCallbackRegistry� _callbacks�widgetsZLockDraw� keypresslockZ messagelock�_figure� set_figure�r �figurer r r r 5 s zToolManager.__init__c C s | j s dS | j jS )z Canvas managed by FigureManager.N)r �canvas�r r r r r$ H s zToolManager.canvasc C s | j S )z Figure that holds the canvas.)r r% r r r r# O s zToolManager.figurec C s | � |� d S r )r! r" r r r r# T s Tc C sN | j r| j�| j � || _|r0| j�d| j�| _ |rJ| j�� D ] }||_q>dS )z� Bind the given figure to the tools. Parameters ---------- figure : `.Figure` update_tools : bool Force tools to update figure Zkey_press_eventN) r r$ Zmpl_disconnectr Z mpl_connect� _key_pressr �valuesr# )r r# Z update_toolsr r r r r! X s �zToolManager.set_figurec C s | j �||�S )a% Connect event with string *s* to *func*. Parameters ---------- s : String 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 : function Function to be called with signature def func(event) )r Zconnect)r �s�funcr r r �toolmanager_connectl 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 Zcidr r r �toolmanager_disconnect� s z"ToolManager.toolmanager_disconnectc C s. |dkr | }d}t |||�}| j�||� dS )z!Emit a `ToolManagerMessageEvent`.NZtool_message_event)r r �process)r r r r( �eventr r r � message_event� s zToolManager.message_eventc C s | j S )zCurrently toggled tools.)r r% r r r � active_toggle� s zToolManager.active_togglec s � fdd�| j �� D �}|S )z� Get the keymap associated with the specified tool. Parameters ---------- name : string Name of the Tool Returns ------- list : list of keys associated with the Tool c s g | ]\}}|� kr|�qS r r )�.0�k�i�r r r � <listcomp>� s z/ToolManager.get_tool_keymap.<locals>.<listcomp>)r �items)r r �keysr r3 r �get_tool_keymap� s zToolManager.get_tool_keymapc C s | � |�D ] }| j|= q d S r )r7 r )r r r1 r r r � _remove_keys� s zToolManager._remove_keysc G sj || j krtd| ��| �|� |D ]@}t|�D ]2}|| jkrXt�d|| j| |f � || j|< q0q$dS )z� Set the keymap to associate with the specified tool. Parameters ---------- name : string Name of the Tool keys : keys to associate with the Tool z%s not in Toolsz Key %s changed from %s to %sN)r �KeyErrorr8 r r r �_warn_external)r r r6 �keyr1 r r r � update_keymap� s �zToolManager.update_keymapc C s^ | � |�}|�� t|dd�r*| �|d� | �|� d}t|| |�}| j�||� | j|= dS )z� Remove tool named *name*. Parameters ---------- name : string Name of the Tool �toggledFZ toolmanagerZtool_removed_eventN) �get_toolZdestroy�getattr� trigger_toolr8 r r r, r )r r r r( r- r r r � remove_tool� s zToolManager.remove_toolc O s� | � |�}|s tdt|� ��|| jkr<t�d� | j| S || |f|�|�}|| j|< |jdk rp| �||j� t|t j �r�|j dkr�| j � dt� � n| j � |j d� |jr�| �|ddd� |�| 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 everytime 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 addedN)�_get_cls_to_instantiate� ValueError�strr r r: Zdefault_keymapr< � isinstance�tools�ToolToggleBase� radio_groupr � setdefault�setr= �_handle_toggler! r# �_tool_added_event)r r r �args�kwargsZtool_clsZtool_objr r r �add_tool� s& zToolManager.add_toolc C s" d}t || |�}| j�||� d S )NZtool_added_event)r r r, )r r r( r- r r r rL s z ToolManager._tool_added_eventc C s� |j }|dkrH|j| jd kr2| jd �|j� n| jd �|j� dS | j| |jkr^d}n2| j| dkrt|j}n | �| j| | ||� |j}|| j|< dS )a� Toggle tools, need to untoggle prior to using other Toggle tool. Called from trigger_tool. Parameters ---------- tool : Tool object 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�addr@ )r r r r r rH r= r r r rK " s" �zToolManager._handle_togglec C sZ t |t�rF|t� kr t� | }n&d}t|t� t� |gd�}t||d�}t|�rR|S d S d S )N� backend_tools� F)rE rD �globals� __import__�localsr? �callable)r Zcallback_class�modZcurrent_moduler r r rB O s � z#ToolManager._get_cls_to_instantiatec C s\ | � |�}|dkrdS |dkr"| }| �||||� d| }t|||||�}| j�||� dS )a� Trigger a tool and emit the ``tool_trigger_{name}`` event. Parameters ---------- name : string 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 Nztool_trigger_%s)r> � _trigger_toolr r r, )r r r r r r r( r- r r r r@ ` s zToolManager.trigger_toolc C s8 | � |�}t|tj�r&| �||||� |�|||� dS )zActually trigger a tool.N)r> rE rF rG rK Ztrigger)r r r r r r r r r rY | s zToolManager._trigger_toolc C sF |j d ks| j�� rd S | j�|j d �}|d kr4d S | j||d� d S )N)r )r; r �lockedr �getr@ )r r- r r r r r& � s zToolManager._key_pressc C s | j S )z,A dict mapping tool name -> controlled tool.)r r% r r r rF � s zToolManager.toolsc C sF t |tj�r |j| jkr |S || jkr<|r8t�d| � dS | j| S )a Return the tool object, also accepts the actual tool for convenience. Parameters ---------- name : str, ToolBase Name of the tool, or the tool itself warn : bool, optional If this method should give warnings. z$ToolManager does not control tool %sN)rE rF ZToolBaser r r r: )r r �warnr r r r> � s �zToolManager.get_tool)N)T)N)NNN)NNN)T)r r r r r �propertyr$ r# �setterr! r* r+ r. r/ r7 r8 r< rA rO rL rK rB r@ rY r& rF r> r r r r r ' s8 8- r )ZloggingZmatplotlib.cbookr Zmatplotlib.widgetsr Zmatplotlib.rcsetupr Zmatplotlib.backend_toolsrR rF Z getLoggerr r �objectr r r r r r r r �<module> s