OwlCyberSecurity - MANAGER
Edit File: axes_grid.cpython-38.pyc
U ��] k � @ s� d dl mZ d dlmZ d dlmZ d dlmZ ddl m Z m Z m Z ddl mZ ddlmZ dd � Zd d � ZG d d � d e�ZG dd� dee�ZG dd� de�ZG dd� de�ZeZdS )� )�NumberN)� SubplotSpec� )�Size�SubplotDivider�Divider)�Colorbar)�Axesc C s | }t |d�s| | f}|S )N� __getitem__)�hasattr)�valueZret� r �C/usr/lib/python3/dist-packages/mpl_toolkits/axes_grid1/axes_grid.py�_extend_axes_pad s r c C s8 | }| }| j d j||d� | j d j||d� d S )N�bottom�Z ticklabelsZlabel�left)�axis�toggle)�ax� bottom_on�left_onZ bottom_offZleft_offr r r � _tick_only s r c @ s* e Zd Zdd�dd�Zdd� Zdd� ZdS ) � CbarAxesBaseN)�locatorc s� |d kr d|kr t �d�|d< |d k r@d|kr8td��n||d< | jdkrPd}nd}t| |fd|i|��� | �� � fdd �}|j�d |�| _� |_ � j � � | _ � S ) NZticks� z6Either *locator* or *ticks* need to be given, not both��topr � horizontal�vertical� orientationc s* � � | �� � � �| �� � � �| � d S �N)Zset_cmapZget_cmapZset_climZget_climZupdate_bruteforce)�m��cbr r � on_changed5 s z)CbarAxesBase.colorbar.<locals>.on_changedZchanged) �tickerZ MaxNLocator� ValueErrorr r � _config_axesZ callbacksSMZconnectZcbid�colorbarZ cbar_axisZget_major_locatorr )�selfZmappabler �kwargsr r% r r# r r) ! s" zCbarAxesBase.colorbarc C sB | }|� d� |jdd� jdd� | j}|j| j j|d� dS )z Make an axes patch and outline.FN)�all)Z set_navigater r �_default_label_onr )r* r �br r r r( A s zCbarAxesBase._config_axesc C s$ || _ | j| j }|j||d� d S )Nr )r- r r r )r* r. r r r r � toggle_label] s zCbarAxesBase.toggle_label)�__name__� __module__� __qualname__r) r( r/ r r r r r s r c s( e Zd Z� fdd�Z� fdd�Z� ZS )�CbarAxesc s$ || _ d| _d | _t� j||� d S )NT)r r- r �super�__init__)r* r �argsr+ �� __class__r r r5 g s zCbarAxes.__init__c s t � �� | �� d S r! )r4 �clar( �r* r7 r r r9 m s z CbarAxes.cla)r0 r1 r2 r5 r9 � __classcell__r r r7 r r3 f s r3 c @ s� e Zd ZdZeZd(dd �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'� ZdS ))�Grida� A class that creates a grid of Axes. In matplotlib, the axes location (and size) is specified in the normalized figure coordinates. This may not be ideal for images that needs to be displayed with a given aspect ratio. For example, displaying images of a same size with some fixed padding between them cannot be easily done in matplotlib. AxesGrid is used in such case. N�row�{�G�z�?TF�Lc C s� |\| _ | _|dkr"| j | j }n&d| k r>| j | j ksHn td��|| _| �|� |dkrhtd��|| _| dkr�| j} i } n&t| t�r�t | | jj �r�i } n| \} } g | _ dd� t | j�D �| _ dd� t | j �D �| _g }g }t|ttf��rt||||dd �| _n~t|t��r*t||||dd �| _n\t|�d k�rZt||dd �}t|f|�|�| _n,t|�d k�r~t||||dd �| _ntd��| j�� }d d� t | j�D �| _d d� t | j �D �| _d| _t | j�D ]�}| �|�\}}|�r�| j}| j}n,| �r| j| }nd}| �r| j| }nd}| ||f||d�| ��}|�rP| jdk�rx|| _n(|dk�rd|| j|< |dk�rx|| j|< | j � |� | j | � |� | j| � |� �q�| j d d | _ | � � |�r�| j D ]}|� |� �q�| � | � dS )aS Parameters ---------- fig : `.Figure` The parent figure. rect : (float, float, float, float) or int The axes position, as a ``(left, bottom, width, height)`` tuple or as a three-digit subplot position code (e.g., "121"). direction : {"row", "column"}, default: "row" axes_pad : float or (float, float), default: 0.02 Padding or (horizontal padding, vertical padding) between axes, in inches. add_all : bool, default: True share_all : bool, default: False share_x : bool, default: True share_y : bool, default: True label_mode : {"L", "1", "all"}, default: "L" Determines which axes will get tick labels: - "L": All axes on the left column get vertical tick labels; all axes on the bottom row get horizontal tick labels. - "1": Only the bottom left axes is labelled. - "all": all axes are labelled. axes_class : a type that is a subclass of `matplotlib.axes.Axes`, default: None Nr � ��columnr= c S s g | ]}g �qS r r ��.0�_r r r � <listcomp>� s z!Grid.__init__.<locals>.<listcomp>c S s g | ]}g �qS r r rC r r r rF � s F�r r �aspect� � c S s g | ]}d �qS r! r rC r r r rF � s c S s g | ]}d �qS r! r rC r r r rF � s ��sharex�sharey���)!�_nrows�_ncols� Exception�ngrids�_init_axes_pad� _direction�_defaultAxesClass� isinstance�type� issubclassr �axes_all�range� axes_column�axes_row�strr r �_dividerr �len�dictr � get_position� _column_refax� _row_refax�_refax� _get_col_row�append�axes_llc�_update_locators�add_axes�set_label_mode)r* �fig�rect� nrows_ncolsrR � direction�axes_pad�add_all� share_allZshare_xZshare_y� label_mode� axes_class�axes_class_args�h�v�kw�i�colr= rL rM r r r r r5 ~ s� ) �� � � � � z Grid.__init__c C s2 t |�}|| _t�|d �| _t�|d �| _d S )Nr r )r � _axes_padr ZFixed�_horiz_pad_size�_vert_pad_size�r* ro r r r rS s zGrid._init_axes_padc C s g }g }| j D ]6}|r"|�| j� |�t|�� t�d�}|�|� qg }g }| jd d d� D ]6}|rr|�| j� |�t|�� t�d�}|�|� q^t| j �D ]D}| � |�\}} | j j || || j d | d�} | j| �| � q�| j �|� | j �|� d S )Nr rN ��nx�ny)rb rf r{ r_ r ZScaledrc r| rZ rR re r^ � new_locatorrO rY �set_axes_locator�set_horizontal� set_vertical) r* ru �h_ax_posrE �szrv �v_ax_posrx ry r= r r r r rh s0 � zGrid._update_locatorsc C s4 | j dkr t|| j�\}}nt|| j�\}}||fS )NrB )rT �divmodrO rP )r* �nry r= r r r re 7 s zGrid._get_col_rowc C s t | j�S r! )r_ rY r: r r r �__len__@ s z Grid.__len__c C s | j | S r! )rY )r* rx r r r r C s zGrid.__getitem__c C s | j | jfS )zV Return the number of rows and columns of the grid as (nrows, ncols). )rO rP r: r r r � get_geometryF s zGrid.get_geometryc C s" || _ |d | j_|d | j_dS )z� Set the padding between the axes. Parameters ---------- axes_pad : (float, float) The padding (horizontal pad, vertical pad) in inches. r r N)rz r{ Z fixed_sizer| r} r r r � set_axes_padL s zGrid.set_axes_padc C s | j S )z� Return the axes padding. Returns ------- hpad, vpad Padding (horizontal pad, vertical pad) in inches. )rz r: r r r � get_axes_pad[ s zGrid.get_axes_padc C s | j �|� dS )z%Set the aspect of the SubplotDivider.N)r^ � set_aspect)r* rH r r r r� f s zGrid.set_aspectc C s | j �� S )z(Return the aspect of the SubplotDivider.)r^ � get_aspectr: r r r r� j s zGrid.get_aspectc C s� |dkr"| j D ]}t|dd� qn�|dkr�| jd dd� D ]}t|ddd� q<| jd d }t|ddd� | jd d� D ]:}|dd� D ]}t|ddd� q�|d }t|ddd� qzn6|d kr�| j D ]}t|ddd� q�| j}t|ddd� dS ) a� Define which axes have tick labels. Parameters ---------- mode : {"L", "1", "all"} The label mode: - "L": All axes on the left column get vertical tick labels; all axes on the bottom row get horizontal tick labels. - "1": Only the bottom left axes is labelled. - "all": all axes are labelled. r, Fr? r NrN T)r r r �1)rY r r[ rg )r* �moder ry r r r rj n s$ zGrid.set_label_modec C s | j S r! )r^ r: r r r � get_divider� s zGrid.get_dividerc C s | j �|� d S r! )r^ Z set_locator)r* r r r r r� � s zGrid.set_axes_locatorc C s | j �� S r! )r^ Z get_locatorr: r r r �get_axes_locator� s zGrid.get_axes_locatorc C s | j �� S r! )r^ �get_vsize_hsizer: r r r r� � s zGrid.get_vsize_hsize) Nr= r> TFTTr? N)r0 r1 r2 �__doc__r rU r5 rS rh re r� r r� r� r� r� r� rj r� r� r� r� r r r r r< r s8 � % )r< c @ s& e Zd ZdZeZdd d �Zd d � ZdS )� ImageGrida� A class that creates a grid of Axes. In matplotlib, the axes location (and size) is specified in the normalized figure coordinates. This may not be ideal for images that needs to be displayed with a given aspect ratio. For example, displaying images of a same size with some fixed padding between them cannot be easily done in matplotlib. ImageGrid is used in such case. Nr= r> TFr? �right�5%c C s� |\| _ | _|dkr"| j | j }n"d| k r>| j | j ksDn t�|| _t|�}|| _| | _| | _| dkr�| dkr�|d | _q�|d | _n| | _|| _ | � |� |dkr�td��|| _ |dkr�| j }i }nt |tj�r�i }n|\}}g | _dd� t| j�D �| _d d� t| j �D �| _g | _g }g }t |ttf��rLt||||| d �| _n~t |t��rnt||||| d �| _n\t|�d k�r�t||| d �}t|f|�|�| _n,t|�d k�r�t ||||| d �| _ntd��| j� � }d d� t| j�D �| _ dd� t| j �D �| _ d| _ t| j�D ]�}| �!|�\}}|�rP| j�rF| jd }| jd }nd}d}n| j | }| j | }|||f||d�|��}| j�"|� | j| �"|� | j| �"|� |�r�| j dk�r�|| _ |dk�r�|| j |< |dk�r�|| j |< | j#||| jd�}| j�"|� �q| jd d | _$| �%� |�rF| j| j D ]}|�&|� �q4|�r�| jdk�rt| jD ]}| jd |_'�q^nx| jdk�r�t(| j�D ]>\} }| �!| �\}}| jdk�r�| j| |_'n | j| |_'�q�n t)| j| j�D ]\}}||_'�q�| �*| � dS )a� Parameters ---------- fig : `.Figure` The parent figure. rect : (float, float, float, float) or int The axes position, as a ``(left, bottom, width, height)`` tuple or as a three-digit subplot position code (e.g., "121"). direction : {"row", "column"}, default: "row" axes_pad : float or (float, float), default: 0.02 Padding or (horizontal padding, vertical padding) between axes, in inches. add_all : bool, default: True share_all : bool, default: False aspect : bool, default: True label_mode : {"L", "1", "all"}, default: "L" Determines which axes will get tick labels: - "L": All axes on the left column get vertical tick labels; all axes on the bottom row get horizontal tick labels. - "1": Only the bottom left axes is labelled. - "all": all axes are labelled. cbar_mode : {"each", "single", "edge", None }, default: None cbar_location : {"left", "right", "bottom", "top"}, default: "right" cbar_pad : float, default: None cbar_size : size specification (see `.Size.from_any`), default: "5%" cbar_set_cax : bool, default: True If True, each axes in the grid has a *cax* attribute that is bound to associated *cbar_axes*. axes_class : a type that is a subclass of `matplotlib.axes.Axes`, default: None Nr )r r� r rA r@ c S s g | ]}g �qS r r rC r r r rF s z&ImageGrid.__init__.<locals>.<listcomp>c S s g | ]}g �qS r r rC r r r rF s rG rI rJ c S s g | ]}d �qS r! r rC r r r rF - s c S s g | ]}d �qS r! r rC r r r rF . s rK )r rN �single�edge)+rO rP rQ rR r rz �_colorbar_mode�_colorbar_location� _colorbar_pad�_colorbar_sizerS rT rU rV �maxesr rY rZ r[ r\ � cbar_axesr] r r r^ r r_ r` r ra rb rc rd re rf �_defaultCbarAxesClassrg rh ri �cax� enumerate�ziprj ) r* rk rl rm rR rn ro rp rq rH rr Z cbar_modeZ cbar_locationZcbar_padZ cbar_sizeZ cbar_set_caxrs rt ru rv rw rx ry r= rL rM r r� �indexr r r r5 � s� 2 � � � � � zImageGrid.__init__c C sx g }g }g }g }| j dk�r | jdk�r | jdkr�t�| jt�| j��}|�t�| j |�� |�t�| j |�� | j j dddd�}nZ| jdkr�t�| j t�| j��}|�t�| j |�� |�t�| j |�� | j j dddd�}t| j�D ]}| j| �d � q�| jd �|� | jd �d � t| jd �D �]<\}} |�rH|�| j� | �rftj| d | jd d �}n tj| jd d | jd d �}| j d k�s�| j dk�r�|dk�r�| jdk�r�|�t|�� |�t�| j |�� |�t�| j |�� |�t|�� |�|� | j d k�s&| j dk�r,|| j d k�r,| jdk�r,|�t�| j |�� |�t|�� |�t�| j |�� �q,g } g } t| jd d d d� �D �]<\} } |�r�|�| j� | �r�tj| d | jd d �}n tj| jd d | jd d �}| j d k�s| j dk�rF| dk�rF| jdk�rF| �t|�� |�t�| j |�� |�t�| j |�� | �t|�� |�|� | j d k�s�| j dk�r�| | jd k�r�| jdk�r�|�t�| j |�� | �t|�� |�t�| j |�� �q�t| j�D �]�}| �|�\}} | j j || | | jd | d�}| j| �|� | j d k�r�| jdk�rX| j j || | | jd | d�}n.| jdk�r�| j j || | | jd | d�}| j| �|� n�| j dk�r�| jdk�r�|dk�s�| jdk�r || j d k�r | j j |d | | jd | d�}| j| �|� nZ| jdk�r&| | jd k�s<| jdk�r�| dk�r�| j j || | d d�}| j| �|� �q�| j dk�r~| jdk�r�t�| jt�| j��}|�t�| j |�� |�t�| j |�� | j j dddd�}n\| jdk�r.t�| j t�| j��}|�t�| j |�� |�t�| j |�� | j j dddd�}| jdk�r\t| j�D ]}| j| �d � �qD| jd �|� | jd �d � n�| j d k�r�t| j�D ]}| j| �d � �q�n�| j dk�r | jdk�r�| j} n| j } t| �D ]}| j| �d � �q�t|d | j�D ]}| j| �d � �qn>t| j�D ]2}| j| �d � | j| j ddddgdd� �q(| j � |� | j � |� d S )Nr� )r r r r rN )r r� Zny1r )r Znx1r� FT�axes)rH Zref_axZeachr� r r� r r~ )r� r r ���)r� r g �?g����MbP?Zactive)Zwhich) r� r� r ZFractionrO ZAxesXrg rf Zfrom_anyr� r� r^ r� rP ZAxesYrZ rR r� Z set_visibler� r� r\ r{ rY r_ r[ r| re Z set_positionr� r� )r* ru rv r� Zh_cb_posr� r rx ry r r� Zv_cb_posr= �count�jr r r rh l s> � � ��� � ��" � ��� � �� � � � � � � ��� � � zImageGrid._update_locators) Nr= r> TFTr? Nr� Nr� TN)r0 r1 r2 r� r3 r� r5 rh r r r r r� � s$ � 6r� )Znumbersr Zmatplotlib.axesr� r� Zmatplotlib.tickerr& Zmatplotlib.gridspecr Z axes_dividerr r r r) r Zmpl_axesr r r �objectr r3 r< r� ZAxesGridr r r r �<module> s" G ; `