a Ebg@sdZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ddlmZGdd d ZGd d d ZGd d d eZGdddeZddZddZddZGdddeZGdddeZddZdddZdS) zJ Helper classes to adjust the positions of multiple axes at drawing time. N)_api) SubplotBase) SubplotSpecGridSpec) 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. NCcCs:||_||_||_||_||_||_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 : {'C', 'SW', 'S', 'SE', 'E', 'NE', 'N', 'NW', 'W'} Placement of the reduced rectangle, when *aspect* is True. rN) _fig_pos _horizontal _vertical_anchor_aspect _xrefindex _yrefindex_locator)selffigpos horizontalverticalaspectanchorrd/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/mpl_toolkits/axes_grid1/axes_divider.py__init__szDivider.__init__csfdd|DS)Ncsg|]}|qSrget_size.0srendererrr 9z0Divider.get_horizontal_sizes..)get_horizontalrr#rr"rget_horizontal_sizes8szDivider.get_horizontal_sizescsfdd|DS)Ncsg|]}|qSrrrr"rrr$<r%z.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)l total_sizeZrs_sumZas_sum_rs_askrrr_calc_kDs   zDivider._calc_kcCs2dg}|D]"\}}||d|||q |SNr/)append)r0r4offsetsr2r3rrr _calc_offsetsSs 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_positionZs zDivider.set_positioncCs|jS)z%Return the position of the rectangle.r;rrrr get_positioneszDivider.get_positioncCs(t|dkrtjtjj|d||_dS)a Parameters ---------- anchor : (float, float) or {'C', 'SW', 'S', 'SE', 'E', 'NE', ...} 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 )rN)lenr check_in_list mtransformsBboxZcoefsr)rrrrr set_anchoris zDivider.set_anchorcCs|jS)zReturn the anchor.)rr=rrr get_anchorzszDivider.get_anchorcCs ||_dS)z Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for horizontal division Nr )rhrrrset_horizontal~szDivider.set_horizontalcCs|jS)zReturn horizontal sizes.rFr=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.rIr=rrrr)szDivider.get_verticalFcCs ||_dS)zE Parameters ---------- aspect : bool Nrrrrrr set_aspectszDivider.set_aspectcCs|jS)zReturn aspect.rLr=rrr get_aspectszDivider.get_aspectcCs ||_dSr,r)rrrrr set_locatorszDivider.set_locatorcCs|jSr,rPr=rrr get_locatorszDivider.get_locatorcCs$|jdur|S|||jSdSr,)rr>bounds)raxr#rrrget_position_runtimes zDivider.get_position_runtimecCs|jjj|jj\}}|||\} } } } ||} ||}|| || }|||| }|rt ||}| | |}| ||}|d|d|}|d|d|}t j | | | | }t j | | ||}|||}|j|j}}n"| | |}| ||}| | }}|dur6tjddd|d}|durVtjddd|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 r7rNr+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 bboxsizedpirUr(r*r5rOminr:rBrC from_boundsanchoredrEx0y0rwarn_deprecated)rnxnynx1ny1axesr#fig_wfig_hxywrGZhsizesZvsizesZk_hZk_vr4oxoywwhhpbpb1 pb1_anchoredr`rax1w1y1h1rrrlocates@          &&zDivider.locatecCs0t||||dur|n|d|dur&|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)rrcrdrerfrrr new_locators zDivider.new_locatorcCs|dkr&|jd||jd7_nd|dkr<|j|nN|dkrb|jd||jd7_n(|dkrx|j|ntjgd|ddS) Nleftrrrightbottomtopr|r}r~rposition)r insertrr8r rrrA)rrr[rrr append_sizes zDivider.append_size皙?c CsZ|durgd}ddlm}m}m}|D],}|||}||} || |} ||| q(dS)Nrr)Padded SizeFromFuncGetExtentHelper)rrrrr) ruse_axespad adjust_dirsrrrdhelperr[Z padded_sizerrradd_auto_adjustable_area s  z Divider.add_auto_adjustable_area)Nr )F)NNNN)NN)rN)__name__ __module__ __qualname____doc__rr(r*r deprecatedr. staticmethodr5r:r<r>rDrErHr&rKr)rNrOrQrRrUrxr{rrrrrrrs8         : rc@s*eZdZdZd ddZddZddZdS) rzz` A callable object which returns the position and size of a given AxesDivider cell. NcCs|||_|j}|j}|||||_|_|durFtjddd|d}|durdtjddd|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+rVrWrrY) _axes_dividerrr_nx_nyrrb_nx1_ny1)rZ axes_dividerrcrdrerfrrrrrrs  zAxesLocator.__init__cCs>|jj}|jj}|j|j||j||j||j|||Sr,)rrrrxrrrr)rrgr#rrrrr__call__AszAxesLocator.__call__cCst|jdr|jSdSdSNget_subplotspec)hasattrrrr=rrrrMs  zAxesLocator.get_subplotspec)NN)rrrrrrrrrrrrzs $ rzcseZdZdZdddddfdd ZddZed ed d Z ed d d Z ejd dddddZ ejd ddddZ ddZ ddZZS)SubplotDividerzT The Divider class whose rectangle area is specified as a subplot geometry. Nr rrrrcs@||_tj|gd|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)). )rrrrrN)figuresuperrset_subplotspecrZ_from_subplot_args)rrrrrrargs __class__rrrYs  zSubplotDivider.__init__cCs||jjS)z%Return the bounds of the subplot box.)rr>rrSr=rrrr>pszSubplotDivider.get_positionz3.4cCs||jSr,)rr>rr=rrrfigboxtszSubplotDivider.figboxcCsdSr,rr=rrr update_paramsyszSubplotDivider.update_paramsrz1(get_subplotspec returns a SubplotSpec instance.)) alternativeZaddendumcCs"|\}}}}|||dfS)z*Get the subplot geometry, e.g., (2, 2, 3).r)r get_geometry)rrowscolsZnum1Znum2rrrr}szSubplotDivider.get_geometryrrcCs,t|||d|_|||jdS)z;Change subplot geometry, e.g., from (1, 1, 1) to (2, 2, 3).rN)r _subplotspecrr<r)rZnumrowsZnumcolsnumrrrchange_geometryszSubplotDivider.change_geometrycCs|jS)zGet the SubplotSpec instance.)rr=rrrrszSubplotDivider.get_subplotspeccCs||_|||jdS)zSet the SubplotSpec instance.N)rr<r>r)rZ subplotspecrrrrszSubplotDivider.set_subplotspec)rrrrrr>rrpropertyrrrrrr __classcell__rrrrrTs&     rcs~eZdZdZdfdd ZddddZdd d Zdd d Zej d ddddddZ ddZ ddZ ddZ ddZZS) AxesDividerz1 Divider based on the pre-existing axes. Ncsf||_|durt||_n||_|dur8t||_n||_tj|d|jg|jgddddS)zw Parameters ---------- axes : :class:`~matplotlib.axes.Axes` xref yref Nr )rrrrrr) _axesr-ZAxesX_xrefZAxesY_yrefrr get_figure)rrgZxrefZyrefrrrrs zAxesDivider.__init__) axes_classcKsF|j}|dur(t|tr |j}nt|}|||jddfi|SNT)original)r isinstancerZ _axes_classtyperr>)rrkwargsrgrrr _get_new_axess zAxesDivider._get_new_axesFcKs|durtjd|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}|jfi|}|||S)ak Add a new axes on the right (or left) side of the main axes. Parameters ---------- size : :mod:`~mpl_toolkits.axes_grid1.axes_size` or float or str The axes width. float or str arguments are interpreted as ``axes_size.from_any(size, AxesX())``. pad : :mod:`~mpl_toolkits.axes_grid1.axes_size` or float or str Padding between the axes. float or str arguments are interpreted as ``axes_size.from_any(size, AxesX())``. Defaults to :rc:`figure.subplot.wspace` times the main axes width. pack_start : bool If False, the new axes is appended at the end of the list, i.e., it became the right-most axes. If True, it is inserted at the start of the list, and becomes the left-most axes. **kwargs All extra keywords arguments are passed to the created axes. If *axes_class* is given, the new axes will be created as an instance of the given class. Otherwise, the same class of the main axes will be used. Nzfigure.subplot.wspaceZ fraction_refrrrcrd)mplrcParamsrrr-_Basefrom_anyr rrr8r{rr@rset_axes_locatorrr[r pack_startrlocatorrTrrrnew_horizontals,     zAxesDivider.new_horizontalcKs|durtjd|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}|jfi|}|||S)am Add a new axes on the top (or bottom) side of the main axes. Parameters ---------- size : :mod:`~mpl_toolkits.axes_grid1.axes_size` or float or str The axes height. float or str arguments are interpreted as ``axes_size.from_any(size, AxesY())``. pad : :mod:`~mpl_toolkits.axes_grid1.axes_size` or float or str Padding between the axes. float or str arguments are interpreted as ``axes_size.from_any(size, AxesY())``. Defaults to :rc:`figure.subplot.hspace` times the main axes height. pack_start : bool If False, the new axes is appended at the end of the list, i.e., it became the right-most axes. If True, it is inserted at the start of the list, and becomes the left-most axes. **kwargs All extra keywords arguments are passed to the created axes. If *axes_class* is given, the new axes will be created as an instance of the given class. Otherwise, the same class of the main axes will be used. Nzfigure.subplot.hspacerrrr)rrrrr-rrr rrr8r{rr@rrrrrr new_verticals,     zAxesDivider.new_verticalr+ add_to_figurez ax.remove()rTcKs|dkr"|j||fddi|}nx|dkrD|j||fddi|}nV|dkrf|j||fddi|}n4|dkr|j||fddi|}ntjgd|d |r|j||S) z Create an axes at the given *position* with the same height (or width) of the main axes. *position* ["left"|"right"|"bottom"|"top"] *size* and *pad* should be axes_grid.axes_size compatible. r|rTr}Fr~rrr)rrrrAr Zadd_axes)rrr[rrrrTrrr append_axess   zAxesDivider.append_axescCs0|jdur&|j}|dkr dSdSn|jSdS)NautoFT)rrrOrMrrrrO4s   zAxesDivider.get_aspectcCs(|jdur|jjdd}|jS|jSdSr)r rr>rS)rrZrrrr>>s zAxesDivider.get_positioncCs|jdur|jS|jSdSr,)rrrEr=rrrrEEs  zAxesDivider.get_anchorcCst|jdr|jSdSdSr)rrrr=rrrrKs  zAxesDivider.get_subplotspec)NN)NF)NF)NT)rrrrrrrrrZdelete_parameterrrOr>rErrrrrrrs / /  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)Nrr7) r@npZasarrayTZzerosZ fill_diagonalsumZlinalgZsolve) summed_widths equal_heights total_width max_heightnZeq_rsZeq_asZsm_rsZsm_asABZkarray_and_heightkarrayheightrrr_determine_karrayUs   rcCs<dg}t||D]&\\}}}||d|||q|Sr6)zipr8)Z summed_sizesrr9rar4rrrr:msr:c Cst||||||d} t|| } | d| d|} |d\} } | d| | |}tj||||}tj||| |}|||}|j|j}}||| |fS)N)rrr7r)rr:rBrCr^r_r`ra)rjrkrlrGrrrhrirrrmroZh0_rZh0_arprqrrrsr`rarrr_locateus    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|dur|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. rNrry)rrcrerrrr{s zHBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|durt j ddd|d}|||||||||}}||}}t j ||||S)Nr+rVrWr)r rZr[r\rUr(r*rrErrbrBrCr^)rrcrdrerfrgr#rhrirjrkrlrGZ summed_wsZequal_hsr`rarmrprtrurvrwrrrrxs   & zHBoxDivider.locate)N)NNNNrrrrr{rxrrrrrs rc@s$eZdZdZdddZdddZdS) VBoxDividerzl A `SubplotDivider` for laying out axes vertically, while ensuring that they have equal widths. NcCs t|d|d|dur|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. rrNry)rrdrfrrrr{s zVBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|durt j ddd|d}||}}|||||||||}}t j ||||S)Nr+rYrWr)r rZr[r\rUr*r(rrErrbrBrCr^)rrcrdrerfrgr#rhrirjrkrlrGZ summed_hsZequal_wsrar`rnrortrurvrwrrrrxs    &zVBoxDivider.locate)N)NNNNrrrrrrs rcCs$t|}|jddd}|||S)Nrr)rr{r)rgdividerrrrrmake_axes_locatables rrcCs8|durgd}t|}|dur$|}|j|||ddS)Nr)rrr)rr)rTrrrrrrrmake_axes_area_auto_adjustablesr)NrN)rnumpyrZ matplotlibrrZmatplotlib.axesrZmatplotlib.gridspecrrZmatplotlib.transformsZ transformsrBrr-rrzrrrr:rrrrrrrrrs.     =B@+'