OwlCyberSecurity - MANAGER
Edit File: _secondary_axes.cpython-38.pyc
U ��]r= � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl m Z d dl mZ d dlmZ d dlmZ d dlmZmZmZmZmZmZmZmZm Z d dl m Z dd� Z G dd� de�Z d Z ej!j"e d � dS ) � N)� _AxesBase) � AutoLocator�AutoMinorLocator� FixedLocator� FuncFormatter�LogFormatterSciNotation� LogLocator� NullLocator� NullFormatter�ScalarFormatter)�Log10Transformc s t jj| � � � �fdd�}|S )a Helper function to locate the secondary axes. A locator gets used in `Axes.set_aspect` to override the default locations... It is a function that takes an axes object and a renderer and tells `set_aspect` where it is to be placed. This locator make the transform be in axes-relative co-coordinates because that is how we specify the "location" of the secondary axes. Here *rect* is a rectangle [l, b, w, h] that specifies the location for the axes in the transform given by *trans* on the *parent*. c s* t �� �j�}�jj�� }t �||�}|S �N)� mtransformsZTransformedBboxZ transAxes�figureZ transFigure�inverted)Zax�renderer�bbZtr�Z_rect�parent� �A/usr/lib/python3/dist-packages/matplotlib/axes/_secondary_axes.py�secondary_locator. s z2_make_secondary_locator.<locals>.secondary_locator)r �BboxZ from_bounds)Zrectr r r r r �_make_secondary_locator s r c s� e Zd ZdZ� fdd�Zdd� Zdd� Zd!� fd d � Zd"d d �Zdd� Z d#� fdd� Z dd� Z dd� Z d$dd�Z dd� Zd%dd �Zd&d d �Zd d � Z� ZS )'� SecondaryAxisz4 General class to hold a Secondary_X/Yaxis. c s^ || _ || _|| _d| _| jdkr^t� j| jjddddgf|� | j| _ddg| _ d d g| _ nD| jd kr�t� j| jjddddgf|� | j | _d d g| _ ddg| _ d | _ d | _ | �|� | �|� | jdkr�| j }n| j}|�t�� � |�d � | j D ]}| j| �d� q�| j D ]}| j| �d� �q| jdk �rJ| j d d d� | _ | �| j d � d S )z� See `.secondary_xaxis` and `.secondary_yaxis` for the doc string. While there is no need for this to be private, it should really be called by those higher level functions. F�xr � �?� g-C ��6?�top�bottom�left�right�yNZnoneTg �?���)� _functions�_parent� _orientation� _ticks_set�super�__init__r �xaxis�_axis� _locstringsZ _otherstrings�yaxisZ _layoutboxZ _poslayoutbox� set_location� set_functions�set_major_locator�mtickerr �set_ticks_position�spines� set_visible�_pos� set_alignment)�selfr Z orientation�location� functions�kwargsZ otheraxis�st�� __class__r r r) > s< zSecondaryAxis.__init__c C s� || j kr�|| j d kr,| j ddd� | _ n,|| j d krXtd�|| j d | j d ���| j| j d �d� | j| j d �d� | j�|� | j�|� dS )a! Set if axes spine and labels are drawn at top or bottom (or left/right) of the axes. Parameters ---------- align :: string either 'top' or 'bottom' for orientation='x' or 'left' or 'right' for orientation='y' axis r Nr# r zW"{}" is not a valid axis orientation, not changing the orientation;choose "{}" or "{}""TF)r, � ValueError�formatr3 r4 r+ r2 Zset_label_position)r7 Zalignr r r r6 o s � zSecondaryAxis.set_alignmentc C s� t |t�rJ|dkrd| _qP|dkr*d| _qPtd�|| jd | jd ���n|| _|| _| jdkrpd| jdd g}n| jdd dg}t|| j �}| � |� d S ) a Set the vertical or horizontal location of the axes in parent-normalized co-ordinates. Parameters ---------- location : string or scalar The position to put the secondary axis. Strings can be 'top' or 'bottom' for orientation='x' and 'right' or 'left' for orientation='y', scalar can be a float indicating the relative position on the parent axes to put the new axes, 0.0 being the bottom (or left) and 1.0 being the top (or right). )r r! r )r r g z1location must be '{}', '{}', or a float, not '{}'r r r g�����|�=N) � isinstance�strr5 r> r? r, Z_locr& r r% Zset_axes_locator)r7 r8 Zboundsr r r r r. � s" � zSecondaryAxis.set_locationNc s | � � t� �|� d S r )� _set_limsr( � apply_aspect)r7 Zpositionr< r r rC � s zSecondaryAxis.apply_aspectFc C s | j j||d�}d| _d| _|S )a( Set the x ticks with list of *ticks* Parameters ---------- ticks : list List of x-axis tick locations. minor : bool, optional If ``False`` sets major ticks, if ``True`` sets minor ticks. Default is ``False``. )�minorT)r+ � set_ticksZstaler' )r7 �ticksrD Zretr r r rE � s zSecondaryAxis.set_ticksc C s� | j dkr | j}| j�� }n| j}| j�� }|dkr:d}nd}t|t�rtt|�dkrtt |d �rtt |d �rt|| _ n$|dkr�d d � d d � f| _ nt d ��||| j ddd � d � dS )a' Set how the secondary axis converts limits from the parent axes. Parameters ---------- functions : 2-tuple of func, or `Transform` with an inverse. Transform between the parent axis values and the secondary axis values. If supplied as a 2-tuple of functions, the first function is the forward transform function and the second is the inverse transform. If a transform is supplied, then the transform must have an inverse. r �log� functionlog�function� r r Nc S s | S r r )r r r r �<lambda>� � z-SecondaryAxis.set_functions.<locals>.<lambda>z�functions argument of secondary axes must be a two-tuple of callable functions with the first function being the transform and the second being the inverser# �r9 ) r& � set_xscaler% Z get_xscale� set_yscaleZ get_yscaler@ �tuple�len�callabler$ r> )r7 r9 � set_scaleZ parent_scale�defscaler r r r/ � s$ � �zSecondaryAxis.set_functionsc s$ | � � | �� t� j||d� dS )a Draw the secondary axes. Consults the parent axes for its limits and converts them using the converter specified by `~.axes._secondary_axes.set_functions` (or *functions* parameter when axes initialized.) )r �inframeN)rB � _set_scaler( �draw)r7 r rU r< r r rW � s zSecondaryAxis.drawc C s� | j dkr | jj�� }| j}| j dkr8| jj�� }| j}|dkrFd}nd}| jrZ| j� � }||| j ddd� d� | jr�| j� t |�� dS ) z3 Check if parent has set its scale r r" rG rH rI Nr# rM ) r& r% r* Z get_scalerN r- rO r' r+ Z get_ticklocsr$ r0 r )r7 ZpscalerS rT rF r r r rV s zSecondaryAxis._set_scalec C s� | j dkr$| j�� }| j}| j�� }| j dkrH| j�� }| j}| j�� }|d |d k }| j d t � |��}|d |d k }||kr�|ddd� }||� dS )z� Set the limits based on parent limits and the convert method between the parent and this secondary axes r r" r r Nr# ) r& r% Zget_xlimZset_xlimr* Z get_transformZget_ylimZset_ylimr- r$ �npZarray)r7 ZlimsZset_limZtrans�orderZneworderr r r rB $ s zSecondaryAxis._set_limsTc C s� g }| � � sdS | �� | �� }|r>|r>|| |�}| �|� n| �� | jdkr^| j�|�}n | j�|�}|rx|�|� |�| � |�� t j � dd� |D ��}|S )a� Return the tight bounding box of the axes. The dimension of the Bbox in canvas coordinate. If *call_axes_locator* is *False*, it does not call the _axes_locator attribute, which is necessary to get the correct bounding box. ``call_axes_locator==False`` can be used if the caller is only intereted in the relative size of the tightbbox compared to the axes bbox. Nr c S s$ g | ] }|j d ks |jd kr|�qS )r )�widthZheight)�.0�br r r � <listcomp>[ s z/SecondaryAxis.get_tightbbox.<locals>.<listcomp>) Z get_visiblerB Zget_axes_locatorrC r& r* � get_tightbboxr- �appendZget_window_extentr r �union)r7 r Zcall_axes_locatorr Zlocator�posZbb_axisZ_bboxr r r r^ 9 s& �zSecondaryAxis.get_tightbboxc O s t �d� dS )zj Secondary axes cannot set the aspect ratio, so calling this just sets a warning. z)Secondary axes can't set the aspect ratioN)�cbookZ_warn_external)r7 �argsr: r r r � set_aspect_ s zSecondaryAxis.set_aspectc K s" |dk r|| j _| j j||f|�S )a Set the label for the x-axis. Parameters ---------- xlabel : str The label text. labelpad : scalar, optional, default: None Spacing in points between the label and the x-axis. Other Parameters ---------------- **kwargs : `.Text` properties `.Text` properties control the appearance of the label. See also -------- text : for information on how override and the optional args work N)r* �labelpad�set_label_text)r7 Zxlabel�fontdictre r: r r r � set_xlabelf s zSecondaryAxis.set_xlabelc K s" |dk r|| j _| j j||f|�S )a Set the label for the x-axis. Parameters ---------- ylabel : str The label text. labelpad : scalar, optional, default: None Spacing in points between the label and the x-axis. Other Parameters ---------------- **kwargs : `.Text` properties `.Text` properties control the appearance of the label. See also -------- text : for information on how override and the optional args work N)r- re rf )r7 Zylabelrg re r: r r r � set_ylabel s zSecondaryAxis.set_ylabelc C s� | j dkrH| jd|d� | jd �|� | jd �|� | jj�|� n<| jd|d� | jd �|� | jd �|� | jj�|� dS ) z� Change the color of the secondary axes and all decorators Parameters ---------- color : Matplotlib color r )ZaxisZcolorsr r r" r r! N)r& Z tick_paramsr3 � set_colorr* Zlabelr- )r7 Zcolorr r r rj � s zSecondaryAxis.set_color)N)F)NF)T)NN)NN)�__name__� __module__� __qualname__�__doc__r) r6 r. rC rE r/ rW rV rB r^ rd rh ri rj � __classcell__r r r< r r 9 s 1* 0 & r a^ Warnings -------- This method is experimental as of 3.1, and the API may change. Parameters ---------- location : string or scalar The position to put the secondary axis. Strings can be 'top' or 'bottom', for x-oriented axises or 'left' or 'right' for y-oriented axises or a scalar can be a float indicating the relative position on the axes to put the new axes (0 being the bottom (left), and 1.0 being the top (right).) functions : 2-tuple of func, or Transform with an inverse If a 2-tuple of functions, the user specifies the transform function and its inverse. i.e. `functions=(lambda x: 2 / x, lambda x: 2 / x)` would be an reciprocal transform with a factor of 2. The user can also directly supply a subclass of `.transforms.Transform` so long as it has an inverse. See :doc:`/gallery/subplots_axes_and_figures/secondary_axis` for examples of making these conversions. Other Parameters ---------------- **kwargs : `~matplotlib.axes.Axes` properties. Other miscellaneous axes parameters. Returns ------- ax : axes._secondary_axes.SecondaryAxis )�_secax_docstring)#� collectionsZnumpyrX Znumbers�warningsZmatplotlib.docstringZ docstringZmatplotlib.tickerZtickerr1 Zmatplotlib.transformsZ transformsr Zmatplotlib.scaleZscaleZmscaleZmatplotlib.cbookrb Zmatplotlib.axes._baser r r r r r r r r r r r r rp Zinterpd�updater r r r �<module> s" , u&