OwlCyberSecurity - MANAGER
Edit File: _constrained_layout.cpython-38.pyc
U ��]�u � @ s� d Z ddlZddlZddlmZ ddlmZ e� e �Z dd� Z dd� Z dd� Zdd d �Zd d � Zd d� Zdd� Zddd�Zd dd�Zdd� Zd dd�ZdS ) a� This module provides the routine to adjust subplot layouts so that there are no overlapping axes or axes decorations. All axes decorations are dealt with (labels, ticks, titles, ticklabels) and some dependent artists are also dealt with (colorbar, suptitle, legend). Layout is done via :meth:`~matplotlib.gridspec`, with one constraint per gridspec, so it is possible to have overlapping axes if the gridspecs overlap (i.e. using :meth:`~matplotlib.gridspec.GridSpecFromSubplotSpec`). Axes placed using ``figure.subplots()`` or ``figure.add_subplots()`` will participate in the layout. Axes manually placed via ``figure.add_axes()`` will not. See Tutorial: :doc:`/tutorials/intermediate/constrainedlayout_guide` � Nc C s0 || ko|kn p.|| ko*|kS S �N� )� colnum0min� colnum0max� colnumCmin� colnumCmaxr r �@/usr/lib/python3/dist-packages/matplotlib/_constrained_layout.py�_in_same_column: s �r c C s0 || ko|kn p.|| ko*|kS S r r )� rownum0min� rownum0max� rownumCmin� rownumCmaxr r r � _in_same_row? s �r c C sB | j D ]6}|jdk r|j�� }|d dks6|d dkr dS qdS )zw helper function to make sure all axes in the figure have a finite width and height. If not, return False N� r � FT)�axes� _layoutbox� _poslayoutbox�get_rect)�fig�ax�newposr r r �_axes_all_finite_sizedD s r c C s | j �� j}t� }| jD ].}t|d�r|�� �� } | jdk r|� | � qt |�dkr^t � d� | jj dk r~|D ]} t| | � qntd�D �]�} | jD ](}t�|j� |jdk r�t||||� q�| j} | dk o�| jdk o�| �� } | �r || j|d��} | j| j }t�|��r | j�||d � | jj dk �rn| j}|jD ] }|� � �r8t |||d� �q8|D ]} t | | � �q\| j j d7 _ | j� � t | ��r| jD ]*}|jdk �r�|j!�"� }|j#|d d � �q�| �r�| j�"� }| �$d | � n"| dk �r| jdk �r| j�d� q�t � d � q�dS ) a� Do the constrained_layout. Called at draw time in ``figure.constrained_layout()`` Parameters ---------- fig : Figure is the ``figure`` instance to do the layout in. renderer : Renderer the renderer to use. h_pad, w_pad : float are in figure-normalized units, and are a padding around the axes elements. hspace, wspace : float are in fractions of the subplot sizes. �get_subplotspecNr zgThere are no gridspecs with layoutboxes. Possibly did not call parent GridSpec with the figure= keyword� r ��renderer��hspace�wspace�original)Zwhichg �?zUconstrained_layout not applied. At least one axes collapsed to zero width or height.)%� transFigure�inverted�transform_bbox�setr �hasattrr � get_gridspecr �add�len�cbookZ_warn_external�constrained_layout_called�_make_ghost_gridspec_slots�range�_log�debug�_make_layout_marginsZ _suptitleZ get_in_layoutZget_window_extent�y1�y0�np�isfiniteZ edit_height�children�_is_gridspec_layoutbox�_arrange_subplotspecs� _align_spinesZupdate_variablesr r r Z _set_positionZset_y)r r �h_pad�w_padr r � invTransFigZgssr �gsZnnnZsuptitleZ do_suptitle�bbox�heightZfiglb�childr r r r �do_constrained_layoutR sd A �� � r? c C s� |� � \}}tj|| td�}g }| jD ]2}t|d�r(|jdk r(|�� �� |kr(||g7 }q(|D ]2}|�� }|j dkr~|j |_ d||j |j d �< q`t |�D ]B\}} | s�| � || �}|� d� |�g � |�g � |�d� q�dS )a+ Check for unoccupied gridspec slots and make ghost axes for these slots... Do for each gs separately. This is a pretty big kludge but shouldn't have too much ill effect. The worst is that someone querying the figure will wonder why there are more axes than they thought. �Zdtyper NTr F)r r r r )� get_geometryr2 �zeros�boolr r% r r r& �num2�num1� enumerateZ add_subplotZ set_frame_onZ set_xticksZ set_yticksZ set_facecolor) r r; �nrows�ncolsZhassubplotspec�axsr �ss0�nnZhssr r r r+ � s, �� r+ c C s� | j }|j�� j}| jdd�}| j|d�}|dkr8|}n||�}t�|j�rXt�|j �s\dS | j j } | dkrp|} | j j } | dkr�|} | j � |j |j | � | j �|j|j | � | j �|j |j | � | j �|j|j | � t�d|j |j | � t�d|j|j | � t�d|j |j | � t�d|j� t�d |j� |jjd k �r�| j jd d d � | j jd d d � | jjd d d � | jjd d d � | j jdd d � | j j dd d � | j j dd d � | j j dd d � dS )z� For each axes, make a margin between the *pos* layoutbox and the *axes* layoutbox be a minimum size that can accommodate the decorations on the axis. T)r r Nzleft %fzright %fz bottom %fz bbox.y0 %fz pos.y0 %fr � Zweak��strengthr ) Zfigurer! r" r# Z get_positionZ get_tightbboxr2 r3 �widthr= r r8 r9 Zedit_left_margin_minZx0Zedit_right_margin_minZx1Zedit_bottom_margin_minr1 Zedit_top_margin_minr0 r- r. r r* �constrain_height_min�constrain_width_minZconstrain_top_marginZconstrain_bottom_marginZconstrain_right_marginZconstrain_left_margin) r r r8 r9 r r: �posZ tightbboxr< Zh_padtZw_padtr r r r/ sZ �� ����r/ c ) C s� |� � \}}|�� }|�� }|dkr.t�|�}|dkr@t�|�}g }| jD ]2}t|d�rJ|jdk rJ|�� � � |krJ||g7 }qJtj t |�tj d�}tj t |�tj d�} tj t |�tj d�} tj t |�tj d�} t� t |��} t� t |��} t |�D ]�\}}|�� }|jdk�r|j|_t|j|�\||< | |< t|j|�\| |< | |< t�|| | | | d � �| |< t�||| | | d � �| |< q�t |dd� �D �]\}}|| | | }}| | | | }}| | | | }}d}d}d}d}d}d} t|d t |��D �]�} || } || | | } } | | | | }!}"| | | | }#}$|�sr|| k�rrt�|j| jgd� d}|�s�||"k�r�t�|j| jgd � d}|�s�|| k�r�t�d � t�|j| jgd � d}|�s�||!k�r�t�d � t�|j| jgd � d}|!| d }%|| d }&|"| d }'|| d }(|�s\|&|%k�r\|j�| jj| |$ � d}n�t||| |"��r�||$k�r�|j�| jj| |$ � | j�|jj|$ |d � n>||$k �r�| j�|jj|$ | � |j�|jj| |$d � | �s|(|'k�r|j� | jj | |# � d} n�t ||| |!��r||#k�rh|j� | jj | |# � | j� |jj |# |d � n>||#k �r| j� |jj |# | � |j� | jj | |#d � �q�q�dS )z� - Align right/left and bottom/top spines of appropriate subplots. - Compare size of subplotspec including height and width ratios and make sure that the axes spines are at least as large as they should be. Nr r@ r ���F�leftT�rightzrownum0min == rownumCmin�topzrownum0max == rownumCmax�bottomg������?) rA Zget_width_ratiosZget_height_ratiosr2 Zonesr r% r r r& rB r( Zint8rF rD rE �divmod�sumr, � layoutbox�alignr r- r. �constrain_heightr= r rP �constrain_widthrO r rQ ))r r; rG rH Z width_ratiosZ height_ratiosrI r Z rownumminZ rownummaxZ colnumminZ colnummaxrO r= �nrJ rK r r r r Zwidth0Zheight0Z alignleftZ alignrightZalignbotZaligntopZ alignheightZ alignwidthZmmZaxcr r r r ZwidthCZheightCZdrowsCZdrows0ZdcolsCZdcols0r r r r7 H s � �� ���� � � � � � �� � ��� � � �� � ��r7 c C sx g }| j D ]8}|�� r |j D ]}|�� r t|||d� q ||g7 }q |D �](}|j}|�� �� \}} |jdkrv|j|_t |j| �\} } t |j| �\} } |dd� }|D ]�}|j}t |j| �\}}|jdkr�|j|_t |j| �\}}|| }| |k �r t j |j |j g|d� || k �r*t j |j |j g|d� || }| |k �rRt j |j |j g|d� || k r�t j |j |j g|d� q�qHdS )z� arrange the subplotspec children of this gridspec, and then recursively do the same of any gridspec children of those gridspecs... r Nr ��padding)r4 Z_is_subplotspec_layoutboxr5 r6 �artistr& rA rD rE rX rZ �hstackr �vstack)r; r r Z sschildrenr> Zchild2Zchild0rJ rG rH Z rowNum0minZ colNum0minZ rowNum0maxZ colNum0maxZchildcZsscZ rowNumCminZ colNumCminZ rowNumCmaxZ colNumCmaxZthepadr r r r6 � sV � � ����r6 皙�����?c C sx | j }| j}| �� j }tj||jd |d�} |dkr�tj| | jd ddd|d�} |dkrttj|| g||j d d � ntj| |g||j d � t�|| gd |g� t� || gd � | j ||j d | d d� n�|dk�rptj| | jd ddd|d�} |dk�rtj || g||j d � ntj | |g||j d � tj || gd |gd d� t� || gd� | j|j| | dd� | | fS )�~ Do the layout for a colorbar, to not overly pollute colorbar.py `pad` is in fraction of the original axis size. �.cbar��parent�namera �rT rU �.posFT�rh ri Z tightwidthrR �subplotra rU �strong�r` rN r_ r Zv_centerrM �rW rV �rh ri Z tightheightrR rm ra rW Zh_center�medium)r r r rZ � LayoutBoxri rb rO Z match_heightsr[ r] r= rc Z match_widthsr\ ) r �cax�shrink�aspect�location�padZaxlbZaxposZaxsslb�lb�lbposr r r �layoutcolorbarsingle s` ���� � ��r{ c C s� d}d}d }d }d}d}d }d }| D ]`} | � � } | �� \} } } }}}||krT|}| }| |k rd| }| }||krt|}| }||k r$|}| }q$||||||||fS )Ni`y��i@B )r �get_rows_columns)rI �maxrow�minrowZmaxaxZminax�maxcol�mincol� maxax_col� minax_colr �subspecrG rH � row_start�row_stop� col_start�col_stopr r r �_getmaxminrowcolumnR s2 �r� c C s$ | d � � �� }|j}tj|j|jjd |d�}t| �\} } } } } }}}|dk�r�tj||jd ddd|d�}| D ]8}|d kr�|j|g}n ||jg}tj|||j d d � qr|j D ]�}|j }|� � \}}}}}}|d kr�||kr�|j|g}||kr�||jg}n2|d k�r0|| k�r ||jg}|| k �r0|j|g}tj|||j d d � q�| j } | j } |j| j| j | d d � |j| j| j d| d | j d d � |j|j|| d d � �nj|dk�r tj||jd ddd|d�}| D ]<}|dk�r�|j|g}n ||jg}tj|||j d d � �q�|j D ]�}|j }|� � \}}}}}}|dk�rr|| k�r\|j|g}|| k�r�||jg}n2|dk�r�|| k �r�|j|g}|| k�r�||jg}tj|||j d d � �q |j } |j } |�| j| j | � |�| j| j d| d | j � |j|j | | dd � ||fS )re r rf rg rj rk FTrl rU rn ro rT rM r r rp rq rW rV rr )r r& r rZ rs rh ri r� rb rO r4 ra r| r r\ rV rW Zconstrain_bottomr] r= rc rU rT Zconstrain_left) �parentsrt ru rv rw rx r; Zgslbry r~ r} Z minax_rowZ maxax_rowr� r r� r� rz r �orderZchr� rG rH r� r� r� r� ZmaxposlbZminposlbr r r �layoutcolorbargridspecp s� �� � � � � �� ����� � � � � � �����r� )NN)r r )rd )rd )�__doc__ZloggingZnumpyr2 Zmatplotlib.cbookr) Zmatplotlib._layoutboxr rZ Z getLogger�__name__r- r r r r? r+ r/ r7 r6 r{ r� r� r r r r �<module> s( / � 7 3 ;