o liX@sdZddlZddlZddlmZddlmZddlm Z ddl m Z GdddZGd d d ZGd d d eZGd ddeZddZGdddeZGdddeZddZ dddZdS)zJ Helper classes to adjust the positions of multiple axes at drawing time. N)_api) SubplotSpec) axes_sizec@seZdZdZ d2ddZddZdd 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 horizontalverticalaspectanchorrb/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/axes_divider.py__init__s  zDivider.__init__ctfdd|DS)Ncg|]}|qSrget_size.0srendererrr :z0Divider.get_horizontal_sizes..)nparrayget_horizontalrr$rr#rget_horizontal_sizes9zDivider.get_horizontal_sizescr)Ncrrrr r#rrr%=r&z.Divider.get_vertical_sizes..)r'r( get_verticalr*rr#rget_vertical_sizes<r,zDivider.get_vertical_sizescC ||_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_positioncC|jS)z%Return the position of the rectangle.r0rrrr get_positionJzDivider.get_positioncCsJt|trtjtjj|dnt|ttfrt |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 mtransformsBboxcoefstuplelistlen TypeErrorr )rrrrrr Ns  zDivider.set_anchorcCr2)zReturn the anchor.)r r3rrr get_anchorbr5zDivider.get_anchorcCr/)z Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for horizontal division Nr )rhrrrset_horizontalf zDivider.set_horizontalcCr2)zReturn horizontal sizes.rBr3rrrr)or5zDivider.get_horizontalcCr/)z Parameters ---------- v : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for vertical division Nr )rvrrr set_verticalsrEzDivider.set_verticalcCr2)zReturn vertical sizes.rFr3rrrr-|r5zDivider.get_verticalFcCr/)zE Parameters ---------- aspect : bool Nrrrrrr set_aspects zDivider.set_aspectcCr2)zReturn aspect.rIr3rrr get_aspectr5zDivider.get_aspectcCs ||_dSNr)rrrrr set_locator zDivider.set_locatorcCr2rMrNr3rrr get_locatorszDivider.get_locatorcCs |jdur |S|||jSrM)rr4bounds)raxr$rrrget_position_runtimes zDivider.get_position_runtimecCs"|d\}}|r|||SdS)Nr)sum)sizestotalrel_sumabs_sumrrr_calc_kszDivider._calc_kcCstdg||dgS)Nrr)r'cumsum)rVkrrr _calc_offsetsszDivider._calc_offsetscCs|jjj|jj\}}|||\} } } } ||} ||}|| || }|||| }|rvt ||}| | |}| ||}|d|d|}|d|d|}t j | | | | }t j | | ||}|||j\}}n| | |}| ||}| | }}|durd}|durd}|||||||||}}|||||||||}}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)rbboxsizedpirTr+r.rZrLminr]r:r; from_boundsanchoredrAp0)rnxnynx1ny1axesr$fig_wfig_hxywrChsizesvsizesk_hk_vr\oxoywwhhpbpb1x0y0x1w1y1h1rrrlocates2        &&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)rrfrgrhrirrr new_locators zDivider.new_locatorcCstjgd|d|dkr|jd||jd7_dS|dkr)|j|dS|dkr=|jd||jd7_dS|j|dS)Nleftrightbottomtoppositionrrrrr)rr9r insertrappendr r)rrr`rrr append_sizes zDivider.append_size皙?cCs6|durgd}|D]}||t|||q dS)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. Nr)rSize_AxesDecorationsSize)ruse_axespad adjust_dirsdrrradd_auto_adjustable_areas z Divider.add_auto_adjustable_area)Nr)FNNNNNN)rN)__name__ __module__ __qualname____doc__rr+r.r1r4r rArDr)rHr-rKrLrOrQrT staticmethodrZr]rrrrrrrrrs6        1rc@s*eZdZdZd ddZddZddZdS) rzc A callable object which returns the position and size of a given `.AxesDivider` cell. NcCsh||_|j}|j}|||||_|_|durt|j}|dur(t|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_nyr?_nx1_ny1)r axes_dividerrfrgrhrirrrrrrs   zAxesLocator.__init__cCs>|jj}|jj}|j|j||j||j||j|||SrM)rrrrrrrr)rrjr$rrrrr__call__-szAxesLocator.__call__cC |jSrM)rget_subplotspecr3rrrr9rPzAxesLocator.get_subplotspecr)rrrrrrrrrrrr s   rcsDeZdZdZdddddfdd ZddZd d Zd d ZZS) SubplotDividerzT The Divider class whose rectangle area is specified as a subplot geometry. Nrrrrrcs@||_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__rrrBs  zSubplotDivider.__init__cCs||jjS)z%Return the bounds of the subplot box.)rr4rrRr3rrrr4YszSubplotDivider.get_positioncCr2)zGet the SubplotSpec instance.) _subplotspecr3rrrr]r5zSubplotDivider.get_subplotspeccCs||_|||jdS)zSet the SubplotSpec instance.N)rr1r4r)r subplotspecrrrraszSubplotDivider.set_subplotspec) rrrrrr4rr __classcell__rrrrr=srcsteZdZdZdfdd ZddddZdd d Zdd d Zdddd dZddZ ddZ ddZ ddZ Z S) AxesDividerz0 Divider based on the preexisting axes. Ncsf||_|durt||_n||_|durt||_n||_tj|d|jg|jgddddS)zw Parameters ---------- axes : :class:`~matplotlib.axes.Axes` xref yref Nr)rrrrrr) _axesrAxesX_xrefAxesY_yrefrr get_figure)rrjxrefyrefrrrrls  zAxesDivider.__init__) axes_classcKs4|j}|dur t|}|||jddfi|SNT)original)rtyperr4)rrkwargsrjrrr _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)z Helper method for ``append_axes("left")`` and ``append_axes("right")``. See the documentation of `append_axes` for more details. :meta private: Nzfigure.subplot.wspace fraction_refrrrfrgr)mplrcParamsrr7r_Basefrom_anyr rrrrrr?rset_axes_locatorrr`r pack_startrlocatorrSrrrnew_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)z Helper method for ``append_axes("top")`` and ``append_axes("bottom")``. See the documentation of `append_axes` for more details. :meta private: Nzfigure.subplot.hspacerrrrr)rrrr7rrrr rrrrrr?rrrrrr new_verticalrzAxesDivider.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. TFrr)rr)r check_getitemrrradd_axes) rrr`rrr create_axesrrSrrr append_axess"  zAxesDivider.append_axescCs*|jdur|j}|dkrdSdS|jS)NautoFT)rrrLrJrrrrLs  zAxesDivider.get_aspectcCs$|jdur|jjdd}|jS|jSr)r rr4rR)rr_rrrr4s zAxesDivider.get_positioncCs|jdur |jS|jSrM)r rrAr3rrrrAs  zAxesDivider.get_anchorcCrrM)rrr3rrrrrPzAxesDivider.get_subplotspecr)NFrM)rrrrrrrrrrLr4rArrrrrrrgs   " rc Cs*||} ||} t|} |j\} } |j\}}tg| d}d|d| df<||dddf<g| | |}tj||^}}|| krN| | | }tdg|||}|d|d|}|d\}}|d|||}tj ||||}tj ||||}| ||j \}}||||fS)Nrr^) r?Tr'diagrUlinalgsolver[r:r;rcrdre)rmrnrorC summed_widths equal_heightsrkrlr total_width max_heightneq_relseq_absssm_relssm_abssABkarrayheightrtrvh0_relh0_absrwrxryrzr{rrr_locate s(     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)rrfrhrrrr4s$ zHBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|dur7d}|||||||||}}||}}t j ||||SNr^) rr_r`rarTr+r.rrAr:r;rc)rrfrgrhrirjr$rkrlrmrnrorC summed_wsequal_hsrzr{rtrwr|r}r~rrrrrBs   & zHBoxDivider.locaterMrrrrrrrrrrrr*s rc@r) VBoxDividerzm 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)rrgrirrrrWs" zVBoxDivider.new_locatorc Cs|jjj|jj\}}|||\} } } } ||} ||}t| | | | | |||| \}}}}|dur7d}||}}|||||||||}}t j ||||Sr) rr_r`rarTr.r+rrAr:r;rc)rrfrgrhrirjr$rkrlrmrnrorC summed_hsequal_wsr{rzrurvr|r}r~rrrrres    &zVBoxDivider.locaterMrrrrrrrQs rcCs$t|}|jddd}|||S)Nrr)rrr)rjdividerrrrrmake_axes_locatablets rrcCs8|durgd}t|}|dur|}|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. Nr)rrr)rr)rSrrrrrrrmake_axes_area_auto_adjustable|s  r)NrN)rnumpyr' matplotlibrrmatplotlib.gridspecrmatplotlib.transforms transformsr:rrrrrrrrrrrrrrrs&    ~2*#!'#