U d{jb@sdZddlZddlZddlmZddlmZddlm Z ddl m Z ddl mZGdd d ZGd d d ZGd d d eZGdddeZddZddZddZGdddeZGdddeZddZdddZdS) zJ Helper classes to adjust the positions of multiple axes at drawing time. N)_api) SubplotBase) SubplotSpec) axes_sizec@seZdZdZd5ddZddZdd Zed d d Z e d dZ e ddZ ddZ ddZddZddZddZddZddZdd Zd6d"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd7d,d-Zd8d.d/Zd0d1Zd9d3d4ZdS):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. 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 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'} Placement of the reduced rectangle, when *aspect* is True. rN) _fig_pos _horizontal _vertical_anchor set_anchor_aspect _xrefindex _yrefindex_locator)selffigpos horizontalverticalaspectanchorrc/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mpl_toolkits/axes_grid1/axes_divider.py__init__s zDivider.__init__csfdd|DS)Ncsg|]}|qSrget_size.0srendererrr ;sz0Divider.get_horizontal_sizes..)get_horizontalrr#rr"rget_horizontal_sizes:szDivider.get_horizontal_sizescsfdd|DS)Ncsg|]}|qSrrrr"rrr$>sz.Divider.get_vertical_sizes..) get_verticalr&rr"rget_vertical_sizes=szDivider.get_vertical_sizes3.5cCs$t|}t|}||fSN)SizeZAddListr(r%)rZvsizeZhsizerrrget_vsize_hsize@szDivider.get_vsize_hsizecCsFd\}}|D]\}}||7}||7}q |dkr>|||}|SdSdS)N)r.r.r)lZ total_sizeZrs_sumZas_sum_rs_askrrr_calc_kFs   zDivider._calc_kcCs2dg}|D]"\}}||d|||q |SNr.)append)r/r2offsetsr0r1rrr _calc_offsetsUs zDivider._calc_offsetscCs ||_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_position\s zDivider.set_positioncCs|jS)z%Return the position of the rectangle.r9rrrr get_positiongszDivider.get_positioncCsJt|trtjtjj|dn"t|ttfr8t |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 mtransformsBboxZcoefstuplelistlen TypeErrorr )rrrrrrks  zDivider.set_anchorcCs|jS)zReturn the anchor.)r r;rrr get_anchorszDivider.get_anchorcCs ||_dS)z Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for horizontal division Nr )rhrrrset_horizontalszDivider.set_horizontalcCs|jS)zReturn horizontal sizes.rHr;rrrr%szDivider.get_horizontalcCs ||_dS)z Parameters ---------- v : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for vertical division Nr )rvrrr set_verticalszDivider.set_verticalcCs|jS)zReturn vertical sizes.rKr;rrrr(szDivider.get_verticalFcCs ||_dS)zE Parameters ---------- aspect : bool Nrrrrrr set_aspectszDivider.set_aspectcCs|jS)zReturn aspect.rNr;rrr get_aspectszDivider.get_aspectcCs ||_dSr+r)rrrrr set_locatorszDivider.set_locatorcCs|jSr+rRr;rrr get_locatorszDivider.get_locatorcCs$|jdkr|S|||jSdSr+)rr<bounds)raxr#rrrget_position_runtimes zDivider.get_position_runtimecCs|jjj|jj\}}|||\} } } } ||} ||}|| || }|||| }|rt ||}| | |}| ||}|d|d|}|d|d|}t j | | | | }t j | | ||}|||j\}}n"| | |}| ||}| | }}|dkr.tjddd|d}|dkrNtjddd|d}|||||||||}}|||||||||}}t j ||||S) a 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. axes renderer r5rNr*Support for passing nx1=None to mean nx+1 is deprecated since %(since)s; in a future version, nx1=None will mean 'up to the last cell'.messagerSupport for passing ny1=None to mean ny+1 is deprecated since %(since)s; in a future version, ny1=None will mean 'up to the last cell'.)r bboxsizedpirWr'r)r3rQminr8rArB from_boundsanchoredrGp0rwarn_deprecated)rnxnynx1ny1axesr#fig_wfig_hxywrIZhsizesZvsizesZk_hZk_vr2oxoywwhhpbpb1x0y0x1w1y1h1rrrlocatesB          &&zDivider.locatecCs0t||||dk r|n|d|dk r&|n|dS)a Return a new `AxesLocator` 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 AxesLocator)rrdrerfrgrrr new_locatorszDivider.new_locatorcCstjddddg|d|dkr<|jd||jd7_nH|dkrR|j|n2|dkrx|jd||jd7_n |j|dS)Nleftrightbottomtoppositionrr)rr@r insertrr6r r)rrr]rrr append_sizeszDivider.append_size皙?cCs:|dkrddddg}|D]}||t|||qdS)a Add auto-adjustable padding around *use_axes* to take their decorations (title, labels, ticks, ticklabels) into account during layout. Parameters ---------- use_axes : `~.axes.Axes` or list of `~.axes.Axes` The Axes whose decorations are taken into account. pad : float, optional 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~rrr)rr,Z_AxesDecorationsSize)ruse_axespad adjust_dirsdrrradd_auto_adjustable_areas z Divider.add_auto_adjustable_area)Nr)F)NNNN)NN)rN)__name__ __module__ __qualname____doc__rr'r)r deprecatedr- staticmethodr3r8r:r<rrGrJr%rMr(rPrQrSrTrWrzr}rrrrrrrs:         9 rc@s*eZdZdZd ddZddZddZdS) r|z` A callable object which returns the position and size of a given AxesDivider cell. NcCs|||_|j}|j}|||||_|_|dkrFtjddd|d}|dkrdtjddd|d}|||_|||_dS)a Parameters ---------- 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*rXrYrr[) _axes_dividerrr_nx_nyrrc_nx1_ny1)rZ axes_dividerrdrerfrgrrrrrr)s$  zAxesLocator.__init__cCs>|jj}|jj}|j|j||j||j||j|||Sr+)rrrrzrrrr)rrhr#rrrrr__call__MszAxesLocator.__call__cCst|jdr|jSdSdSNget_subplotspec)hasattrrrr;rrrrYs  zAxesLocator.get_subplotspec)NN)rrrrrrrrrrrr|#s $ r|csDeZdZdZdddddfdd ZddZd d Zd d ZZS) SubplotDividerzT The Divider class whose rectangle area is specified as a subplot geometry. NrrrrrcsD||_tj|ddddg|pg|p$g||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)). rrrN)figuresuperrset_subplotspecrZ_from_subplot_args)rrrrrrargs __class__rrreszSubplotDivider.__init__cCs||jjS)z%Return the bounds of the subplot box.)rr<rrUr;rrrr<|szSubplotDivider.get_positioncCs|jS)zGet the SubplotSpec instance.) _subplotspecr;rrrrszSubplotDivider.get_subplotspeccCs||_|||jdS)zSet the SubplotSpec instance.N)rr:r<r)rZ subplotspecrrrrszSubplotDivider.set_subplotspec) rrrrrr<rr __classcell__rrrrr`srcseZdZdZdfdd ZddddZdd d Zdd d Zej d ddddddddZ ddZ ddZ ddZ ddZZS) AxesDividerz0 Divider based on the preexisting axes. Ncsf||_|dkrt||_n||_|dkr8t||_n||_tj|d|jg|jgddddS)zw Parameters ---------- axes : :class:`~matplotlib.axes.Axes` xref yref Nr)rrrrrr) _axesr,ZAxesX_xrefZAxesY_yrefrr get_figure)rrhZxrefZyrefrrrrszAxesDivider.__init__) axes_classcKsB|j}|dkr(t|tr |j}nt|}|||jddf|SNT)original)rr>rZ _axes_classtyperr<)rrkwargsrhrrr _get_new_axess zAxesDivider._get_new_axesFcKs|dkrtjd|j}|rft|tjs8tj||jd}|rZ|jd||j d7_ n |j |t|tjstj||jd}|r|jd||j d7_ |j d|j d}n&|j ||j t |jd|j d}|jf|}|||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.wspaceZ fraction_refrrrdre)mplrcParamsrr>r,_Basefrom_anyr rrr6r}rrErset_axes_locatorrr]r pack_startrlocatorrVrrrnew_horizontals.       zAxesDivider.new_horizontalcKs|dkrtjd|j}|rft|tjs8tj||jd}|rZ|jd||j d7_ n |j |t|tjstj||jd}|r|jd||j d7_ |j |j dd}n&|j ||j |j t |jdd}|jf|}|||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.hspacerrrr)rrrr>r,rrr rrr6r}rrErrrrrr new_verticals.       zAxesDivider.new_verticalr* add_to_figurez ax.remove()) alternativeTc Ks`tj|jdf|jdf|jdf|jdfd|d\}}|||f||d|} |r\|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. TF)r~rrrr)rr)rZ check_getitemrrr Zadd_axes) rrr]rrrrZ create_axesrrVrrr append_axess(  zAxesDivider.append_axescCs0|jdkr&|j}|dkr dSdSn|jSdS)NautoFT)rrrQrOrrrrQs   zAxesDivider.get_aspectcCs(|jdkr|jjdd}|jS|jSdSr)r rr<rU)rr\rrrr<s zAxesDivider.get_positioncCs|jdkr|jS|jSdSr+)r rrGr;rrrrG$s  zAxesDivider.get_anchorcCst|jdr|jSdSdSr)rrrr;rrrr*s  zAxesDivider.get_subplotspec)NN)NF)NF)NT)rrrrrrrrrZdelete_parameterrrQr<rGrrrrrrrs # rcCst|}t|j\}}t|j\}}t|d|df} t|d} t| d|d|f|d| d|df<|| dddf<| | d|<|t|| d<tj| | } | dd} | d} | |kr|||} | S)Nrr5) rEnpZasarrayTZzerosZ fill_diagonalsumZlinalgZsolve) summed_widths equal_heights total_width max_heightnZeq_rsZeq_asZsm_rsZsm_asABZkarray_and_heightkarrayheightrrr_determine_karray4s   rcCs<dg}t||D]&\\}}}||d|||q|Sr4)zipr6)Z summed_sizesrr7rar2rrrr8Lsr8c Cst||||||d} t|| } | d| d|} |d\} } | d| | |}tj||||}tj||| |}|||j\}}||| |fS)N)rrr5r)rr8rArBr`rarb)rkrlrmrIrrrirjrrrnrpZh0_rZh0_arqrrrsrtrurrr_locateTs  rc@s$eZdZdZdddZdddZdS) 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 NcCs t||d|dk r|n|ddS)a Create a new `AxesLocator` 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. rNrr{)rrdrfrrrr}ns zHBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|dkrt j ddd|d}|||||||||}}||}}t j ||||S)Nr*rXrYr)r r\r]r^rWr'r)rrGrrcrArBr`)rrdrerfrgrhr#rirjrkrlrmrIZ summed_wsZequal_hsrtrurnrqrvrwrxryrrrrz|s0   & zHBoxDivider.locate)N)NNNNrrrrr}rzrrrrrds rc@s$eZdZdZdddZdddZdS) VBoxDividerzl A `SubplotDivider` for laying out axes vertically, while ensuring that they have equal widths. NcCs t|d|d|dk r|n|dS)a Create a new `AxesLocator` 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. rrNr{)rrergrrrr}s zVBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|dkrt j ddd|d}||}}|||||||||}}t j ||||S)Nr*r[rYr)r r\r]r^rWr)r'rrGrrcrArBr`)rrdrerfrgrhr#rirjrkrlrmrIZ summed_hsZequal_wsrurtrorprvrwrxryrrrrzs0    &zVBoxDivider.locate)N)NNNNrrrrrrs rcCs$t|}|jddd}|||S)Nrr)rr}r)rhdividerrrrrmake_axes_locatables rrcCs<|dkrddddg}t|}|dkr(|}|j|||ddS)aF 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)rrr)rr)rVrrrrrrrmake_axes_area_auto_adjustables  r)NrN)rnumpyrZ matplotlibrrZmatplotlib.axesrZmatplotlib.gridspecrZmatplotlib.transformsZ transformsrArr,rr|rrrr8rrrrrrrrrs0     =*++'