o tBhg@sdZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ddlmZGdd d ZGd d d ZGd d d eZGdddeZddZddZddZGdddeZGdddeZddZ  dddZdS) zJ Helper classes to adjust the positions of multiple axes at drawing time. N)_api) SubplotBase) SubplotSpecGridSpec) axes_sizec@seZdZdZ d5ddZddZdd 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 horizontalverticalaspectanchorr{/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/axes_divider.py__init__s zDivider.__init__cfdd|DS)Ncg|]}|qSrget_size.0srendererrr 9z0Divider.get_horizontal_sizes..)get_horizontalrr%rr$rget_horizontal_sizes8zDivider.get_horizontal_sizescr)Ncrrrr!r$rrr&<r'z.Divider.get_vertical_sizes..) get_verticalr)rr$rget_vertical_sizes;r+zDivider.get_vertical_sizes3.5cCs$t|}t|}||fSN)SizeAddListr,r()rvsizehsizerrrget_vsize_hsize>szDivider.get_vsize_hsizecCsBd\}}|D] \}}||7}||7}q|dkr|||}|SdS)N)r5r5r)l total_sizers_sumas_sum_rs_askrrr_calc_kDs   zDivider._calc_kcCs2dg}|D]\}}||d|||q|SNr5)append)r6r<offsetsr:r;rrr _calc_offsetsSs zDivider._calc_offsetscC ||_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_positioncC|jS)z%Return the position of the rectangle.rDrrrr get_positionezDivider.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 mtransformsBboxcoefsr)rrrrr set_anchoris  zDivider.set_anchorcCrF)zReturn the anchor.)rrGrrr get_anchorzrIzDivider.get_anchorcCrC)z Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for horizontal division Nr )rhrrrset_horizontal~ zDivider.set_horizontalcCrF)zReturn horizontal sizes.rRrGrrrr(rIzDivider.get_horizontalcCrC)z Parameters ---------- v : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for vertical division Nr )rvrrr set_verticalrUzDivider.set_verticalcCrF)zReturn vertical sizes.rVrGrrrr,rIzDivider.get_verticalFcCrC)zE Parameters ---------- aspect : bool Nrrrrrr set_aspects zDivider.set_aspectcCrF)zReturn aspect.rYrGrrr get_aspectrIzDivider.get_aspectcCs ||_dSr/r)rrrrr set_locators zDivider.set_locatorcCrFr/r]rGrrr get_locatorszDivider.get_locatorcCs |jdur |S|||jSr/)rrHbounds)raxr%rrrget_position_runtimes zDivider.get_position_runtimecCs|jjj|jj\}}|||\} } } } ||} ||}|| || }|||| }|rzt ||}| | |}| ||}|d|d|}|d|d|}t j | | | | }t j | | ||}|||}|j|j}}n| | |}| ||}| | }}|durtjddd|d}|durtjddd|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 r?rNr.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 bboxsizedpirbr*r-r=r\minrBrMrN from_boundsanchoredrQx0y0rwarn_deprecated)rnxnynx1ny1axesr%fig_wfig_hxywrShsizesvsizesk_hk_vr<oxoywwhhpbpb1 pb1_anchoredrmrnx1w1y1h1rrrlocates@        &&zDivider.locatecCs2t||||dur |n|d|dur|S|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)rrprqrrrsrrr new_locators zDivider.new_locatorcCs|dkr|jd||jd7_dS|dkr |j|dS|dkr4|jd||jd7_dS|dkr@|j|dStjgd|ddS) Nleftrrrightbottomtoprrrrposition)r insertrr@r rrrL)rrrhrrr append_sizes  zDivider.append_size皙?c CsZ|durgd}ddlm}m}m}|D]}|||}||} || |} ||| qdS)Nrr)Padded SizeFromFuncGetExtentHelper)rrrrr) ruse_axespad adjust_dirsrrrdhelperrh padded_sizerrradd_auto_adjustable_area s  z Divider.add_auto_adjustable_area)Nr )FNNNNNN)rN)__name__ __module__ __qualname____doc__rr*r-r deprecatedr4 staticmethodr=rBrErHrPrQrTr(rXr,r[r\r^r_rbrrrrrrrrrs:         :rc@s*eZdZdZd ddZddZddZdS) rz` A callable object which returns the position and size of a given AxesDivider cell. NcCs|||_|j}|j}|||||_|_|dur#tjddd|d}|dur2tjddd|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.rcrdrrf) _axes_dividerrr_nx_nyrro_nx1_ny1)r axes_dividerrprqrrrsrrrrrrs   zAxesLocator.__init__cCs>|jj}|jj}|j|j||j||j||j|||Sr/)rrrrrrrr)rrtr%rrrrr__call__AszAxesLocator.__call__cCt|jdr |jSdSNget_subplotspec)hasattrrrrGrrrrM  zAxesLocator.get_subplotspecr)rrrrrrrrrrrrs  $ rcseZdZdZdddddfdd 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|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)). )rrrrrN)figuresuperrset_subplotspecr_from_subplot_args)rrrrrrargs __class__rrrYs  zSubplotDivider.__init__cCs||jjS)z%Return the bounds of the subplot box.)rrHrr`rGrrrrHpszSubplotDivider.get_positionz3.4cCs||jSr/)rrHrrGrrrfigboxtszSubplotDivider.figboxcCsdSr/rrGrrr update_paramsyszSubplotDivider.update_paramsrz1(get_subplotspec returns a SubplotSpec instance.)) alternativeaddendumcCs"|\}}}}|||dfS)z*Get the subplot geometry, e.g., (2, 2, 3).r)r get_geometry)rrowscolsnum1num2rrrr}szSubplotDivider.get_geometryrrcCs,t|||d|_|||jdS)z;Change subplot geometry, e.g., from (1, 1, 1) to (2, 2, 3).rN)r _subplotspecrrEr)rnumrowsnumcolsnumrrrchange_geometryszSubplotDivider.change_geometrycCrF)zGet the SubplotSpec instance.)rrGrrrrrIzSubplotDivider.get_subplotspeccCs||_|||jdS)zSet the SubplotSpec instance.N)rrErHr)r subplotspecrrrrszSubplotDivider.set_subplotspec)rrrrrrHrrpropertyrrrrrr __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||_|durt||_n||_tj|d|jg|jgddddS)zw Parameters ---------- axes : :class:`~matplotlib.axes.Axes` xref yref Nr )rrrrrr) _axesr0AxesX_xrefAxesY_yrefrr get_figure)rrtxrefyrefrrrrs  zAxesDivider.__init__) axes_classcKsF|j}|durt|tr|j}nt|}|||jddfi|SNT)original)r isinstancer _axes_classtyperrH)rrkwargsrtrrr _get_new_axess zAxesDivider._get_new_axesFcKs|dur tjd|j}|r3t|tjstj||jd}|r-|jd||j d7_ n|j |t|tjsAtj||jd}|rZ|jd||j d7_ |j d|j d}n|j ||j t |jd|j d}|jdi|}|||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.wspace fraction_refrrrprqr)mplrcParamsrrr0_Basefrom_anyr rrr@rrrKrset_axes_locatorrrhr pack_startrlocatorrarrrnew_horizontal,     zAxesDivider.new_horizontalcKs|dur tjd|j}|r3t|tjstj||jd}|r-|jd||j d7_ n|j |t|tjsAtj||jd}|rZ|jd||j d7_ |j |j dd}n|j ||j |j t |jdd}|jdi|}|||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.hspacerrrrr)rrrrr0rrr rrr@rrrKrrrrrr new_verticalrzAxesDivider.new_verticalr. add_to_figurez ax.remove()rTcKs|dkr|j||fddi|}n<|dkr"|j||fddi|}n+|dkr3|j||fddi|}n|dkrD|j||fddi|}n tjgd|d |rU|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. rrTrFrrrr)rrrrLr add_axes)rrrhrrrrarrr append_axess   zAxesDivider.append_axescCs*|jdur|j}|dkrdSdS|jS)NautoFT)rrr\rZrrrr\4s  zAxesDivider.get_aspectcCs$|jdur|jjdd}|jS|jSr)r rrHr`)rrgrrrrH>s zAxesDivider.get_positioncCs|jdur |jS|jSr/)rrrQrGrrrrQEs  zAxesDivider.get_anchorcCrr)rrrrGrrrrKrzAxesDivider.get_subplotspecr)NF)NT)rrrrrrrrrdelete_parameterrr\rHrQrrrrrrrs //  rcCst|}t|j\}}t|j\}}t|d|df} t|d} t| d|d|f|d| d|df<|| dddf<| | d|<|t|| d<tj| | } | dd} | d} | |krn|||} | S)Nrr?) rKnpasarrayTzeros fill_diagonalsumlinalgsolve) summed_widths equal_heights total_width max_heightneq_rseq_assm_rssm_asABkarray_and_heightkarrayheightrrr_determine_karrayUs   r cCs<dg}t||D]\\}}}||d|||q|Sr>)zipr@) summed_sizesrrArar<rrrrBmsrBc Cst||||||d} t|| } | d| d|} |d\} } | d| | |}tj||||}tj||| |}|||}|j|j}}||| |fS)N)rrr?r)r rBrMrNrkrlrmrn)rwrxryrSrrrurvrrr~rh0_rh0_arrrrrmrnrrr_locateus     rc@$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 |dS|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)rrprrrrrrs$ zHBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|dur@t j ddd|d}|||||||||}}||}}t j ||||S)Nr.rcrdr)r rgrhrirbr*r-rrQrrorMrNrk)rrprqrrrsrtr%rurvrwrxryrS summed_wsequal_hsrmrnr~rrrrrrrrrs   & zHBoxDivider.locater/rrrrrrrrrrrrs rc@r) VBoxDividerzl A `SubplotDivider` for laying out axes vertically, while ensuring that they have equal widths. NcCs"t|d|d|dur |S|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)rrqrsrrrrs" zVBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|dur@t j ddd|d}||}}|||||||||}}t j ||||S)Nr.rfrdr)r rgrhrirbr-r*rrQrrorMrNrk)rrprqrrrsrtr%rurvrwrxryrS summed_hsequal_wsrnrmrrrrrrrrrrs    &zVBoxDivider.locater/rrrrrrrs rcCs$t|}|jddd}|||S)Nrr)rrr)rtdividerrrrrmake_axes_locatables rrcCs8|durgd}t|}|dur|}|j|||ddS)Nr)rrr)rr)rarrrrrrrmake_axes_area_auto_adjustables r)NrN)rnumpyr matplotlibrrmatplotlib.axesrmatplotlib.gridspecrrmatplotlib.transforms transformsrMrr0rrrrr rBrrrrrrrrrs0     =B@+'