jXzdZddlZddlZddlmZddlZddlZddlm Z m Z m Z ddl m ZddlmZejeZGddZGd d eZGd d eZGd dZGddZdS)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)UNSET)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 AC:\Projects\DSCons\.venv\Lib\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__rr)r height_arg width_args r__repr__zGridSpecBase.__repr__:sS!899::a??E)@DDDEG C 67788A==Bt'=AAACE 7>>N+++ 9, ? rc|jSN)rrs rzGridSpecBase.F$+rzThe number of rows in the grid.)docc|jSr*)rr+s rr,zGridSpecBase.Hr-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. )rrr+s r get_geometryzGridSpecBase.get_geometryKs{DK''rcdSr*)rfigures rget_subplot_paramszGridSpecBase.get_subplot_paramsQs 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. r3)rlocrowspancolspanloc1loc2 subplotspecs rnew_subplotspeczGridSpecBase.new_subplotspecUs4 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_ratiosesX  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_ratiosss %%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)rrr r!)rrs rrzGridSpecBase.set_height_ratios{sX  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|]}|zSr3r3.0rnorms r z3GridSpecBase.get_grid_positions..sBBBQDBBBrrcg|]}|zSr3r3rFs rrJz3GridSpecBase.get_grid_positions..s@@@Aq4x@@@r))r1r5leftrightbottomtopwspacehspacesumr!npcumsum column_stackflatr"reshapeT)rfigrrsubplot_paramsrNrOrPrQrRrS tot_width tot_heightcell_hsep_h cell_heights sep_heightscell_hscell_wsep_w cell_widths sep_widthscell_wsfig_tops fig_bottoms fig_lefts fig_rightsrIs @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_subplotspecr4)get_axes get_gridspechasattrror1GridSpec)r4rraxgss 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)r1r tupler rUravel_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)r4r r strr check_in_listcopyrUemptyrrobjectrange add_subplotrX_label_outer_xaxis_label_outer_yaxisr~itemr) rrrrrr4axarrrr shared_withrus 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(propertyrrr1r5r=rr@rrCrm staticmethodrwrrr3rrr r sT ,,,,2    H--: < < z/GridSpec.get_subplot_params..s7...S\"3A"56...rc2i|]}|t|Sr3getattrrGrrs rrz/GridSpec.get_subplot_params..s%MMMaaq!1!1MMMrr3) _AllowedKeys SubplotParamsr subplotparsr)rr4kwrs` rr5zGridSpec.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||Sr3rrs rrJz.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_padrectr3) _get_rendererrget_tight_layout_figurerget_subplotspec_listr)rr4rendererrrrrkwargss r tight_layoutzGridSpec.tight_layouts,  ++--H6 FK  .v{d K K K #U%dDDD  " DK ! !& ! ! ! ! ! " "r) NNNNNNNNNr*)NrNNN) r%rrrrrrrr5rr __classcell__r$s@rrtrt0s-19=%)26464646464646lJIIL#F&fV#Q#Q#Q#Q#QJ.BBB-1<@""""""""rrtc6eZdZdZ 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 TypeErrorrrr4rr) rrr subplot_specrRrSrrr$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)rNrOrPrQrRrS) rrrSrrrrRrrextentsr) rr4rSrRfigboxrNrPrOrQs rr5z*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. )rror+s rroz/GridSpecFromSubplotSpec.get_topmost_subplotspec s!99;;;r)NNNNr*)r%rrrrr5rorrs@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)takesgivenrprMc3@K|]}t|tVdSr*)r r)rGns r z1SubplotSpec._from_subplot_args..Ps,<<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.num2as J.tyyDJ>rc||_dSr*)r)rvalues rrzSubplotSpec.num2es  rc|jSr*)rr+s rrrzSubplotSpec.get_gridspecis ~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``). )rrr1rr)rrrs rr1zSubplotSpec.get_geometryls8&&((5577 dT49di//rcz|j}t|j|z|j|zdzS)z6The rows spanned by this subplot, as a `range` object.r)rrrrrr)rrs rr8zSubplotSpec.rowspanws;!!##)TY%'e);a)?@@@rc|j}t|j|z|j|zg\}}t ||dzS)z9The columns spanned by this subplot, as a `range` object.r)rrrsortedrrr)rrc1c2s rr9zSubplotSpec.colspan}sR!!##)U*DI,=>??BRa   rc"|jjdkSNr)r8rr+s r is_first_rowzSubplotSpec.is_first_row|!Q&&rcP|jj|jkSr*)r8rrrrr+s r is_last_rowzSubplotSpec.is_last_row!| D$5$5$7$7$===rc"|jjdkSr)r9rr+s r is_first_colzSubplotSpec.is_first_colrrcP|jj|jkSr*)r9rrrrr+s r is_last_colzSubplotSpec.is_last_colrrc|}|\}}tj|j|jg||f\}}||\}}} } ||} ||} | |} | |}tj | | || S)zJ Update the subplot position from ``figure.subplotpars``. ) rrr1rU unravel_indexrrrmminmaxr from_extents)rr4rrrrrrjrirkrl fig_bottomfig_topfig_left fig_rights rrzSubplotSpec.get_positions$$&&,,.. u%ty$)&>> **\*^??X? [[ 0 0 0AAXA !!X!'''>>>'''>>>KKK  7 7 7<<<(E(E(E(E(Errc6eZdZdZ ddZ ddZdZdZdS)rzM Parameters defining the positioning of a subplots grid in a figure. NcdD]&}t||tjd|'|||||||dS)a Defaults are given by :rc:`figure.subplot.[name]`. Parameters ---------- left : float, optional The position of the left edge of the subplots, as a fraction of the figure width. right : float, optional The position of the right edge of the subplots, as a fraction of the figure width. bottom : float, optional The position of the bottom edge of the subplots, as a fraction of the figure height. top : float, optional The position of the top edge of the subplots, as a fraction of the figure height. wspace : float, optional The width of the padding between subplots, as a fraction of the average Axes width. hspace : float, optional The height of the padding between subplots, as a fraction of the average Axes height. rrN)setattrrrr)rrNrPrOrQrRrSr}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)rNrOr rPrQrRrS)rrNrPrOrQrRrSs rrzSubplotParams.updates %TT49".EEDJ @ @677 7)VVt{?CC : :677 7  DI  DJ   DK ?DH   DK   DKKK  rc||D]&}t||tjd|'dS)zIRestore the subplot positioning parameters to the default rcParams valuesrN)to_dictr2rr)rr}s rresetzSubplotParams.reset sL<<>> F FC D#s|,Cc,C,CD E E E E F Frc4|jS)z2Return a copy of the subplot parameters as a dict.)__dict__rr+s rr5zSubplotParams.to_dict%s}!!###r)NNNNNN)r%rrrrrr6r5r3rrrrsz@D%)>>>><>B#'!!!!0FFF $$$$$rr)rrloggingnumbersrnumpyrU matplotlibrrrrmatplotlib._apirrmatplotlib.transformsr getLoggerr%_logr rtrrrr3rrrAs   ::::::::::++++++&&&&&&w""RRRRRRRRjd"d"d"d"d"|d"d"d"N9<9<9<9<9