ehHWddlmZddlZddlmZddlZddlmZm Z ddl m Z ddl m Z mZmZddlmZmZGd d Ze j(ed ZGd d ZGddeZeZy))NumberN) MethodType)_apicbook) SubplotSpec)SizeSubplotDividerDivider)AxesSimpleAxisArtistc$eZdZfdZdZxZS) CbarAxesBasec2||_t||i|yN) orientationsuper__init__)selfrargskwargs __class__s _/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/mpl_toolkits/axes_grid1/axes_grid.pyrzCbarAxesBase.__init__s& $)&)c b|jdj|f||jd|S)NFroot)caxlocation) get_figurecolorbarr)rmappablers rr!zCbarAxesBase.colorbarsA3tE*33 E)9)9E=CE Er)__name__ __module__ __qualname__rr! __classcell__rs@rrrs*ErrzCbar{}ceZdZdZeZ dddddddddZdZd Zd Z d Z d Z d Z dZ dZdZdZdZdZdZy)GridaL A grid of Axes. In Matplotlib, the Axes location (and size) is specified in normalized figure coordinates. This may not be ideal for images that needs to be displayed with a given aspect ratio; for example, it is difficult to display multiple images of a same size with some fixed padding between them. AxesGrid can be used in such case. Attributes ---------- axes_all : list of Axes A flat list of Axes. Note that you can also access this directly from the grid. The following is equivalent :: grid[i] == grid.axes_all[i] len(grid) == len(grid.axes_all) axes_column : list of list of Axes A 2D list of Axes where the first index is the column. This results in the usage pattern ``grid.axes_column[col][row]``. axes_row : list of list of Axes A 2D list of Axes where the first index is the row. This results in the usage pattern ``grid.axes_row[row][col]``. axes_llc : Axes The Axes in the lower left corner. ngrids : int Number of Axes in the grid. NFTL) share_allshare_xshare_y label_mode axes_classaspectc|\|_|_||j|jz}n9d|cxkr&|j|jzkstdtd||_t t j tj|d\|_ |_ tjddg|||_ | |j} n1t| t t"fr| \} }t%j&| fi|} t)gg| }t|t*t,frt/||fi||_nOt3|d k(rt/|g|i||_n,t3|d k(rt5||fi||_n t7d |j0j9}tj:|j|jfdt< }t?|jD]G}|jA|\}}|r|d x}}n|r|d|fnd}| r||dfnd}| |||||||f<I|jC|jdk(rdndjE|_#|jHjE|_%|jE|_&|jJdd|_'|jQ|jFD]}|jS||jU| y)a Parameters ---------- fig : `.Figure` The parent figure. rect : (float, float, float, float), (int, int, int), int, or `~.SubplotSpec` The axes position, as a ``(left, bottom, width, height)`` tuple, as a three-digit subplot position code (e.g., ``(1, 2, 1)`` or ``121``), or as a `~.SubplotSpec`. nrows_ncols : (int, int) Number of rows and columns in the grid. ngrids : int or None, default: None If not None, only the first *ngrids* axes in the grid are created. direction : {"row", "column"}, default: "row" Whether axes are created in row-major ("row by row") or column-major order ("column by column"). This also affects the order in which axes are accessed using indexing (``grid[index]``). axes_pad : float or (float, float), default: 0.02 Padding or (horizontal padding, vertical padding) between axes, in inches. share_all : bool, default: False Whether all axes share their x- and y-axis. Overrides *share_x* and *share_y*. share_x : bool, default: True Whether all axes of a column share their x-axis. share_y : bool, default: True Whether all axes of a row share their y-axis. label_mode : {"L", "1", "all", "keep"}, default: "L" Determines which axes will get tick labels: - "L": All axes on the left column get vertical tick labels; all axes on the bottom row get horizontal tick labels. - "1": Only the bottom left axes is labelled. - "all": All axes are labelled. - "keep": Do not do anything. axes_class : subclass of `matplotlib.axes.Axes`, default: `.mpl_axes.Axes` The type of Axes to create. aspect : bool, default: False Whether the axes aspect ratio follows the aspect ratio of the data limits. Nrz7ngrids must be positive and not larger than nrows*ncolscolumnrow) direction) horizontalverticalr0zIncorrect rect format)dtype)rr)sharexshareyCF)order)+_nrows_ncols ValueErrorngridsmapr Fixednp broadcast_to_horiz_pad_size_vert_pad_sizer check_in_list _direction_defaultAxesClass isinstancelisttuple functoolspartialdictrrr _dividerlenr TypeError get_positionfullobjectrange _get_col_rowraveltolistaxes_allT axes_columnaxes_rowaxes_llc_init_locatorsadd_axesset_label_mode)rfigrect nrows_ncolsrDr5axes_padr+r,r-r.r/r0clsrkw axes_arrayicolr4r;r<axs rrz Grid.__init__<sr$/  T[ >[[4;;.Fv:t{{!:: MOO; MOO 47 JJ!4561d1 He, B#  //J  T5M 2$KC"**39&9J R"V < dV[1 2*3;;DM Y!^*3<<S9T)U W Wrc|jdk(rt||j\}}||fSt||j\}}||fS)Nr3)rLdivmodrArB)rnrnr4s rr[zGrid._get_col_rowsL ??h &a-HCCxa-HCCxrc,t|jSr)rUr^rs r__len__z Grid.__len__s4==!!rc |j|Sr)r^)rrms r __getitem__zGrid.__getitem__s}}Qrc2|j|jfS)zV Return the number of rows and columns of the grid as (nrows, ncols). )rArBr}s r get_geometryzGrid.get_geometrys{{DKK''rcT|d|j_|d|j_y)z Set the padding between the axes. Parameters ---------- axes_pad : (float, float) The padding (horizontal pad, vertical pad) in inches. rrNrI fixed_sizerJ)rris r set_axes_padzGrid.set_axes_pads(+31+')1!&rcZ|jj|jjfS)z Return the axes padding. Returns ------- hpad, vpad Padding (horizontal pad, vertical pad) in inches. rr}s r get_axes_padzGrid.get_axes_pads,$$//##..0 0rc:|jj|y)z%Set the aspect of the SubplotDivider.N)rT set_aspect)rr0s rrzGrid.set_aspects   (rc6|jjS)z(Return the aspect of the SubplotDivider.)rT get_aspectr}s rrzGrid.get_aspects}}''))rc\tjgd|tjd|jd|j f|jdz ggdgggk(\}}|dk(r/tj |j|j fdx}}n|dk(r|}|}n|d k(r||zx}}nyt|jD]}t|j D]}|j||}t|jtrIt|jd|jd } t|jd|jd } n|jd } |jd } | j!|||f|||f | j!|||f|||f y) a Define which axes have tick labels. Parameters ---------- mode : {"L", "1", "all", "keep"} The label mode: - "L": All axes on the left column get vertical tick labels; all axes on the bottom row get horizontal tick labels. - "1": Only the bottom left axes is labelled. - "all": All axes are labelled. - "keep": Do not do anything. )allr*1keep)modeNrrrTr*rbottomleft) ticklabelslabel)rrKrGmgridrArBrXrZrarNaxisrr xaxisspinesyaxistoggle) rr is_last_row is_first_colrrrmjro bottom_axis left_axiss rrezGrid.set_label_modes 44@ HH\dkk\E EFT S[ FD S[',6 6FT t{{# JA4;;' J]]1%a(bggz2"2288Q (@S"TK 01bii>O PI"$''("3K "I""fQTl&A,"O  DAJd1a4j I J Jrc|jSr)rTr}s r get_dividerzGrid.get_dividers }}rc:|jj|yr)rT set_locator)rlocators rrwzGrid.set_axes_locators !!'*rc6|jjSr)rT get_locatorr}s rget_axes_locatorzGrid.get_axes_locators}}((**rNr4g{Gz?)r#r$r%__doc__r rMrrcr[r~rrrrrrrerrwrrrr)r)s<   q(! q(fW" ( 5 0)*%JN++rr)c FeZdZdZ d dddddddddd fd Zd ZxZS) ImageGridaR A grid of Axes for Image display. This class is a specialization of `~.axes_grid1.axes_grid.Grid` for displaying a grid of images. In particular, it forces all axes in a column to share their x-axis and all axes in a row to share their y-axis. It further provides helpers to add colorbars to some or all axes. NFTr*rightz5%) r+r0r. cbar_mode cbar_locationcbar_pad cbar_size cbar_set_caxr/c tjgd| tjgd| | |_| |_| |_| |_t ||||||||dd|| | |jD]}|j||r|jdk(r&|jD]}|jd|_ y |jd k(rit|jD]P\}}|j|\}}|jd vr|j||_ =|j||_ Ry t|j|jD] \}}||_ y y ) a Parameters ---------- fig : `.Figure` The parent figure. rect : (float, float, float, float) or int The axes position, as a ``(left, bottom, width, height)`` tuple or as a three-digit subplot position code (e.g., "121"). nrows_ncols : (int, int) Number of rows and columns in the grid. ngrids : int or None, default: None If not None, only the first *ngrids* axes in the grid are created. direction : {"row", "column"}, default: "row" Whether axes are created in row-major ("row by row") or column-major order ("column by column"). This also affects the order in which axes are accessed using indexing (``grid[index]``). axes_pad : float or (float, float), default: 0.02in Padding or (horizontal padding, vertical padding) between axes, in inches. share_all : bool, default: False Whether all axes share their x- and y-axis. Note that in any case, all axes in a column share their x-axis and all axes in a row share their y-axis. aspect : bool, default: True Whether the axes aspect ratio follows the aspect ratio of the data limits. label_mode : {"L", "1", "all"}, default: "L" Determines which axes will get tick labels: - "L": All axes on the left column get vertical tick labels; all axes on the bottom row get horizontal tick labels. - "1": Only the bottom left axes is labelled. - "all": all axes are labelled. cbar_mode : {"each", "single", "edge", None}, default: None Whether to create a colorbar for "each" axes, a "single" colorbar for the entire grid, colorbars only for axes on the "edge" determined by *cbar_location*, or no colorbars. The colorbars are stored in the :attr:`cbar_axes` attribute. cbar_location : {"left", "right", "bottom", "top"}, default: "right" cbar_pad : float, default: None Padding between the image axes and the colorbar axes. .. versionchanged:: 3.10 ``cbar_mode="single"`` no longer adds *axes_pad* between the axes and the colorbar if the *cbar_location* is "left" or "bottom". cbar_size : size specification (see `.Size.from_any`), default: "5%" Colorbar size. cbar_set_cax : bool, default: True If True, each axes in the grid has a *cax* attribute that is bound to associated *cbar_axes*. axes_class : subclass of `matplotlib.axes.Axes`, default: None )eachsingleedgeN)r)rrrtop)rT)r5rir+r,r-r0r.r/rrrrrN)rrK_colorbar_mode_colorbar_location _colorbar_pad_colorbar_sizerr cbar_axesrdr^r enumerater[zip)rrfrgrhrDr5rir+r0r.rrrrrr/roindexrnr4rrs rrzImageGrid.__init__)srN ;%. 0 =)6 8'"/%'  {F(tF!j  : .. B LL   ""h.--/B!^^A.BF/$$.!*4==!95IE2#007HC..2CC!%!4!%!4 5 #4==$..A!GB BF! rc|jE|jdvr|jj|_n|jj|_t |j Dcgc]_}t|j|jdjd|jj|jac}|_ |j}|j}g}g}g}g}|dk(r|dvr|dk(r|jt!j"|j$z}|j't!j(|j*||j't!j(|j||jj-ddd } n|d k(r|j.t!j0|j$z}|j't!j(|j*||j't!j(|j||jj-dd d } t |j D] } |j| j3d"|jdj5 |jdj3d t7|j8dD]\} } | dk7r|j'|j| r&t!j"| d|jd}n2t!j"|jdd|jd}|dk(r|dk(s |dk(r}| dk(rx|j't;||j't!j(|j*||j't!j(|j||j't;||j'||dk(s?|dk(s|dk(sK| |j.dz k(s_|j't!j(|j||j't;||j't!j(|j*|g} g}t7|j<dddd D]\}} |dk7r|j'|j| r&t!j0| d|jd}n2t!j0|jdd|jd}|d k(r|dk(s |dk(r}|dk(rx|j't;||j't!j(|j*||j't!j(|j|| j't;||j'||dk(s?|dk(s|dk(sK||jdz k(s_|j't!j(|j||j't;||j't!j(|j*|t |j D]} |j?| \} }|jj-|| | |jdz |z } |j| j5| |dk(r|dvr4|jj-|| | |jdz |z } n7|dvr3|jj-|| ||jdz |z } |j| j5| |dk(s|dk(r| dk(s|dk(re| |j.dz k(rS|jj-|d| |jdz |z } |j|j5| w|d k(r||jdz k(s|dk(s|dk(s|jj-|| |d} |j| j5| |dk(r|dk(r|jt!j"|j$z}|j't!j(|j||j't!j(|j*||jj-ddd } n|dk(r|j.t!j0|j$z}|j't!j(|j||j't!j(|j*||jj-dd d } |dvrt |j D] } |j| j3d"|jdj5 |jdj3d n%|dk(r9t |j D] } |j| j3d "n|dk(r|dvr |j}n |j.}t |D] } |j| j3d "t  dz|j D] }|j|j3d"nZt |j D]B} |j| j3d|j| jAgddD|jjC||jjE|ycc}w)NrrFr)rr)rrrr@)rrrsny1r)rrnx1rsTaxes)r0ref_axrrrrrrq)rr)rr)rr)?rMbP?ractive)which)#rrrIrrJrZrD_cbaraxes_class_factoryrMr^r rTrWrrrAr AxesXrbappendfrom_anyrrxrBAxesY set_visiblerwrrarUr`r[ set_positionrtrv)r_cb_mode cb_locationhvh_ax_posh_cb_posszrrmrnrov_ax_posv_cb_posr4countrs rrczImageGrid._init_locatorss     %&&*;;%)%9%9%D%D"%)%8%8%C%C" 4;;' ) < #D$:$: ; a +++79S9S9U 33 5) %%--   h ;2D#Df$[[4::dmm#<<t':':B?@t'9'92>?--33qQB3G([[4::dmm#<<t':':B?@t'9'92>?--33qbQ3G4;;' 5q!--e4 5 NN1  . .w 7 NN1  ) )$ / q!12 AGCax--.ZZ6$--:JKZZ a 0'-dmmA6FHv% F*$.3!8A't':':B?@t'9'92>? OOCF # HHRLw& F*$.3$++/3It'9'92>?A't':':B?@3 A6 !1!1!!4TrT!:; AGCax,,-ZZ6$--:JKZZ a 0'-dmmA6FHx' F*$.3!8A't':':B?@t'9'92>? OOCF # HHRLu$ F*$.3$++/3It'9'92>?A't':':B?@3 A6t{{# BA((+HCmm//8C=3;DKKM#?t':':B?@--33raR3H%[[4::dmm#<<t'9'92>?t':':B?@--33qbR3H..t{{+9ANN1%11%89q!227;q!--d3  4;;' 4q!--d3 4  //  5\ 4q!--d3 41q5$++. 5q!--e4 54;;' ?q!--e4q!../E5=/? ? $$Q' ""1%k)s)A$o*r)r#r$r%rrrcr&r's@rrrsC  g!!&" g!R^&rr)numbersrrQtypesrnumpyrG matplotlibrrmatplotlib.gridspecr axes_dividerr r r mpl_axesr r r_make_class_factoryrr)rAxesGridrrrrsf"+77,EE4%33L(KA+A+HQ&Q&h r