OwlCyberSecurity - MANAGER
Edit File: axes_divider.cpython-38.pyc
U ��]�s � @ s6 d Z ddlmZ ddlmZ ddlmZmZ ddlm Z ddl m Z G dd � d e�ZG d d � d e�ZG d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�Zejddd�G dd� de��Zejddd�dd� �Zd d � Zd(d d �Zdd!lmZ ejdd"d#�G d$d%� d%e ��Zejdd"d#�G d&d'� d'e ��Z dS ))a The axes_divider module provides helper classes to adjust the positions of multiple axes at drawing time. Divider: this is the class that is used to calculate the axes position. It divides the given rectangular area into several sub rectangles. You initialize the divider by setting the horizontal and vertical lists of sizes that the division will be based on. You then use the new_locator method, whose return value is a callable object that can be used to set the axes_locator of the axes. � )�cbook)� SubplotBase)� SubplotSpec�GridSpecN� )� axes_sizec @ s� e Zd ZdZd4dd�Zdd� Zdd � Zd d � Zed d � �Z edd� �Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd5d!d"�Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd6d+d,�Zd7d-d.�Zd/d0� Zd8d2d3�ZdS )9�Dividera� This class calculates the axes position. It divides the given rectangular area into several sub-rectangles. You initialize the divider by setting the horizontal and vertical lists of sizes (:mod:`mpl_toolkits.axes_grid.axes_size`) that the division will be based on. You then use the new_locator method to create a callable object that can be used as the axes_locator of the axes. N�Cc C s: || _ || _|| _|| _|| _|| _d| _d| _d| _dS )a� Parameters ---------- fig : Figure pos : tuple of 4 floats position of the rectangle that will be divided horizontal : list of :mod:`~mpl_toolkits.axes_grid.axes_size` sizes for horizontal division vertical : list of :mod:`~mpl_toolkits.axes_grid.axes_size` sizes for vertical division aspect : bool if True, the overall rectangular area is reduced so that the relative part of the horizontal and vertical scales have the same scale. anchor : {'C', 'SW', 'S', 'SE', 'E', 'NE', 'N', 'NW', 'W'} placement of the reduced rectangle when *aspect* is True r N) �_fig�_pos� _horizontal� _vertical�_anchor�_aspect� _xrefindex� _yrefindex�_locator)�self�fig�pos� horizontal�vertical�aspect�anchor� r �F/usr/lib/python3/dist-packages/mpl_toolkits/axes_grid1/axes_divider.py�__init__ s zDivider.__init__c s � fdd�| � � D �S )Nc s g | ]}|� � ��qS r �Zget_size��.0�s��rendererr r � <listcomp>? s z0Divider.get_horizontal_sizes.<locals>.<listcomp>)�get_horizontal�r r" r r! r �get_horizontal_sizes> s z Divider.get_horizontal_sizesc s � fdd�| � � D �S )Nc s g | ]}|� � ��qS r r r r! r r r# B s z.Divider.get_vertical_sizes.<locals>.<listcomp>)� get_verticalr% r r! r �get_vertical_sizesA s zDivider.get_vertical_sizesc C s, ddl m} || �� �}|| �� �}||fS )Nr )�AddList)r r) r' r$ )r r) ZvsizeZhsizer r r �get_vsize_hsizeD s zDivider.get_vsize_hsizec C sF d\}}| D ]\}}||7 }||7 }q |dkr>|| | }|S dS d S )N)� r+ r+ r )�lZ total_sizeZrs_sumZas_sum�_rs�_as�kr r r �_calc_kM s zDivider._calc_kc C s2 dg}| D ]"\}}|� |d || | � q |S �Nr+ ���)�append)r, r/ �offsetsr- r. r r r � _calc_offsets\ s zDivider._calc_offsetsc C s || _ dS )z� set the position of the rectangle. Parameters ---------- pos : tuple of 4 floats position of the rectangle that will be divided N�r )r r r r r � set_positionh s zDivider.set_positionc C s | j S )z%return the position of the rectangle.r6 �r r r r � get_positions s zDivider.get_positionc C s: |t jjkst|�dkr || _ntdd�t jj� ��dS )a� Parameters ---------- anchor : {'C', 'SW', 'S', 'SE', 'E', 'NE', 'N', 'NW', 'W'} anchor position ===== ============ value description ===== ============ 'C' Center 'SW' bottom left 'S' bottom 'SE' bottom right 'E' right 'NE' top right 'N' top 'NW' top left 'W' left ===== ============ � zargument must be among %sz, N)� mtransforms�BboxZcoefs�lenr � ValueError�joinZBBox)r r r r r � set_anchorw s �zDivider.set_anchorc C s | j S )zreturn the anchor)r r8 r r r � get_anchor� s zDivider.get_anchorc C s || _ dS )z� Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid.axes_size` sizes for horizontal division N�r )r �hr r r �set_horizontal� s zDivider.set_horizontalc C s | j S )zreturn horizontal sizesrB r8 r r r r$ � s zDivider.get_horizontalc C s || _ dS )z� Parameters ---------- v : list of :mod:`~mpl_toolkits.axes_grid.axes_size` sizes for vertical division N�r )r �vr r r � set_vertical� s zDivider.set_verticalc C s | j S )zreturn vertical sizesrE r8 r r r r' � s zDivider.get_verticalFc C s || _ dS )zE Parameters ---------- aspect : bool N�r �r r r r r � set_aspect� s zDivider.set_aspectc C s | j S )z return aspectrH r8 r r r � get_aspect� s zDivider.get_aspectc C s || _ d S �N�r )r r r r r � set_locator� s zDivider.set_locatorc C s | j S rL rM r8 r r r � get_locator� s zDivider.get_locatorc C s$ | j d kr| �� S | � ||�jS d S rL )r r9 �bounds)r �axr" r r r �get_position_runtime� s z Divider.get_position_runtimec C s� | j �� \}}| �||�\} } } } | �|�} | �|�}| �| || �}| �||| �}| �� r�t||�}| �| |�}| �||�}|d |d | }|d |d | }t j � | | | | �}t j � | | ||�}|� | � � |�}|j|j }}n"| �| |�}| �||�}| | }}|dk�r |d }|dk�r2|d }||| | || || | }} ||| | || || | } } t j � || | | �S )a� Parameters ---------- nx, nx1 : int Integers specifying the column-position of the cell. When *nx1* is None, a single *nx*-th column is specified. Otherwise location of columns spanning between *nx* to *nx1* (but excluding *nx1*-th column) is specified. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. axes renderer r2 r Nr )r �get_size_inchesrR r& r( r0 rK �minr5 r; r< � from_bounds�anchoredrA �x0�y0) r �nx�ny�nx1�ny1�axesr" �figW�figH�x�y�wrC ZhsizesZvsizesZk_hZk_vr/ �ox�oy�ww�hh�pb�pb1� pb1_anchoredrW rX �x1�w1�y1�h1r r r �locate� s4 &&zDivider.locatec C s t | ||||�S )a Returns a new locator (:class:`mpl_toolkits.axes_grid.axes_divider.AxesLocator`) for specified cell. Parameters ---------- nx, nx1 : int Integers specifying the column-position of the cell. When *nx1* is None, a single *nx*-th column is specified. Otherwise location of columns spanning between *nx* to *nx1* (but excluding *nx1*-th column) is specified. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. �� AxesLocator)r rY rZ r[ r\ r r r � new_locator� s zDivider.new_locatorc C s� |dkr&| j �d|� | jd7 _nh|dkr<| j �|� nR|dkrb| j�d|� | jd7 _n,|dkrx| j�|� ntjddddg|d� d S )N�leftr r �right�bottom�top��position)r �insertr r3 r r r �_check_in_list)r rw �sizer r r � append_size s �zDivider.append_size皙�����?c C s^ |d krddddg}ddl m}m}m} |D ],}|||�}||�} || |�} | �|| � q,d S )Nrr rs rt ru r )�Padded� SizeFromFunc�GetExtentHelper)r r} r~ r r{ ) r �use_axes�pad� adjust_dirsr} r~ r �d�helperrz Z padded_sizer r r �add_auto_adjustable_area s z Divider.add_auto_adjustable_area)Nr )F)NNNN)NN)r| N) �__name__� __module__� __qualname__�__doc__r r& r( r* � staticmethodr0 r5 r7 r9 r@ rA rD r$ rG r' rJ rK rN rO rR rn rq r{ r� r r r r r s< � 2 �r c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS ) rp z| A simple callable object, initialized with AxesDivider class, returns the position and size of the given cell. Nc C s` || _ |j}|j}|| || | _| _|dkr8|d }|dkrH|d }|| | _|| | _dS )a� Parameters ---------- axes_divider : AxesDivider nx, nx1 : int Integers specifying the column-position of the cell. When *nx1* is None, a single *nx*-th column is specified. Otherwise location of columns spanning between *nx* to *nx1* (but excluding *nx1*-th column) is specified. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. Nr )� _axes_dividerr r �_nx�_ny�_nx1�_ny1)r Z axes_dividerrY rZ r[ r\ r r r r r r / s zAxesLocator.__init__c C s> | j j}| j j}| j �| j| | j| | j| | j| ||�S rL )r� r r rn r� r� r� r� )r r] r" r r r r r �__call__K s �zAxesLocator.__call__c C s t | jd�r| j�� S d S d S �N�get_subplotspec)�hasattrr� r� r8 r r r r� W s zAxesLocator.get_subplotspec)NN)r� r� r� r� r r� r� r r r r rp * s rp c @ sT e Zd ZdZddddd�dd�Zdd� Zd d � Zd d � Zd d� Zdd� Z dd� Z dS )�SubplotDivider�T The Divider class whose rectangle area is specified as a subplot geometry. Nr )r r r r c G sR || _ t|�dkr�t|d t�r,|d | _nVz$tt|d ��}tt|�\}} } W n tk rl td��Y nX t || �| d | _n�t|�dk�r |\}} } t|�}t| �} t| t �r�t| �dkr�dd� | D �} t || �| d d | d � | _nt || �t| �d | _ntd|� ���| � � | j j } tj| || |�p<g |�pDg ||d � d S ) a, Parameters ---------- fig : :class:`matplotlib.figure.Figure` *args : tuple (*numRows*, *numCols*, *plotNum*) The array of subplots in the figure has dimensions *numRows*, *numCols*, and *plotNum* is the number of the subplot being created. *plotNum* starts at 1 in the upper left corner and increases to the right. If *numRows* <= *numCols* <= *plotNum* < 10, *args* can be the decimal integer *numRows* * 100 + *numCols* * 10 + *plotNum*. r r z4Single argument to subplot must be a 3-digit integer� r: c S s g | ] }t |��qS r )�int)r �nr r r r# � s z+SubplotDivider.__init__.<locals>.<listcomp>z Illegal argument(s) to subplot: )r r N)�figurer= � isinstancer � _subplotspec�strr� �mapr> r �tuple� update_params�figboxrP r r ) r r r r r r �argsr �rows�cols�numr r r r r c s6 � " �zSubplotDivider.__init__c C s | � � | jjS )z$return the bounds of the subplot box)r� r� rP r8 r r r r9 � s zSubplotDivider.get_positionc C s | � � �| j�| _dS )z0update the subplot position from fig.subplotparsN)r� r9 r� r� r8 r r r r� � s z SubplotDivider.update_paramsc C s" | � � �� \}}}}|||d fS )z%get the subplot geometry, e.g., 2,2,3r )r� � get_geometry)r r� r� Znum1Znum2r r r r� � s zSubplotDivider.get_geometryc C s, t ||�|d | _| �� | �| j� dS )z2change subplot geometry, e.g., from 1,1,1 to 2,2,3r N)r r� r� r7 r� )r ZnumrowsZnumcolsr� r r r �change_geometry� s z SubplotDivider.change_geometryc C s | j S )z get the SubplotSpec instance�r� r8 r r r r� � s z SubplotDivider.get_subplotspecc C s || _ dS )z set the SubplotSpec instanceNr� )r Z subplotspecr r r �set_subplotspec� s z SubplotDivider.set_subplotspec) r� r� r� r� r r9 r� r� r� r� r� r r r r r� ^ s � >%r� c @ sf e Zd ZdZddd�Zdd�dd�Zdd d �Zdd d �Zddd�Zdd� Z dd� Z dd� Z dd� Z dS ) � AxesDividerz1 Divider based on the pre-existing axes. Nc C sf || _ |dkr t�|�| _n|| _|dkr8t�|�| _n|| _tj| |�� d| jg| jgddd� dS )zw Parameters ---------- axes : :class:`~matplotlib.axes.Axes` xref yref Nr )r r r r r r ) �_axes�SizeZAxesX�_xrefZAxesY�_yrefr r � get_figure)r r] ZxrefZyrefr r r r � s �zAxesDivider.__init__�� axes_classc K sB | j }|d kr(t|t�r |j}nt|�}||�� |jdd�f|�S �NT)Zoriginal)r� r� r Z _axes_class�typer� r9 )r r� �kwargsr] r r r � _get_new_axes� s �zAxesDivider._get_new_axesFc K s� |rNt |tj�s tj|| jd�}|rB| j�d|� | jd7 _n | j�|� t |tj�sjtj|| jd�}|r�| j�d|� | jd7 _| j d| j d�}n&| j�|� | j t | j�d | j d�}| j f |�}|� |� |S )a Add a new axes on the right (or left) side of the main axes. Parameters ---------- size : :mod:`~mpl_toolkits.axes_grid.axes_size` or float or string A width of the axes. If float or string is given, *from_any* function is used to create the size, with *ref_size* set to AxesX instance of the current axes. pad : :mod:`~mpl_toolkits.axes_grid.axes_size` or float or string Pad between the axes. It takes same argument as *size*. pack_start : bool If False, the new axes is appended at the end of the list, i.e., it became the right-most axes. If True, it is inserted at the start of the list, and becomes the left-most axes. kwargs All extra keywords arguments are passed to the created axes. If *axes_class* is given, the new axes will be created as an instance of the given class. Otherwise, the same class of the main axes will be used. �Z fraction_refr r �rY rZ )r� r� �_Base�from_anyr� r rx r r3 rq r r= r� �set_axes_locator�r rz r� � pack_startr� �locatorrQ r r r �new_horizontal s* � zAxesDivider.new_horizontalc K s� |rNt |tj�s tj|| jd�}|rB| j�d|� | jd7 _n | j�|� t |tj�sjtj|| jd�}|r�| j�d|� | jd7 _| j | j dd�}n&| j�|� | j | j t | j�d d�}| j f |�}|� |� |S )a Add a new axes on the top (or bottom) side of the main axes. Parameters ---------- size : :mod:`~mpl_toolkits.axes_grid.axes_size` or float or string A height of the axes. If float or string is given, *from_any* function is used to create the size, with *ref_size* set to AxesX instance of the current axes. pad : :mod:`~mpl_toolkits.axes_grid.axes_size` or float or string Pad between the axes. It takes same argument as *size*. pack_start : bool If False, the new axes is appended at the end of the list, i.e., it became the right-most axes. If True, it is inserted at the start of the list, and becomes the left-most axes. kwargs All extra keywords arguments are passed to the created axes. If *axes_class* is given, the new axes will be created as an instance of the given class. Otherwise, the same class of the main axes will be used. r� r r r� )r� r� r� r� r� r rx r r3 rq r r= r� r� r� r r r � new_vertical1 s$ zAxesDivider.new_verticalTc K s� |dkr"| j ||fddi|��}n||dkrD| j ||fddi|��}nZ|dkrf| j||fddi|��}n8|dkr�| j||fddi|��}ntjddddg|d� |r�| j�|� |S ) z� Create an axes at the given *position* with the same height (or width) of the main axes. *position* ["left"|"right"|"bottom"|"top"] *size* and *pad* should be axes_grid.axes_size compatible. rr r� Trs Frt ru rv )r� r� r ry r Zadd_axes)r rw rz r� Z add_to_figurer� rQ r r r � append_axes\ s � zAxesDivider.append_axesc C s0 | j d kr&| j�� }|dkr dS dS n| j S d S )N�autoFT)r r� rK rI r r r rK v s zAxesDivider.get_aspectc C s( | j d kr | jjdd�}|jS | j S d S r� )r r� r9 rP )r Zbboxr r r r9 � s zAxesDivider.get_positionc C s | j d kr| j�� S | j S d S rL )r r� rA r8 r r r rA � s zAxesDivider.get_anchorc C s t | jd�r| j�� S d S d S r� )r� r� r� r8 r r r r� � s zAxesDivider.get_subplotspec)NN)NF)NF)NT) r� r� r� r� r r� r� r� r� rK r9 rA r� r r r r r� � s , + r� c @ sH e Zd Zdd� Zedd� �Zedd� �Zddd �Zd d � Zdd d �Z dS )� HBoxDividerc O s t j| |f|�|� d S rL )r� r )r r r� r� r r r r � s zHBoxDivider.__init__c C s t | �}dd l}|�|j|d |d fdd��}|j|d dd�}t| �D ].\}\} } | |||f< d||df< | ||< qJdd� |D �|dd d�f< |tdd� |D �� |d< |j|�|�j j} | d d� } | d } | |k�r ||� d d� | D �� |� d d� | D �� } | S ) Nr r r� )Zdtyper2 c S s g | ] \}}|�qS r r �r �r�ar r r r# � s z1HBoxDivider._determine_karray.<locals>.<listcomp>c S s g | ] \}}|�qS r r )r Zrsr� r r r r# � s c S s g | ] \}}|�qS r r r� r r r r# � s c S s g | ] \}}|�qS r r r� r r r r# � s ) r= ZnumpyZmatZzeros� enumerate�sum�I�TZA1Zarray)�equivalent_sizes�appended_sizes�max_equivalent_size�total_appended_sizer� Znp�A�B�ir� r� Zkarray_H�karray�Hr r r �_determine_karray� s( ��z HBoxDivider._determine_karrayc C s< dg}t | |�D ]&\\}}}|�|d || | � q|S r1 )�zipr3 )r� r� r4 r� r� r/ r r r r5 � s zHBoxDivider._calc_offsetsNc C s t | |d|d�S )a Create a new `~mpl_toolkits.axes_grid.axes_divider.AxesLocator` for the specified cell. Parameters ---------- nx, nx1 : int Integers specifying the column-position of the cell. When *nx1* is None, a single *nx*-th column is specified. Otherwise location of columns spanning between *nx* to *nx1* (but excluding *nx1*-th column) is specified. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. r Nro )r rY r[ r r r rq � s zHBoxDivider.new_locatorc C s� |} |} || } || } | � | | | | �} | �| | �}|d |d | }| d }| d |d |d | }tj�||||�}tj�||||�}|�| �� |�}|j|j }}||||fS )z� Parameters ---------- x y w h y_equivalent_sizes x_appended_sizes figW figH r2 r r ) r� r5 r; r< rU rV rA rW rX )r r` ra rb rC �y_equivalent_sizes�x_appended_sizesr^ r_ r� r� r� r� r� rc re Zref_hrf rg rh ri rW rX r r r �_locate� s" � zHBoxDivider._locatec C s� | j �� \}}| �||�\} } } } | �|�} | �|�}| �| | | | | |||�\}}}}|dkrf|d }||| | || || | }}|| }}tj�||||�S �a� Parameters ---------- axes_divider : AxesDivider nx, nx1 : int Integers specifying the column-position of the cell. When *nx1* is None, a single *nx*-th column is specified. Otherwise location of columns spanning between *nx* to *nx1* (but excluding *nx1*-th column) is specified. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. axes renderer Nr ) r rS rR r( r& r� r; r< rU )r rY rZ r[ r\ r] r" r^ r_ r` ra rb rC r� r� rW rX rc rf rj rk rl rm r r r rn � s � & zHBoxDivider.locate)N)NNNN) r� r� r� r r� r� r5 rq r� rn r r r r r� � s %r� c @ s$ e Zd ZdZddd�Zddd�ZdS ) � VBoxDividerr� Nc C s t | d|d|�S )a� Create a new `~mpl_toolkits.axes_grid.axes_divider.AxesLocator` for the specified cell. Parameters ---------- ny, ny1 : int Integers specifying the row-position of the cell. When *ny1* is None, a single *ny*-th row is specified. Otherwise location of rows spanning between *ny* to *ny1* (but excluding *ny1*-th row) is specified. r Nro )r rZ r\ r r r rq s zVBoxDivider.new_locatorc C s� | j �� \}}| �||�\} } } } | �|�} | �|�}| �| | | | | |||�\}}}}|dkrf|d }|| }}||| | || || | }}tj�||||�S r� ) r rS rR r& r( r� r; r< rU )r rY rZ r[ r\ r] r" r^ r_ r` ra rb rC Zx_equivalent_sizesZy_appended_sizesrX rW rd re rj rk rl rm r r r rn + s � &zVBoxDivider.locate)N)NNNN)r� r� r� r� rq rn r r r r r� s r� z3.0za There is no alternative. Deriving from matplotlib.axes.Axes provides this functionality already.)Zaddendumc @ s e Zd ZdS )�LocatableAxesBaseN�r� r� r� r r r r r� M s r� zg There is no alternative. Classes derived from matplotlib.axes.Axes provide this functionality already.c C s | S rL r r� r r r �locatable_axes_factoryU s r� c C s$ t | �}|jddd�}| �|� |S )Nr r� )r� rq r� )r] �dividerr� r r r �make_axes_locatable] s r� r| c C s< |d krddddg}t | �}|d kr(| }|j|||d� d S )Nrr rs rt ru )r� r� r� )r� r� )rQ r� r� r� r� r r r � make_axes_area_auto_adjustablee s �r� )�Axesz%mpl_toolkits.axes_grid1.mpl_axes.Axes)Z alternativec @ s e Zd ZdS )r� Nr� r r r r r� v s r� c @ s e Zd ZdS )� LocatableAxesNr� r r r r r� | s r� )Nr| N) r� Z matplotlibr Zmatplotlib.axesr Zmatplotlib.gridspecr r Zmatplotlib.transformsZ transformsr; � r r� �objectr rp r� r� r� r� Z deprecatedr� r� r� r� Zmpl_axesr� Z_Axesr� r r r r �<module> sL 4 5 6�� � ��