U mdX@sdZddlZddlZddlmZddlmZddlm Z ddl m Z GdddZGd d d ZGd d d eZGd ddeZddZGdddeZGdddeZddZdddZdS)zJ Helper classes to adjust the positions of multiple axes at drawing time. N)_api) SubplotSpec) axes_sizec@seZdZdZd2ddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZd3ddZddZdd Zd!d"Zd#d$Zed%d&Zed'd(Zd4d)d*Zd5d+d,Zd-d.Zd6d0d1ZdS)7DivideraQ 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 horizontalverticalaspectanchorr]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/mpl_toolkits/axes_grid1/axes_divider.py__init__s zDivider.__init__cstfdd|DS)Ncsg|]}|qSrget_size.0srendererrr :sz0Divider.get_horizontal_sizes..)nparrayget_horizontalrr"rr!rget_horizontal_sizes9szDivider.get_horizontal_sizescstfdd|DS)Ncsg|]}|qSrrrr!rrr#=sz.Divider.get_vertical_sizes..)r$r% get_verticalr'rr!rget_vertical_sizes<szDivider.get_vertical_sizescCs ||_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.r+rrrr get_positionJszDivider.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 )rrrrrr Ns  zDivider.set_anchorcCs|jS)zReturn the anchor.)r r-rrr get_anchorbszDivider.get_anchorcCs ||_dS)z Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for horizontal division Nr )rhrrrset_horizontalfszDivider.set_horizontalcCs|jS)zReturn horizontal sizes.r:r-rrrr&oszDivider.get_horizontalcCs ||_dS)z Parameters ---------- v : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for vertical division Nr )rvrrr set_verticalsszDivider.set_verticalcCs|jS)zReturn vertical sizes.r=r-rrrr)|szDivider.get_verticalFcCs ||_dS)zE Parameters ---------- aspect : bool Nrrrrrr set_aspectszDivider.set_aspectcCs|jS)zReturn aspect.r@r-rrr get_aspectszDivider.get_aspectcCs ||_dSNr)rrrrr set_locatorszDivider.set_locatorcCs|jSrDrEr-rrr get_locatorszDivider.get_locatorcCs$|jdkr|S|||jSdSrD)rr.bounds)raxr"rrrget_position_runtimes zDivider.get_position_runtimecCs"|d\}}|r|||SdS)Nr)sum)sizestotalZrel_sumZabs_sumrrr_calc_kszDivider._calc_kcCstd||dgS)Nrr)r)r$cumsum)rLkrrr _calc_offsetsszDivider._calc_offsetscCs|jjj|jj\}}|||\} } } } ||} ||}|| || }|||| }|rt ||}| | |}| ||}|d|d|}|d|d|}t j | | | | }t j | | ||}|||j\}}n"| | |}| ||}| | }}|dkrd}|dkr*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, the 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 rN)rbboxsizedpirJr(r*rNrCminrQr3r4 from_boundsanchoredr9p0)rnxnynx1ny1axesr"fig_wfig_hxywr;ZhsizesZvsizesZk_hZk_vrPoxoywwhhpbpb1x0y0x1w1y1h1rrrlocates2          &&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)rrZr[r\r]rrr new_locatorszDivider.new_locatorcCstjddddg|d|dkr<|jd||jd7_nH|dkrR|j|n2|dkrx|jd||jd7_n |j|dS)Nleftrightbottomtoppositionrr)rr2r insertrappendr r)rryrTrrr append_sizeszDivider.append_size皙?cCs:|dkrddddg}|D]}||t|||qdS)a1 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, 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. Nrtrurvrw)r|SizeZ_AxesDecorationsSize)ruse_axespad adjust_dirsdrrradd_auto_adjustable_areas z Divider.add_auto_adjustable_area)Nr)F)NNNN)NN)r}N)__name__ __module__ __qualname____doc__rr(r*r,r.r r9r<r&r?r)rBrCrFrGrJ staticmethodrNrQrprsr|rrrrrrs6        1 rc@s*eZdZdZd ddZddZddZdS) rrzc A callable object which returns the position and size of a given `.AxesDivider` cell. NcCsh||_|j}|j}|||||_|_|dkr:t|j}|dkrPt|jd}|||_|||_dS)a Parameters ---------- axes_divider : `~mpl_toolkits.axes_grid1.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_nyr7_nx1_ny1)rZ axes_dividerrZr[r\r]rrrrrrs   zAxesLocator.__init__cCs>|jj}|jj}|j|j||j||j||j|||SrD)rrrrprrrr)rr^r"rrrrr__call__-szAxesLocator.__call__cCs |jSrD)rget_subplotspecr-rrrr9szAxesLocator.get_subplotspec)NN)rrrrrrrrrrrrr s  rrcsDeZdZdZdddddfdd 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__rrrBszSubplotDivider.__init__cCs||jjS)z%Return the bounds of the subplot box.)rr.rrHr-rrrr.YszSubplotDivider.get_positioncCs|jS)zGet the SubplotSpec instance.) _subplotspecr-rrrr]szSubplotDivider.get_subplotspeccCs||_|||jdS)zSet the SubplotSpec instance.N)rr,r.r)rZ subplotspecrrrraszSubplotDivider.set_subplotspec) rrrrrr.rr __classcell__rrrrr=srcsteZdZdZdfdd ZddddZdd d Zdd d Zdddd dZddZ ddZ ddZ ddZ Z S) 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)rr^ZxrefZyrefrrrrlszAxesDivider.__init__) axes_classcKs0|j}|dkrt|}|||jddf|SNT)original)rtyperr.)rrkwargsr^rrr _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_refrrrZr[)mplrcParamsrr0r~_Basefrom_anyr rzrr{rsrr7rset_axes_locatorrrTr pack_startrlocatorrIrrrnew_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)rrrr0r~rrr rzrr{rsrr7rrrrrr 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. TF)rtrurvrwrx)rr)rZ check_getitemrrrZadd_axes) rryrTrrrZ create_axesrrIrrr append_axess&  zAxesDivider.append_axescCs0|jdkr&|j}|dkr dSdSn|jSdS)NautoFT)rrrCrArrrrCs   zAxesDivider.get_aspectcCs(|jdkr|jjdd}|jS|jSdSr)r rr.rH)rrSrrrr.s zAxesDivider.get_positioncCs|jdkr|jS|jSdSrD)r rr9r-rrrr9s  zAxesDivider.get_anchorcCs |jSrD)rrr-rrrrszAxesDivider.get_subplotspec)NN)NF)NF)N)rrrrrrrrrrCr.r9rrrrrrrgs " rc Cs"||} ||} t|} |j\} } |j\}}t| d}d|d| df<||dddf<| | |f}tj||^}}|| kr| | | }td|||}|d|d|}|d\}}|d|||}tj ||||}tj ||||}| ||j \}}||||fS)NrrR)r)r) r7Tr$ZdiagrKZlinalgZsolverOr3r4rWrXrY)rarbrcr;Z summed_widthsZ equal_heightsr_r`r total_width max_heightnZeq_relsZeq_abssZsm_relsZsm_abssABZkarrayheightrdrfZh0_relZh0_absrgrhrirjrkrrr_locate s(    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. rNrrq)rrZr\rrrrs4s zHBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|dkrnd}|||||||||}}||}}t j ||||SNrR) rrSrTrUrJr(r*rr9r3r4rW)rrZr[r\r]r^r"r_r`rarbrcr;Z summed_wsZequal_hsrjrkrdrgrlrmrnrorrrrpBs(   & zHBoxDivider.locate)N)NNNNrrrrrsrprrrrr*s rc@s$eZdZdZdddZdddZdS) VBoxDividerzm 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. rrNrq)rr[r]rrrrsWs zVBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|dkrnd}||}}|||||||||}}t j ||||Sr) rrSrTrUrJr*r(rr9r3r4rW)rrZr[r\r]r^r"r_r`rarbrcr;Z summed_hsZequal_wsrkrjrerfrlrmrnrorrrrpes(    &zVBoxDivider.locate)N)NNNNrrrrrrQs rcCs$t|}|jddd}|||S)Nrr)rrsr)r^dividerrrrrmake_axes_locatablets rr}cCs<|dkrddddg}t|}|dkr(|}|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. Nrtrurvrw)rrr)rr)rIrrrrrrrmake_axes_area_auto_adjustable|s  r)Nr}N)rnumpyr$Z matplotlibrrZmatplotlib.gridspecrZmatplotlib.transformsZ transformsr3rr~rrrrrrrrrrrrrrs(    ~2*#!'#