o :aa)@snddlZddlmZddlmZddlmZddl m Z m Z ddl m Z Gddde ZdZejjeddS) N)_api) _AxesBase_TransformedBoundsLocator)AxiscseZdZdZfddZddZddZdfd d Ze e j dd d d dZ ddZ fddZ ddZddZddZddZZS) SecondaryAxisz4 General class to hold a Secondary_X/Yaxis. c s@||_||_||_d|_|jdkr/tj|jjgdfi||j|_ddg|_ ddg|_ n"|jdkrQtj|jjgd fi||j |_ddg|_ ddg|_ d |_ | ||||jdkrg|j }n|j}|t|d |j|j d|j|j d |jd kr|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. Fx)r?-C6?topbottomleftrighty)rrr r NZnoneTg?r) _functions_parent _orientation _ticks_setsuper__init__Zfigurexaxis_axis _locstringsZ _otherstringsyaxis _parentscale set_location set_functionsset_major_locatormtickerZ NullLocatorset_ticks_positionspines set_visible_pos set_alignment)selfparentZ orientationlocation functionskwargsZ otheraxis __class__A/usr/lib/python3/dist-packages/matplotlib/axes/_secondary_axes.pyrs6           zSecondaryAxis.__init__cCsxtj|j|d||jdkr|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. )alignr NrrTF)rZ check_in_listrr!r"rr Zset_label_position)r%r.r,r,r-r$<s  zSecondaryAxis.set_alignmentcCst|tr(|dvr d|_n|dvrd|_ntd|jdd|jdd |||_||_|jd kr;d|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 rr)r r gzlocation must be rz, r z, or a float, not rg|=N) isinstancestrr# ValueErrorrZ_locrZset_axes_locatorrrZ transAxes)r%r'Zboundsr,r,r-rOs(   zSecondaryAxis.set_locationNcs|t|dSN) _set_limsr apply_aspect)r%Zpositionr*r,r-r4zszSecondaryAxis.apply_aspectF)minorcKs*|jj||fd|i|}d|_d|_|S)Nr5T)r set_ticksZstaler)r%tickslabelsr5r)retr,r,r-r6szSecondaryAxis.set_tickscCsft|trt|dkrt|drt|dr||_n|dur)ddddf|_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|Sr2r,)rr,r,r-sz-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|i|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)r3r?rdrawr%argsr)r*r,r-r@s zSecondaryAxis.drawcCs|jdkr|jj}|j}|jdkr|jj}|j}||jkr#dS|dkr*d}nd}|jr4|j }|||j dddd|jrL|j t |||_dS) z3 Check if parent has set its scale rrNlogZ functionlogfunctionr)r()rrrZ get_scaleZ set_xscalerZ set_yscalerrrZ get_ticklocsrrrZ FixedLocator)r%ZpscaleZ set_scaleZdefscaler7r,r,r-r?s"       zSecondaryAxis._set_scalecCs|jdkr |j}|j}|jdkr|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. rrrr Nr) rrZget_xlimZset_xlimZget_ylimZset_ylimrnpZarray)r%ZlimsZset_limZorderZneworderr,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_externalrAr,r,r- set_aspectszSecondaryAxis.set_aspectcCs|jdkr#|jd|d|jj||jj||jj|dS|jd|d|jj||jj ||j j|dS)z Change the color of the secondary axes and all decorators. Parameters ---------- color : color r)ZaxisZcolorsrN) rZ tick_paramsr!r set_colorr rZlabelr rr)r%Zcolorr,r,r-rGs zSecondaryAxis.set_colorr2)__name__ __module__ __qualname____doc__rr$rr4 docstringcopyrr6rr@r?r3rFrG __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)ZnumpyrEZ matplotlibrZmatplotlib.docstringrLZmatplotlib.tickerZtickerrZmatplotlib.axes._baserrZmatplotlib.axisrrrOZinterpdupdater,r,r,r-s    {$