OwlCyberSecurity - MANAGER
Edit File: blocking_input.cpython-38.pyc
U ��]�+ � @ sr d Z ddlZddlmZ ddlmZ e�e�Z G dd� de �Z G dd� de �Z G dd � d e �Z G d d � d e �ZdS ) a� This provides several classes used for blocking interaction with figure windows: `BlockingInput` Creates a callable object to retrieve events in a blocking way for interactive sessions. Base class of the other classes listed here. `BlockingKeyMouseInput` Creates a callable object to retrieve key or mouse clicks in a blocking way for interactive sessions. Used by `waitforbuttonpress`. `BlockingMouseInput` Creates a callable object to retrieve mouse clicks in a blocking way for interactive sessions. Used by `ginput`. `BlockingContourLabeler` Creates a callable object to retrieve mouse clicks in a blocking way that will then be used to place labels on a `ContourSet`. Used by `clabel`. � N)�Integralc @ sR e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zddd�Z e Z ddd�Z dS )� BlockingInputz1Callable for retrieving events in a blocking way.� c C s || _ || _d S �N��fig� eventslist)�selfr r r r �;/usr/lib/python3/dist-packages/matplotlib/blocking_input.py�__init__! s zBlockingInput.__init__c C sT | � |� t�dt| j�� | �� t| j�| j kr@dkrPn n | jj� � dS )zY Event handler; will be passed to the current figure to retrieve events. zEvent %ir N) � add_event�_log�info�len�events� post_event�nr �canvas�stop_event_loop�r �eventr r r �on_event% s zBlockingInput.on_eventc C s dS )z-For baseclass, do nothing but collect events.Nr �r r r r r 5 s zBlockingInput.post_eventc C s$ | j D ]}| jj�|� qg | _ dS )zDisconnect all callbacks.N)� callbacksr r Zmpl_disconnect)r �cbr r r �cleanup8 s zBlockingInput.cleanupc C s | j �|� dS )z5For base class, this just appends an event to events.N)r �appendr r r r r > s zBlockingInput.add_event���c C s | j �|� dS )z� Remove an event from the event list -- by default, the last. Note that this does not check that there are events, much like the normal pop method. If no events exist, this will throw an exception. N)r �pop)r �indexr r r � pop_eventB s zBlockingInput.pop_event� � c sr t |t�std��|� _g � _t� jjd�r6� j�� � fdd�� j D �� _ z� jjj |d� W 5 � � � X � jS )z%Blocking call to retrieve *n* events.z Requires an integer argumentZmanagerc s g | ]}� j j�|� j��qS r )r r Z mpl_connectr )�.0�namer r r � <listcomp>X s �z*BlockingInput.__call__.<locals>.<listcomp>)�timeout) � isinstancer � ValueErrorr r �hasattrr r Zshowr r r Zstart_event_loop)r r r& r r r �__call__M s � zBlockingInput.__call__N)r )r )r! r" ) �__name__� __module__� __qualname__�__doc__r r r r r r r r* r r r r r s r c @ s� e Zd ZdZdZdZdZd!dd�Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zd"dd�Zd#dd�Zd$dd �Zd%d d �ZdS )&�BlockingMouseInputa Callable for retrieving mouse clicks in a blocking way. This class will also retrieve keypresses and map them to mouse clicks: delete and backspace are like mouse button 3, enter is like mouse button 2 and all others are like mouse button 1. r! � � c C s& t j| |dd� || _|| _|| _d S �N)Zbutton_press_event�key_press_eventr )r r � button_add� button_pop� button_stop)r r Z mouse_addZ mouse_popZ mouse_stopr r r r q s �zBlockingMouseInput.__init__c C s@ t | j�dkrt�d� n"| jd jdkr4| �� n| �� dS )zProcess an event.r z No events yetr r3 N)r r r �warningr$ � key_event� mouse_eventr r r r r y s z BlockingMouseInput.post_eventc C sT | j d }|j}|| jkr&| �|� n*|| jkr<| �|� n|| jkrP| �|� dS )z Process a mouse click event.r N)r �buttonr5 �mouse_event_popr6 �mouse_event_stopr4 �mouse_event_add)r r r: r r r r9 � s z BlockingMouseInput.mouse_eventc C sX | j d }|jdkrdS |j�� }|dkr6| �|� n |dkrJ| �|� n | �|� dS )zV Process a key press event, mapping keys to appropriate mouse clicks. r N)Z backspace�delete)�escapeZenter)r �key�lowerr; r< r= )r r r@ r r r r8 � s z BlockingMouseInput.key_eventc C s |j r| �|� n t�| � dS )z� Process an button-1 event (add a click if inside axes). Parameters ---------- event : `~.backend_bases.MouseEvent` N)�inaxes� add_clickr r r r r r r= � s z"BlockingMouseInput.mouse_event_addc C s t �| � | jj�� dS )z� Process an button-2 event (end blocking input). Parameters ---------- event : `~.backend_bases.MouseEvent` N)r r r r r r r r r r< � s z#BlockingMouseInput.mouse_event_stopc C s t �| � | jr| �|� dS )z� Process an button-3 event (remove the last click). Parameters ---------- event : `~.backend_bases.MouseEvent` N)r r r r r r r r; � s z"BlockingMouseInput.mouse_event_popc C sv | j �|j|jf� t�dt| j �|j|j� | jrrtj |jg|jgddd�}|j � |� | j �|� | j j�� dS )z� Add the coordinates of an event to the list of clicks. Parameters ---------- event : `~.backend_bases.MouseEvent` zinput %i: %f, %f�+�r)ZmarkerZcolorN)�clicksr ZxdataZydatar r r � show_clicks�mlinesZLine2DrB Zadd_line�marksr r �draw)r r �liner r r rC � s � � z BlockingMouseInput.add_clickr c C s2 | j �|� | jr.| j�|��� | jj�� dS )z� Remove a click (by default, the last) from the list of clicks. Parameters ---------- event : `~.backend_bases.MouseEvent` N)rF r rG rI �remover r rJ �r r r r r r � pop_click� s z BlockingMouseInput.pop_clickc C s | � ||� t�| |� dS )zx Removes a click and the associated event from the list of clicks. Defaults to the last click. N)rN r r rM r r r r � s zBlockingMouseInput.popNc C s: | j r,| jD ] }|�� q g | _| jj�� t�| � dS )z{ Parameters ---------- event : `~.backend_bases.MouseEvent`, optional Not used N)rG rI rL r r rJ r r )r r Zmarkr r r r � s zBlockingMouseInput.cleanupr" Tc C s( || _ g | _g | _tj| ||d� | jS )zV Blocking call to retrieve *n* coordinate pairs through mouse clicks. �r r& )rG rF rI r r* )r r r& rG r r r r* s zBlockingMouseInput.__call__)r! r0 r1 )r )r )N)r! r" T)r+ r, r- r. r4 r5 r6 r r r9 r8 r= r< r; rC rN r r r* r r r r r/ d s r/ c @ sD e Zd ZdZdd� Zdd� Zddd�Zd d � Zd d � Zddd�Z dS )�BlockingContourLabelerzt Callable for retrieving mouse clicks and key presses in a blocking way. Used to place contour labels. c C s || _ tj| |jjd� d S )N)r )�csr/ r �ax�figure)r rQ r r r r s z BlockingContourLabeler.__init__c C s | � |� d S r )�button1r r r r rC s z BlockingContourLabeler.add_clickr c C s | � |� d S r )�button3rM r r r rN s z BlockingContourLabeler.pop_clickc C sH |j | jjkr:| jj|j|j| j| jdd� | jj � � n t � | � dS )z� Process an button-1 event (add a label to a contour). Parameters ---------- event : `~.backend_bases.MouseEvent` F)�inline_spacingZ transformN) rB rQ rR Zadd_label_near�x�y�inlinerV r r rJ r r r r r r rT s �z BlockingContourLabeler.button1c C s& | j rn| j�� | jjjj�� dS )a� Process an button-3 event (remove a label if not in inline mode). Unfortunately, if one is doing inline labels, then there is currently no way to fix the broken contour - once humpty-dumpty is broken, he can't be put back together. In inline mode, this does nothing. Parameters ---------- event : `~.backend_bases.MouseEvent` N)rY rQ Z pop_labelrR rS r rJ r r r r rU - s z BlockingContourLabeler.button3� c C s" || _ || _tj| ||dd� d S )NF)r r& rG )rY rV r/ r* )r rY rV r r& r r r r* ? s �z BlockingContourLabeler.__call__N)r )rZ r r ) r+ r, r- r. r rC rN rT rU r* r r r r rP s rP c @ s* e Zd ZdZdd� Zdd� Zd dd�Zd S ) �BlockingKeyMouseInputzQ Callable for retrieving mouse clicks and key presses in a blocking way. c C s t j| |dd� d S r2 )r r )r r r r r r K s z BlockingKeyMouseInput.__init__c C s( | j r| j d jdk| _n t�d� dS )z Determine if it is a key event.r r3 zNo events yet.N)r r$ � keyormouser r7 r r r r r O s z BlockingKeyMouseInput.post_eventr" c C s d| _ tj| d|d� | j S )z� Blocking call to retrieve a single mouse click or key press. Returns ``True`` if key press, ``False`` if mouse click, or ``None`` if timed out. Nr! rO )r\ r r* )r r& r r r r* V s z BlockingKeyMouseInput.__call__N)r" )r+ r, r- r. r r r* r r r r r[ F s r[ )r. ZloggingZnumbersr Zmatplotlib.lines�linesrH Z getLoggerr+ r �objectr r/ rP r[ r r r r �<module> s F (;