P/PhZtdZddlZddlZddlmZddlZddlZddlm Z m Z m Z ddl m Z ejeZGddZGdd eZGd d eZGd d ZGddZdS)a :mod:`~matplotlib.gridspec` contains classes that help to layout multiple `~.axes.Axes` in a grid-like pattern within a figure. The `GridSpec` specifies the overall grid structure. Individual cells within the grid are referenced by `SubplotSpec`\s. Often, users need not access this module directly, and can use higher-level methods like `~.pyplot.subplots`, `~.pyplot.subplot_mosaic` and `~.Figure.subfigures`. See the tutorial :ref:`arranging_axes` for a guide. N)Integral)_api_pylab_helpers _tight_layout)BboxceZdZdZddZdZeddZedd Zd Z dd Z dd Z dZ dZ dZdZdZedZdZddddddZdS) GridSpecBasezm A base class of GridSpec that specifies the geometry of the grid that a subplot will be placed. Nc,t|tr|dkrtd|t|tr|dkrtd|||c|_|_||||dS)a Parameters ---------- nrows, ncols : int The number of rows and columns of the grid. width_ratios : array-like of length *ncols*, optional Defines the relative widths of the columns. Each column gets a relative width of ``width_ratios[i] / sum(width_ratios)``. If not given, all columns will have the same width. height_ratios : array-like of length *nrows*, optional Defines the relative heights of the rows. Each row gets a relative height of ``height_ratios[i] / sum(height_ratios)``. If not given, all rows will have the same height. rz/Number of rows must be a positive integer, not z2Number of columns must be a positive integer, not N) isinstancer ValueError_nrows_ncolsset_height_ratiosset_width_ratios)selfnrowsncols height_ratios width_ratioss S/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/matplotlib/gridspec.py__init__zGridSpecBase.__init__ s%** MeqjjK%KKMM M%** PeqjjNUNNPP P#(%  T[ }--- l+++++c&tt|jdkr d|jnd}tt|jdkr d|jnd}d|jj|j|j||zS)Nz, height_ratios=z, width_ratios=z&{clsname}({nrows}, {ncols}{optionals}))clsnamerr optionals) lenset_row_height_ratios_col_width_ratiosformat __class____name__r r)r height_arg width_args r__repr__zGridSpecBase.__repr__9sS!899::a??E)@DDDEG C 67788A==Bt'=AAACE 7>>N+++ 9, ? rc|jSN)r rs rzGridSpecBase.E$+rzThe number of rows in the grid.)docc|jSr))rr*s rr+zGridSpecBase.Gr,rz"The number of columns in the grid.c|j|jfS)zW Return a tuple containing the number of rows and columns in the grid. )r rr*s r get_geometryzGridSpecBase.get_geometryJs{DK''rcdSr))rfigures rget_subplot_paramszGridSpecBase.get_subplot_paramsPs rrc8|\}}||||z|||zf}|S)a[ Create and return a `.SubplotSpec` instance. Parameters ---------- loc : (int, int) The position of the subplot in the grid as ``(row_index, column_index)``. rowspan, colspan : int, default: 1 The number of rows and columns the subplot should span in the grid. r2)rlocrowspancolspanloc1loc2 subplotspecs rnew_subplotspeczGridSpecBase.new_subplotspecTs4 d4W ,d4<.??@ rc~| dg|jz}n't||jkrtd||_dS)z Set the relative widths of the columns. *width_ratios* must be of length *ncols*. Each column gets a relative width of ``width_ratios[i] / sum(width_ratios)``. NrzTExpected the given number of width ratios to match the number of columns of the grid)rrr r!)rrs rrzGridSpecBase.set_width_ratiosdsX  3,LL   $+ - -GHH H!-rc|jS)zo Return the width ratios. This is *None* if no width ratios have been set explicitly. )r!r*s rget_width_ratioszGridSpecBase.get_width_ratiosrs %%rc~| dg|jz}n't||jkrtd||_dS)z Set the relative heights of the rows. *height_ratios* must be of length *nrows*. Each row gets a relative height of ``height_ratios[i] / sum(height_ratios)``. NrzRExpected the given number of height ratios to match the number of rows of the grid)r rr r )rrs rrzGridSpecBase.set_height_ratioszsX  C$+-MM   4; . .DEE E"/rc|jS)zq Return the height ratios. This is *None* if no height ratios have been set explicitly. )r r*s rget_height_ratioszGridSpecBase.get_height_ratioss &&rc8|\}}||}|j}|j}|j}|j}|j} |j} ||z } ||z } | || |dz zzz } | | z}| |zt|j z fd|j D}dg|g|dz zz}tj tj ||gj }| || |dz zzz }| |z}||zt|jz fd|jD}dg|g|dz zz}tj tj ||gj }||z dj\}}||zdj\}}||||fS)a Return the positions of the grid cells in figure coordinates. Parameters ---------- fig : `~matplotlib.figure.Figure` The figure the grid should be applied to. The subplot parameters (margins and spacing between subplots) are taken from *fig*. Returns ------- bottoms, tops, lefts, rights : array The bottom, top, left, right positions of the grid cells in figure coordinates. rcg|]}|zSr2r2.0rnorms r z3GridSpecBase.get_grid_positions..sBBBQDBBBrrcg|]}|zSr2r2rEs rrIz3GridSpecBase.get_grid_positions..s@@@Aq4x@@@r))r0r4leftrightbottomtopwspacehspacesumr npcumsum column_stackflatr!reshapeT)rfigrrsubplot_paramsrMrNrOrPrQrR tot_width tot_heightcell_hsep_h cell_heights sep_heightscell_hscell_wsep_w cell_widths sep_widthscell_wsfig_tops fig_bottoms fig_lefts fig_rightsrHs @rget_grid_positionszGridSpecBase.get_grid_positionss ((** u0055"$& &&DL 6\ uvuQw'778~D$; < <<BBBB$*ABBB ceWa01 )BO[,,GHHMNNefeAg&667~D$: ; ;;@@@@)?@@@ SUGuQw/0 )BOZ,EFFKLL!$w 7 7 @ @ B+!% 8 8 A A C :Hi;;rc(|D]l}|}|Tt|dr&|}|||fkr|cSmt |||S)zo Check if the figure already has a gridspec with these dimensions, or create a new one Nget_topmost_subplotspecr3)get_axes get_gridspechasattrrnr0GridSpec)r3rraxgss r_check_gridspec_existsz#GridSpecBase._check_gridspec_existss //##  B""B~2899E 3355BBDDB??$$66IIIuV4444rc`|\}}d}t|tr\ |\}}n"#t$r}td|d}~wwxYwt j|||d|||dg||f\}} n||||zd\}} t ||| S)z,Create and return a `.SubplotSpec` instance.c6|}t|tr5||\}}}||kr||dz fStd|dkr||z}d|cxkr|krnn||fS|td|d|d|td|d|)Nrz=GridSpec slice would result in no space allocated for subplotrzindex z is out of bounds for axis z with size z) is out of bounds for GridSpec with size )r sliceindices IndexError)keysizeaxisorig_keystartstop_s r _normalizez,GridSpecBase.__getitem__.._normalizesH#u%% C!$T!2!2tQ%<< $(?* "9:::77*C????d?????8O%$&Eh&E&E-1&E&E>B&E&EFFF%&Bh&B&B;?&B&BCCCrzUnrecognized subplot specNrr)r0r tupler rTravel_multi_index SubplotSpec) rr|rrrk1k2errnum1num2s r __getitem__zGridSpecBase.__getitem__s((** u C C C( c5 ! ! > GBB G G G !<==3F G-Bq))::b%+C+CD  JD$$$C==JD$4t,,,s7 AAAFT)sharexshareysqueeze subplot_kwcD|j}|tdt|ts|rdnd}t|ts|rdnd}t jgd|||i}|}tj|j |j ft}t|j D]g}t|j D]P}d|d||d f|d |fd } | ||d <| ||d <|j |||ffi||||f<Qh|d vr |jD]} | d|dvr |jD]} | d|r3|jdkr|n|S|S)z Add all subplots specified by this `GridSpec` to its parent figure. See `.Figure.subplots` for detailed documentation. NzIGridSpec.subplots() only works for GridSpecs created with a parent figureallnone)rrowcolrFT)rr)dtype)rrr)rrrrrr)rrT)skip_non_rectangular_axes)rrr)r3r r strr check_in_listcopyrTemptyr robjectrange add_subplotrW_label_outer_xaxis_label_outer_yaxisr}itemr) rrrrrr3axarrrr shared_withrts rsubplotszGridSpecBase.subplotss( ><== =&#&& 1$0UU&F&#&& 1$0UU&F EEE"( 9 9 9 9  J__&& $+t{36BBB%% 2 2CT[)) 2 2'+E$K&+CFmE!S&MKK '26': 8$'26': 8$"4&"4cN#2#2&0#2#2c3h  2 ^ # #j F F%%%EEEE ^ # #j F F%%%EEEE  $):??5::<<<  GLr)NNr))rr)r$ __module__ __qualname____doc__rr'propertyrrr0r4r<rr?rrBrl staticmethodrvrrr2rrr r sT ,,,,2    H--: < < z/GridSpec.get_subplot_params..s7...S\"3A"56...rc2i|]}|t|Sr2getattrrFrrs rrz/GridSpec.get_subplot_params..s%MMMaaq!1!1MMMrr2)r SubplotParamsr subplotparsr)rr3kwrs` rr4zGridSpec.get_subplot_paramss >..,...B'--"--KK)F$677K NNMMMM4;LMMMNNNrc*fdjDS)z Return a list of the names of the subplot parameters explicitly set in the GridSpec. This is a subset of the attributes of `.SubplotParams`. c4g|]}t||Sr2rrs rrIz.s(AAAaa0@0@AAAAr)rr*s`rlocally_modified_subplot_paramsz(GridSpec.locally_modified_subplot_paramss#BAAA4,AAAArHzG?c ||}tj||jtj|j||||||}|r|jdi|dSdS)a Adjust subplot parameters to give specified padding. Parameters ---------- figure : `.Figure` The figure. renderer : `.RendererBase` subclass, optional The renderer to be used. pad : float Padding between the figure edge and the edges of subplots, as a fraction of the font-size. h_pad, w_pad : float, optional Padding (height/width) between edges of adjacent subplots. Defaults to *pad*. rect : tuple (left, bottom, right, top), default: None (left, bottom, right, top) rectangle in normalized figure coordinates that the whole subplots area (including labels) will fit into. Default (None) is the whole figure. N) grid_spec)padh_padw_padrectr2) _get_rendererrget_tight_layout_figurerget_subplotspec_listr)rr3rendererrrrrrs r tight_layoutzGridSpec.tight_layouts,  ++--H6 FK  .v{d K K K #U%dDDD  " DK ! !& ! ! ! ! ! " "r) NNNNNNNNNr))NrNNN) r$rrrrrrr4rr __classcell__r#s@rrsrs/s-19=%)26464646464646lJIILQQQ8.BBB-1<@""""""""rrsc6eZdZdZ dfd ZddZdZxZS)GridSpecFromSubplotSpeczx GridSpec whose subplot layout parameters are inherited from the location specified by a given SubplotSpec. Nc ||_||_t|tr||_nt d|jj|_t ||||dS)ak Parameters ---------- nrows, ncols : int Number of rows and number of columns of the grid. subplot_spec : SubplotSpec Spec from which the layout parameters are inherited. wspace, hspace : float, optional See `GridSpec` for more details. If not specified default values (from the figure or rcParams) are used. height_ratios : array-like of length *nrows*, optional See `GridSpecBase` for details. width_ratios : array-like of length *ncols*, optional See `GridSpecBase` for details. zVsubplot_spec must be type SubplotSpec, usually from GridSpec, or axes.get_subplotspec.rN) _wspace_hspacer r _subplot_spec TypeErrorrqr3rr) rrr subplot_specrQrRrrr#s rrz GridSpecFromSubplotSpec.__init__s&  lK 0 0 O!-D  NOO O(5577>  &2'4  6 6 6 6 6rc0|j|jn| |jjntjd}|j|jn| |jjntjd}|j|}|j \}}}}t||||||S)z1Return a dictionary of subplot layout parameters.Nzfigure.subplot.hspacezfigure.subplot.wspace)rMrNrOrPrQrR) rrrRrrrrQrrextentsr) rr3rRrQfigboxrMrOrNrPs rr4z*GridSpecFromSubplotSpec.get_subplot_paramss"&,":$,,4:4Fv)00|$;< #',":$,,4:4Fv)00|$;< #0088#)> feS$e$*$*6;;; ;rc4|jS)zY Return the topmost `.SubplotSpec` instance associated with the subplot. )rrnr*s rrnz/GridSpecFromSubplotSpec.get_topmost_subplotspecs!99;;;r)NNNNr))r$rrrrr4rnrrs@rrrss &*26666666@;;;; <<<<<<rcft|dkr|\}t|tr|St|tst d| t t t|\}}}n_#t$rt d|dwxYwt|dkr|\}}}n$tj ddt|t |||}|t|||}t|trDt|d kr1td |Dst d ||\}}n@t|tr|dks |||zkrt d ||zd ||x}}||dz |S)z Construct a `.SubplotSpec` from a parent `.Figure` and either - a `.SubplotSpec` -- returned as is; - one or three numbers -- a MATLAB-style subplot specifier. rz>Single argument to subplot must be a three-digit integer, not Nsubplotz1 or 3)takesgivenrorLc3@K|]}t|tVdSr))r r)rFns r z1SubplotSpec._from_subplot_args..Fs,<<1z!X..<<<<<>DC#{++ - X.. - ,$',,--- 7"%c3s88"4"4dCC 7 7 7 ,$',,--267 7YY!^^"OD$"9HCIINNN N  , ,VT4 @ @ :$V444B c5 ! ! c#hh!mm<<<<<<<  O#OODAqqc8,, a3d?? #tDy####KA!A#a%ys &A>>Bc,|j|jn|jSr))_num2rr*s rrzSubplotSpec.num2Ws J.tyyDJ>rc||_dSr))r )rvalues rrzSubplotSpec.num2[s  rc|jSr))rr*s rrqzSubplotSpec.get_gridspec_s ~rct|\}}|||j|jfS)a Return the subplot geometry as tuple ``(n_rows, n_cols, start, stop)``. The indices *start* and *stop* define the range of the subplot within the `GridSpec`. *stop* is inclusive (i.e. for a single cell ``start == stop``). )rqr0rr)rrrs rr0zSubplotSpec.get_geometrybs8&&((5577 dT49di//rcz|j}t|j|z|j|zdzS)z6The rows spanned by this subplot, as a `range` object.r)rqrrrr)rrs rr7zSubplotSpec.rowspanms;!!##)TY%'e);a)?@@@rc|j}t|j|z|j|zg\}}t ||dzS)z9The columns spanned by this subplot, as a `range` object.r)rqrsortedrrr)rrc1c2s rr8zSubplotSpec.colspanssR!!##)U*DI,=>??BRa   rc"|jjdkSNr)r7rr*s r is_first_rowzSubplotSpec.is_first_row||!Q&&rcP|jj|jkSr))r7rrqrr*s r is_last_rowzSubplotSpec.is_last_row!| D$5$5$7$7$===rc"|jjdkSr)r8rr*s r is_first_colzSubplotSpec.is_first_colrrcP|jj|jkSr))r8rrqrr*s r is_last_colzSubplotSpec.is_last_colrrc|}|\}}tj|j|jg||f\}}||\}}} } ||} ||} | |} | |}tj | | || S)zJ Update the subplot position from ``figure.subplotpars``. ) rqr0rT unravel_indexrrrlminmaxr from_extents)rr3rrrrrrirhrjrk fig_bottomfig_topfig_left fig_rights rrzSubplotSpec.get_positions$$&&,,.. u%ty$)&>> **\*^??X? [[ 0 0 0AAXA !!X!'''>>>'''>>>KKK  7 7 7<<<(E(E(E(E(Errc*eZdZdZ ddZ ddZdS)rzM Parameters defining the positioning of a subplots grid in a figure. NcdD]&}t||tjd|'|||||||dS)au Defaults are given by :rc:`figure.subplot.[name]`. Parameters ---------- left : float The position of the left edge of the subplots, as a fraction of the figure width. right : float The position of the right edge of the subplots, as a fraction of the figure width. bottom : float The position of the bottom edge of the subplots, as a fraction of the figure height. top : float The position of the top edge of the subplots, as a fraction of the figure height. wspace : float The width of the padding between subplots, as a fraction of the average Axes width. hspace : float The height of the padding between subplots, as a fraction of the average Axes height. rrN)rrrr)rrMrOrNrPrQrRr|s rrzSubplotParams.__init__s]4J F FC D#s|,Cc,C,CD E E E E D&%ff=====rc||n|j||n|jkrtd||n|j||n|jkrtd|||_|||_|||_|||_|||_| ||_dSdS)zY Update the dimensions of the passed parameters. *None* means unchanged. Nzleft cannot be >= rightzbottom cannot be >= top)rMrNr rOrPrQrR)rrMrOrNrPrQrRs rrzSubplotParams.updates %TT49".EEDJ @ @677 7)VVt{?CC : :677 7  DI  DJ   DK ?DH   DK   DKKK  r)NNNNNN)r$rrrrrr2rrrrsY@D%)>>>><>B#'!!!!!!rr)rrloggingnumbersrnumpyrT matplotlibrrrrmatplotlib.transformsr getLoggerr$_logr rsrrrr2rrr<s   ::::::::::&&&&&&w""RRRRRRRRj["["["["["|["["["|9<9<9<9<9