o id@sdZddlZddlmZddlmZddlmZm Z ddl m Z ddl mZGdd d ZGd d d ZGd d d eZGdddeZGdddeZGdddeZddZ  dddZdS)zJ Helper classes to adjust the positions of multiple axes at drawing time. N)cbook) SubplotBase) SubplotSpecGridSpec) axes_sizec@seZdZdZ d4ddZddZdd 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 horizontalverticalaspectanchorr_/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/axes_divider.py__init__s zDivider.__init__cfdd|DS)Ncg|]}|qSrget_size.0srendererrr 8z0Divider.get_horizontal_sizes..)get_horizontalrr%rr$rget_horizontal_sizes7zDivider.get_horizontal_sizescr)Ncrrrr!r$rrr&;r'z.Divider.get_vertical_sizes..) get_verticalr)rr$rget_vertical_sizes:r+zDivider.get_vertical_sizescCs$t|}t|}||fSN)SizeAddListr,r()rvsizehsizerrrget_vsize_hsize=szDivider.get_vsize_hsizecCsBd\}}|D] \}}||7}||7}q|dkr|||}|SdS)N)r4r4r)l total_sizers_sumas_sum_rs_askrrr_calc_kBs   zDivider._calc_kcCs2dg}|D]\}}||d|||q|SNr4)append)r5r;offsetsr9r:rrr _calc_offsetsQs 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_positionXs zDivider.set_positioncC|jS)z%Return the position of the rectangle.rCrrrr get_positionczDivider.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 mtransformsBboxcoefsr)rrrrr set_anchorgs  zDivider.set_anchorcCrE)zReturn the anchor.)rrFrrr get_anchorrHzDivider.get_anchorcCrB)z Parameters ---------- h : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for horizontal division Nr )rhrrrset_horizontal zDivider.set_horizontalcCrE)zReturn horizontal sizes.rQrFrrrr(rHzDivider.get_horizontalcCrB)z Parameters ---------- v : list of :mod:`~mpl_toolkits.axes_grid1.axes_size` sizes for vertical division Nr )rvrrr set_verticalrTzDivider.set_verticalcCrE)zReturn vertical sizes.rUrFrrrr,rHzDivider.get_verticalFcCrB)zE Parameters ---------- aspect : bool Nrrrrrr set_aspects zDivider.set_aspectcCrE)zReturn aspect.rXrFrrr get_aspectrHzDivider.get_aspectcCs ||_dSr.r)rrrrr set_locators zDivider.set_locatorcCrEr.r\rFrrr get_locatorszDivider.get_locatorcCs |jdur |S|||jSr.)rrGbounds)raxr%rrrget_position_runtimes zDivider.get_position_runtimecCs|j\}}|||\} } } } ||} ||}|| || }|||| }|rvt||}|| |}|||}|d|d|}|d|d|}t j | | | | }t j | | ||}| | |}|j|j}}n|| |}|||}| | }}|dur|d}|dur|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)r get_size_inchesrar*r-r<r[minrArLrM from_boundsanchoredrPx0y0)rnxnynx1ny1axesr%figWfigHxywrRhsizesvsizesk_hk_vr;oxoywwhhpbpb1 pb1_anchoredrfrgx1w1y1h1rrrlocates4        &&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)rrhrirjrkrrr new_locatorszDivider.new_locatorcCs|dkr|jd||jd7_dS|dkr |j|dS|dkr4|jd||jd7_dS|dkr@|j|dStjgd|ddS) Nleftrrrightbottomtoprrrrposition)r insertrr?r rrrK)rrsizerrr append_sizes  zDivider.append_size皙?c CsZ|durgd}ddlm}m}m}|D]}|||}||} || |} ||| qdS)Nrr)Padded SizeFromFuncGetExtentHelper)rrrrr) ruse_axespad adjust_dirsrrrdhelperr padded_sizerrradd_auto_adjustable_areas  z Divider.add_auto_adjustable_area)Nr )FNNNNNN)rN)__name__ __module__ __qualname____doc__rr*r-r3 staticmethodr<rArDrGrOrPrSr(rWr,rZr[r]r^rarrrrrrrrrs8        2rc@s*eZdZdZd ddZddZddZdS) rz| A simple callable object, initialized with AxesDivider class, returns the position and size of the given cell. NcCs`||_|j}|j}|||||_|_|dur|d}|dur$|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)r axes_dividerrhrirjrkrrrrrrs  zAxesLocator.__init__cCs>|jj}|jj}|j|j||j||j||j|||Sr.)rrrrrrrr)rrlr%rrrrr__call__4szAxesLocator.__call__cCt|jdr |jSdSNget_subplotspec)hasattrrrrFrrrr@  zAxesLocator.get_subplotspecr)rrrrrrrrrrrrs   rc@sTeZdZdZdddddddZddZd d Zd d Zd dZddZ ddZ dS)SubplotDividerT The Divider class whose rectangle area is specified as a subplot geometry. Nr )rrrrc GsD||_t|||_|tj|||jj|pg|pg||ddS)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) figurer_from_subplot_args _subplotspec update_paramsrrfigboxr_)rrrrrrargsrrrrLs  zSubplotDivider.__init__cCs||jjS)z%Return the bounds of the subplot box.)rrr_rFrrrrGdszSubplotDivider.get_positioncCs||j|_dS)z1Update the subplot position from fig.subplotpars.N)rrGrrrFrrrriszSubplotDivider.update_paramscCs"|\}}}}|||dfS)z*Get the subplot geometry, e.g., (2, 2, 3).r)r get_geometry)rrowscolsnum1num2rrrrmszSubplotDivider.get_geometrycCs,t|||d|_|||jdS)z;Change subplot geometry, e.g., from (1, 1, 1) to (2, 2, 3).rN)rrrrDr)rnumrowsnumcolsnumrrrchange_geometrysszSubplotDivider.change_geometrycCrE)zGet the SubplotSpec instance.rrFrrrryrHzSubplotDivider.get_subplotspeccCrB)zSet the SubplotSpec instance.Nr)r subplotspecrrrset_subplotspec}s zSubplotDivider.set_subplotspec) rrrrrrGrrrrrrrrrrGs  rc@sfeZdZdZdddZddddZdd d Zdd d ZdddZddZ ddZ ddZ ddZ dS) AxesDividerz1 Divider based on the pre-existing axes. Nc Csf||_|durt||_n||_|durt||_n||_tj||d|jg|jgddddS)zw Parameters ---------- axes : :class:`~matplotlib.axes.Axes` xref yref Nr )rrrrrr) _axesr/AxesX_xrefAxesY_yrefrr get_figure)rrlxrefyrefrrrrs  zAxesDivider.__init__) axes_classcKsF|j}|durt|tr|j}nt|}|||jddfi|SNT)original)r isinstancer _axes_classtyperrG)rrkwargsrlrrr _get_new_axess zAxesDivider._get_new_axesFcKs|dur tjddd|r2t|tjstj||jd}|r,|jd||j d7_ n|j |t|tjs@tj||jd}|rY|jd||j d7_ |j d|j d}n|j ||j t |jd|j d}|jd i|}|||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.message fraction_refrrrhrir)rwarn_deprecatedrr/_Basefrom_anyrr rrr?rrrJrset_axes_locatorrrr pack_startrlocatorr`rrrnew_horizontal0     zAxesDivider.new_horizontalcKs|dur tjddd|r2t|tjstj||jd}|r,|jd||j d7_ n|j |t|tjs@tj||jd}|rY|jd||j d7_ |j |j dd}n|j ||j |j t |jdd}|jd i|}|||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.rrrrrr)rrrr/rrrr rrr?rrrJrrrrrr new_verticalrzAxesDivider.new_verticalTcKs|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)rrrrKr add_axes)rrrr add_to_figurerr`rrr append_axes s   zAxesDivider.append_axescCs*|jdur|j}|dkrdSdS|jS)NautoFT)rrr[rYrrrr[#s  zAxesDivider.get_aspectcCs$|jdur|jjdd}|jS|jSr)r rrGr_)rbboxrrrrG-s zAxesDivider.get_positioncCs|jdur |jS|jSr.)rrrPrFrrrrP4s  zAxesDivider.get_anchorcCrr)rrrrFrrrr:rzAxesDivider.get_subplotspecr)NF)NT) rrrrrrrrrr[rGrPrrrrrrs  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} | |krn|||} | S)Nrr>) rJnpasarrayTzeros fill_diagonalsumlinalgsolve)equivalent_sizesappended_sizesmax_equivalent_sizetotal_appended_sizeneq_rseq_asap_rsap_asABkarray_HkarrayHrrr_determine_karrayCs   zHBoxDivider._determine_karraycCs<dg}t||D]\\}}}||d|||q|Sr=)zipr?)rrr@rar;rrrrA[szHBoxDivider._calc_offsetsNcCst||d|dS)a  Create a new `~mpl_toolkits.axes_grid.axes_divider.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. rNr)rrhrjrrrrbszHBoxDivider.new_locatorc Cs|} |} ||} ||} || | | | } || | }|d|d|}| d}| d|d|d|}tj||||}tj||||}|||}|j|j}}||||fS)Nr>rr) rrArLrMrdrerPrfrg)rrorprqrRy_equivalent_sizesx_appended_sizesrmrnrrrrrrvrxref_hryrzr{r|rfrgrrr_locatess"  zHBoxDivider._locatec Cs|j\}}|||\} } } } ||} ||}|| | | | | |||\}}}}|dur3|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 rbrar-r*rrLrMrd)rrhrirjrkrlr%rmrnrorprqrRrrrfrgrvryr}r~rrrrrrs    & zHBoxDivider.locater.r) rrrrrrArrrrrrrrAs   rc@s$eZdZdZdddZdddZdS) VBoxDividerrNcCst|d|d|S)a Create a new `~mpl_toolkits.axes_grid.axes_divider.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. rNr)rrirkrrrrs zVBoxDivider.new_locatorc Cs|j\}}|||\} } } } ||} ||}|| | | | | |||\}}}}|dur3|d}||}}|||||||||}}tj||||Sr) r rbrar*r-rrLrMrd)rrhrirjrkrlr%rmrnrorprqrRx_equivalent_sizesy_appended_sizesrgrfrwrxr}r~rrrrrrs     &zVBoxDivider.locater.r)rrrrrrrrrrrs rcCs$t|}|jddd}|||S)Nrr)rrr)rldividerrrrrmake_axes_locatables r rcCs8|durgd}t|}|dur|}|j|||ddS)Nr)rrr)r r)r`rrrr rrrmake_axes_area_auto_adjustables r )NrN)rnumpyr matplotlibrmatplotlib.axesrmatplotlib.gridspecrrmatplotlib.transforms transformsrLrr/rrrrrrr r rrrrs(    4;@k6