U \ ad@sdZddlZddlmZddlmZddlmZm Z ddl m Z ddl mZGdd d ZGd d d ZGd d d eZGdddeZGdddeZGdddeZddZdddZdS)zJ Helper classes to adjust the positions of multiple axes at drawing time. N)_api) SubplotBase) SubplotSpecGridSpec) axes_sizec@seZdZdZd4ddZddZdd Zd d Zed d Z eddZ ddZ ddZ ddZ ddZddZddZddZddZd5d!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd6d+d,Zd7d-d.Zd/d0Zd8d2d3ZdS)9DivideraQ 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 horizontalverticalaspectanchorrH/tmp/pip-unpacked-wheel-wjyw_3jo/mpl_toolkits/axes_grid1/axes_divider.py__init__szDivider.__init__csfdd|DS)Ncsg|]}|qSrget_size.0srendererrr 8sz0Divider.get_horizontal_sizes..)get_horizontalrr#rr"rget_horizontal_sizes7szDivider.get_horizontal_sizescsfdd|DS)Ncsg|]}|qSrrrr"rrr$;sz.Divider.get_vertical_sizes..) get_verticalr&rr"rget_vertical_sizes:szDivider.get_vertical_sizescCs$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_kBs   zDivider._calc_kcCs2dg}|D]"\}}||d|||q |SNr-)append)r.r1offsetsr/r0rrr _calc_offsetsQs 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_positionXs zDivider.set_positioncCs|jS)z%Return the position of the rectangle.r8rrrr get_positioncszDivider.get_positioncCs(t|dkrtjtjj|d||_dS)a Parameters ---------- anchor : {'C', 'SW', 'S', 'SE', 'E', 'NE', 'N', 'NW', 'W'} anchor position ===== ============ value description ===== ============ 'C' Center 'SW' bottom left 'S' bottom 'SE' bottom right 'E' right 'NE' top right 'N' top 'NW' top left 'W' left ===== ============ )rN)lenr check_in_list mtransformsBboxZcoefsr)rrrrr set_anchorgs zDivider.set_anchorcCs|jS)zReturn the anchor.)rr: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.rCr: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.rFr:rrrr(szDivider.get_verticalFcCs ||_dS)zE Parameters ---------- aspect : bool Nrrrrrr set_aspectszDivider.set_aspectcCs|jS)zReturn aspect.rIr:rrr get_aspectszDivider.get_aspectcCs ||_dSr*r)rrrrr set_locatorszDivider.set_locatorcCs|jSr*rMr:rrr get_locatorszDivider.get_locatorcCs$|jdkr|S|||jSdSr*)rr;bounds)raxr#rrrget_position_runtimes zDivider.get_position_runtimecCs|j\}}|||\} } } } ||} ||}|| || }|||| }|rt||}|| |}|||}|d|d|}|d|d|}t j | | | | }t j | | ||}| | |}|j|j}}n"|| |}|||}| | }}|dkr |d}|dkr2|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 r4rNr)r get_size_inchesrRr'r)r2rLminr7r?r@ from_boundsanchoredrBx0y0)rnxnynx1ny1axesr#figWfigHxywrDZhsizesZvsizesZk_hZk_vr1oxoywwhhpbpb1 pb1_anchoredrWrXx1w1y1h1rrrlocates4          &&zDivider.locatecCst|||||S)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.  AxesLocator)rrYrZr[r\rrr new_locatorszDivider.new_locatorcCs|dkr&|jd||jd7_nh|dkr<|j|nR|dkrb|jd||jd7_n,|dkrx|j|ntjddddg|ddS)Nleftrrrightbottomtopposition)r insertrr5r rrr>)rrwsizerrr append_sizeszDivider.append_size皙?c Cs^|dkrddddg}ddlm}m}m}|D],}|||}||} || |} ||| q,dS)Nrrrsrtrur)Padded SizeFromFuncGetExtentHelper)rr|r}r~rz) ruse_axespad adjust_dirsr|r}r~dhelperryZ padded_sizerrradd_auto_adjustable_areas   z Divider.add_auto_adjustable_area)Nr )F)NNNN)NN)r{N)__name__ __module__ __qualname____doc__rr'r)r, staticmethodr2r7r9r;rArBrEr%rHr(rKrLrNrOrRrnrqrzrrrrrrs8        2 rc@s*eZdZdZd ddZddZddZdS) rpz| A simple callable object, initialized with AxesDivider class, returns the position and size of the given cell. NcCs`||_|j}|j}|||||_|_|dkr8|d}|dkrH|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) _axes_dividerrr_nx_ny_nx1_ny1)rZ axes_dividerrYrZr[r\rrrrrrs  zAxesLocator.__init__cCs>|jj}|jj}|j|j||j||j||j|||Sr*)rrrrnrrrr)rr]r#rrrrr__call__4szAxesLocator.__call__cCst|jdr|jSdSdSNget_subplotspec)hasattrrrr:rrrr@s  zAxesLocator.get_subplotspec)NN)rrrrrrrrrrrrps  rpcseZdZdZdddddfdd ZddZed ed d Z ed d d Z ejd dddddZ ejd ddddZ ddZ ddZZS)SubplotDividerT The Divider class whose rectangle area is specified as a subplot geometry. Nr rrrrcsD||_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__rrrLszSubplotDivider.__init__cCs||jjS)z%Return the bounds of the subplot box.)rr;rrPr:rrrr;cszSubplotDivider.get_positionz3.4cCs||jSr*)rr;rr:rrrfigboxgszSubplotDivider.figboxcCsdSr*rr:rrr update_paramslszSubplotDivider.update_paramsrz1(get_subplotspec returns a SubplotSpec instance.)) alternativeZaddendumcCs"|\}}}}|||dfS)z*Get the subplot geometry, e.g., (2, 2, 3).r)r get_geometry)rZrowscolsZnum1Znum2rrrrpszSubplotDivider.get_geometryr)rcCs,t|||d|_|||jdS)z;Change subplot geometry, e.g., from (1, 1, 1) to (2, 2, 3).rN)r _subplotspecrr9r)rZnumrowsZnumcolsnumrrrchange_geometryxszSubplotDivider.change_geometrycCs|jS)zGet the SubplotSpec instance.)rr:rrrrszSubplotDivider.get_subplotspeccCs||_|||jdS)zSet the SubplotSpec instance.N)rr9r;r)rZ subplotspecrrrrszSubplotDivider.set_subplotspec)rrrrrr;r deprecatedpropertyrrrrrr __classcell__rrrrrGs*     rcsneZdZdZdfdd ZddddZdd d Zdd d ZdddZddZ ddZ ddZ ddZ Z S) AxesDividerz1 Divider based on the pre-existing 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)rr]ZxrefZyrefrrrrszAxesDivider.__init__) axes_classcKsB|j}|dkr(t|tr |j}nt|}|||jddf|SNT)original)r isinstancerZ _axes_classtyperr;)rrkwargsr]rrr _get_new_axess zAxesDivider._get_new_axesFcKs|dkrtjddd|rdt|tjs6tj||jd}|rX|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) a  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 A width of the axes. If float or string is given, *from_any* function is used to create the size, with *ref_size* set to AxesX instance of the current axes. pad : :mod:`~mpl_toolkits.axes_grid1.axes_size` or float or str Pad between the axes. It takes same argument as *size*. 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. N3.2zrIn a future version, 'pad' will default to rcParams['figure.subplot.wspace']. Set pad=0 to keep the old behavior.messageZ fraction_refrrrYrZ)rwarn_deprecatedrr+_Basefrom_anyrr rxrr5rqrr=rset_axes_locatorrryr pack_startrlocatorrQrrrnew_horizontals4       zAxesDivider.new_horizontalcKs|dkrtjddd|rdt|tjs6tj||jd}|rX|jd||j d7_ n |j |t|tjstj||jd}|r|jd||j d7_ |j |j dd}n&|j ||j |j t |jdd}|jf|}|||S) a  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 A height of the axes. If float or string is given, *from_any* function is used to create the size, with *ref_size* set to AxesX instance of the current axes. pad : :mod:`~mpl_toolkits.axes_grid1.axes_size` or float or str Pad between the axes. It takes same argument as *size*. 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. NrzrIn a future version, 'pad' will default to rcParams['figure.subplot.hspace']. Set pad=0 to keep the old behavior.rrrrr)rrrr+rrrr rxrr5rqrr=rrrrrr new_verticals4       zAxesDivider.new_verticalTcKs|dkr"|j||fddi|}n||dkrD|j||fddi|}nZ|dkrf|j||fddi|}n8|dkr|j||fddi|}ntjddddg|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. rrrTrsFrtrurv)rrrr>r Zadd_axes)rrwryrZ add_to_figurerrQrrr append_axess  zAxesDivider.append_axescCs0|jdkr&|j}|dkr dSdSn|jSdS)NautoFT)rrrLrJrrrrL*s   zAxesDivider.get_aspectcCs(|jdkr|jjdd}|jS|jSdSr)r rr;rP)rZbboxrrrr;4s zAxesDivider.get_positioncCs|jdkr|jS|jSdSr*)rrrBr:rrrrB;s  zAxesDivider.get_anchorcCst|jdr|jSdSdSr)rrrr:rrrrAs  zAxesDivider.get_subplotspec)NN)NF)NF)NT)rrrrrrrrrrLr;rBrrrrrrrs 1 1  rc@s@eZdZeddZeddZd ddZdd Zd d d ZdS) HBoxDividercCst|}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)Nrr4) r=npZasarrayTzerosZ fill_diagonalsumZlinalgZsolve)equivalent_sizesappended_sizesmax_equivalent_sizetotal_appended_sizenZeq_rsZeq_asZap_rsZap_asABZkarray_HkarrayHrrr_determine_karrayJs   zHBoxDivider._determine_karraycCs<dg}t||D]&\\}}}||d|||q|Sr3)zipr5)rrr6rar1rrrr7bszHBoxDivider._calc_offsetsNcCst||d|dS)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. ny, ny1 : int Same as *nx* and *nx1*, but for row positions. rNro)rrYr[rrrrqiszHBoxDivider.new_locatorc Cs|} |} ||} ||} || | | | } || | }|d|d|}| d}| d|d|d|}tj||||}tj||||}|||}|j|j}}||||fS)Nr4rr) rr7r?r@rUrVrBrWrX)rr`rarbrDy_equivalent_sizesx_appended_sizesr^r_rrrrrrcreZref_hrfrgrhrirWrXrrr_locateys" zHBoxDivider._locatec Cs|j\}}|||\} } } } ||} ||}|| | | | | |||\}}}}|dkrf|d}|||||||||}}||}}tj||||Sa 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. axes renderer Nr) r rSrRr)r'rr?r@rU)rrYrZr[r\r]r#r^r_r`rarbrDrrrWrXrcrfrjrkrlrmrrrrns    & zHBoxDivider.locate)N)NNNN) rrrrrr7rqrrnrrrrrHs   rc@s$eZdZdZdddZdddZdS) VBoxDividerrNcCst|d|d|S)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. rNro)rrZr\rrrrqs zVBoxDivider.new_locatorc Cs|j\}}|||\} } } } ||} ||}|| | | | | |||\}}}}|dkrf|d}||}}|||||||||}}tj||||Sr) r rSrRr'r)rr?r@rU)rrYrZr[r\r]r#r^r_r`rarbrDZx_equivalent_sizesZy_appended_sizesrXrWrdrerjrkrlrmrrrrns     &zVBoxDivider.locate)N)NNNN)rrrrrqrnrrrrrs rcCs$t|}|jddd}|||S)Nrr)rrqr)r]dividerrrrrmake_axes_locatables rr{cCs<|dkrddddg}t|}|dkr(|}|j|||ddS)Nrrrsrtru)rrr)rr)rQrrrrrrrmake_axes_area_auto_adjustables r)Nr{N)rZnumpyrZ matplotlibrZmatplotlib.axesrZmatplotlib.gridspecrrZmatplotlib.transformsZ transformsr?rr+rrprrrrrrrrrrs(    4B@j5