OwlCyberSecurity - MANAGER
Edit File: widgets.cpython-38.pyc
U ��]�t � @ s� d Z ddlZddlmZ ddlZddlmZmZ ddl m Z ddl m Z m Z mZ ddlmZ G d d � d e�ZG d d � d e�ZG d d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d e�Z G d d � d e�Z G d!d"� d"e �Z G d#d$� d$e�Z G d%d&� d&e �Z G d'd(� d(e �Z!G d)d*� d*e �Z"G d+d,� d,e �Z#G d-d.� d.e�Z$dS )/a� GUI neutral widgets =================== Widgets that are designed to work for any of the GUI backends. All of these widgets require you to predefine a :class:`matplotlib.axes.Axes` instance and pass that as the first arg. matplotlib doesn't try to be too smart with respect to layout -- you will have to figure out how wide and tall you want your Axes to be to accommodate your widget. � N)�Integral� )�cbook�rcParams)�Line2D)�Circle� Rectangle�Ellipse)�blended_transform_factoryc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�LockDrawa� Some widgets, like the cursor, draw onto the canvas, and this is not desirable under all circumstances, like when the toolbar is in zoom-to-rect mode and drawing a rectangle. To avoid this, a widget can acquire a canvas' lock with ``canvas.widgetlock(widget)`` before drawing on the canvas; this will prevent other widgets from doing so at the same time (if they also try to acquire the lock first). c C s d | _ d S �N��_owner��self� r �4/usr/lib/python3/dist-packages/matplotlib/widgets.py�__init__! s zLockDraw.__init__c C s | � |�std��|| _dS )zReserve the lock for *o*.zalready lockedN�� available� ValueErrorr �r �or r r �__call__$ s zLockDraw.__call__c C s | � |�std��d| _dS )zRelease the lock from *o*.zyou do not own this lockNr r r r r �release* s zLockDraw.releasec C s | � � p| �|�S )z+Return whether drawing is available to *o*.)�locked�isownerr r r r r 0 s zLockDraw.availablec C s | j |kS )z"Return whether *o* owns this lock.r r r r r r 4 s zLockDraw.isownerc C s | j dk S )z6Return whether the lock is currently held by an owner.Nr r r r r r 8 s zLockDraw.lockedN) �__name__� __module__� __qualname__�__doc__r r r r r r r r r r r s r c @ sF e Zd ZdZdZdZdZdd� Zdd� Ze edd� d d �Z d d � Z d S )�Widgetz5 Abstract base class for GUI neutral widgets Tc C s || _ dS )z*Set whether the widget is active. N��_active�r �activer r r � set_activeE s zWidget.set_activec C s | j S )z*Get whether the widget is active. r" r r r r � get_activeJ s zWidget.get_activec C s | � |�S r )r& r$ r r r �<lambda>P � zWidget.<lambda>zIs the widget active?)�docc C s | j S )z�Return True if event should be ignored. This method (or a version of it) should be called at the beginning of any event callback. )r% �r �eventr r r �ignoreS s z Widget.ignoreN) r r r r �drawon�eventsonr# r& r' �propertyr% r- r r r r r! = s �r! c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � AxesWidgeta Widget that is connected to a single :class:`~matplotlib.axes.Axes`. To guarantee that the widget remains responsive and not garbage-collected, a reference to the object should be maintained by the user. This is necessary because the callback registry maintains only weak-refs to the functions, which are member functions of the widget. If there are no references to the widget object it may be garbage collected which will disconnect the callbacks. Attributes: *ax* : :class:`~matplotlib.axes.Axes` The parent axes for the widget *canvas* : :class:`~matplotlib.backend_bases.FigureCanvasBase` subclass The parent figure canvas for the widget. *active* : bool If False, the widget does not respond to events. c C s || _ |jj| _g | _d S r )�ax�figure�canvas�cids)r r2 r r r r r s zAxesWidget.__init__c C s | j �||�}| j�|� dS )z�Connect callback with an event. This should be used in lieu of `figure.canvas.mpl_connect` since this function stores callback ids for later clean up. N)r4 � mpl_connectr5 �append)r r, �callback�cidr r r � connect_eventw s zAxesWidget.connect_eventc C s | j D ]}| j�|� qdS )z-Disconnect all events created by this widget.N)r5 r4 �mpl_disconnect)r �cr r r �disconnect_events� s z AxesWidget.disconnect_eventsN)r r r r r r: r= r r r r r1 \ s r1 c @ sB e Zd ZdZddd�Zdd� Zd d � Zd d � Zd d� Zdd� Z dS )�Buttona� A GUI neutral button. For the button to remain responsive you must keep a reference to it. Call :meth:`on_clicked` to connect to the button. Attributes ---------- ax The :class:`matplotlib.axes.Axes` the button renders into. label A :class:`matplotlib.text.Text` instance. color The color of the button when not hovering. hovercolor The color of the button when hovering. N�0.85�0.95c C s� t �| |� |dk r |�|� |jdd|dd|jd�| _d| _i | _| �d| j � | �d| j � | �d| j � |� d � |� |� |�g � |�g � || _|| _|| _dS ) af Parameters ---------- ax : `~matplotlib.axes.Axes` The `~.axes.Axes` instance the button will be placed into. label : str The button text. Accepts string. image : array, mpl image, Pillow Image The image to place in the button, if not *None*. Can be any legal arg to imshow (numpy array, matplotlib Image instance, or Pillow Image). color : color The color of the button when not activated. hovercolor : color The color of the button when the mouse is over it. N� �?�center��verticalalignment�horizontalalignment� transformr �button_press_event�button_release_event�motion_notify_eventF)r1 r Zimshow�text� transAxes�label�cnt� observersr: �_click�_release�_motion� set_navigate� set_facecolor� set_xticks� set_yticks�color� hovercolor� _lastcolor)r r2 rL ZimagerV rW r r r r � s( � zButton.__init__c C sH | � |�rd S |j| jkr d S | js(d S |jj| jkrD|j�| j� d S r )r- �inaxesr2 r/ r4 � mouse_grabber� grab_mouser+ r r r rO � s z Button._clickc C sh | � |�rd S |jj| jkr d S |j�| j� | js8d S |j| jkrHd S | j�� D ]\}}||� qRd S r ) r- r4 rZ r2 � release_mouser/ rY rN �items)r r, r9 �funcr r r rP � s zButton._releasec C s\ | � |�rd S |j| jkr"| j}n| j}|| jkrX| j�|� || _| jrX| jjj � � d S r � r- rY r2 rW rV rX rS r. r3 r4 �draw�r r, r< r r r rQ � s zButton._motionc C s" | j }|| j|< | j d7 _ |S �z� Connect the callback function *func* to button click events. Returns a connection id, which can be used to disconnect the callback. r �rM rN �r r^ r9 r r r � on_clicked� s zButton.on_clickedc C s& z | j |= W n tk r Y nX dS )z6Remove the callback function with connection id *cid*.N�rN �KeyError�r r9 r r r � disconnect� s zButton.disconnect)Nr? r@ ) r r r r r rO rP rQ re ri r r r r r> � s � . r> c @ sJ e Zd ZdZddd�Zd d � Zd d � Zd d� Zdd� Zdd� Z dd� Z dS )�SlideraB A slider representing a floating point range. Create a slider from *valmin* to *valmax* in axes *ax*. For the slider to remain responsive you must maintain a reference to it. Call :meth:`on_changed` to connect to the slider event. Attributes ---------- val : float Slider value. rA �%1.2fTN� horizontalc K s6 t �| |� | dk r0t| d�s0td�t| ����| dk rTt| d�sTtd�t| ����| dkrjtd�| ���| | _|| _|| _| | _ | | _ d| _ || _ || _ | | _| �|�}|dkr�|}|| _|| _| dkr�|j||d d f|�| _|j|d d d d d �| _n,|j||d d f|�| _|j|d d d d d �| _|| _|�g � | dk�rN|�||f� n|� ||f� |� g � |� d� | � d | j � | � d| j � | �r�| � d| j � | dk�r�|j!dd||j"ddd�| _#|j!dd|| |j"ddd�| _$n8|j!dd||j"ddd�| _#|j!dd|| |j"ddd�| _$d | _%i | _&| �'|� dS )a� Parameters ---------- ax : Axes The Axes to put the slider in. label : str Slider label. valmin : float The minimum value of the slider. valmax : float The maximum value of the slider. valinit : float, optional, default: 0.5 The slider initial position. valfmt : str, optional, default: "%1.2f" Used to format the slider value, fprint format string. closedmin : bool, optional, default: True Indicate whether the slider interval is closed on the bottom. closedmax : bool, optional, default: True Indicate whether the slider interval is closed on the top. slidermin : Slider, optional, default: None Do not allow the current slider to have a value less than the value of the Slider `slidermin`. slidermax : Slider, optional, default: None Do not allow the current slider to have a value greater than the value of the Slider `slidermax`. dragging : bool, optional, default: True If True the slider can be dragged by the mouse. valstep : float, optional, default: None If given, the slider will snap to multiples of `valstep`. orientation : str, 'horizontal' or 'vertical', default: 'horizontal' The orientation of the slider. Notes ----- Additional kwargs are passed on to ``self.poly`` which is the :class:`~matplotlib.patches.Rectangle` that draws the slider knob. See the :class:`~matplotlib.patches.Rectangle` documentation for valid property names (e.g., `facecolor`, `edgecolor`, `alpha`). N�valz$Argument slidermin ({}) has no 'val'z$Argument slidermax ({}) has no 'val')rl �verticalzBArgument orientation ({}) must be either'horizontal' or 'vertical'Frn r r �r)rV �lwrG rH rI rA gR� ��Q�?�bottomrB )rF rD rE g{�G�z���top�right�left)(r1 r �hasattrr �format�type� orientation� closedmin� closedmax� slidermin� slidermax� drag_active�valmin�valmax�valstep�_value_in_boundsrm �valinitZaxhspan�poly�axhlineZhlineZaxvspan�axvlineZvline�valfmtrU �set_ylim�set_xlimrT rR r: �_updaterJ rK rL �valtextrM rN �set_val)r r2 rL r~ r r� r� ry rz r{ r| Zdraggingr� rx �kwargsr r r r s� 7 ��� �� ��zSlider.__init__c C s� | j r,t�|| j | j �| j }|| j7 }|| jkrH| js@dS | j}n|| jkrb| js\dS | j}| jdk r�|| jjkr�| js�dS | jj}| j dk r�|| j jkr�| js�dS | j j}|S )z&Makes sure *val* is with given bounds.N) r� �np�roundr~ ry r rz r{ rm r| �r rm r r r r� � s( zSlider._value_in_boundsc C s� | � |�s|jdkrdS |jdkrB|j| jkrBd| _|j�| j� | jsLdS |jdksl|jdkr�|j| jkr�d| _|j�| j� dS | j dkr�| � |j �}n | � |j �}|d| j fkr�| �|� dS )zUpdate the slider position.r NrG TrH Frn )r- �button�namerY r2 r} r4 r[ r\ rx r� �ydata�xdatarm r� )r r, rm r r r r� � s( � � zSlider._updatec C s� | j j}| jdkr,d|f|d< d|f|d< n|df|d< |df|d< || j _| j�| j| � | jrr| jjj � � || _ | j s�dS | j �� D ]\}}||� q�dS )zf Set slider value to *val* Parameters ---------- val : float rn r r � � N)r� �xyrx r� Zset_textr� r. r2 r3 r4 � draw_idlerm r/ rN r] )r rm r� r9 r^ r r r r� � s zSlider.set_valc C s" | j }|| j|< | j d7 _ |S )a� When the slider value is changed call *func* with the new slider value Parameters ---------- func : callable Function to call when slider is changed. The function must accept a single float as its arguments. Returns ------- cid : int Connection id (which can be used to disconnect *func*) r rc rd r r r � on_changed� s zSlider.on_changedc C s& z | j |= W n tk r Y nX dS )z� Remove the observer with connection id *cid* Parameters ---------- cid : int Connection id of the observer to be removed Nrf rh r r r ri � s zSlider.disconnectc C s | j | jkr| �| j� dS )z%Reset the slider to the initial valueN)rm r� r� r r r r �reset� s z Slider.reset) rA rk TTNNTNrl ) r r r r r r� r� r� r� ri r� r r r r rj � s � }rj c @ sB e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� CheckButtonsa� A GUI neutral set of check buttons. For the check buttons to remain responsive you must keep a reference to this object. The following attributes are exposed *ax* The :class:`matplotlib.axes.Axes` instance the buttons are located in *labels* List of :class:`matplotlib.text.Text` instances *lines* List of (line1, line2) tuples for the x's in the check boxes. These lines exist for each box, but have ``set_visible(False)`` when its box is not checked. *rectangles* List of :class:`matplotlib.patches.Rectangle` instances Connect to the CheckButtons with the :meth:`on_clicked` method Nc C s� t �| |� |�g � |�g � |�d� |dkr@dgt|� }t|�dkrtdt|�d }t�d| |t|��}n d}dg}|�� }g | _ g | _ g | _ dd|j d d �}t |||�D ]�\}} } |jd|| |j d d d �} |d |d } } d|| d }}t||f| | d||j d�}t||| g|| |gf|�}t||| g||| gf|�}|�| � |�| � | j �| � | j �|� | j �||f� |�|� |�|� |�|� q�| �d| j� d| _i | _dS )a� Add check buttons to :class:`matplotlib.axes.Axes` instance *ax* Parameters ---------- ax : `~matplotlib.axes.Axes` The parent axes for the widget. labels : List[str] The labels of the check buttons. actives : List[bool], optional The initial check states of the buttons. The list must have the same length as *labels*. If not given, all buttons are unchecked. FNr � �?� �?rA �kg �?Zbutt)rV � linewidthrF Zsolid_capstylert rB �rF rE rD r� 皙�����?�black)r� �width�height� edgecolor� facecolorrF rG r )r1 r rT rU rR �lenr� �linspace� get_facecolor�labels�lines� rectanglesrK �ziprJ r r � set_visibler7 � add_patch�add_liner: �_clickedrM rN )r r2 r� Zactives�dy�ys�axcolorZ lineparams�yrL r% �t�w�h�x�p�l1�l2r r r r sV �� � zCheckButtons.__init__c C s~ | � |�s |jdks |j| jkr$d S tt| j| j��D ]B\}\}}|�� � |j |j �sj|�� � |j |j �r6| � |� qzq6d S �Nr ) r- r� rY r2 � enumerater� r� r� �get_window_extent�containsr� r� r& )r r, �ir� r� r r r r� [ s � zCheckButtons._clickedc C s� d| krt | j�kr*n n td| ��| j| \}}|�|�� � |�|�� � | jrl| jjj � � | j svdS | j � � D ]\}}|| j| �� � q�dS )a% Directly (de)activate a check button by index. *index* is an index into the original label list that this object was constructed with. Raises ValueError if *index* is invalid. Callbacks will be triggered if :attr:`eventson` is True. r z Invalid CheckButton index: %dN)r� r� r r� r� � get_visibler. r2 r3 r4 r` r/ rN r] �get_text)r �indexr� r� r9 r^ r r r r& d s zCheckButtons.set_activec C s dd� | j D �S )zX Return a tuple of the status (True/False) of all of the check buttons. c S s g | ]\}}|� � �qS r )r� )�.0r� r� r r r � <listcomp>� s z+CheckButtons.get_status.<locals>.<listcomp>)r� r r r r � get_status~ s zCheckButtons.get_statusc C s" | j }|| j|< | j d7 _ |S rb rc rd r r r re � s zCheckButtons.on_clickedc C s& z | j |= W n tk r Y nX dS )z,remove the observer with connection id *cid*Nrf rh r r r ri � s zCheckButtons.disconnect)N) r r r r r r� r& r� re ri r r r r r� � s D r� c @ s� e Zd ZdZd)dd�Zdd � Zd d � Zd d � Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(S )*�TextBoxao A GUI neutral text input box. For the text box to remain responsive you must keep a reference to it. The following attributes are accessible: *ax* The :class:`matplotlib.axes.Axes` the button renders into. *label* A :class:`matplotlib.text.Text` instance. *color* The color of the text box when not hovering. *hovercolor* The color of the text box when hovering. Call :meth:`on_text_change` to be updated whenever the text changes. Call :meth:`on_submit` to be updated whenever the user hits enter or leaves the text entry field. � �.95�1�{�G�z�?c C s. t �| |� d| _dd� tD �| _|| _|j| d|dd|jd�| _| �| j�| _ d| _ i | _ i | _ | j �dd � | j �dd � d| _| j �ddd�| _| j�d � | �d | j� | �d | j� | �d | j� | �d| j� | �d| j� |�d � |�|� |� g � |� g � || _ || _ || _ d | _!dS )a! Parameters ---------- ax : `~matplotlib.axes.Axes` The `~.axes.Axes` instance the button will be placed into. label : str Label for this text box. Accepts string. initial : str Initial value in the text box. color : color The color of the box. hovercolor : color The color of the box when the mouse is over it. label_pad : float The distance between the label and the right side of the textbox. r� c S s g | ]}d |kr|�qS )Zkeymapr )r� �keyr r r r� � s z$TextBox.__init__.<locals>.<listcomp>rA rB rs rC r r FrG rH rI �key_press_eventZ resize_eventN)"r1 r �DIST_FROM_LEFTr �params_to_disablerJ rK rL �_make_text_disp� text_disprM �change_observers�submit_observersr2 r� r� � cursor_index�vlines�cursorr� r: rO rP rQ � _keypress�_resizerR rS rT rU rV rW rX �capturekeystrokes)r r2 rL �initialrV rW Z label_padr r r r � s> � zTextBox.__init__c C s | j j| jd|dd| j jd�S )NrA rB rt rC )r2 rJ r� rK )r �stringr r r r� � s �zTextBox._make_text_dispc C s� | j d | j� }d}|dks,|dks,|dkr8|dk}d}| �|�}| jjj�� |�� }| jj� � }|� |�}|� d� |r�|d |d< | j � d� | j� |d |d |d �| _ | jjj�� d S ) NFr� � z �,�r r �r r �r r )r r )rJ r� r� r2 r3 r4 r` r� � transData�invertedrF r� r� r� )r Z widthtextZno_textZwt_disp�bb�invr r r � _rendercursor� s zTextBox._rendercursorc C s( | j r$| j�� D ]\}}|| j� qd S r )r/ r� r] rJ �r r9 r^ r r r �_notify_submit_observers s z TextBox._notify_submit_observersc C s2 | � |�rd S |jj| jkr d S |j�| j� d S r )r- r4 rZ r2 r\ r+ r r r rP s zTextBox._releasec C s� | � |�rd S | j�r�|j}t|�dkr^| jd | j� | | j| jd � | _| jd7 _n�|dkr�| jt| j�kr�| jd7 _n�|dkr�| jdkr�| jd8 _n�|dkr�d| _n�|dkr�t| j�| _n�|dk�r| jdk�r\| jd | jd � | j| jd � | _| jd8 _nB|dk�r\| jt| j�k�r\| jd | j� | j| jd d � | _| j�� | �| j�| _| � � | � � |d k�r�| � � d S ) Nr rs rt r �home�endZ backspace�deleteZenter) r- r� r� r� rJ r� r� �remover� r� �_notify_change_observersr� �r r, r� r r r r� " sH � � � zTextBox._keypressc C sP t |�}| j|krd S || _| j�� | �| j�| _| �� | �� | �� d S r )�strrJ r� r� r� r� r� r� )r rm Znewvalr r r r� G s zTextBox.set_valc C s( | j r$| j�� D ]\}}|| j� qd S r )r/ r� r] rJ r� r r r r� R s z TextBox._notify_change_observersc C s2 d| _ i | _| jD ]}t| | j|< g t|< qd S �NT)r� � reset_paramsr� r )r r� r� r r r � begin_typingW s zTextBox.begin_typingc C sX d}| j r(| jD ]}| j| t|< qd}d| _ | j�d� | jjj� � |rT| � � d S )NFT) r� r� r� r r� r� r2 r3 r4 r` r� )r Z notifysubmitr� r r r � stop_typing` s zTextBox.stop_typingc C s� t | j�dkrd| _nz| j�� }| jj}| jj�� }|�|�|��}|d }|d }|| || }|dk rpd}|dkr|d}t t | j�| �| _| � � d S )Nr r� r� r ) r� rJ r� r� r� r2 r� r� rF �intr� )r r� r� �transr� Z text_startZtext_endZratior r r �position_cursorq s zTextBox.position_cursorc C sn | � |�rd S |j| jkr&| �� d S | js0d S |jj| jkrL|j�| j� | js^| � |j � | � |j � d S r ) r- rY r2 r� r/ r4 rZ r[ r� r� r� r� r+ r r r rO � s zTextBox._clickc C s | � � d S r )r� r+ r r r r� � s zTextBox._resizec C s\ | � |�rd S |j| jkr"| j}n| j}|| jkrX| j�|� || _| jrX| jjj � � d S r r_ ra r r r rQ � s zTextBox._motionc C s" | j }|| j|< | j d7 _ |S )z� When the text changes, call this *func* with event. A connection id is returned which can be used to disconnect. r )rM r� rd r r r �on_text_change� s zTextBox.on_text_changec C s" | j }|| j|< | j d7 _ |S )z� When the user hits enter or leaves the submission box, call this *func* with event. A connection id is returned which can be used to disconnect. r )rM r� rd r r r � on_submit� s zTextBox.on_submitc C s6 | j | jfD ]$}z ||= W q tk r. Y q X q dS �z-Remove the observer with connection id *cid*.N)r� r� rg )r r9 Zregr r r ri � s zTextBox.disconnectN)r� r� r� r� )r r r r r r� r� r� rP r� r� r� r� r� r� rO r� rQ r� r� ri r r r r r� � s, � D% r� c @ s: e Zd ZdZddd�Zdd� Zdd � Zd d � Zd d � ZdS )� RadioButtonsaI A GUI neutral radio button. For the buttons to remain responsive you must keep a reference to this object. Connect to the RadioButtons with the :meth:`on_clicked` method. Attributes ---------- ax The containing `~.axes.Axes` instance. activecolor The color of the selected button. labels A list of `~.text.Text` instances containing the button labels. circles A list of `~.patches.Circle` instances defining the buttons. value_selected : str The label text of the currently selected button. r �bluec C s0 t �| |� || _d| _|�g � |�g � |�d� dt|�d }t� d| |t|��}d}|� � }|d d } | dkr�d} g | _ g | _ t ||�D ]z\} } |jd | | |jd d d �} ||kr�| | _|} n|} td | f| d| |jd�}| j �| � | j �|� |�|� |d7 }q�| �d| j� d| _i | _dS )a� Add radio buttons to an `~.axes.Axes`. Parameters ---------- ax : `~matplotlib.axes.Axes` The axes to add the buttons to. labels : list of str The button labels. active : int The index of the initially selected button. activecolor : color The color of the selected button. NFr� r r r� r� r� r� rt rB r� �333333�?r� )r� �radiusr� r� rF rG )r1 r � activecolor�value_selectedrT rU rR r� r� r� r� r� �circlesr� rJ rK r r7 r� r: r� rM rN )r r2 r� r% r� r� r� rM r� Z circle_radiusr� rL r� r� r� r r r r � sF � � zRadioButtons.__init__c C s� | � |�s |jdks |j| jkr$d S | jj�� �|j|jf�}t � |d |d g�}i }t t | j | j��D ]N\}\}}|�� �|j|j�s�t j�||j �|jk rjt j�||j �||< qjt|�dkr�t||jd�}| �|� d S )Nr r )r� )r- r� rY r2 rK r� �transform_pointr� r� r� �arrayr� r� r� r� r� r� ZlinalgZnormrB r� r� �min�getr& ) r r, r� ZpclickedZ distancesr� r� r� �closestr r r r� s � zRadioButtons._clickedc C s� d| krt | j�kr*n n td| ��| j| �� | _t| j�D ],\}}||kr\| j}n | j� � }|� |� qD| j r�| jj j �� | js�dS | j�� D ]\}}|| j| �� � q�dS )zv Select button with number *index*. Callbacks will be triggered if :attr:`eventson` is True. r z Invalid RadioButton index: %dN)r� r� r r� r� r� r� r� r2 r� rS r. r3 r4 r` r/ rN r] )r r� r� r� rV r9 r^ r r r r& + s zRadioButtons.set_activec C s" | j }|| j|< | j d7 _ |S rb rc rd r r r re E s zRadioButtons.on_clickedc C s& z | j |= W n tk r Y nX dS r� rf rh r r r ri P s zRadioButtons.disconnectN)r r� ) r r r r r r� r& re ri r r r r r� � s < r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )� SubplotToolzU A tool to adjust the subplot params of a :class:`matplotlib.figure.Figure`. c sF |� _ �jddd� ��d�� _� j�d� � j�d� t� jddd |jjdd �� _ � j � � j � ��d �� _ � j �d� t� j d dd |jj dd �� _� j� � j� ��d �� _� j�d� t� jddd |jjdd�� _� j� � j� ��d�� _� j�d� t� jddd |jjdd�� _� j� � j� ��d�� _� j�d� t� jddd |jjdd �� _� j� � j� ��d�� _ � j �d� t� j ddd |jj dd �� _ � j � � j � � j� j _ � j � j_!� j� j_ � j� j_!��"ddddg�}t#|d�� _$� j � j� j� j� j� j f�� ��fdd �}�jj%}d�j_%� j$�&|� |�j_%d S ) a *targetfig* The figure instance to adjust. *toolfig* The figure instance to embed the subplot tool into. If *None*, a default figure will be created. If you are using this from the GUI 皙�����?g�������?)rt rs i� z'Click on slider to adjust subplot paramFrt r r )rz i� rq i� rs )ry i� rr i� �wspacei� �hspaceg�������?r� r� g333333�?ZResetc s~ � j }d� _ g }�D ]}|�|j � d|_ q�D ] }|�� q0t�|�D ]\}}||_ qH|� _ � j rz�j�� � jj�� d S �NF)r. r7 r� r� r4 r` � targetfig)r, Z thisdrawonZbsZslider�b�r Zsliders�toolfigr r r^ � s z"SubplotTool.__init__.<locals>.funcN)'r �subplots_adjustZ add_subplotZaxleftZ set_titlerR rj Z subplotparsrt Z sliderleftr� �funcleftZaxbottomrq Z sliderbottom� funcbottomZaxrightrs Z sliderright� funcrightZaxtoprr Z slidertop�functopZaxwspacer Z sliderwspace� funcwspaceZaxhspacer Z sliderhspace� funchspacer| r{ Zadd_axesr> Z buttonreset�validatere )r r r Zbaxr^ r r r r r \ s� � � � � � � � zSubplotTool.__init__c C s$ | j j|d� | jr | j j�� d S )N)rt �r r r. r4 r` r� r r r r � s zSubplotTool.funcleftc C s$ | j j|d� | jr | j j�� d S )N)rs r r� r r r r � s zSubplotTool.funcrightc C s$ | j j|d� | jr | j j�� d S )N)rq r r� r r r r � s zSubplotTool.funcbottomc C s$ | j j|d� | jr | j j�� d S )N)rr r r� r r r r � s zSubplotTool.functopc C s$ | j j|d� | jr | j j�� d S )N)r r r� r r r r � s zSubplotTool.funcwspacec C s$ | j j|d� | jr | j j�� d S )N)r r r� r r r r � s zSubplotTool.funchspaceN) r r r r r r r r r r r r r r r r X s ir c @ s2 e Zd ZdZd dd�Zdd� Zdd � Zd d � Zd S )�Cursora A crosshair cursor that spans the axes and moves with mouse cursor. For the cursor to remain responsive you must keep a reference to it. Parameters ---------- ax : `matplotlib.axes.Axes` The `~.axes.Axes` to attach the cursor to. horizOn : bool, optional, default: True Whether to draw the horizontal line. vertOn : bool, optional, default: True Whether to draw the vertical line. useblit : bool, optional, default: False Use blitting for faster drawing if supported by the backend. Other Parameters ---------------- **lineprops `.Line2D` properties that control the appearance of the lines. See also `~.Axes.axhline`. Examples -------- See :doc:`/gallery/widgets/cursor`. TFc K s� t �| |� | �d| j� | �d| j� d| _|| _|| _|oD| jj | _ | j rVd|d< |j |� � d fddi|��| _ |j|�� d fddi|��| _d | _d| _d S )NrI � draw_eventT�animatedr �visibleF)r1 r r: �onmove�clearr �horizOn�vertOnr4 � supports_blit�useblitr� � get_ybound�linehr� � get_xbound�linev� background� needclear)r r2 r r r � linepropsr r r r s zCursor.__init__c C sB | � |�rdS | jr&| j�| jj�| _| j�d� | j �d� dS )z+Internal event handler to clear the cursor.NF) r- r r4 �copy_from_bboxr2 �bboxr% r$ r� r" r+ r r r r s z Cursor.clearc C s� | � |�rdS | jj�| �s dS |j| jkr^| j�d� | j�d� | j rZ| j� � d| _ dS d| _ | j sndS | j� |j |j f� | j�|j|jf� | j�| j o�| j� | j�| j o�| j� | �� dS )z?Internal event handler to draw the cursor when the mouse moves.NFT)r- r4 � widgetlockr rY r2 r$ r� r" r& r` r � set_xdatar� � set_ydatar� r r r� r+ r r r r s& z Cursor.onmovec C sZ | j rL| jd k r | j�| j� | j�| j� | j�| j� | j�| jj � n | j� � dS r ) r r% r4 �restore_regionr2 � draw_artistr$ r" �blitr) r� r r r r r� 6 s zCursor._updateN)TTF)r r r r r r r r� r r r r r � s r c @ sB e Zd ZdZddd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dS )� MultiCursora� Provide a vertical (default) and/or horizontal line cursor shared between multiple axes. For the cursor to remain responsive you must keep a reference to it. Example usage:: from matplotlib.widgets import MultiCursor import matplotlib.pyplot as plt import numpy as np fig, (ax1, ax2) = plt.subplots(nrows=2, sharex=True) t = np.arange(0.0, 2.0, 0.01) ax1.plot(t, np.sin(2*np.pi*t)) ax2.plot(t, np.sin(4*np.pi*t)) multi = MultiCursor(fig.canvas, (ax1, ax2), color='r', lw=1, horizOn=False, vertOn=True) plt.show() TFc s� || _ || _|| _|| _|d �� \}}|d �� \} } d|| �d| | �d| _|o`| j j| _d | _ d| _ | jr~d� d< |r�� �fdd�|D �| _ ng | _ |r�� �fdd�|D �| _ ng | _ | � � d S ) N���rA TFr c s" g | ]}|j �fd di� ���qS �r F)r� �r� r2 )r' �xmidr r r� p s �z(MultiCursor.__init__.<locals>.<listcomp>c s" g | ]}|j �fd di� ���qS r2 )r� r3 )r' �ymidr r r� v s �)r4 �axesr r �get_xlim�get_ylimr r r r% r& r� �hlines�connect) r r4 r6 r r r r' �xmin�xmax�ymin�ymaxr )r' r4 r5 r r Z s2 � � zMultiCursor.__init__c C s( | j �d| j�| _| j �d| j�| _dS )zconnect eventsrI r N)r4 r6 r � _cidmotionr �_ciddrawr r r r r: } s �zMultiCursor.connectc C s | j �| j� | j �| j� dS )zdisconnect eventsN)r4 r; r? r@ r r r r ri � s zMultiCursor.disconnectc C sH | � |�rdS | jr(| j�| jjj�| _| j| j D ]}|� d� q4dS )zclear the cursorNF) r- r r4 r( r3 r) r% r� r9 r� �r r, �liner r r r � s �zMultiCursor.clearc C s� | � |�rd S |jd kr d S | jj�| �s.d S d| _| js>d S | jrn| jD ]"}|� |j |j f� |� | j� qJ| j r�| j D ]"}|�|j|jf� |� | j� qz| �� d S r� )r- rY r4 r* r r& r r r� r+ r� r� r r9 r, r� r� rA r r r r � s$ zMultiCursor.onmovec C s� | j r�| jd k r | j�| j� | jrFt| j| j�D ]\}}|�|� q2| j rnt| j| j �D ]\}}|�|� qZ| j� | jj j � n | j�� d S r )r r% r4 r- r r� r6 r� r. r r9 r/ r3 r) r� )r r2 rB r r r r� � s zMultiCursor._updateN)TFT) r r r r r r: ri r r r� r r r r r0 B s # r0 c @ s� e Zd Zd-dd�Zdd� Zdd� Zd d � Zd d � Zd d� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� ZdS ).�_SelectorWidgetFNc C s� t �| |� d| _|| _|o"| jj| _| �� tddddd�| _ | j � |pLi � d | _ g | _ t |t�rp|g| _n|| _d | _d | _d | _t� | _d S )NTr� �escape�shift�control)�mover �squarerB )r1 r r �onselectr4 r r �connect_default_events�dict�state_modifier_keys�updater% �artists� isinstancer � validButtons� eventpress� eventrelease� _prev_event�set�state)r r2 rI r r� rL r r r r � s& � z_SelectorWidget.__init__c C s t �| |� |r| �d � d S r )r1 r& �update_backgroundr$ r r r r& � s z_SelectorWidget.set_activec C s | j r| j�| jj�| _dS )z!force an update of the backgroundN)r r4 r( r2 r) r% r+ r r r rV � s z!_SelectorWidget.update_backgroundc C sf | � d| j� | � d| j� | � d| j� | � d| j� | � d| j� | � d| j� | � d| j� dS ) z+Connect the major canvas events to methods.rI rG rH r r� Zkey_release_eventZ scroll_eventN)r: r �pressr rV � on_key_press�on_key_release� on_scrollr r r r rJ � s z&_SelectorWidget.connect_default_eventsc C s� | j r| j�� sdS | jj�| �s&dS t|d�s6d|_| jdk rP|j| jkrPdS | j dkrf|j | jkS |j| j jkrxdS |j | jkp�|j| j jkS )�*return *True* if *event* should be ignoredTr� NF) r% r2 r� r4 r* r ru r� rP rQ rY r+ r r r r- � s �z_SelectorWidget.ignorec C sd | j �� sdS | jrV| jdk r,| j�| j� | jD ]}| j �|� q2| j�| j j � n | j� � dS )zTdraw using newfangled blit or oldfangled draw depending on useblit FN) r2 r� r r% r4 r- rN r. r/ r) r� )r �artistr r r rM s z_SelectorWidget.updatec C s^ |j dkrdS | j�� \}}| j�� \}}t||j �}t||�}t||j�}t||�}||fS )z.Get the xdata and ydata for event, with limitsN)NN)r� r2 r# r! �maxr r� )r r, �x0�x1�y0�y1r� r� r r r � _get_data s z_SelectorWidget._get_datac C s8 |j dkr| j}n t�|�}| �|�\|_ |_|| _|S )z�Clean up an event Use prev event if there is no xdata Limit the xdata and ydata to the axes limits Set the prev event N)r� rS �copyrb r� r+ r r r � _clean_event* s z _SelectorWidget._clean_eventc C sb | � |�s^| �|�}|| _|| _|jp(d}|�dd�}|| jd krP| j�d� | � |� dS dS )z"Button press handler and validatorr� �ctrlrF rG TF) r- rd rQ rS r� �replacerL rU �add�_pressr� r r r rW : s z_SelectorWidget.pressc C s dS )zButton press handlerNr r+ r r r rh I s z_SelectorWidget._pressc C sJ | � |�sF| jrF| �|�}|| _| �|� d| _d| _| j�d� dS dS )z*Button release event handler and validatorNrG TF)r- rQ rd rR rP rU �discardr+ r r r r M s z_SelectorWidget.releasec C s dS )� Button release event handlerNr r+ r r r rP Y s z_SelectorWidget._releasec C s, | � |�s(| jr(| �|�}| �|� dS dS �z'Cursor move event handler and validatorTF)r- rQ rd �_onmover+ r r r r ] s z_SelectorWidget.onmovec C s dS )�Cursor move event handlerNr r+ r r r rl e s z_SelectorWidget._onmovec C s | � |�s| �|� dS )z(Mouse scroll event handler and validatorN)r- � _on_scrollr+ r r r rZ i s z_SelectorWidget.on_scrollc C s dS )zMouse scroll event handlerNr r+ r r r rn n s z_SelectorWidget._on_scrollc C s� | j r~|jpd}|�dd�}|| jd krL| jD ]}|�d� q0| �� dS | j�� D ] \}}||krV| j� |� qV| � |� dS )z?Key press event handler and validator for all selection widgetsr� re rF r FN) r% r� rf rL rN r� rM r] rU rg � _on_key_press)r r, r� r\ rU �modifierr r r rX r s z _SelectorWidget.on_key_pressc C s dS )zMKey press event handler - use for widget-specific key press actions. Nr r+ r r r ro � s z _SelectorWidget._on_key_pressc C sF | j rB|jpd}| j�� D ] \}}||kr| j�|� q| �|� dS )z(Key release event handler and validator.r� N)r% r� rL r] rU ri �_on_key_release)r r, r� rU rp r r r rY � s z _SelectorWidget.on_key_releasec C s dS )zKey release event handler.Nr r+ r r r rq � s z _SelectorWidget._on_key_releasec C s || _ | jD ]}|�|� q dS )z"Set the visibility of our artists.N)r rN r� )r r r\ r r r r� � s z_SelectorWidget.set_visible)FNN)r r r r r& rV rJ r- rM rb rd rW rh r rP r rl rZ rn rX ro rY rq r� r r r r rC � s. � ! rC c @ s` e Zd ZdZddd�Zdd� Zdd � Zd d � Zd d � Ze � d�e dd� ��Z dd� Z dd� ZdS )� SpanSelectora( Visually select a min/max range on a single axis and call a function with those values. To guarantee that the selector remains responsive, keep a reference to it. In order to turn off the SpanSelector, set `span_selector.active=False`. To turn it back on, set `span_selector.active=True`. Parameters ---------- ax : :class:`matplotlib.axes.Axes` object onselect : func(min, max), min/max are floats direction : "horizontal" or "vertical" The axis along which to draw the span selector minspan : float, default is None If selection is less than *minspan*, do not call *onselect* useblit : bool, default is False If True, use the backend-dependent blitting features for faster canvas updates. rectprops : dict, default is None Dictionary of :class:`matplotlib.patches.Patch` properties onmove_callback : func(min, max), min/max are floats, default is None Called on mouse move while the span is being selected span_stays : bool, default is False If True, the span stays visible after the mouse is released button : int or list of ints Determines which mouse buttons activate the span selector 1 = left mouse button 2 = center mouse button (scroll wheel) 3 = right mouse button Examples -------- >>> import matplotlib.pyplot as plt >>> import matplotlib.widgets as mwidgets >>> fig, ax = plt.subplots() >>> ax.plot([1, 2, 3], [10, 50, 100]) >>> def onselect(vmin, vmax): ... print(vmin, vmax) >>> rectprops = dict(facecolor='blue', alpha=0.5) >>> span = mwidgets.SpanSelector(ax, onselect, 'horizontal', ... rectprops=rectprops) >>> fig.show() See also: :doc:`/gallery/widgets/span_selector` NFc C s� t j| |||| d� |d kr(tddd�}| j|d< tjddg|d� || _d | _d | _|| _ || _ || _ || _ d | _ d | _| �|� d S ) N�r r� �redrA )r� �alphar rl rn )� directionr� )rC r rK r r �_check_in_listrv �rect�pressv� rectprops�onmove_callback�minspan� span_stays�prevr4 �new_axes) r r2 rI rv r| r rz r{ r} r� r r r r � s" � zSpanSelector.__init__c C s� || _ | j|jjk r8| jdk r&| �� |jj| _| �� | jdkr^t| j j| j j�}d\}}nt| j j| j j�}d\}}t d||f|dd�| j ��| _ | j r�t d||f|dd�| j ��| _ | j �d� | j �| j � | j �| j � | j g| _dS )z)Set SpanSelector to operate on a new AxesNrl r� r� r� F)rF r )r2 r4 r3 r= rJ rv r r� rK r rz rx r} � stay_rect� set_animatedr� rN )r r2 r� r� r� r r r r � s@ � � �� �� zSpanSelector.new_axesc C s t �| |�p| j S )r[ )rC r- r r+ r r r r- s zSpanSelector.ignorec C sd | j �| j� | jr0| j�d� | jr0| j�� | �|�\}}| j dkrP|| _ n|| _ | � |� dS )�on button press eventFrl ) rx r� r r} r� r r4 r` rb rv ry � _set_span_xy)r r, r� r� r r r rh s zSpanSelector._pressc C s | j dkrdS | j�d� | jrt| j�| j�� � | j�| j�� � | j� | j� � � | j� | j� � � | j�d� | j �� | j }| �|�\}}| jdkr�|p�| jd }n|p�| jd }||kr�|| }}|| }| jdk r�|| jk r�dS | �||� d| _ dS )�on button release eventNFTrl r r )ry rx r� r} r� �set_x�get_x�set_y�get_y� set_width� get_width� set_height� get_heightr4 r� rb rv r~ r| rI )r r, �vminr� r� �vmax�spanr r r rP ( s. zSpanSelector._releasez3.1c C s dS r r r r r r � buttonDownG s zSpanSelector.buttonDownc C s� | j dkrdS | �|� | jdk r|| j }| �|�\}}| jdkrP|pL| jd }n|p\| jd }||krp|| }}| �||� | �� dS )zon motion notify eventNrl r r F)ry r� r{ rb rv r~ rM )r r, r� r� r� r� r r r rl L s zSpanSelector._onmovec C s� | � |�\}}|dkrdS ||f| _| jdkr4|}n|}|| j }}||krV|| }}| jdkr~| j�|� | j�|| � n | j�|� | j�|| � dS )z Setting the span coordinatesNrl ) rb r~ rv ry rx r� r� r� r� )r r, r� r� �vZminvZmaxvr r r r� b s zSpanSelector._set_span_xy)NFNNFN)r r r r r r r- rh rP r Z deprecatedr0 r� rl r� r r r r rr � s 9 � ! rr c @ sT e Zd ZdZddd�Zedd� �Zed d � �Zdd d �Zd d� Z dd� Z dd� Z dS )� ToolHandlesa� Control handles for canvas tools. Parameters ---------- ax : :class:`matplotlib.axes.Axes` Matplotlib axes where tool handles are displayed. x, y : 1D arrays Coordinates of control handles. marker : str Shape of marker used to display handle. See `matplotlib.pyplot.plot`. marker_props : dict Additional marker properties. See :class:`matplotlib.lines.Line2D`. r NTc C sd || _ t|ddddddd�}|�|d k r,|ni � t||fd|i|��| _| j �| j� | j| _d S ) N� r� �nonerA FZ _nolegend_)�markerZ markersize�mfcZlsru r rL r )r2 rK rM r �_markersr� r\ )r r2 r� r� r� � marker_propsr �propsr r r r � s �zToolHandles.__init__c C s | j �� S r )r� Z get_xdatar r r r r� � s z ToolHandles.xc C s | j �� S r )r� Z get_ydatar r r r r� � s z ToolHandles.yc C s* |dk r|}t �||g�}| j�|� dS )z Set x and y positions of handlesN)r� r r� �set_data)r �ptsr� r� r r r r� � s zToolHandles.set_datac C s | j �|� d S r )r� r� r� r r r r� � s zToolHandles.set_visiblec C s | j �|� d S r )r� r� r� r r r r� � s zToolHandles.set_animatedc C s~ t �| j| jf�}| jj�|�}|||f }|jdkrbt �t j |d dd��}t � |�t � |�fS dt �t � |d ��fS dS )z1Return index and pixel distance to closest index.r� r )Zaxisr N) r� Z transposer� r� r2 r� rF �ndimZsqrt�sumZargminr )r r� r� r� ZdiffZdistr r r r � s zToolHandles.closest)r NT)N) r r r r r r0 r� r� r� r� r� r r r r r r� y s r� c @ s� e Zd ZdZeZd dd�Zd d � Zd d � Zd d� Z e dd� �Z e dd� �Z e dd� �Z e dd� �Ze dd� �Zejdd� �Zdd� Zd d � Ze d d � �ZdS )!�RectangleSelectora Select a rectangular region of an axes. For the cursor to remain responsive you must keep a reference to it. Example usage:: import numpy as np import matplotlib.pyplot as plt from matplotlib.widgets import RectangleSelector def onselect(eclick, erelease): "eclick and erelease are matplotlib events at press and release." print('startposition: (%f, %f)' % (eclick.xdata, eclick.ydata)) print('endposition : (%f, %f)' % (erelease.xdata, erelease.ydata)) print('used button : ', eclick.button) def toggle_selector(event): print('Key pressed.') if event.key in ['Q', 'q'] and toggle_selector.RS.active: print('RectangleSelector deactivated.') toggle_selector.RS.set_active(False) if event.key in ['A', 'a'] and not toggle_selector.RS.active: print('RectangleSelector activated.') toggle_selector.RS.set_active(True) x = np.arange(100.) / 99 y = np.sin(x) fig, ax = plt.subplots() ax.plot(x, y) toggle_selector.RS = RectangleSelector(ax, onselect, drawtype='line') fig.canvas.mpl_connect('key_press_event', toggle_selector) plt.show() �boxNF�data� c C s t j| |||| |d� d| _d| _| | _|dkr:d}d| _|dkr�|dkrZtdd d dd �}| j|d < || _| jd&ddi| j��| _| j � | j� |dkr�|dkr�td dddd�}| j|d < || _ t ddgddgfddi| j ��| _| j � | j� || _|| _tjddg| d� | | _|| _| | _|dk�r6tdd�}nt|�dd�d�}dd d d g| _| j\}}t| j |||| jd �| _d d!d"d#g| _| j\}}t| j ||d$|| jd%�| _ | j \}}t| j |g|gd$|| jd%�| _ d| _ | j| j j | jj | j j g| _!| j�s| jg| _!d| _"dS )'a� Create a selector in *ax*. When a selection is made, clear the span and call onselect with:: onselect(pos_1, pos_2) and clear the drawn box/line. The ``pos_1`` and ``pos_2`` are arrays of length 2 containing the x- and y-coordinate. If *minspanx* is not *None* then events smaller than *minspanx* in x direction are ignored (it's the same for y). The rectangle is drawn with *rectprops*; default:: rectprops = dict(facecolor='red', edgecolor = 'black', alpha=0.2, fill=True) The line is drawn with *lineprops*; default:: lineprops = dict(color='black', linestyle='-', linewidth = 2, alpha=0.5) Use *drawtype* if you want the mouse to draw a line, a box or nothing between click and actual position by setting ``drawtype = 'line'``, ``drawtype='box'`` or ``drawtype = 'none'``. Drawing a line would result in a line from vertex A to vertex C in a rectangle ABCD. *spancoords* is one of 'data' or 'pixels'. If 'data', *minspanx* and *minspanx* will be interpreted in the same coordinates as the x and y axis. If 'pixels', they are in pixels. *button* is a list of integers indicating which mouse buttons should be used for rectangle selection. You can also specify a single integer if only a single button is desired. Default is *None*, which does not limit which button can be used. Note, typically: 1 = left mouse button 2 = center mouse button (scroll wheel) 3 = right mouse button *interactive* will draw a set of handles and allow you interact with the widget after it is drawn. *state_modifier_keys* are keyboard modifiers that affect the behavior of the widget. The defaults are: dict(move=' ', clear='escape', square='shift', center='ctrl') Keyboard modifiers, which: 'move': Move the existing shape. 'clear': Clear the current shape. 'square': Makes the shape square. 'center': Make the initial point the center of the shape. 'square' and 'center' can be combined. )r r� rL NTr� rB Fr� rt r� r )r� r� ru Zfillr r� r r r �-r� rA �rV � linestyler� ru r� �pixels�� spancoordsro )�mecr� �NW�NEZSE�SW)r� r �W�N�E�S�s)r� r� r )r� r r )#rC r �to_drawr � interactiverK r rz � _shape_klassr2 r� r' r r� �minspanx�minspanyr rw r� �drawtype�maxdistr � _corner_order�cornersr� �_corner_handles� _edge_order� edge_centers� _edge_handlesrB �_center_handle� active_handler\ rN �_extents_on_press)r r2 rI r� r� r� r r' rz r� r� r� r� r� rL r� �xc�yc�xe�yer r r r � s� A � � � � � � � � � zRectangleSelector.__init__c C sj | j r | j�� r | �|� nd| _| jdks2| j s:| �� | j sZ|j}|j}||||f| _| � | j � dS )r� N) r� r� r� �_set_active_handler� rM r� r� �extentsr� r )r r, r� r� r r r rh d s zRectangleSelector._pressc C s� | j s| j�d� | j\}}}}|| j_|| j_| jj� ||g�}|\| j_ | j_ || j _|| j _| jj� ||g�}|\| j _ | j _ | j dkr�| jj| jj }} | j j| j j } } nD| j dkr�| jj | jj }} | j j | j j } } ntjddg| j d� || k�r | | }} | | k�r | | } } | | } | | } | jdk �oD| | jk }| jdk �oZ| | jk }| jdk�r�|�st|�r�| jD ]}|�d� �qz| �� dS | �| j| j � | �� dS )r� Fr� r� r� Nr� )r� r� r� r� rQ r� r� r2 r� r r� r� rR r� r rw r� r� r� rN rM rI )r r, r_ �x2ra �y2Zxy1Zxy2r; r= r< r> ZspanxZspanyZ xproblemsZ yproblemsr\ r r r rP x sH � zRectangleSelector._releasec C s | j rV| j dksV| j\}}}}| j ddg| j kr8|j}| j ddg| j krR|j}�n�d| jksj| j dkr�| jdk r�| j\}}}}|j| jj }|j| jj }||7 }||7 }||7 }||7 }�nF| jj| jjg}| jj| jjg} |j|d d }|j|d d }d | jk�r�t |j| d �} t |j| d �} | �s@dS t t | �t | ��} t | �| k �rt|| t | �d 9 }t | �| k �r�|| t | �d 9 }d | jk�r�|d9 }|d9 }n |d |7 < |d |7 < |d | |d | |d | |d | f\}}}}||||f| _ dS )z,on motion notify event if box/line is wanted�Cr� r� r� r� rG Nr � @r rH g���ư>rB r� ) r� r� r� r� r� rU rQ r� r� �absr] r� ) r r, r_ r� ra r� �dxr� rB Z center_pixZdx_pixZdy_pixZmaxdr r r rl � sP � � zRectangleSelector._onmovec C s� | j dkr>| j�� }| j�� }| j�� }| j�� }||||fS | j�� \}}t|�t|� }}t|�t|� }}|||| || fS d S )Nr� ) r� r� r� r� r� r� �get_datar r] ) r r^ r` r� r� r� r� r_ ra r r r � _rect_bbox� s z RectangleSelector._rect_bboxc C s> | j \}}}}||| || |f}|||| || f}||fS )z7Corners of rectangle from lower left, moving clockwise.�r� )r r^ r` r� r� r� r� r r r r� � s zRectangleSelector.cornersc C sV | j \}}}}|d }|d }||| || || f}|| ||| || f}||fS )z8Midpoint of rectangle edges from left, moving clockwise.r� r� ) r r^ r` r� r� r� r� r� r� r r r r� � s z RectangleSelector.edge_centersc C s& | j \}}}}||d ||d fS )zCenter of rectangler� r� )r r^ r` r� r� r r r rB s zRectangleSelector.centerc C sB | j \}}}}t||| g�\}}t||| g�\}}||||fS )z Return (xmin, xmax, ymin, ymax).)r� �sorted) r r^ r` r� r� r; r< r= r> r r r r� s zRectangleSelector.extentsc C sL | � |� | jj| j� | jj| j� | jj| j� | �| j � | � � d S r ) � draw_shaper� r� r� r� r� r� rB r� r rM )r r� r r r r� s c C s� |\}}}}t ||g�\}}t ||g�\}} t | j�� �} t | j�� �} t| d |�}t| d |�}t|| d �}t| | d �} | jdkr�| j�|� | j� |� | j� || � | j� | | � n | jdkr�| j� ||g|| g� d S )Nr r r� rB ) r� r2 r7 r8 r] r r� r� r� r� r� r� r� ) r r� r^ r_ r` ra r; r< r= r> ZxlimZylimr r r r� s z RectangleSelector.draw_shapec C s� | j �|j|j�\}}| j�|j|j�\}}| j�|j|j�\}}d| jkr\d| _| j| _ nV|| j d k rrd| _n@|| j kr�|| j kr�d| _dS ||k r�| j | | _n | j | | _| j\}} } } | jdkr�| |j }} | jdkr�| |j } } || | | f| _ dS )z:Set active handle based on the location of the mouse eventrG r� r� N)r� r� r� )r� r� r� )r� r r� r� r� r� rU r� r� r� r� r� r� r� r� ) r r, Zc_idxZc_distZe_idxZe_distZm_idxZm_distr_ r� ra r� r r r r� 1 s( z$RectangleSelector._set_active_handlec C sP t | jd�r<| jj�� }|�| j�� �j\}}t� ||g�S t� | j� � �S dS )a Returns numpy.ndarray of shape (2,5) containing x (``RectangleSelector.geometry[1,:]``) and y (``RectangleSelector.geometry[0,:]``) coordinates of the four corners of the rectangle starting and ending in the top left corner. � get_vertsN) ru r� r2 r� r� rF r� �Tr� r r� )r Zxfmr� r� r r r �geometryP s zRectangleSelector.geometry) r� NNFNNr� Nr� NFN)r r r r r r� r rh rP rl r0 r� r� r� rB r� �setterr� r� r� r r r r r� � sF % � 47 r� c @ s( e Zd ZdZeZdd� Zedd� �ZdS )�EllipseSelectora Select an elliptical region of an axes. For the cursor to remain responsive you must keep a reference to it. Example usage:: import numpy as np import matplotlib.pyplot as plt from matplotlib.widgets import EllipseSelector def onselect(eclick, erelease): "eclick and erelease are matplotlib events at press and release." print('startposition: (%f, %f)' % (eclick.xdata, eclick.ydata)) print('endposition : (%f, %f)' % (erelease.xdata, erelease.ydata)) print('used button : ', eclick.button) def toggle_selector(event): print(' Key pressed.') if event.key in ['Q', 'q'] and toggle_selector.ES.active: print('EllipseSelector deactivated.') toggle_selector.RS.set_active(False) if event.key in ['A', 'a'] and not toggle_selector.ES.active: print('EllipseSelector activated.') toggle_selector.ES.set_active(True) x = np.arange(100.) / 99 y = np.sin(x) fig, ax = plt.subplots() ax.plot(x, y) toggle_selector.ES = EllipseSelector(ax, onselect, drawtype='line') fig.canvas.mpl_connect('key_press_event', toggle_selector) plt.show() c C s� |\}}}}t ||g�\}}t ||g�\}} ||| d ||| d g} || d } | | d } | jdkr�| | j_d| | j_d| | j_nNt�t�d�d �} | t� | � | d }| t� | � | d }| j� ||� d S )Nr� r� r� � � r r ) r� r� r� rB r� r� r� Zdeg2radZarangeZcosZsinr� )r r� r_ r� ra r� r; r< r= r> rB �ar Zradr� r� r r r r� � s zEllipseSelector.draw_shapec C s� | j dkrB| jj\}}| jj}| jj}||d ||d ||fS | j�� \}}t|�t|� }}t|�t|� }}|||| || fS d S )Nr� r� )r� r� rB r� r� r� r r] ) r r� r� r� r� r^ r_ r` ra r r r r� � s zEllipseSelector._rect_bboxN) r r r r r r� r� r0 r� r r r r r� a s $r� c @ sB e Zd ZdZddd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dS )� LassoSelectora Selection curve of an arbitrary shape. For the selector to remain responsive you must keep a reference to it. The selected path can be used in conjunction with `~.Path.contains_point` to select data points from an image. In contrast to `Lasso`, `LassoSelector` is written with an interface similar to `RectangleSelector` and `SpanSelector`, and will continue to interact with the axes until disconnected. Example usage:: ax = subplot(111) ax.plot(x,y) def onselect(verts): print(verts) lasso = LassoSelector(ax, onselect) Parameters ---------- ax : :class:`~matplotlib.axes.Axes` The parent axes for the widget. onselect : function Whenever the lasso is released, the *onselect* function is called and passed the vertices of the selected path. button : List[Int], optional A list of integers indicating which mouse buttons should be used for rectangle selection. You can also specify a single integer if only a single button is desired. Default is ``None``, which does not limit which button can be used. Note, typically: - 1 = left mouse button - 2 = center mouse button (scroll wheel) - 3 = right mouse button NTc C sl t j| ||||d� d | _|d kr(t� }|r4d|d< tg g f|�| _| j�d� | j�| j� | jg| _ d S )Nrs Tr F) rC r �vertsrK r rB r� r2 r� rN )r r2 rI r r' r� r r r r � s � zLassoSelector.__init__c C s | � |� d S r )rW r+ r r r �onpress� s zLassoSelector.onpressc C s | � |�g| _| j�d� d S r� )rb r� rB r� r+ r r r rh � s zLassoSelector._pressc C s | � |� d S r )r r+ r r r � onrelease� s zLassoSelector.onreleasec C sN | j d k r(| j �| �|�� | �| j � | j�g g g� | j�d� d | _ d S r )r� r7 rb rI rB r� r� r+ r r r rP � s zLassoSelector._releasec C sB | j d krd S | j �| �|�� | j�tt| j � �� | �� d S r )r� r7 rb rB r� �listr� rM r+ r r r rl � s zLassoSelector._onmove)NTNN) r r r r r r� rh r� rP rl r r r r r� � s * � r� c @ s^ e Zd ZdZddd�Zdd� Zd d � Zd d � Zd d� Zdd� Z dd� Z dd� Z e dd� �Z dS )�PolygonSelectoraq Select a polygon region of an axes. Place vertices with each mouse click, and make the selection by completing the polygon (clicking on the first vertex). Hold the *ctrl* key and click and drag a vertex to reposition it (the *ctrl* key is not necessary if the polygon has already been completed). Hold the *shift* key and click and drag anywhere in the axes to move all vertices. Press the *esc* key to start a new polygon. For the selector to remain responsive you must keep a reference to it. Parameters ---------- ax : :class:`~matplotlib.axes.Axes` The parent axes for the widget. onselect : function When a polygon is completed or modified after completion, the `onselect` function is called and passed a list of the vertices as ``(xdata, ydata)`` tuples. useblit : bool, optional lineprops : dict, optional The line for the sides of the polygon is drawn with the properties given by `lineprops`. The default is ``dict(color='k', linestyle='-', linewidth=2, alpha=0.5)``. markerprops : dict, optional The markers for the vertices of the polygon are drawn with the properties given by `markerprops`. The default is ``dict(marker='o', markersize=7, mec='k', mfc='k', alpha=0.5)``. vertex_select_radius : float, optional A vertex is selected (to complete the polygon or to move a vertex) if the mouse click is within `vertex_select_radius` pixels of the vertex. The default radius is 15 pixels. Examples -------- :doc:`/gallery/widgets/polygon_selector_demo` FN� c C s� t ddddddd�}tj| ||||d� dgdg | _| _d| _|d krXt d d d d d �}| j|d< t| j| jf|�| _| j � | j� |d kr�t d |� dd �d�}t | j | j| j| j|d�| _ d| _|| _| j| j jg| _| �d� d S )NrD rF rE znot-applicable)r � move_vertex�move_allrG rH rB )r rL r Fr� r� r� rA r� r rV )r� r� )r r� r1 T)rK rC r �_xs�_ys�_polygon_completedr r rB r2 r� r r� �_polygon_handles�_active_handle_idx�vertex_select_radiusr\ rN r� )r r2 rI r r' Z markerpropsr� rL r r r r ' s4 � � �zPolygonSelector.__init__c C sj | j sd| jkrDt| j�dkrD| j�|j|j�\}}|| jk rD|| _ | jdd� | j dd� | _ | _ dS )zButton press event handlerr� r N) r� rU r� r� r� r r� r� r� r� r� � _xs_at_press� _ys_at_press)r r, Zh_idxZh_distr r r rh J s � zPolygonSelector._pressc C s� | j dkrd| _ nxt| j�dkrP| jd | jd krP| jd | jd krPd| _n:| js�d| jkr�d| jkr�| j�d|j� | j�d|j� | jr�| � | j � dS )rj r r1 r� Tr� r� N) r� r� r� r� r� rU �insertr� r� rI r� r+ r r r rP V s" ����zPolygonSelector._releasec C s& | � |�s"| �|�}| �|� dS dS rk )r- rd rl r+ r r r r l s zPolygonSelector.onmovec C s� | j dkrV| j }|j|j | j|< | j|< |dkrR| jrR|j|j | jd< | jd< �n d| jkr�| jr�|j| jj }|j| jj }tt | j��D ],}| j | | | j|< | j | | | j|< q�n�| js�d| jks�d| jkr�dS | j � � �| jd | jd f�\}}t�||j ||j �}t | j�dk�rZ|| jk �rZ| jd | jd | jd< | jd< n|j|j | jd< | jd< | �� dS )rm r r1 r� r� Nr� )r� r� r� r� r� r� rU rQ �ranger� r� r� rB Z get_transformrF r� Zhypotr� r� r� � _draw_polygon) r r, �idxr� r� r� r^ r` Zv0_distr r r rl x s2 ��� $zPolygonSelector._onmovec C sH | j sDd| jksd| jkrD| jdd� | jdd� | _| _| �� dS )zKey press event handlerr� r� Nr1 )r� rU r� r� r� r+ r r r ro � s ��"z PolygonSelector._on_key_pressc C s� | j sP|j| j�d�ks*|j| j�d�krP| j�|j� | j�|j� | � � nB|j| j�d�kr�| � |�}|jg|jg | _| _d| _ | � d� dS )zKey release event handlerr� r� r FTN) r� r� rL r r� r7 r� r� r� r� rd r� r+ r r r rq � s �� z PolygonSelector._on_key_releasec C s� | j �| j| j� | jsNt| j�dkrr| jd | jd krr| jd | jd krr| j�| jdd� | jdd� � n| j�| j| j� | �� dS )z5Redraw the polygon based on the new vertex positions.r� r1 r N)rB r� r� r� r� r� r� rM r r r r r� � s ���$z PolygonSelector._draw_polygonc C s" t t| jdd� | jdd� ��S )z�Get the polygon vertices. Returns ------- list A list of the vertices of the polygon as ``(xdata, ydata)`` tuples. Nr1 )r� r� r� r� r r r r r� � s zPolygonSelector.verts)FNNr� )r r r r r rh rP r rl ro rq r� r0 r� r r r r r� � s ' � # & r� c @ s* e Zd ZdZd dd�Zdd� Zdd � ZdS ) �Lassoa� Selection curve of an arbitrary shape. The selected path can be used in conjunction with :func:`~matplotlib.path.Path.contains_point` to select data points from an image. Unlike :class:`LassoSelector`, this must be initialized with a starting point `xy`, and the `Lasso` events are destroyed upon release. Parameters ---------- ax : `~matplotlib.axes.Axes` The parent axes for the widget. xy : (float, float) Coordinates of the start of the lasso. callback : callable Whenever the lasso is released, the `callback` function is called and passed the vertices of the selected path. NTc C s� t �| |� |o| jj| _| jr2| j�| jj�| _|\}}||fg| _ t |g|gdddd�| _ | j� | j � || _ | �d| j� | �d| j� d S )Nr� r� r� )r� rV rp rH rI )r1 r r4 r r r( r2 r) r% r� r rB r� r8 r: r� r )r r2 r� r8 r r� r� r r r r � s zLasso.__init__c C sh | � |�rd S | jd k rV| j�|j|jf� t| j�dkrF| �| j� | jj� | j � d | _| � � d S )Nr� ) r- r� r7 r� r� r� r8 r2 r� r� rB r= r+ r r r r� � s zLasso.onreleasec C s� | � |�rd S | jd kr d S |j| jkr,d S |jdkr:d S | j�|j|jf� | j� t t | j� �� | j r�| j �| j� | j�| j� | j �| jj� n | j �� d S r� )r- r� rY r2 r� r7 r� r� rB r� r� r� r r4 r- r% r. r/ r) r� r+ r r r r s z Lasso.onmove)NT)r r r r r r� r r r r r r� � s r� )%r rc Znumbersr Znumpyr� r� r r r� r Zpatchesr r r Z transformsr �objectr r! r1 r> rj r� r� r� r r r0 rC rr r� r� r� r� r� r� r r r r �<module> sN & *x 4 ^s e a; 0GW V