o - hU@sdZddlZddlZddlZddlmZddlmZddl m Z ddl m ZGdddZGd d d eZGd d d eZd dZGdddeZGdddeZddZ dddZdS)zJ Helper classes to adjust the positions of multiple axes at drawing time. N)_api) SubplotSpec) axes_sizec@seZdZdZ d4ddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZd5ddZdd Zd!d"Zd#d$Zd%d&Zed'd(Zed)d*Zd6d+d,Zd-d.Zd/d0Zd7d2d3ZdS)8DivideraQ 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. NCcCsD||_||_||_||_||_||||_d|_d|_d|_ dS)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 horizontalverticalaspectanchorrX/var/www/vscode/kcb/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/axes_divider.py__init__s  zDivider.__init__ctfdd|DS)Ncg|]}|qSrget_size.0srendererrr <z0Divider.get_horizontal_sizes..)nparrayget_horizontalrr$rr#rget_horizontal_sizes;zDivider.get_horizontal_sizescr)Ncrrrr r#rrr%?r&z.Divider.get_vertical_sizes..)r'r( get_verticalr*rr#rget_vertical_sizes>r,zDivider.get_vertical_sizescC ||_dS)z Set the position of the rectangle. Parameters ---------- pos : tuple of 4 floats position of the rectangle that will be divided Nr )rrrrr set_positionAs zDivider.set_positioncC|jS)z%Return the position of the rectangle.r0rrrr get_positionLzDivider.get_positioncCsJt|trtjtjj|dnt|ttfrt |dkr t d||_ dS)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 )rrrrrr Ps  zDivider.set_anchorcCr2)zReturn the anchor.)r r3rrr get_anchordr5zDivider.get_anchorcCsdSNrr3rrrget_subplotspechszDivider.get_subplotspeccCr/)z Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for horizontal division Nr )rhrrrset_horizontalk zDivider.set_horizontalcCr2)zReturn horizontal sizes.rDr3rrrr)tr5zDivider.get_horizontalcCr/)z Parameters ---------- v : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for vertical division Nr )rvrrr set_verticalxrGzDivider.set_verticalcCr2)zReturn vertical sizes.rHr3rrrr-r5zDivider.get_verticalFcCr/)zE Parameters ---------- aspect : bool Nrrrrrr set_aspects zDivider.set_aspectcCr2)zReturn aspect.rKr3rrr get_aspectr5zDivider.get_aspectcCs ||_dSrBr)rrrrr set_locator zDivider.set_locatorcCr2rBrOr3rrr get_locatorszDivider.get_locatorcCs |jdur |S|||jSrB)rr4bounds)raxr$rrrget_position_runtimes zDivider.get_position_runtimecCs"|d\}}|r|||SdS)Nr)sum)sizestotalrel_sumabs_sumrrr_calc_kszDivider._calc_kcCstdg||dgS)Nrr)r'cumsum)rWkrrr _calc_offsetsszDivider._calc_offsetscCs\|dur|d}|dur|d}|j}|j}t|j||||||||}|j|_|S)a Return 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. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. Nr)rr functoolspartial_locaterC)rnxnynx1ny1xrefyreflocatorrrr new_locatorszDivider.new_locatorcCs||j7}||j7}||j7}||j7}|jjj|jj\}}|||\} } } } ||} ||}| | || }| ||| }| rt ||}| | |}| ||}|d|d|}|d|d|}t j| | | | }t j| | ||}|||j\}}n| | |}| ||}| | }}|durd}|durd}|||||||||}}|||||||||}}t j||||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. rN)rrrbboxsizedpirUr+r.r[rNminr^r:r; from_boundsanchoredrAp0)rrbrcrdreaxesr$fig_wfig_hxywrEhsizesvsizesk_hk_vr]oxoywwhhpbpb1x0y0x1w1y1h1rrrras:            &&zDivider._locatecCstjgd|d|dkr|jd||jd7_dS|dkr)|j|dS|dkr=|jd||jd7_dS|j|dS)Nleftrightbottomtoppositionrrrrr)rr9r insertrappendr r)rrrlrrr append_sizes zDivider.append_size皙?cCs6|durgd}|D]}||t|||q dS)a5 Add auto-adjustable padding around *use_axes* to take their decorations (title, labels, ticks, ticklabels) into account during layout. Parameters ---------- use_axes : `~matplotlib.axes.Axes` or list of `~matplotlib.axes.Axes` The Axes whose decorations are taken into account. pad : float, default: 0.1 Additional padding in inches. adjust_dirs : list of {"left", "right", "bottom", "top"}, optional The sides where padding is added; defaults to all four sides. Nr)rSize_AxesDecorationsSize)ruse_axespad adjust_dirsdrrradd_auto_adjustable_areas z Divider.add_auto_adjustable_area)Nr)FNN)rN)__name__ __module__ __qualname____doc__rr+r.r1r4r rArCrFr)rJr-rMrNrPrRrU staticmethodr[r^rirarrrrrrrs8        /rcsDeZdZdZdddddfdd ZddZd d Zd d ZZS) SubplotDividerzT The Divider class whose rectangle area is specified as a subplot geometry. Nrrrrrcs@||_tj|gd|p g|pg||d|t||dS)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__rrr s  zSubplotDivider.__init__cCs||jjS)z%Return the bounds of the subplot box.)rCr4rrSr3rrrr4AszSubplotDivider.get_positioncCr2)zGet the SubplotSpec instance.) _subplotspecr3rrrrCEr5zSubplotDivider.get_subplotspeccCs||_|||jdS)zSet the SubplotSpec instance.N)rr1r4r)r subplotspecrrrrIszSubplotDivider.set_subplotspec) rrrrrr4rCr __classcell__rrrrrs!rcsteZdZdZdfdd ZddddZdd d Zdd d Zdddd dZddZ ddZ ddZ ddZ Z S) AxesDividerz0 Divider based on the preexisting axes. Ncsf||_|durt||_n||_|durt||_n||_tj|d|jg|jgddddS)zw Parameters ---------- axes : :class:`~matplotlib.axes.Axes` xref yref Nr)rrrrrr) _axesrAxesX_xrefAxesY_yrefrr get_figure)rrrrfrgrrrrTs  zAxesDivider.__init__) axes_classcKs4|j}|dur t|}|||jddfi|SNT)original)rtyperr4)rrkwargsrrrrr _get_new_axesjs zAxesDivider._get_new_axesFcKs|dur tjd|j}|rdnd}|r(t|tjs"tj||jd}|||t|tjs6tj||jd}||||j|rBdnt |j d|j d}|j d i|}| ||S) z Helper method for ``append_axes("left")`` and ``append_axes("right")``. See the documentation of `append_axes` for more details. :meta private: Nzfigure.subplot.wspacerr fraction_refrrrbrcr)mplrcParamsrr7r_Basefrom_anyrrir?r rrset_axes_locatorrrlr pack_startrrrhrTrrrnew_horizontalqs"      zAxesDivider.new_horizontalcKs|dur tjd|j}|rdnd}|r(t|tjs"tj||jd}|||t|tjs6tj||jd}||||j|j |rDdnt |j dd}|j d i|}| ||S) z Helper method for ``append_axes("top")`` and ``append_axes("bottom")``. See the documentation of `append_axes` for more details. :meta private: Nzfigure.subplot.hspacerrrrrrr)rrrr7rrrrrirr?r rrrrrr new_verticals"      zAxesDivider.new_verticalc Ks\tj|jdf|jdf|jdf|jdfd|d\}}|||f||d|}|j||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_getitemrrradd_axes) rrrlrrr create_axesrrTrrr append_axess"  zAxesDivider.append_axescCs*|jdur|j}|dkrdSdS|jS)NautoFT)rrrNrLrrrrNs  zAxesDivider.get_aspectcCs$|jdur|jjdd}|jS|jSr)r rr4rS)rrkrrrr4s zAxesDivider.get_positioncCs|jdur |jS|jSrB)r rrAr3rrrrAs  zAxesDivider.get_anchorcCs |jSrB)rrCr3rrrrCrQzAxesDivider.get_subplotspecr)NFrB)rrrrrrrrrrNr4rArCrrrrrrOs  " rc Cs*||} ||} t|} |j\} } |j\}}tg| d}d|d| df<||dddf<g| | |}tj||^}}|| krN| | | }tdg|||}|d|d|}|d\}}|d|||}tj ||||}tj ||||}| ||j \}}||||fS)Nrrj) r?Tr'diagrVlinalgsolver\r:r;rorprq)rurvrwrE summed_widths equal_heightsrsrtr total_width max_heightneq_relseq_absssm_relssm_abssABkarrayheightr|r~h0_relh0_absrrrrrrrrras(     rac*eZdZdZdfdd ZddZZS) 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 Ncst|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. rrri)rrbrdrrrri zHBoxDivider.new_locatorc Cs||j7}||j7}|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|durAd}|||||||||}}||}}t j ||||SNrj) rrrkrlrmrUr+r.rarAr:r;ro)rrbrcrdrerrr$rsrtrurvrwrE summed_wsequal_hsrrr|rrrrrrrrras     & zHBoxDivider._locaterBrrrrrirarrrrrrs rcr) VBoxDividerzm A `.SubplotDivider` for laying out axes vertically, while ensuring that they have equal widths. Ncstd|d|S)a Create an axes locator callable for the specified cell. Parameters ---------- ny, ny1 : int Integers specifying the row-position of the cell. When *ny1* is None, a single *ny*-th row is specified. Otherwise, location of rows spanning between *ny* to *ny1* (but excluding *ny1*-th row) is specified. rr)rrcrerrrri3rzVBoxDivider.new_locatorc Cs||j7}||j7}|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|durAd}||}}|||||||||}}t j ||||Sr) rrrkrlrmrUr.r+rarAr:r;ro)rrbrcrdrerrr$rsrtrurvrwrE summed_hsequal_wsrrr}r~rrrrrrrraAs      &zVBoxDivider._locaterBrrrrrr-srcCs$t|}|jddd}|||S)Nrr)rrir)rrdividerrhrrrmake_axes_locatableRs rrcCs8|durgd}t|}|dur|}|j|||ddS)aG Add auto-adjustable padding around *ax* to take its decorations (title, labels, ticks, ticklabels) into account during layout, using `.Divider.add_auto_adjustable_area`. By default, padding is determined from the decorations of *ax*. Pass *use_axes* to consider the decorations of other Axes instead. Nr)rrr)rr)rTrrrrrrrmake_axes_area_auto_adjustableZs  r)NrN)rr_numpyr' matplotlibrrmatplotlib.gridspecrmatplotlib.transforms transformsr:rrrrrrarrrrrrrrs(     4!)%