U \ a *@sbddlZddlmZddlmZddlmZddl m Z m Z Gddde Z dZ ejje ddS)N)_api) _AxesBase_TransformedBoundsLocatorcseZdZdZfddZddZddZdfd d Zd d d dZddZ fddZ ddZ ddZ ddZ ddZZS) SecondaryAxisz4 General class to hold a Secondary_X/Yaxis. c sB||_||_||_d|_|jdkr^tj|jjddddgf||j|_ddg|_ d d g|_ nD|jd krtj|jjddddgf||j |_d d g|_ ddg|_ d |_ | ||||jdkr|j }n|j}|t|d |j|j d|j|j d|jdkr.|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. Fxr?g-C6?topbottomleftrightyNnoneTg?) _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)selfparentZ orientationlocation functionskwargsZ otheraxis __class__C/tmp/pip-unpacked-wheel-wjyw_3jo/matplotlib/axes/_secondary_axes.pyrs6           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 : str either 'top' or 'bottom' for orientation='x' or 'left' or 'right' for orientation='y' axis. )alignrNrrTF)rZ check_in_listrr!r"rr Zset_label_position)r%r.r,r,r-r$;s  zSecondaryAxis.set_alignmentcCst|trR|dkrd|_qX|dkr*d|_qXtd|jdd|jdd |n||_||_|jd krxd|jdd g}n|jdd dg}|t||j j d 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 r)r r gzlocation must be rz, rz, or a float, not rg|=N) isinstancestrr# ValueErrorrZ_locrZset_axes_locatorrrZ transAxes)r%r'Zboundsr,r,r-rNs     zSecondaryAxis.set_locationNcs|t|dSN) _set_limsr apply_aspect)r%positionr*r,r-r4yszSecondaryAxis.apply_aspectFminorcCs |jj||d}d|_d|_|S)a  Set the x ticks with list of *ticks* Parameters ---------- ticks : list List of x-axis tick locations. minor : bool, default: False If ``False`` sets major ticks, if ``True`` sets minor ticks. r6T)r set_ticksstaler)r%ticksr7retr,r,r-r8~s zSecondaryAxis.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. rrNcSs|Sr2r,)rr,r,r-z-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)r/tuplelencallablerr1 _set_scale)r%r(r,r,r-rs  zSecondaryAxis.set_functionscs"||tj||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)r3rBrdrawr%argsr)r*r,r-rCs zSecondaryAxis.drawcCs|jdkr|jj}|j}|jdkr8|jj}|j}||jkrFdS|dkrTd}nd}|jrh|j }|||j dddd|jr|j t |||_dS) z3 Check if parent has set its scale rr NlogZ functionlogfunctionr)r()rrrZ get_scaleZ set_xscalerZ set_yscalerrrZ get_ticklocsrrrZ FixedLocator)r%ZpscaleZ set_scaleZdefscaler:r,r,r-rBs"      zSecondaryAxis._set_scalecCs|jdkr|j}|j}|jdkr4|j}|j}|d|dk}|jdt|}|d|dk}||kr~|ddd}||dS)z Set the limits based on parent limits and the convert method between the parent and this secondary axes. rr rrNr) rrZget_xlimZset_xlimZget_ylimZset_ylimrnparray)r%ZlimsZset_limorderZneworderr,r,r-r3s    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_externalrDr,r,r- 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 rlabelr r r)r%colorr,r,r-rMs zSecondaryAxis.set_color)N)__name__ __module__ __qualname____doc__rr$rr4r8rrCrBr3rKrM __classcell__r,r,r*r-r s -+! !rao 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. 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)ZnumpyrHZ matplotlibrZmatplotlib.docstringZ docstringZmatplotlib.tickerZtickerrZmatplotlib.axes._baserrrrUZinterpdupdater,r,r,r-s   $