U Ev f(@snddlZddlZddlmZmZddlmZddl m Z m Z ddl m Z Gddde ZdZejjeddS) N)_api _docstring) _AxesBase_TransformedBoundsLocator)AxiscseZdZdZfddZddZddZdfd d Ze e j dd d d dZ ddZ fddZ ddZddZddZddZZS) SecondaryAxisz4 General class to hold a Secondary_X/Yaxis. c sHtjddg|d||_||_||_d|_|jdkrptj|jjddddgf||j |_ d d g|_ d d g|_ n:tj|jjddddgf||j |_ d d g|_ d d g|_ d |_|||||jdkr|j n|j }|t|d|j|j d|j|j d|jdkr4|j d d d|_ ||j dd 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. xy) orientationFr?g-C6?topbottomleftrightNnoneTg?)r check_in_list _functions_parent _orientation _ticks_setsuper__init__figurexaxis_axis _locstringsZ _otherstringsyaxis _parentscale set_location set_functionsset_major_locatormtickerZ NullLocatorset_ticks_positionspines set_visible_pos set_alignment)selfparentr location functionskwargsZ otheraxis __class__C/tmp/pip-unpacked-wheel-7vhvci0g/matplotlib/axes/_secondary_axes.pyrs2         zSecondaryAxis.__init__cCsxtj|j|d||jdkr0|jddd|_|j|jdd|j|jdd|j||j|dS)a< Set if axes spine and labels are drawn at top or bottom (or left/right) of the axes. Parameters ---------- align : {'top', 'bottom', 'left', 'right'} Either 'top' or 'bottom' for orientation='x' or 'left' or 'right' for orientation='y' axis. )alignr NrrTF)rrrr%r&rr$Zset_label_position)r)r2r0r0r1r(9s  zSecondaryAxis.set_alignmentcCst|tr.tj|j|d|dkr&dnd|_n:t|tjrB||_n&td|jdd|jdd |||_ |j d krd|jdd g}n|jdd dg}| t ||j jd S) a+ Set the vertical or horizontal location of the axes in parent-normalized coordinates. Parameters ---------- location : {'top', 'bottom', 'left', 'right'} or float The position to put the secondary axis. Strings can be 'top' or 'bottom' for orientation='x' and 'right' or 'left' for orientation='y'. A float indicates 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 rr gzlocation must be rz, r z, or a float, not rg|=N) isinstancestrrrrr'numbersReal ValueErrorZ_locrZset_axes_locatorrrZ transAxes)r)r+Zboundsr0r0r1r Ls     zSecondaryAxis.set_locationNcs|t|dSN) _set_limsr apply_aspect)r)positionr.r0r1r:vszSecondaryAxis.apply_aspectF)minorcKs*|jj||fd|i|}d|_d|_|S)Nr<T)r set_ticksstaler)r)tickslabelsr<r-retr0r0r1r={szSecondaryAxis.set_tickscCsft|tr6t|dkr6t|dr6t|dr6||_n$|dkrRddddf|_ntd|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. rr NcSs|Sr8r0)rr0r0r1z-SecondaryAxis.set_functions..zfunctions argument of secondary axes must be a two-tuple of callable functions with the first function being the transform and the second being the inverse)r3tuplelencallablerr7 _set_scale)r)r,r0r0r1r!s  zSecondaryAxis.set_functionscs ||t|dS)z 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.) N)r9rHrdraw)r)Zrendererr.r0r1rIs zSecondaryAxis.drawcCs|jdkr|jj}|j}n|jj}|j}||jkr>dS|jrN|j }||dkr\dnd|j dddd|jr|j t |||_dS)z3 Check if parent has set its scale rNlogZ functionlogfunctionr)r,)rrrZ get_scaleZ set_xscalerZ set_yscalerrrZ get_ticklocsrr"r#Z FixedLocator)r)ZpscaleZ set_scaler?r0r0r1rHs     zSecondaryAxis._set_scalecCs|jdkr|j}|j}n|j}|j}|d|dk}|jdt|}|d|dk}||krv|ddd}||dS)z Set the limits based on parent limits and the convert method between the parent and this secondary axes. rrr Nr) rrZget_xlimZset_xlimZget_ylimZset_ylimrnparray)r)ZlimsZset_limorderZneworderr0r0r1r9s   zSecondaryAxis._set_limscOstddS)zj Secondary axes cannot set the aspect ratio, so calling this just sets a warning. z)Secondary axes can't set the aspect ratioN)rZ warn_external)r)argsr-r0r0r1 set_aspectszSecondaryAxis.set_aspectcCs|jdkrD|jd|d|jj||jj||jj|n8|jd|d|jj||jj ||j j|dS)z Change the color of the secondary axes and all decorators. Parameters ---------- color : color r)Zaxiscolorsr N) rZ tick_paramsr%r set_colorr rlabelrrr)r)colorr0r0r1rRs zSecondaryAxis.set_color)N)N)__name__ __module__ __qualname____doc__rr(r r:rcopyrr=r!rIrHr9rPrR __classcell__r0r0r.r1r s )*  ra Warnings -------- This method is experimental as of 3.1, and the API may change. Parameters ---------- location : {'top', 'bottom', 'left', 'right'} or float The position to put the secondary axis. Strings can be 'top' or 'bottom' for orientation='x' and 'right' or 'left' for orientation='y'. A float indicates 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). 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. Both functions must accept numpy arrays as input. 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. Returns ------- ax : axes._secondary_axes.SecondaryAxis Other Parameters ---------------- **kwargs : `~matplotlib.axes.Axes` properties. Other miscellaneous axes parameters. )_secax_docstring)r5ZnumpyrLZ matplotlibrrZmatplotlib.tickerZtickerr#Zmatplotlib.axes._baserrZmatplotlib.axisrrr[Zinterpdupdater0r0r0r1s  p%