OwlCyberSecurity - MANAGER
Edit File: offsetbox.cpython-38.pyc
U ��]�� � @ s� d Z ddlZddlmZmZmZ ddlmZ ddl m Z ddl mZ ddlmZ ddlmZ ddlmZ ddlmZmZmZ ddl m Z ddlm Z m Z m Z d Z d d � Zd0d d�Z!d1dd�Z"G dd� de j#�Z$G dd� de$�Z%G dd� de%�Z&G dd� de%�Z'G dd� de$�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 j#e �Z/G d*d+� d+e0�Z1G d,d-� d-e1�Z2G d.d/� d/e1�Z3dS )2a� The OffsetBox is a simple container artist. The child artist are meant to be drawn at a relative position to its parent. The [VH]Packer, DrawingArea and TextArea are derived from the OffsetBox. The [VH]Packer automatically adjust the relative positions of their children, which should be instances of the OffsetBox. This is used to align similar artists together, e.g., in legend. The DrawingArea can contain any Artist as a child. The DrawingArea has a fixed width and height. The position of children relative to the parent is fixed. The TextArea is contains a single Text instance. The width and height of the TextArea instance is the width and height of the its child text. � N)�cbook� docstring�rcParams)�FontProperties)� BboxImage)�FancyBboxPatch�FancyArrowPatch� bbox_artist)�_AnnotationBase)�Bbox�BboxBase�TransformedBboxFc O s t rt| |� d S �N)�DEBUG� mbbox_artist)�args�kwargs� r �6/usr/lib/python3/dist-packages/matplotlib/offsetbox.pyr $ s r �fixedc s6 t | � \}}|dkrZt�dg� fdd�|D � �}|dd� }|dkrR|d � }||fS |dkr�|dkrnd}t|�dkr�|t|� t|�d � nd� t�dg� fd d�|D � �}|dd� }||fS |d k�r$t|�}|dkr�|� t|� }n|t|� | � |� t�t|�� }||fS td |f ��dS ) a� Given a list of (width, xdescent) of each boxes, calculate the total width and the x-offset positions of each items according to *mode*. xdescent is analogous to the usual descent, but along the x-direction. xdescent values are currently ignored. *wd_list* : list of (width, xdescent) of boxes to be packed. *sep* : spacing between boxes *total* : Intended total length. None if not used. *mode* : packing mode. 'fixed', 'expand', or 'equal'. r r c s g | ] }|� �qS r r ��.0�w��sepr r � <listcomp>> s z'_get_packed_offsets.<locals>.<listcomp>N����expand� c s g | ] }|� �qS r r r r r r r M s ZequalzUnknown mode : %s)�zip�npZcumsum�len�sum�maxZarange� ValueError) �wd_listZtotalr �modeZw_listZd_listZoffsets_�offsetsZmaxhr r r �_get_packed_offsets- s0 r( �baselinec s� � dkrt dd� | D ��� |dkr^t dd� | D ��}t dd� | D ��}|| � dd� | D �}nj|d krzd }d d� | D �}nN|d kr�d }� fd d�| D �}n.|dkr�d }� fdd�| D �}ntd|f ��� ||fS )a) Given a list of (height, descent) of each boxes, align the boxes with *align* and calculate the y-offsets of each boxes. total width and the offset positions of each items according to *mode*. xdescent is analogous to the usual descent, but along the x-direction. xdescent values are currently ignored. *hd_list* : list of (width, xdescent) of boxes to be aligned. *sep* : spacing between boxes *height* : Intended total length. None if not used. *align* : align mode. 'baseline', 'top', 'bottom', or 'center'. Nc s s | ]\}}|V qd S r r �r �h�dr r r � <genexpr>m s z'_get_aligned_offsets.<locals>.<genexpr>r) c s s | ]\}}|| V qd S r r r* r r r r- p s c s s | ]\}}|V qd S r r r* r r r r- q s c S s g | ] \}}d �qS )� r r* r r r r s s z(_get_aligned_offsets.<locals>.<listcomp>)�left�topr. c S s g | ] \}}|�qS r r r* r r r r v s )�rightZbottomc s g | ]\}}� | | �qS r r r* ��heightr r r y s �centerc s g | ]\}}� | d | �qS )� �?r r* r2 r r r | s zUnknown Align mode : %s)r# r$ )�hd_listr3 �align�height_descentZdescentr' r r2 r �_get_aligned_offsets^ s$ r9 c s� e Zd ZdZ� fdd�Zdd� Zejjj dd� �Zdd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Z� ZS ) � OffsetBoxz� The OffsetBox is a simple container artist. The child artist are meant to be drawn at a relative position to its parent. c s( t � j||� | �d� g | _d| _d S )NF�r r )�super�__init__Z set_clip_on� _children�_offset)�selfr r �� __class__r r r= � s zOffsetBox.__init__c C s* t j�| |� | �� D ]}|�|� qdS )z� Set the `.Figure` for the `.OffsetBox` and all its children. Parameters ---------- fig : `~matplotlib.figure.Figure` N)�martist�Artist� set_figure� get_children)r@ �fig�cr r r rE � s zOffsetBox.set_figurec C s0 t jj�| |� | �� D ]}|d k r||_qd S r )rC rD �axes�fsetrF )r@ ZaxrH r r r rI � s zOffsetBox.axesc C s4 | � � D ]"}|�|�\}}|r||f S qdi fS )a� Delegate the mouse event contains-check to the children. As a container, the `.OffsetBox` does not respond itself to mouseevents. Parameters ---------- mouseevent : `matplotlib.backend_bases.MouseEvent` Returns ------- contains : bool Whether any values are within the radius. details : dict An artist-specific dictionary of details of the event context, such as which points are contained in the pick radius. See the individual Artist subclasses for details. See Also -------- .Artist.contains F)rF �contains)r@ � mouseeventrH �a�br r r rK � s zOffsetBox.containsc C s || _ d| _dS )a� Set the offset. Parameters ---------- xy : (float, float) or callable The (x, y) coordinates of the offset in display units. These can either be given explicitly as a tuple (x, y), or by providing a function that converts the extent into the offset. This function must have the signature:: def offset(width, height, xdescent, ydescent, renderer) -> (float, float) TN)r? �stale�r@ �xyr r r � set_offset� s zOffsetBox.set_offsetc C s" t | j�r | �|||||�S | jS )a� Return the offset as a tuple (x, y). The extent parameters have to be provided to handle the case where the offset is dynamically determined by a callable (see `~.OffsetBox.set_offset`). Parameters ---------- width, height, xdescent, ydescent Extent parameters. renderer : `.RendererBase` subclass )�callabler? )r@ �widthr3 �xdescent�ydescent�rendererr r r � get_offset� s ��zOffsetBox.get_offsetc C s || _ d| _dS )zh Set the width of the box. Parameters ---------- width : float TN)rT rO )r@ rT r r r � set_width� s zOffsetBox.set_widthc C s || _ d| _dS )zj Set the height of the box. Parameters ---------- height : float TN)r3 rO )r@ r3 r r r � set_height� s zOffsetBox.set_heightc C s dd� | j D �S )z/Return a list of the visible child `.Artist`\s.c S s g | ]}|� � r|�qS r )� get_visible�r rH r r r r s z2OffsetBox.get_visible_children.<locals>.<listcomp>�r> �r@ r r r �get_visible_children s z OffsetBox.get_visible_childrenc C s | j S )z'Return a list of the child `.Artist`\s.r] r^ r r r rF s zOffsetBox.get_childrenc C s t d��dS )a- Update offset of the children and return the extent of the box. Parameters ---------- renderer : `.RendererBase` subclass Returns ------- width height xdescent ydescent list of (xoffset, yoffset) pairs z9get_extent_offsets must be overridden in derived classes.N)�NotImplementedError�r@ rW r r r �get_extent_offsets s �z OffsetBox.get_extent_offsetsc C s | � |�\}}}}}||||fS )z@Return a tuple ``width, height, xdescent, ydescent`` of the box.)rb )r@ rW r r+ �xd�ydr' r r r � get_extent s zOffsetBox.get_extentc C sD | � |�\}}}}}| �|||||�\}}tj�|| || ||�S )z3Return the bounding box (`.Bbox`) in display space.)rb rX � mtransformsr � from_bounds) r@ rW r r+ rc rd r' �px�pyr r r �get_window_extent! s zOffsetBox.get_window_extentc C s� | � |�\}}}}}| �|||||�\}}t| �� |�D ],\} \} } | �|| || f� | �|� q8t| |dtdd�d� d| _dS )�m Update the location of children if necessary and draw them to the given *renderer*. Fr. ��pad�ZfillZpropsN) rb rX r r_ rR �drawr �dictrO � r@ rW rT r3 rU rV r' rh ri rH �ox�oyr r r ro ' s � zOffsetBox.draw)�__name__� __module__� __qualname__�__doc__r= rE rC rD rI �setterrK rR rX rY rZ r_ rF rb re rj ro � __classcell__r r rA r r: � s r: c s e Zd Zd� fdd� Z� ZS )� PackerBaseNc s8 t � �� || _|| _|| _|| _|| _|| _|| _dS �aW Parameters ---------- pad : float, optional The boundary padding in points. sep : float, optional The spacing between items in points. width, height : float, optional Width and height of the container box in pixels, calculated if *None*. align : {'top', 'bottom', 'left', 'right', 'center', 'baseline'} Alignment of boxes. mode : {'fixed', 'expand', 'equal'} The packing mode. - 'fixed' packs the given `.Artists` tight with *sep* spacing. - 'expand' uses the maximal available space to distribute the artists with equal spacing in between. - 'equal': Each artist an equal fraction of the available space and is left-aligned (or top-aligned) therein. children : list of `.Artist` The artists to pack. Notes ----- *pad* and *sep* need to given in points and will be scale with the renderer dpi, while *width* and *height* need to be in pixels. N) r< r= r3 rT r rm r& r7 r> �r@ rm r rT r3 r7 r& �childrenrA r r r= : s % zPackerBase.__init__)NNNNNNN)rt ru rv r= ry r r rA r rz 9 s �rz c s* e Zd ZdZd � fdd� Zdd� Z� ZS ) �VPackerz� The VPacker has its children packed vertically. It automatically adjust the relative positions of children in the drawing time. Nr) r c s t � �|||||||� dS r{ �r< r= r| rA r r r= p s %zVPacker.__init__c s � � d�}| j| }| j| }| jd k rV| �� D ]$}t|t�r0|jdkr0|�| j� q0� fdd�| �� D �}dd� |D �}dd� |D �}t || j| j �\}} } dd� |D �} t | | j || j�\} } | dd� |D � }| |d }| | }|| }|d | | d | | | || t t| |��fS ) N� �?r c s g | ]}|� � ��qS r �re r\ �rW r r r � s �z.VPacker.get_extent_offsets.<locals>.<listcomp>c S s$ g | ] \}}}}||||| f�qS r r �r r r+ rc rd r r r r � s c S s g | ]\}}}}||f�qS r r r� r r r r � s c S s g | ]\}}}}||f�qS r r r� r r r r � s c S s g | ]\}}}}|�qS r r r� r r r r � s r � )�points_to_pixelsrm r rT r_ � isinstancerz r& rY r9 r7 r( r3 �listr )r@ rW �dpicorrm r rH �whd_listr% rT rU �xoffsets� pack_listr3 Z yoffsets_�yoffsetsrV r r� r rb � s< �� � �zVPacker.get_extent_offsets)NNNNr) r N�rt ru rv rw r= rb ry r r rA r r~ k s �'r~ c s* e Zd ZdZd � fdd� Zdd� Z� ZS ) �HPackerz� The HPacker has its children packed horizontally. It automatically adjusts the relative positions of children at draw time. Nr) r c s t � �|||||||� dS r{ r r| rA r r r= � s %zHPacker.__init__c s2 � � d�}| j| }| j| }� fdd�| �� D �}|sNd| d| ||g fS | jd kr�tdd� |D ��}tdd� |D ��}|| }n| jd| }dd� |D �} t| | j| j�\}}} d d� |D �} t| | j || j �\} } | d d� |D � }|d d }|| }| d| |d| || || t t || ��fS ) Nr� c s g | ]}|� � ��qS r r� r\ r� r r r � s �z.HPacker.get_extent_offsets.<locals>.<listcomp>r� c s s | ]\}}}}|| V qd S r r r� r r r r- � s z-HPacker.get_extent_offsets.<locals>.<genexpr>c s s | ]\}}}}|V qd S r r r� r r r r- � s c S s g | ]\}}}}||f�qS r r r� r r r r � s c S s g | ]\}}}}||f�qS r r r� r r r r � s c S s g | ]\}}}}|�qS r r r� r r r r s r ) r� rm r r_ r3 r# r9 r7 r( rT r& r� r )r@ rW r� rm r r� r8 rV r3 r6 r� r� rT Z xoffsets_r� rU r r� r rb � s> � � � �zHPacker.get_extent_offsets)NNNNr) r Nr� r r rA r r� � s �'r� c sD e Zd ZdZd� fdd� Zdd� Zdd � Zdd d �Zd d � Z� Z S )� PaddedBoxz� A container to add a padding around an `.Artist`. The `.PaddedBox` contains a `.FancyBboxPatch` that is used to visualize it when rendering. NFc s^ t � �� || _|g| _tdddddddd�| _| jjdd d � |d k rT| j�|� || _d S ) a� Parameters ---------- child : `~matplotlib.artist.Artist` The contained `.Artist`. pad : float The padding in points. This will be scaled with the renderer dpi. In contrast *width* and *hight* are in *pixel* and thus not scaled. draw_frame : bool Whether to draw the contained `.FancyBboxPatch`. patch_attrs : dict or None Additional parameters passed to the contained `.FancyBboxPatch`. �r. r. r� r �kr T�rQ rT r3 Z facecolorZ edgecolor�mutation_scaleZsnap�squarer rl N) r< r= rm r> r �patch� set_boxstyle�update� _drawFrame)r@ �childrm � draw_frameZ patch_attrsrA r r r= s � zPaddedBox.__init__c C sT |� d�}| j| }| jd �|�\}}}}|d| |d| || || dgfS )Nr� r r� r; )r� rm r> re )r@ rW r� rm r r+ rc rd r r r rb 5 s �z PaddedBox.get_extent_offsetsc C s� | � |�\}}}}}| �|||||�\}}t| �� |�D ]"\} \} } | �|| || f� q8| �|� | �� D ]} | �|� qnd| _dS )rk FN)rb rX r r_ rR r� ro rO rq r r r ro @ s � zPaddedBox.drawc C s4 | j �|j|j|j|j� |r*| j �|� d| _d S �NT)r� � set_bounds�x0�y0rT r3 �set_mutation_scalerO �r@ �bbox�fontsizer r r � update_frameU s � zPaddedBox.update_framec C s* | � |�}| �|� | jr&| j�|� d S r )rj r� r� r� ro )r@ rW r� r r r r� ] s zPaddedBox.draw_frame)NFN)N) rt ru rv rw r= rb ro r� r� ry r r rA r r� s ! r� c s| e Zd ZdZd� fdd� Zedd� �Zejdd� �Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Z� ZS )� DrawingAreaz� The DrawingArea can contain any Artist as a child. The DrawingArea has a fixed width and height. The position of children relative to the parent is fixed. The children can be clipped at the boundaries of the parent. r. Fc sX t � �� || _|| _|| _|| _|| _t�� | _ | j � � | j � dd� t�� | _ dS )z� *width*, *height* : width and height of the container box. *xdescent*, *ydescent* : descent of the box in x- and y-direction. *clip* : Whether to clip the children r N) r< r= rT r3 rU rV �_clip_childrenrf �Affine2D�offset_transform�clear� translate� dpi_transform)r@ rT r3 rU rV ZcliprA r r r= n s zDrawingArea.__init__c C s | j S )zl If the children of this DrawingArea should be clipped by DrawingArea bounding box. )r� r^ r r r � clip_children� s zDrawingArea.clip_childrenc C s t |�| _d| _d S r� )�boolr� rO )r@ �valr r r r� � s c C s | j | j S �zf Return the :class:`~matplotlib.transforms.Transform` applied to the children )r� r� r^ r r r � get_transform� s zDrawingArea.get_transformc C s dS �z+ set_transform is ignored. Nr �r@ �tr r r � set_transform� s zDrawingArea.set_transformc C s0 || _ | j�� | j�|d |d � d| _dS �z� Set the offset of the container. Parameters ---------- xy : (float, float) The (x,y) coordinates of the offset in display units. r r TN�r? r� r� r� rO rP r r r rR � s zDrawingArea.set_offsetc C s | j S �z1 return offset of the container. �r? r^ r r r rX � s zDrawingArea.get_offsetc C s8 | � |�\}}}}| �� \}}tj�|| || ||�S �z8 get the bounding box in display space. �re rX rf r rg �r@ rW r r+ rc rd rr rs r r r rj � s z DrawingArea.get_window_extentc C s. |� d�}| j| | j| | j| | j| fS )z@ Return with, height, xdescent, ydescent of box r� )r� rT r3 rU rV )r@ rW �dpi_corr r r re � s �zDrawingArea.get_extentc C sP | j �|� |�� s"|�| �� � | jdk r4| j|_| j}|dk rL|�|� dS )�?Add any :class:`~matplotlib.artist.Artist` to the container boxN)r> �appendZis_transform_setr� r� rI �figurerE )r@ rM rG r r r � add_artist� s zDrawingArea.add_artistc C s� |� d�}| j�� | j�||� t�t�ddgd| jg| j | jg| j dgg�| � � �}| j D ]*}| j r~|j s~|js~|�|� |�|� q^t| |dtdd�d� d| _dS )�# Draw the children r� r Fr. rl rn N)r� r� r� Zscalerf ZTransformedPath�mpath�Pathr3 rT r� r> r� ZclipboxZ _clippathZ set_clip_pathro r rp rO )r@ rW r� ZtpathrH r r r ro � s �� zDrawingArea.draw)r. r. F)rt ru rv rw r= �propertyr� rx r� r� rR rX rj re r� ro ry r r rA r r� f s" � r� c @ sz e Zd ZdZd dd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � ZdS ) �TextAreaz� The TextArea is contains a single Text instance. The text is placed at (0,0) with baseline+left alignment. The width and height of the TextArea instance is the width and height of the its child text. NTc C s� |dkr i }d|kr d|d< t jdd|f|�| _t�| � | jg| _t�� | _| j� � | j� dd� t�� | _ | j� | j| j � || _ || _dS )aG Parameters ---------- s : str a string to be displayed. textprops : dictionary, optional, default: None Dictionary of keyword parameters to be passed to the `~matplotlib.text.Text` instance contained inside TextArea. multilinebaseline : bool, optional If `True`, baseline for multiline text is adjusted so that it is (approximately) center-aligned with singleline text. minimumdescent : bool, optional If `True`, the box has a minimum descent of "p". N�var) r )�mtextZText�_textr: r= r> rf r� r� r� r� �_baseline_transformr� �_multilinebaseline�_minimumdescent)r@ �s� textpropsZmultilinebaseline�minimumdescentr r r r= � s �zTextArea.__init__c C s | j �|� d| _dS )z&Set the text of this area as a string.TN)r� �set_textrO �r@ r� r r r r� s zTextArea.set_textc C s | j �� S )z5Returns the string representation of this area's text)r� �get_textr^ r r r r� " s zTextArea.get_textc C s || _ d| _dS )z� Set multilinebaseline . If True, baseline for multiline text is adjusted so that it is (approximately) center-aligned with single-line text. TN)r� rO r� r r r �set_multilinebaseline& s z TextArea.set_multilinebaselinec C s | j S )z) get multilinebaseline . )r� r^ r r r �get_multilinebaseline0 s z TextArea.get_multilinebaselinec C s || _ d| _dS )z� Set minimumdescent . If True, extent of the single line text is adjusted so that it has minimum descent of "p" TN)r� rO r� r r r �set_minimumdescent6 s zTextArea.set_minimumdescentc C s | j S )z% get minimumdescent. )r� r^ r r r �get_minimumdescent@ s zTextArea.get_minimumdescentc C s dS r� r r� r r r r� F s zTextArea.set_transformc C s0 || _ | j�� | j�|d |d � d| _dS r� r� rP r r r rR L s zTextArea.set_offsetc C s | j S r� r� r^ r r r rX [ s zTextArea.get_offsetc C s8 | � |�\}}}}| �� \}}tj�|| || ||�S r� r� r� r r r rj a s zTextArea.get_window_extentc C s� |j d| jjdd�\}}}| j�|�\}}}|j|j }} | j�� t|�dkr�| j r�d| d|| } | j� d|| � | }n,t || | | �} | � � r�t ||�}| | } || d|fS )NZlpF)Zismathr r5 r r. ) Z get_text_width_height_descentr� Z_fontpropertiesZ _get_layoutrT r3 r� r� r! r� r� r# r� ) r@ rW �_Zh_Zd_r� �infor, r r+ Zd_newZh_dr r r re i s" � zTextArea.get_extentc C s, | j �|� t| |dtdd�d� d| _dS �r� Fr. rl rn N)r� ro r rp rO ra r r r ro � s z TextArea.draw)NNT)rt ru rv rw r= r� r� r� r� r� r� r� rR rX rj re ro r r r r r� � s" � , r� c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )�AuxTransformBoxaX Offset Box with the aux_transform. Its children will be transformed with the aux_transform first then will be offseted. The absolute coordinate of the aux_transform is meaning as it will be automatically adjust so that the left-lower corner of the bounding box of children will be set to (0,0) before the offset transform. It is similar to drawing area, except that the extent of the box is not predetermined but calculated from the window extent of its children. Furthermore, the extent of the children will be calculated in the transformed coordinate. c C sJ || _ t�| � t�� | _| j�� | j�dd� t�� | _| j�� d S �Nr ) � aux_transformr: r= rf r� r� r� r� �ref_offset_transform)r@ r� r r r r= � s zAuxTransformBox.__init__c C s$ | j �|� |�| �� � d| _dS )r� TN)r> r� r� r� rO )r@ rM r r r r� � s zAuxTransformBox.add_artistc C s | j | j | j S r� )r� r� r� r^ r r r r� � s ��z AuxTransformBox.get_transformc C s dS r� r r� r r r r� � s z AuxTransformBox.set_transformc C s0 || _ | j�� | j�|d |d � d| _dS r� r� rP r r r rR � s zAuxTransformBox.set_offsetc C s | j S r� r� r^ r r r rX � s zAuxTransformBox.get_offsetc C s8 | � |�\}}}}| �� \}}tj�|| || ||�S r� r� r� r r r rj � s z!AuxTransformBox.get_window_extentc s| | j �� }| j�� | j �� � fdd�| jD �}tj�|�}| j�|j |j � | j j |� }| j � |� |j |jddfS )Nc s g | ]}|� � ��qS r )rj r\ r� r r r � s z.AuxTransformBox.get_extent.<locals>.<listcomp>r. )r� Z to_valuesr� r� r> rf r �unionr� r� r� Zmatrix_from_valuesZ set_matrixrT r3 )r@ rW Z_offZbboxesZubZmtxr r� r re � s zAuxTransformBox.get_extentc C s6 | j D ]}|�|� qt| |dtdd�d� d| _dS r� )r> ro r rp rO )r@ rW rH r r r ro � s zAuxTransformBox.drawN) rt ru rv rw r= r� r� r� rR rX rj re ro r r r r r� � s r� c s� e Zd ZdZdZdddddddd d d d � Zd)� fdd� Zdd� Zdd� Zdd� Z dd� Z dd � Z d*d d �Z d d � Z d+d!d"�Zd,d#d$�Zd%d&� Zd'd(� Z� ZS )-�AnchoredOffsetboxa7 An offset box placed according to the legend location loc. AnchoredOffsetbox has a single child. When multiple children is needed, use other OffsetBox class to enclose them. By default, the offset box is anchored against its parent axes. You may explicitly specify the bbox_to_anchor. � r r� � � � � � � � ) z upper rightz upper leftz lower leftz lower rightr1 z center leftz center rightz lower centerz upper centerr4 皙�����?r5 NTc s t � jf | � | �||� | �|� t|t�rjz| j| }W n, tk rh td|d� | j�f ��Y nX || _ || _ || _ |dkr�t td d�| _n6t|t�r�t f |�| _d|kr�| j�td � n|| _tdddd d | j�� d d �| _| jjd dd� || _dS )a� loc is a string or an integer specifying the legend location. The valid location codes are:: 'upper right' : 1, 'upper left' : 2, 'lower left' : 3, 'lower right' : 4, 'right' : 5, (same as 'center right', for back-compatibility) 'center left' : 6, 'center right' : 7, 'lower center' : 8, 'upper center' : 9, 'center' : 10, pad : pad around the child for drawing a frame. given in fraction of fontsize. borderpad : pad between offsetbox frame and the bbox_to_anchor, child : OffsetBox instance that will be anchored. prop : font property. This is only used as a reference for paddings. frameon : draw a frame box if True. bbox_to_anchor : bbox to anchor. Use self.axes.bbox if None. bbox_transform : with which the bbox_to_anchor will be transformed. z4Unrecognized location "%s". Valid locations are %s z N�legend.fontsize��sizer� r� r� r r� Tr� r� r rl )r< r= �set_bbox_to_anchor� set_childr� �str�codes�KeyErrorr$ �join�loc� borderpadrm r r �proprp Zset_sizer �get_size_in_pointsr� r� r� ) r@ r� rm r� r� r� �frameon�bbox_to_anchorZbbox_transformr rA r r r= s> % � �zAnchoredOffsetbox.__init__c C s || _ |dk r| j|_d| _dS )z set the child to be anchoredNT)�_childrI rO )r@ r� r r r r� ^ s zAnchoredOffsetbox.set_childc C s | j S )zreturn the child�r� r^ r r r � get_childe s zAnchoredOffsetbox.get_childc C s | j gS )zreturn the list of childrenr� r^ r r r rF i s z AnchoredOffsetbox.get_childrenc C sT | � � �|�\}}}}|�| j�� �}| j| }|d| |d| || || fS )zq return the extent of the artist. The extent of the child added with the pad is returned r� )r� re r� r� r� rm )r@ rW r r+ rc rd r� rm r r r re m s z AnchoredOffsetbox.get_extentc C s6 | j dkr| jjS | j}|dkr&| j S t| j |�S dS )zB return the bbox that the legend will be anchored N)�_bbox_to_anchorrI r� �_bbox_to_anchor_transformr )r@ � transformr r r �get_bbox_to_anchorx s �z$AnchoredOffsetbox.get_bbox_to_anchorc C s� |dkst |t�r|| _nZz t|�}W n$ tk rJ tdt|� ��Y nX |dkrh|d |d ddg}tj|� | _|| _ d| _ dS )aF set the bbox that the child will be anchored. *bbox* can be a Bbox instance, a list of [left, bottom, width, height], or a list of [left, bottom] where the width and height will be assumed to be zero. The bbox will be transformed to display coordinate by the given transform. Nz Invalid argument for bbox : %sr� r r T) r� r r� r! � TypeErrorr$ r� r rg r� rO )r@ r� r� �lr r r r� � s z$AnchoredOffsetbox.set_bbox_to_anchorc C sJ | � |� | �|�\}}}}| �|||||�\}}t�|| || ||�S r� )�_update_offset_funcre rX r rg r� r r r rj � s z#AnchoredOffsetbox.get_window_extentc C s4 |dkr|� | j�� �}|| fdd�}| �|� dS )zs Update the offset func which depends on the dpi of the renderer (because of the padding). Nc S sH t �dd| |�}|j| }|�� } |�|j|| |�\} } | | | | fS r� )r rg r� r� �_get_anchored_bboxr� ) r r+ rc rd rW r� r@ r� r� r� r� r� r r r r? � s �z6AnchoredOffsetbox._update_offset_func.<locals>._offset)r� r� r� rR )r@ rW r� r? r r r r� � s � z%AnchoredOffsetbox._update_offset_funcc C s. | j �|j|j|j|j� |r*| j �|� d S r )r� r� r� r� rT r3 r� r� r r r r� � s �z AnchoredOffsetbox.update_framec C s� | � � s dS |�| j�� �}| �||� | jrP| �|�}| �||� | j� |� | � |�\}}}}| � |||||�\}} | � � � || f� | � � � |� d| _dS )zdraw the artistNF)r[ r� r� r� r� r� rj r� r� ro re rX r� rR rO ) r@ rW r� r� rT r3 rU rV rh ri r r r ro � s zAnchoredOffsetbox.drawc C s� |t dd�kst�t d�\ }}}}} } } } } }}|d|d|d| d| d| d| d| d |d |d i }|| }|�| �}|j||d �}|j|jfS ) z{ return the position of the bbox anchored at the parentbbox with the loc code, with the borderpad. r � ZNEZNWZSWZSE�E�W�S�N�C)� container)�range�AssertionErrorZpaddedZanchoredr� r� )r@ r� r� Z parentbboxr� ZBESTZURZULZLLZLR�RZCLZCRZLCZUCr Z anchor_coefsrH r Z anchored_boxr r r r � s4 � z$AnchoredOffsetbox._get_anchored_bbox)r� r5 NNTNN)N)N)N)rt ru rv rw �zorderr� r= r� r� rF re r� r� rj r� r� ro r ry r r rA r r� � s@ � �H r� c s" e Zd ZdZd� fdd� Z� ZS )� AnchoredTextz& AnchoredOffsetbox with Text. r� r5 Nc sr |dkr i }ddddh}|t |�@ r2tjddd� t||d d �| _| jj�� }t� j|f||| j|d �|�� dS ) a� Parameters ---------- s : string Text. loc : str Location code. pad : float, optional Pad between the text and the frame as fraction of the font size. borderpad : float, optional Pad between the frame and the axes (or *bbox_to_anchor*). prop : dictionary, optional, default: None Dictionary of keyword parameters to be passed to the `~matplotlib.text.Text` instance contained inside AnchoredText. Notes ----- Other keyword parameters of `AnchoredOffsetbox` are also allowed. NZhaZhorizontalalignmentr� Zverticalalignmentz3.1z�Mixing horizontalalignment or verticalalignment with AnchoredText is not supported, deprecated since %(version)s, and will raise an exception %(removal)s.)�messageF)r� r� )rm r� r� r� ) �setr Zwarn_deprecatedr� Ztxtr� Zget_fontpropertiesr< r= ) r@ r� r� rm r� r� r Z badkwargs�fprA r r r= � s( � � ��zAnchoredText.__init__)r� r5 N)rt ru rv rw r= ry r r rA r r � s r c @ s^ e Zd Zddd�Zdd � Zd d � Zd d � Zdd� Zdd� Zdd� Z dd� Z dd� Z dd� Z dS )� OffsetImager N� @FTc K sV t �| � | | _tf | j||||||| d�| ��| _| jg| _| �|� | �|� d S )N)r� �cmap�norm� interpolation�origin� filternorm� filterrad�resample) r: r= �_dpi_corr rj �imager> �set_zoom�set_data) r@ �arr�zoomr r r r r r r r� r r r r r= ) s �� zOffsetImage.__init__c C s$ t �|�| _| j�| j� d| _d S r� )r Zasarray�_datar r rO )r@ r r r r r I s zOffsetImage.set_datac C s | j S r )r r^ r r r �get_dataN s zOffsetImage.get_datac C s || _ d| _d S r� )�_zoomrO )r@ r r r r r Q s zOffsetImage.set_zoomc C s | j S r )r! r^ r r r �get_zoomU s zOffsetImage.get_zoomc C s | j S r� r� r^ r r r rX j s zOffsetImage.get_offsetc C s | j gS r )r r^ r r r rF p s zOffsetImage.get_childrenc C s8 | � |�\}}}}| �� \}}tj�|| || ||�S r� r� r� r r r rj s s z OffsetImage.get_window_extentc C s^ | j r|�d�}nd}| �� }| �� }|jd d� \}}|| | || | }}||ddfS )Nr� r� r )r r� r" r �shape) r@ rW r� r �dataZnyZnxr r+ r r r re { s zOffsetImage.get_extentc C s | j �|� d| _dS )r� FN)r ro rO ra r r r ro � s zOffsetImage.draw) r NNNNr r FT) rt ru rv r= r r r r" rX rF rj re ro r r r r r ( s& � r c @ s� e Zd ZdZdZdd� Zejd"d d ��Ze d d � �Z e j dd � �Z e dd� �Z e j dd� �Z dd� Z dd� Zdd� Zd#dd�Zd$dd�Zd d � Zd d � Zd d!� ZdS )%�AnnotationBboxzD Annotation-like class, but with offsetbox instead of Text. r� c C s d| j d | j d f S )NzAnnotationBbox(%g,%g)r r )rQ r^ r r r �__str__� s zAnnotationBbox.__str__Nr$ Tr� �r5 r5 c K s� t jj| f| � tj| |||d� || _| | _| �| � |dkrH|| _n|| _|dkr^|| _n|| _| dk r�| j� dd�| _ t d| j�| _ n d| _ d| _ | | _ tdddd d | j�� d d �| _| jjd |d� | r�| jjf | � || _dS )ao *offsetbox* : OffsetBox instance *xycoords* : same as Annotation but can be a tuple of two strings which are interpreted as x and y coordinates. *boxcoords* : similar to textcoords as Annotation but can be a tuple of two strings which are interpreted as x and y coordinates. *box_alignment* : a tuple of two floats for a vertical and horizontal alignment of the offset box w.r.t. the *boxcoords*. The lower-left corner is (0.0) and upper-right corner is (1.1). other parameters are identical to that of Annotation. )�xycoords�annotation_clipN�relposr' r; �r r r� r� r r� Tr� r� rl )r; r+ )rC rD r= r � offsetbox� arrowprops� set_fontsize�xybox� boxcoords�pop� _arrow_relposr � arrow_patch�_box_alignmentr r� r� r� r� r r� )r@ r, rQ r/ r( r0 r� rm r) Z box_alignmentZ bboxpropsr- r� r r r r r= � sH � � �zAnnotationBbox.__init__c C s | j S r )r/ r^ r r r �xyann� s zAnnotationBbox.xyannc C s || _ d| _d S r� )r/ rO )r@ r5 r r r r5 � s c C s | j S r )r0 r^ r r r � anncoords� s zAnnotationBbox.anncoordsc C s || _ d| _d S r� )r0 rO )r@ Zcoordsr r r r6 � s c C s | j �|�\}}||fS r )r, rK )r@ �eventr� Ztinfor r r rK � s zAnnotationBbox.containsc C s" | j | jg}| jr |�| j� |S r )r, r� r3 r� )r@ r} r r r rF � s zAnnotationBbox.get_childrenc C s4 | j d k r| j �|� | j�|� tj�| |� d S r )r3 rE r, rC rD )r@ rG r r r rE s zAnnotationBbox.set_figurec C s& |dkrt d }t|d�| _d| _dS )z( set fontsize in points Nr� r� T)r r r� rO r� r r r r. s zAnnotationBbox.set_fontsizec C s | j �� S )z+ return fontsize in points )r� r� r� r r r � get_fontsize s zAnnotationBbox.get_fontsizec C sF | � |�}| �||� |�| �� �}| j�|� | jrB| j�|� dS )zQ Update the pixel positions of the annotated point and the text. N)�_get_position_xy�_update_position_xyboxr� r8 r� r� r3 )r@ rW �xy_pixelr� r r r �update_positions s z AnnotationBbox.update_positionsc C s� | j \}}t| jt�rT| j\}}| �||||�\}}| �||||�\} } || } } n| �|||| j�\} } | j�|�\} }}}| j\}}| j�| || | | || | f� | j� |�}| j � |j |j |j|j� |\}}|| }}| j�r�| j�� }| j}|j |j|d } |j |j|d } | j�| | f||f� | j�� }|�d|�}|�|�}| j�|� |�d| j �}| j�|� dS )z` Update the pixel positions of the annotation text and the arrow patch. r r r� �patchAN)r/ r� r0 �tupleZ_get_xyr, re r4 rR rj r� r� r� r� rT r3 r- �copyr2 r3 Z set_positionsr� r� r1 r� r� Z set_patchA)r@ rW r; �x�yZxcoordZycoordZx1Zy1Zx2Zy2Zox0Zoy0r r+ rc rd Z_fwZ_fhr� Zox1Zoy1r, r* Zfsr� r= r r r r: ) s: ( � z%AnnotationBbox._update_position_xyboxc C s� |dk r|| _ | �� sdS | �|�}| �||�s4dS | �|� | jdk rt| jjdkrh| jdk rh| j| j_| j�|� | jr�| j �|� | j �|� d| _ dS )zN Draw the :class:`Annotation` object to the given *renderer*. NF) Z _rendererr[ r9 Z _check_xyr< r3 r� ro r� r� r, rO )r@ rW r; r r r ro c s zAnnotationBbox.draw) Nr$ NTr� Nr' NNN)N)N)rt ru rv rw r r& r Zdedent_interpdr= r� r5 rx r6 rK rF rE r. r8 r< r: ro r r r r r% � s> � H :r% c @ sj e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )� DraggableBasea. Helper base class for a draggable artist (legend, offsetbox). Derived classes must override the following methods:: def save_offset(self): ''' Called when the object is picked for dragging; should save the reference position of the artist. ''' def update_offset(self, dx, dy): ''' Called during the dragging; (*dx*, *dy*) is the pixel offset from the point where the mouse drag started. ''' Optionally, you may override the following methods:: def artist_picker(self, artist, evt): '''The picker method that will be used.''' return self.ref_artist.contains(evt) def finalize_offset(self): '''Called when the mouse is released.''' In the current implementation of `DraggableLegend` and `DraggableAnnotation`, `update_offset` places the artists in display coordinates, and `finalize_offset` recalculates their position in axes coordinate and set a relevant attribute. Fc C s` || _ d| _| j jj| _|o"| jj| _| j�d| j�}| j�d| j�}|� | j � ||g| _ d S )NFZ pick_eventZbutton_release_event) � ref_artist� got_artistr� �canvasZ supports_blit� _use_blit� mpl_connect�on_pick� on_releaseZ set_picker� artist_picker�cids)r@ rC �use_blitZc2Zc3r r r r= � s zDraggableBase.__init__c C s@ | � � r<| jr<|j| j }|j| j }| �||� | j�� d S r ) �_check_still_parentedrD r@ �mouse_xrA �mouse_y� update_offsetrE ro �r@ �evt�dx�dyr r r � on_motion� s zDraggableBase.on_motionc C sh | � � rd| jrd|j| j }|j| j }| �||� | j�| j � | j � | j j j � | j�| j j j� d S r )rM rD r@ rN rA rO rP rE Zrestore_region� backgroundrC ro r� �_cachedRenderer�blitr� rQ r r r �on_motion_blit� s z DraggableBase.on_motion_blitc C s� | � � r�|j| jkr�|jj| _|jj| _d| _| j r�| j� d� | j � � | j � | jjj�| _| j� | jjj� | j �| jjj� | j �d| j�| _n| j �d| j�| _| �� d S )NTZmotion_notify_event)rM �artistrC rL r@ rN rA rO rD rF � set_animatedrE ro Zcopy_from_bboxr� r� rV rW rX rG rY �_c1rU � save_offset)r@ rR r r r rH � s& ���zDraggableBase.on_pickc C s@ | � � r<| jr<| �� d| _| j�| j� | jr<| j�d� d S )NF) rM rD �finalize_offsetrE �mpl_disconnectr\ rF rC r[ )r@ r7 r r r rI � s zDraggableBase.on_releasec C s | j jd kr| �� dS dS d S )NFT)rC r� � disconnectr^ r r r rM � s z#DraggableBase._check_still_parentedc C sH | j D ]}| j�|� qz | j}W n tk r6 Y nX | j�|� dS )zDisconnect the callbacks.N)rK rE r_ r\ �AttributeError)r@ ZcidZc1r r r r` � s zDraggableBase.disconnectc C s | j �|�S r )rC rK )r@ rZ rR r r r rJ � s zDraggableBase.artist_pickerc C s d S r r r^ r r r r] � s zDraggableBase.save_offsetc C s d S r r )r@ rS rT r r r rP � s zDraggableBase.update_offsetc C s d S r r r^ r r r r^ � s z DraggableBase.finalize_offsetN)F)rt ru rv rw r= rU rY rH rI rM r` rJ r] rP r^ r r r r rB � s rB c @ s. e Zd Zd dd�Zdd� Zdd� Zdd � Zd S ) �DraggableOffsetBoxFc C s t j| ||d� || _d S �N)rL )rB r= r, )r@ rC r, rL r r r r= � s zDraggableOffsetBox.__init__c C sN | j }|jj}|�|�\}}}}|�|||||�}|\| _| _| j �|� d S r )r, r� rW re rX � offsetbox_x� offsetbox_yrR )r@ r, rW r r+ rc rd �offsetr r r r] s z DraggableOffsetBox.save_offsetc C s$ | j | | j| f}| j�|� d S r )rd re r, rR )r@ rS rT � loc_in_canvasr r r rP s z DraggableOffsetBox.update_offsetc C s> | j }|jj}|�|�\}}}}|j\}}|| || f} | S r )r, r� rW re r? ) r@ r, rW r r+ rc rd rr rs rg r r r �get_loc_in_canvas s z$DraggableOffsetBox.get_loc_in_canvasN)F)rt ru rv r= r] rP rh r r r r rb � s rb c @ s& e Zd Zd dd�Zdd� Zdd� ZdS ) �DraggableAnnotationFc C s t j| ||d� || _d S rc )rB r= � annotation)r@ rj rL r r r r= s z DraggableAnnotation.__init__c C s" | j }|�� �|j�\| _| _d S r )rj r� r� r5 rr rs )r@ �annr r r r] s z DraggableAnnotation.save_offsetc C s. | j }|�� �� �| j| | j| f�|_d S r )rj r� �invertedr� rr rs r5 )r@ rS rT rk r r r rP s �z!DraggableAnnotation.update_offsetN)F)rt ru rv r= r] rP r r r r ri s ri )r )r) )4rw Znumpyr Z matplotlibr r r Zmatplotlib.artistrZ rC Zmatplotlib.path�pathr� Zmatplotlib.text�textr� Zmatplotlib.transformsZ transformsrf Zmatplotlib.font_managerr Zmatplotlib.imager Zmatplotlib.patchesr r r r r r r r r r( r9 rD r: rz r~ r� r� r� r� r� r� r r r% �objectrB rb ri r r r r �<module> sF 1 % 72OSY (n y1i p{