OwlCyberSecurity - MANAGER
Edit File: legend_handler.cpython-38.pyc
U ��]�g � @ s$ d Z ddlmZ ddlZddlmZ ddlmZ ddl m Z ddl m Z dd� ZG dd � d e�ZG d d � d e�ZG d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG d d � d e�ZG d d � d e�Z G d d!� d!e�Z dS )"a� This module defines default legend handlers. It is strongly encouraged to have read the :doc:`legend guide </tutorials/intermediate/legend_guide>` before this documentation. Legend handlers are expected to be a callable object with a following signature. :: legend_handler(legend, orig_handle, fontsize, handlebox) Where *legend* is the legend itself, *orig_handle* is the original plot, *fontsize* is the fontsize in pixels, and *handlebox* is a OffsetBox instance. Within the call, you should create relevant artists (using relevant properties from the *legend* and/or *orig_handle*) and add them into the handlebox. The artists needs to be scaled according to the fontsize (note that the size is in pixel, i.e., this is dpi-scaled value). This module includes definition of several legend handler classes derived from the base class (HandlerBase) with the following method:: def legend_artist(self, legend, orig_handle, fontsize, handlebox) � )�cycleN��Line2D)� Rectanglec C s | � |�� d � d S �Nr )� update_fromZ get_children)Ztgt�src� r �;/usr/lib/python3/dist-packages/matplotlib/legend_handler.py�update_from_first_child% s r c @ sJ e Zd ZdZddd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dS )� HandlerBasea A Base class for default legend handlers. The derived classes are meant to override *create_artists* method, which has a following signature.:: def create_artists(self, legend, orig_handle, xdescent, ydescent, width, height, fontsize, trans): The overridden method needs to create artists of the given transform that fits in the given dimension (xdescent, ydescent, width, height) that are scaled by fontsize if necessary. � Nc C s || | _ | _|| _d S �N)�_xpad�_ypad�_update_prop_func)�selfZxpadZypadZ update_funcr r r �__init__9 s zHandlerBase.__init__c C s( | j d kr| �||� n | � ||� d S r )r �_default_update_prop�r � legend_handle� orig_handler r r � _update_prop= s zHandlerBase._update_propc C s |� |� d S r )r r r r r r C s z HandlerBase._default_update_propc C s. | � ||� |�|� |�d � |�d � d S r )r Z_set_artist_props� set_clip_box� set_clip_path�r r r �legendr r r � update_propF s zHandlerBase.update_propc C sD || j | }|| j| }|| j | }|| j| }||||fS r )r r )r r r �xdescent�ydescent�width�height�fontsizer r r �adjust_drawing_areaN s z HandlerBase.adjust_drawing_areac C s^ | � |||j|j|j|j|�\}}}}| �||||||||�� �} | D ]} |�| � qF| d S )aU Return the artist that this HandlerBase generates for the given original artist/handle. Parameters ---------- legend : :class:`matplotlib.legend.Legend` instance The legend for which these legend artists are being created. orig_handle : :class:`matplotlib.artist.Artist` or similar The object for which these legend artists are being created. fontsize : float or int The fontsize in pixels. The artists being created should be scaled according to the given fontsize. handlebox : :class:`matplotlib.offsetbox.OffsetBox` instance The box which has been created to hold this legend entry's artists. Artists created in the `legend_artist` method must be added to this handlebox inside this method. r )r# r r r r! �create_artistsZ get_transformZ add_artist) r r r r" Z handleboxr r r r! �artists�ar r r � legend_artistW s( � � zHandlerBase.legend_artistc C s t d��d S )NzDerived must override)�NotImplementedError) r r r r r r r! r" �transr r r r$ | s zHandlerBase.create_artists)r r N) �__name__� __module__� __qualname__�__doc__r r r r r# r' r$ r r r r r ) s %r c @ s* e Zd ZdZd dd�Zdd� Zdd � ZdS ) �HandlerNpointszM A legend handler that shows *numpoints* points in the legend entry. �333333�?Nc K s t j| f|� || _|| _dS )a* Parameters ---------- marker_pad : float Padding between points in legend entry. numpoints : int Number of points to show in legend entry. Notes ----- Any other keyword arguments are given to `HandlerBase`. N)r r � _numpoints� _marker_pad�r � marker_pad� numpoints�kwr r r r � s zHandlerNpoints.__init__c C s | j d kr|jS | j S d S r )r0 r4 �r r r r r � get_numpoints� s z HandlerNpoints.get_numpointsc C sn | � |�}|dkr@| j| }t�| | | | | |�} | } n&t�| | | d�} | d| g} | | fS )N� � � �?)r7 r1 �np�linspace) r r r r r r! r" r4 �pad�xdata� xdata_markerr r r � get_xdata� s �zHandlerNpoints.get_xdata)r/ N)r* r+ r, r- r r7 r@ r r r r r. � s r. c @ s" e Zd ZdZddd�Zdd� ZdS )�HandlerNpointsYoffsetsz� A legend handler that shows *numpoints* in the legend, and allows them to be individually offest in the y-direction. Nc K s t j| fd|i|�� || _dS )a^ Parameters ---------- numpoints : int Number of points to show in legend entry. yoffsets : array of floats Length *numpoints* list of y offsets for each point in legend entry. Notes ----- Any other keyword arguments are given to `HandlerNpoints`. r4 N)r. r � _yoffsets)r r4 �yoffsetsr5 r r r r � s z HandlerNpointsYoffsets.__init__c C s* | j d kr||j }n|t�| j � }|S r �rB Z_scatteryoffsetsr; �asarray�r r r r r r! r" �ydatar r r � get_ydata� s z HandlerNpointsYoffsets.get_ydata)NN)r* r+ r, r- r rH r r r r rA � s rA c @ s" e Zd ZdZddd�Zdd� ZdS ) � HandlerLine2Dz* Handler for `.Line2D` instances. r/ Nc K s t j| f||d�|�� dS )a- Parameters ---------- marker_pad : float Padding between points in legend entry. numpoints : int Number of points to show in legend entry. Notes ----- Any other keyword arguments are given to `HandlerNpoints`. �r3 r4 N)r. r r2 r r r r � s ��zHandlerLine2D.__init__c C s� | � ||||||�\} } t�| || d �} t| | �} | �| ||� | �d� | �d� t| | d t| �� �} | �| ||� | �d� |j dkr�| � � |j }| � |� | | _ | � |� | � |� | | gS )Nr9 �default� �Noner8 )r@ r; Z full_liker r � set_drawstyle� set_marker�len� set_linestyle� markerscale�get_markersize�set_markersizeZ _legmarker� set_transform)r r r r r r r! r" r) r> r? rG �legline�legline_marker�newszr r r r$ � s( � z HandlerLine2D.create_artists)r/ N�r* r+ r, r- r r$ r r r r rI � s rI c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS ) � HandlerPatchz) Handler for `.Patch` instances. Nc K s t j| f|� || _dS )a� Parameters ---------- patch_func : callable, optional The function that creates the legend key artist. *patch_func* should have the signature:: def patch_func(legend=legend, orig_handle=orig_handle, xdescent=xdescent, ydescent=ydescent, width=width, height=height, fontsize=fontsize) Subsequently the created artist will have its ``update_prop`` method called and the appropriate transform will be applied. Notes ----- Any other keyword arguments are given to `HandlerBase`. N)r r � _patch_func)r Z patch_funcr5 r r r r s zHandlerPatch.__init__c C s> | j d kr"t| | f||d�}n| j |||||||d�}|S )N�Zxyr r! )r r r r r r! r" )r[ r ) r r r r r r r! r" �pr r r � _create_patch s � �zHandlerPatch._create_patchc C s4 | � |||||||�} | �| ||� | �|� | gS r )r^ r rU � r r r r r r r! r" r) r] r r r r$ ' s � zHandlerPatch.create_artists)N)r* r+ r, r- r r^ r$ r r r r rZ s rZ c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �HandlerLineCollectionz2 Handler for `.LineCollection` instances. c C s | j d kr|jS | j S d S r �r0 Z scatterpointsr6 r r r r7 4 s z#HandlerLineCollection.get_numpointsc C sD |� � d }|jd }|�� d }|�|� |�|� |�|� d S r )�get_linewidthsZ_us_linestylesZ get_colors� set_colorrQ � set_linewidth)r r r ZlwZdashesZcolorr r r r : s z*HandlerLineCollection._default_update_propc C sZ | � ||||||�\} } || d t�| jt� } t| | �} | �| ||� | �|� | gS )N� @)r@ r; �ones�shape�floatr r rU ) r r r r r r r! r" r) r> r? rG rV r r r r$ B s � z$HandlerLineCollection.create_artistsN)r* r+ r, r- r7 r r$ r r r r r` 0 s r` c @ sB e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� HandlerRegularPolyCollectionz0 Handler for `.RegularPolyCollections`. Nc K s t j| fd|i|�� || _d S )NrC )rA r �_sizes)r rC �sizesr5 r r r r T s z%HandlerRegularPolyCollection.__init__c C s | j d kr|jS | j S d S r ra r6 r r r r7 Y s z*HandlerRegularPolyCollection.get_numpointsc C s� | j d kr�|�� }t|�s dg}t|�|jd } t|�|jd } | �|�} | dk rrd| | | | gd | � } q�| | } | t�dd| � | } n| j } | S )Nr8 r9 � r: r ) rj � get_sizesrP �maxrR �minr7 r; r< )r r r r r r r! r" Z handle_sizesZsize_maxZsize_minr4 rk �rngr r r rm _ s" ��z&HandlerRegularPolyCollection.get_sizesc C s0 | � ||� |�|j� |�d � |�d � d S r )r � set_figureZfigurer r r r r r r t s z(HandlerRegularPolyCollection.update_propc C s" t |�|�� |�� |||d�}|S )N)Zrotationrk �offsets� transOffset)�typeZ get_numsidesZ get_rotation�r r rk rr rs r] r r r �create_collection} s �z.HandlerRegularPolyCollection.create_collectionc C sx | � ||||||�\} } | �||||||�} | �|||||||�} | j|| tt| | ��|d�} | �| ||� || _| gS �N)rr rs )r@ rH rm rv �list�zipr Z _transOffset)r r r r r r r! r" r) r> r? rG rk r] r r r r$ � s, � � � �z+HandlerRegularPolyCollection.create_artists)NN) r* r+ r, r- r r7 rm r rv r$ r r r r ri P s ri c @ s e Zd ZdZdd� ZdS )�HandlerPathCollectionzM Handler for `.PathCollections`, which are used by `~.Axes.scatter`. c C s" t |�|�� d g|||d�}|S )Nr )rk rr rs )rt Z get_pathsru r r r rv � s �z'HandlerPathCollection.create_collectionN�r* r+ r, r- rv r r r r rz � s rz c @ s e Zd ZdZdd� ZdS )�HandlerCircleCollectionz+ Handler for `.CircleCollections`. c C s t |�|||d�}|S rw )rt ru r r r rv � s �z)HandlerCircleCollection.create_collectionNr{ r r r r r| � s r| c @ s* e Zd ZdZd dd�Zdd� Zd d � ZdS ) �HandlerErrorbarz Handler for Errorbars. r: Nr/ c K s( || _ || _tj| f||d�|�� d S )NrJ )� _xerr_size� _yerr_sizerI r )r � xerr_size� yerr_sizer3 r4 r5 r r r r � s ��zHandlerErrorbar.__init__c C s, | j | }| jd kr|}n | j| }||fS r )r~ r ) r r r r r r! r" r� r� r r r � get_err_size� s z HandlerErrorbar.get_err_sizec s� |\} } } | � ||||||�\} } || d t�| jt� }t| |�}t�| �} t�|d t| �� �}| �||||||�\� �t| |�}| d kr�|� d� |� d� n^| � || |� |� d� |� d� | � || |� |� d� |jdk�r|�� |j }|�|� g }g }|j�r�� fdd�t| |�D �}t�|�}| � || d |� |�|� | �r�t| � |�}t| � |�}| � || d |� | � || d |� |� d � |� d � |�|� |�|� |j�rt�fd d�t| |�D �}t�|�}| � || d |� |�|� | �rtt| |� �}t| |� �}| � || d |� | � || d |� |� d � |� d � |�|� |�|� g }|�|� |�|� |�|� |�|� |D ]}|�|� �q�|S ) Nre FrK rM r8 c s( g | ] \}}|� |f|� |ff�qS r r ��.0�x�y)r� r r � <listcomp>� s �z2HandlerErrorbar.create_artists.<locals>.<listcomp>r �|c s( g | ] \}}||� f||� ff�qS r r r� )r� r r r� s ��_)r@ r; rf rg rh r rE rP r� Z set_visibler rN rO rQ rR rS rT Zhas_xerrry �mcoll�LineCollection�appendZhas_yerr�extendrU )r r r r r r r! r" r) Z plotlinesZcaplinesZ barlinecolsr> r? rG rV Z ydata_markerrW rX Zhandle_barlinecolsZhandle_caplinesZvertsZcollZ capline_leftZ capline_rightr% �artistr )r� r� r r$ � s� � � � � z HandlerErrorbar.create_artists)r: Nr/ N)r* r+ r, r- r r� r$ r r r r r} � s � r} c @ s2 e Zd ZdZd dd�Zdd� Zdd � Zd d � ZdS ) � HandlerStemz6 Handler for plots produced by `~.Axes.stem`. r/ Nc K s$ t j| f|||d�|�� || _dS )a� Parameters ---------- marker_pad : float Padding between points in legend entry. Default is 0.3. numpoints : int, optional Number of points to show in legend entry. bottom : float, optional yoffsets : array of floats, optional Length *numpoints* list of y offsets for each point in legend entry. Notes ----- Any other keyword arguments are given to `HandlerNpointsYoffsets`. )r3 r4 rC N)rA r �_bottom)r r3 r4 �bottomrC r5 r r r r ) s ��zHandlerStem.__init__c C s2 | j d kr |d|j d }n|t�| j � }|S )Nr: rD rF r r r rH E s zHandlerStem.get_ydatac sJ |\} } } t | tj�} | �||||||�\} }| �||||||�}| jd krRd� n| j� t||d t|�� �}| �|| |� � fdd�t ||�D �}| r�| j }| j | _ |D ]}| �|| |� q�n"t || �D ]\}}| �|||� q�| r�|| _ tt � | �t �| �g� � g�}| �|| |� |}|�|� |�|� |D ]}|�|� �q4|S )Nr c s" g | ]\}}t ||g� |g��qS r r r� �r� r r r� d s �z.HandlerStem.create_artists.<locals>.<listcomp>)� isinstancer� r� r@ rH r� r rP r ry r �_copy_collection_propsr; ro rn r� rU )r r r r r r r! r" r) Z markerlineZ stemlinesZbaselineZusing_linecollr> r? rG Zleg_markerlineZ leg_stemlinesZorig_update_func�lineZlm�mZ leg_baseliner% r� r r� r r$ M sN � � �� zHandlerStem.create_artistsc C s( |� |�� d � |�|�� d � dS )zt Method to copy properties from a LineCollection (orig_handle) to a Line2D (legend_handle). r N)rc Z get_colorrQ Z get_linestyler r r r r� � s z"HandlerStem._copy_collection_props)r/ NNN)r* r+ r, r- r rH r$ r� r r r r r� % s � 7r� c @ s" e Zd ZdZddd�Zdd� ZdS ) � HandlerTuplea� Handler for Tuple. Additional kwargs are passed through to `HandlerBase`. Parameters ---------- ndivide : int, optional The number of sections to divide the legend area into. If None, use the length of the input tuple. Default is 1. pad : float, optional If None, fall back to ``legend.borderpad`` as the default. In units of fraction of font size. Default is None. r8 Nc K s || _ || _tj| f|� d S r )�_ndivide�_padr r )r �ndivider= �kwargsr r r r � s zHandlerTuple.__init__c C s� |� � } | jd kr t|�} n| j} | jd kr8|j| } n | j| } | dkr^|| | d | }t||| t�| � �} g } |D ]6}|�| |�}|� ||t | �|||||�}| � |� q�| S )Nr8 ) Zget_legend_handler_mapr� rP r� Z borderpadr r; ZarangeZget_legend_handlerr$ �nextr� )r r r r r r r! r" r) Z handler_mapr� r= Z xds_cycle�a_listZhandle1ZhandlerZ_a_listr r r r$ � s2 � zHandlerTuple.create_artists)r8 NrY r r r r r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )�HandlerPolyCollectionza Handler for `.PolyCollection` used in `~.Axes.fill_between` and `~.Axes.stackplot`. c C s� dd� }dd� }t |d|�� �}|�||�� t |d|�� �}|�||�� |�|�� � |�|�� � |� ||� � �� |� ||� � �� |� ||�� �� |�|�� � |�|�� � d S )Nc S s. | d kr d S t �| �} t| �r&| d S dS d S )Nr Znone)�mcolorsZ to_rgba_arrayrP )�colorsr r r � first_color� s z7HandlerPolyCollection._update_prop.<locals>.first_colorc S s t | �r| d S d S d S r )rP )Z prop_arrayr r r � get_first� s z5HandlerPolyCollection._update_prop.<locals>.get_firstZ_original_edgecolorZ_original_facecolor)�getattrZ get_edgecolorZ set_edgecolorZ get_facecolorZ set_facecolorZset_fillZget_fillZ set_hatchZ get_hatchrd rb rQ Zget_linestylesrU Zget_transformsrq Z get_figureZ set_alphaZ get_alpha)r r r r� r� Z edgecolorZ facecolorr r r r � s" ��z"HandlerPolyCollection._update_propc C s4 t | | f||d�} | �| ||� | �|� | gS )Nr\ )r r rU r_ r r r r$ � s � z$HandlerPolyCollection.create_artistsN)r* r+ r, r- r r$ r r r r r� � s r� ) r- � itertoolsr Znumpyr; Zmatplotlib.linesr Zmatplotlib.patchesr Zmatplotlib.collections� collectionsr� Zmatplotlib.colorsr� r� r �objectr r. rA rI rZ r` ri rz r| r} r� r� r� r r r r �<module> s( Y, 4. K qh7