a >f[.@s~ddlZddlZddlmZmZmZddlmZ ddl m Z m Z ddl mZddlmZGddde ZdZejjed dS) N)_api _docstring transforms) _AxesBase_TransformedBoundsLocator)Axis) TransformcseZdZdZdfdd ZddZdddZdfd d Ze e j dd d d dZ ddZ fddZ ddZddZddZddZZS) SecondaryAxisz4 General class to hold a Secondary_X/Yaxis. Nc sJtjddg|d||_||_||_d|_|jdkrptj|jjgdfi||j |_ ddg|_ dd g|_ n:tj|jjgd fi||j |_ d dg|_ ddg|_ d |_||||||jdkr|j n|j }|t|d |j|j d|j|j d |jdkr6|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) orientationF)r?-C6?topbottomleftright)rr rrNnoneTg?r)r check_in_list _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)selfparentr location functions transformkwargsZ otheraxis __class__m/var/www/staging/api/virtual_environments/venv/lib/python3.9/site-packages/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. )alignrNrrTF)rrrr'r(rr&Zset_label_position)r+r5r3r3r4r*;s  zSecondaryAxis.set_alignmentcCstjtjdf|dt|trBtj|j|d|dvr:dnd|_n:t|t j rV||_n&t d|jdd |jd d |||_ |j d krd|jdd g}|durt|jj|}n&|jdd d g}|durt||jj}|dur|jj}|t||dS)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). transform : `.Transform`, optional Transform for the location to use. Defaults to the parent's ``transAxes``, so locations are normally relative to the parent axes. .. versionadded:: 3.9 N)r/)r-)rrr gzlocation must be rz, rz, or a float, not r g|=)rZcheck_isinstancerr isinstancestrrrr)numbersReal ValueErrorZ_locrZblended_transform_factoryrZ transAxesZset_axes_locatorr)r+r-r/Zboundsr3r3r4r"Ns8   zSecondaryAxis.set_locationcs|t|dSN) _set_limsr apply_aspect)r+positionr1r3r4r=szSecondaryAxis.apply_aspectF)minorcKs*|jj||fd|i|}d|_d|_|S)Nr?T)r set_ticksstaler)r+tickslabelsr?r0retr3r3r4r@szSecondaryAxis.set_tickscsttr6tdkr6tdr6tdr6|_nDttrVjfddf|_n$durrddddf|_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. rrcs|Sr;)invertedr/r r.r3r4z-SecondaryAxis.set_functions..NcSs|Sr;r3rGr3r3r4rIrJzfunctions 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) r6tuplelencallablerrr/r: _set_scale)r+r.r3rHr4r#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)r<rNrdraw)r+rendererr1r3r4rOs 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 r NlogZ functionlogfunctionrrH)rrrZ get_scaleZ set_xscaler Z set_yscaler!rrZ get_ticklocsrr$r%Z FixedLocator)r+ZpscaleZ set_scalerBr3r3r4rNs     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. r rrNr) rrZget_xlimZset_xlimZget_ylimZset_ylimrnparray)r+ZlimsZset_limorderZneworderr3r3r4r<s   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+argsr0r3r3r4 set_aspectszSecondaryAxis.set_aspectcCsL|j|j}|j|d|jD]}|j|ur"||q"|j|dS)z Change the color of the secondary Axes and all decorators. Parameters ---------- color : :mpltype:`color` )colorsN)Z _axis_maprZset_tick_paramsr'valuesaxis set_colorlabel)r+colorrZZspiner3r3r4r[s     zSecondaryAxis.set_color)N)N)N)N)__name__ __module__ __qualname____doc__rr*r"r=rcopyrr@r#rOrNr<rWr[ __classcell__r3r3r1r4r s* @ % r a 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. transform : `.Transform`, optional If specified, *location* will be placed relative to this transform (in the direction of the axis) rather than the parent's axis. i.e. a secondary x-axis will use the provided y transform and the x transform of the parent. .. versionadded:: 3.9 Returns ------- ax : axes._secondary_axes.SecondaryAxis Other Parameters ---------------- **kwargs : `~matplotlib.axes.Axes` properties. Other miscellaneous Axes parameters. )_secax_docstring)r8numpyrSZ matplotlibrrrZmatplotlib.tickerZtickerr%Zmatplotlib.axes._baserrZmatplotlib.axisrZmatplotlib.transformsrr rdZinterpdupdater3r3r3r4s    -