2i_dZddlZddlZddlZddlmZddlmZddl m Z ddl m ZGddZej d Gd d ZGd d eZGddeZdZGddeZGddeZdZ ddZy)zJ Helper classes to adjust the positions of multiple axes at drawing time. N)_api) SubplotSpec) axes_sizeceZdZdZ ddZdZdZdZdZdZ d Z d Z d Z d Z d ZdZddZdZdZdZdZedZedZd dZej4ddd!dZdZdZd"dZy)#DivideraQ An Axes positioning class. The divider is initialized with lists of horizontal and vertical sizes (:mod:`mpl_toolkits.axes_grid1.axes_size`) based on which a given rectangular area will be divided. The `new_locator` method then creates a callable object that can be used as the *axes_locator* of the axes. Nc||_||_||_||_||_|j |||_d|_d|_d|_ y)a Parameters ---------- fig : Figure pos : tuple of 4 floats Position of the rectangle that will be divided. horizontal : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` Sizes for horizontal division. vertical : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` Sizes for vertical division. aspect : bool, optional Whether overall rectangular area is reduced so that the relative part of the horizontal and vertical scales have the same scale. anchor : (float, float) or {'C', 'SW', 'S', 'SE', 'E', 'NE', 'N', 'NW', 'W'}, default: 'C' Placement of the reduced rectangle, when *aspect* is True. rN) _fig_pos _horizontal _vertical_anchor set_anchor_aspect _xrefindex _yrefindex_locator)selffigpos horizontalverticalaspectanchors [/var/www/html/acx/venv/lib/python3.12/site-packages/mpl_toolkits/axes_grid1/axes_divider.py__init__zDivider.__init__sP(  %!    ctj|jDcgc]}|j|c}Scc}wN)nparrayget_horizontalget_sizerrendererss rget_horizontal_sizeszDivider.get_horizontal_sizes;s1xxt7J7J7LM!H-MNNMActj|jDcgc]}|j|c}Scc}wr)r r! get_verticalr#r$s rget_vertical_sizeszDivider.get_vertical_sizes>s1xxt7H7H7JK!H-KLLKr(c||_y)z Set the position of the rectangle. Parameters ---------- pos : tuple of 4 floats position of the rectangle that will be divided Nr )rrs r set_positionzDivider.set_positionAs  rc|jS)z%Return the position of the rectangle.r-rs r get_positionzDivider.get_positionLs yyrct|tr7tjtj j |||_ yt|ttfrt|dk7r td||_ y)a Parameters ---------- anchor : (float, float) or {'C', 'SW', 'S', 'SE', 'E', 'NE', 'N', 'NW', 'W'} Either an (*x*, *y*) pair of relative coordinates (0 is left or bottom, 1 is right or top), 'C' (center), or a cardinal direction ('SW', southwest, is bottom left, etc.). See Also -------- .Axes.set_anchor )rzanchor must be str or 2-tupleN) isinstancestrr check_in_list mtransformsBboxcoefstuplelistlen TypeErrorr)rrs rrzDivider.set_anchorPsa fc "   {//55f E FUDM2c&kQ6F;< < rc|jS)zReturn the anchor.)rr0s r get_anchorzDivider.get_anchord ||rcyrr0s rget_subplotspeczDivider.get_subplotspechsrc||_y)z Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for horizontal division Nr )rhs rset_horizontalzDivider.set_horizontalks rc|jS)zReturn horizontal sizes.rEr0s rr"zDivider.get_horizontaltsrc||_y)z Parameters ---------- v : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for vertical division Nr )rvs r set_verticalzDivider.set_verticalxs rc|jS)zReturn vertical sizes.rJr0s rr*zDivider.get_verticals ~~rc||_y)zE Parameters ---------- aspect : bool Nrrrs r set_aspectzDivider.set_aspects  rc|jS)zReturn aspect.rOr0s r get_aspectzDivider.get_aspectr@rc||_yrr)rrs r set_locatorzDivider.set_locators   rc|jSrrUr0s r get_locatorzDivider.get_locators }}rcr|j|jS|j||jSr)rr1bounds)raxr%s rget_position_runtimezDivider.get_position_runtimes2 == $$& &==X.55 5rcB|jd\}}|r||z |z SdS)Nr)sum)sizestotalrel_sumabs_sums r_calc_kzDivider._calc_ks,!99Q<.57*<1 I#+Q3$> ( rcX||jz }||jz }||jz }||jz }|jjj|jj z \}}|j ||\} } } } |j|} |j|}|j| || z}|j||| z}|jrt||}|j| |}|j||}|d|dz |z }|d|dz |z }tjj| | | | }tjj| | ||}|j!|j#|j$\}}n(|j| |}|j||}| | }}|d}|d}||||z z||||z |z }}||||z z||||z |z }}tjj||||S)a  Implementation of ``divider.new_locator().__call__``. The axes locator callable returned by ``new_locator()`` is created as a `functools.partial` of this method with *nx*, *ny*, *nx1*, and *ny1* specifying the requested cell. r)rrr bboxsizedpir\r'r+rcrSminrgr7r8 from_boundsanchoredr?p0)rrlrmrnrorwr%fig_wfig_hxywrFhsizesvsizesk_hk_vrfoxoywwhhpbpb1x0y0x1w1y1h1s rrkzDivider._locates0 doo t doo tyy~~**TYY]]: u..tX> 1a**84((2ll6519-ll6519- ?? C A##FA.B##FA.BR&2a5.E)BR&2a5.E)B!!--aAq9B""..q!ROM ( !6 == 11 @T__ CEE.-<^ (Nrrrtc$eZdZdZddZdZdZy) AxesLocatorzc A callable object which returns the position and size of a given `.AxesDivider` cell. Nc||_|j}|j}||z ||z c|_|_|t |j}|t |jd}||z |_||z |_y)a Parameters ---------- axes_divider : `~mpl_toolkits.axes_grid1.axes_divider.AxesDivider` nx, nx1 : int Integers specifying the column-position of the cell. When *nx1* is None, a single *nx*-th column is specified. Otherwise, location of columns spanning between *nx* to *nx1* (but excluding *nx1*-th column) is specified. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. Nr) _axes_dividerrr_nx_nyr<_nx1_ny1)r axes_dividerrlrmrnrorrs rrzAxesLocator.__init__;s*!,, !,, *_b:o$( ;d(()C ;d((+,C*$ *$ rc|jj}|jj}|jj|j|z|j |z|j |z|j|z||Sr)rrrrxrrrr)rrwr%rrs r__call__zAxesLocator.__call__Wsr''22 ''22 !!((J)>)-J)>)-Z)?)-Z)?)-)1 3 3rc6|jjSr)rrCr0s rrCzAxesLocator.get_subplotspeccs!!1133rr)rrrrrrrCrBrrrr4s %8 34rrc@eZdZdZdddddfd ZdZdZdZxZS) SubplotDividerzT The Divider class whose rectangle area is specified as a subplot geometry. Nrrrrrc||_t| |gd|xsg|xsg|||jt j ||y)a Parameters ---------- fig : `~matplotlib.figure.Figure` *args : tuple (*nrows*, *ncols*, *index*) or int The array of subplots in the figure has dimensions ``(nrows, ncols)``, and *index* is the index of the subplot being created. *index* starts at 1 in the upper left corner and increases to the right. If *nrows*, *ncols*, and *index* are all single digit numbers, then *args* can be passed as a single 3-digit number (e.g. 234 for (2, 3, 4)). horizontal : list of :mod:`~mpl_toolkits.axes_grid1.axes_size`, optional Sizes for horizontal division. vertical : list of :mod:`~mpl_toolkits.axes_grid1.axes_size`, optional Sizes for vertical division. aspect : bool, optional Whether overall rectangular area is reduced so that the relative part of the horizontal and vertical scales have the same scale. anchor : (float, float) or {'C', 'SW', 'S', 'SE', 'E', 'NE', 'N', 'NW', 'W'}, default: 'C' Placement of the reduced rectangle, when *aspect* is True. )rrrrrN)figuresuperrset_subplotspecr_from_subplot_args)rrrrrrargs __class__s rrzSubplotDivider.__init__lsR6  l$.$4"x~2 &v  7 [;;CFGrch|jj|jjS)z%Return the bounds of the subplot box.)rCr1rrZr0s rr1zSubplotDivider.get_positions&##%224;;?FFFrc|jS)zGet the SubplotSpec instance.) _subplotspecr0s rrCzSubplotDivider.get_subplotspecs   rcf||_|j|j|jy)zSet the SubplotSpec instance.N)rr.r1r)r subplotspecs rrzSubplotDivider.set_subplotspecs'' +224;;?@r) rrrrrr1rCr __classcell__rs@rrrgs,/3TSHBG!ArrcfeZdZdZd fd ZdddZddZddZddddZd Z d Z d Z d Z xZ S) AxesDividerz0 Divider based on the preexisting axes. Nc||_|tj||_n||_|tj||_n||_t ||jd|jg|j gddy)zw Parameters ---------- axes : :class:`~matplotlib.axes.Axes` xref yref Nr)rrrrrr) _axesrAxesX_xrefAxesY_yrefrr get_figure)rrwrprqrs rrzAxesDivider.__init__sw <D)DJDJ <D)DJDJ T__.D%)ZZLDJJ< $S  2r) axes_classc |j}| t|}||j|jdfi|SNT)original)rtyperr1)rrkwargsrws r _get_new_axeszAxesDivider._get_new_axessHzz  dJ$//+T->->->-M$"$ $rc B| tjd|jz}|rdnd}|rMt|tj s!t j ||j}|j||t|tj s!t j ||j}|j|||j|rdnt|jdz |j}|jdi|}|j||S) z Helper method for ``append_axes("left")`` and ``append_axes("right")``. See the documentation of `append_axes` for more details. :meta private: zfigure.subplot.wspacerr fraction_refrrrlrmrB)mplrcParamsrr4r_Basefrom_anyrrsr<r rrset_axes_locatorrr|r pack_startrrrrr[s rnew_horizontalzAxesDivider.new_horizontals ;,,67$**DC"f c4::.mmCdjjA   S# &$ +==DJJ?D d#""qC(8(8$9A$=#  T   )& ) G$ rc B| tjd|jz}|rdnd}|rMt|tj s!t j ||j}|j||t|tj s!t j ||j}|j|||j|j|rdnt|jdz }|jdi|}|j||S) z Helper method for ``append_axes("top")`` and ``append_axes("bottom")``. See the documentation of `append_axes` for more details. :meta private: zfigure.subplot.hspacerrrrrrrB)rrrr4rrrrrsrr<r rrrs r new_verticalzAxesDivider.new_verticals ;,,67$**DC$h% c4::.mmCdjjA   S# &$ +==DJJ?D d#""qC$7!$;#= T   )& ) G$ rc tj|jdf|jdf|jdf|jdfd|\}}|||f||d|}|jj ||S)am Add a new axes on a given side of the main axes. Parameters ---------- position : {"left", "right", "bottom", "top"} Where the new axes is positioned relative to the main axes. size : :mod:`~mpl_toolkits.axes_grid1.axes_size` or float or str The axes width or height. float or str arguments are interpreted as ``axes_size.from_any(size, AxesX())`` for left or right axes, and likewise with ``AxesY`` for bottom or top axes. pad : :mod:`~mpl_toolkits.axes_grid1.axes_size` or float or str Padding between the axes. float or str arguments are interpreted as for *size*. Defaults to :rc:`figure.subplot.wspace` times the main Axes width (left or right axes) or :rc:`figure.subplot.hspace` times the main Axes height (bottom or top axes). axes_class : subclass type of `~.axes.Axes`, optional The type of the new axes. Defaults to the type of the main axes. **kwargs All extra keywords arguments are passed to the created axes. TFrr)rr)r check_getitemrrr add_axes) rrr|rrr create_axesrr[s r append_axeszAxesDivider.append_axess.#'"4"4(($/))51(($/%%u- 6  # Z  #O",OGMO 2 rct|j!|jj}|dk(ryy|jS)NautoFT)rrrSrPs rrSzAxesDivider.get_aspects6 << ZZ**,F<< rc|j(|jjd}|jS|jSr)r rr1rZ)rr{s rr1zAxesDivider.get_positions7 99 ::**D*9D;; 99 rcf|j|jjS|jSr)rrr?r0s rr?zAxesDivider.get_anchor"s) << ::((* *<< rc6|jjSr)rrCr0s rrCzAxesDivider.get_subplotspec(szz))++rr)NFr)rrrrrrrrrrSr1r?rCrrs@rrrsB2,+/$22 $ D  ,rrc ||z} ||z} t|} |j\} } |j\}}tjg| d}d|d| df<||dddf<g| | |j z }tj j ||^}}|| kDr| | z | z }tjdg||z|z}|d|dz |z }|d\}}|d|z|z|z }tjj||||}tjj||||}|j||j\}}||||fS)Nrrz) r<Tr diagr^linalgsolverer7r8rrr)rrrrF summed_widths equal_heightsrrr total_width max_heightneq_relseq_absssm_relssm_abssABkarrayheightrrh0_relh0_absrrrrrs rrkrk/s!)KJ MA$GW$GW  ' 1 AAbqb"fIAb#2#gJ2G82kGKKM12A iiooa+OVV w&'1 A56)G35 6B R&2a5.E !B"1%NFF )F V #u ,B    % %aAq 1B    & &q!R 4C \\&" % ( (FB r2r>rc*eZdZdZdfd ZdZxZS) HBoxDividerz A `.SubplotDivider` for laying out axes horizontally, while ensuring that they have equal heights. Examples -------- .. plot:: gallery/axes_grid1/demo_axes_hbox_divider.py c(t||d|dS)a Create an axes locator callable for the specified cell. Parameters ---------- nx, nx1 : int Integers specifying the column-position of the cell. When *nx1* is None, a single *nx*-th column is specified. Otherwise, location of columns spanning between *nx* to *nx1* (but excluding *nx1*-th column) is specified. rrrs)rrlrnrs rrszHBoxDivider.new_locatorZsw"2q#q11rc ||jz }||jz }|jjj|jjz \}}|j ||\} } } } |j |} |j|}t| | | | | ||||j \}}}}|d}||||z z||||z |z }}||}}tjj||||SNrz) rr r{r|r}r\r'r+rkr?r7r8r)rrlrmrnrorwr%rrrrrrF summed_wsequal_hsrrrrrrrrs rrkzHBoxDivider._locatehs doo tyy~~**TYY]]: u..tX> 1a--h7 **84 q!Q 8UE4??;LNBB ;Cbfun$r#wB'75&@BRB++BB;;rrrrrrrsrkrrs@rr r Ps 2 1a++H5 ,,X6 q!Q 8UE4??;LNBB ;CRBbfun$r#wB'75&@B++BB;;rrrrs@rrrys 28 !"%G $$hC1<%>r)NrN)rrinumpyr matplotlibrrmatplotlib.gridspecrmatplotlib.transforms transformsr7rrrrrrrrkr rrrrBrrr&s++aNaNH /4/4/4d1AW1AhN,'N,hB&<.&r