o - hHW@sddlmZddlZddlmZddlZddlmZm Z ddl m Z ddl m Z mZmZddlmZmZGd d d Ze ed ZGd d d ZGdddeZeZdS))NumberN) MethodType)_apicbook) SubplotSpec)SizeSubplotDividerDivider)AxesSimpleAxisArtistcs$eZdZfddZddZZS) CbarAxesBasecs||_tj|i|dSN) orientationsuper__init__)selfrargskwargs __class__U/var/www/vscode/kcb/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/axes_grid.pyrszCbarAxesBase.__init__cKs"|jddj|f||jd|S)NFroot)caxlocation) get_figurecolorbarr)rmappablerrrrrs zCbarAxesBase.colorbar)__name__ __module__ __qualname__rr __classcell__rrrrr s r zCbar{}c@seZdZdZeZ   d%dddddddd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ddZddZddZdd Zd!d"Zd#d$ZdS)&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. Nrow{Gz?FTL) share_allshare_xshare_y label_mode axes_classaspectcCsH|\|_|_|dur|j|j}nd|kr"|j|jks'tdtd||_ttjt|d\|_ |_ t j ddg|d||_ | durL|j} nt| ttfr`| \} }tj| fi|} tgg| d}t|ttfryt||fi||_n(t|d krt|g|Ri||_nt|d krt||fi||_ntd |j}tj|j|jfdtd }t|jD]2}| |\}}|r|d }}n|r|d|fnd}| r||dfnd}| ||||d|||f<q|j!|j dkrdndd"|_#|j$"|_%|"|_&|j%dd|_'|(|j#D]}|)|q|*| dS)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*ncolscolumnr%) direction) horizontalverticalr-zIncorrect rect format)dtype)rr)sharexshareyCF)order)+_nrows_ncols ValueErrorngridsmaprFixednp 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_ncolsr?r0axes_padr(r)r*r+r,r-clsrkw axes_arrayicolr%r6r7axrrrr<sj 9       z Grid.__init__cCs|jtd|jg|jdtdg|jtd|jg|jdtdgt |j D] }| |\}}|j | |jjd|d|jd|dq3dS)Nrr.nxny)rOset_horizontalrScaledrDr= set_verticalrEr<rUr?rVrYset_axes_locator new_locator)rrhrir%rrrr^s$$  zGrid._init_locatorscCs:|jdkrt||j\}}||fSt||j\}}||fS)Nr/)rGdivmodr<r=)rnrir%rrrrVs zGrid._get_col_rowcCs t|jSr)rPrYrrrr__len__ z Grid.__len__cCs |j|Sr)rY)rrhrrr __getitem__rwzGrid.__getitem__cCs |j|jfS)zV Return the number of rows and columns of the grid as (nrows, ncols). )r<r=rurrr get_geometrys zGrid.get_geometrycCs|d|j_|d|j_dS)z Set the padding between the axes. Parameters ---------- axes_pad : (float, float) The padding (horizontal pad, vertical pad) in inches. rrNrD fixed_sizerE)rrdrrr set_axes_pads zGrid.set_axes_padcCs|jj|jjfS)z Return the axes padding. Returns ------- hpad, vpad Padding (horizontal pad, vertical pad) in inches. rzrurrr get_axes_pads zGrid.get_axes_padcCs|j|dS)z%Set the aspect of the SubplotDivider.N)rO set_aspect)rr-rrrr~szGrid.set_aspectcC |jS)z(Return the aspect of the SubplotDivider.)rO get_aspectrurrrrs zGrid.get_aspectc CsLtjgd|dtjd|jd|jf|jdggdgggk\}}|dkr4t|j|jfd}}n|dkr=|}|}n |d krH||@}}ndSt|jD]T}t|jD]L}|j||}t |j t rzt |j d|jd } t |jd|jd } n |j d } |j d } | j|||f|||fd | j|||f|||fd qVqOdS) 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)rrFrBmgridr<r=rSrUr\rIaxisrr xaxisspinesyaxistoggle) rr is_last_row is_first_colrrrhjrj bottom_axis left_axisrrrr`s..    zGrid.set_label_modecCs|jSr)rOrurrr get_dividerszGrid.get_dividercCs|j|dSr)rO set_locator)rlocatorrrrrqszGrid.set_axes_locatorcCrr)rO get_locatorrurrrget_axes_locatorrwzGrid.get_axes_locatorNr%r&)r r!r"__doc__r rHrr^rVrvrxryr|r}r~rr`rrqrrrrrr$s6 s    ' r$c sFeZdZdZ   dddddddd ddd fd d Zd dZZS) 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. Nr%r&FTr'rightz5%) r(r-r+ cbar_mode cbar_locationcbar_pad cbar_size cbar_set_caxr,c stjgd| dtjgd| d| |_| |_| |_| |_tj|||||||dd|| |d |jD]}| |q3|r|jdkrP|j D]}|jd|_ qEd S|jd krzt |j D]\}}| |\}}|jd vrq|j||_ qZ|j||_ qZd St|j |jD] \}}||_ qd Sd S) 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)r0rdr(r)r*r-r+r,rrrrrN)rrF_colorbar_mode_colorbar_location _colorbar_pad_colorbar_sizerr cbar_axesr_rYr enumeraterVzip)rrarbrcr?r0rdr(r-r+rrrrrr,rjindexrir%rrrrr)sD G        zImageGrid.__init__cspjdurjdvrjj_njj_fddtjD_j}j}g}g}g}g}|dkr|dvr|dkrcj t j }| t j|| t j|jjdddd }n*|d krjt j }| t j|| t j|jjdddd }tjD] } j| d qjd|jdd tjdD]\} } | dkr| j| rt j | djdd}nt j jddjdd}|dkr |dks|dkr | dkr | t|| t j|| t j|| t|| ||dkrJ|dks/|dkrJ| jdkrJ| t j|| t|| t j|qg} g} tjddddD]\}} |dkrj| j| ryt j| djdd}nt jjddjdd}|d kr|dks|dkr|dkr| t|| t j|| t j|| t|| ||dkr|dks|dkr|j dkr| t j|| t|| t j|q[tjD]} | \} }jj|| | j d|d}j| ||dkrY|dvr9jj|| | j d|d}n|dvrOjj|| | j d|d}j| |q|dkr|dkrh| dksu|dkr| jdkrjj|d| j d|d}j||q|d kr|j dks|dkr|dkrjj|| | dd}j| |q|dkr@|dkrj t j }| t j|| t j|jjdddd }n+|dkrjt j }| t j|| t j|jjdddd }|dvr?tjD] } j| d q#jd|jdd nj|dkrWtjD] } j| d qJnS|dkr|dvrej }nj}t|D] } j| d qlt| djD] }j|d qntjD]} j| d j| jgdddqj|j|dS)Nrcs6g|]}tjjdjddjjdqS)rFr)r)_cbaraxes_class_factoryrHrYrrOrRr).0_rurr sz,ImageGrid._init_locators..r)rrrrr;)rlrmny1r)rlnx1rmFTaxes)r-ref_axrrrrrrk)rr)rr)rr)?rMbP?ractive)which)rrrDr{rErUr?rrr<rAxesXr]appendfrom_anyrrOrrr=AxesY set_visiblerqrr\rYrPr[rV set_positionrnrp)rcb_mode cb_locationhvh_ax_posh_cb_posszrrhrirjv_ax_posv_cb_posr%countrrrurr^s                                     zImageGrid._init_locatorsr)r r!r"rrr^r#rrrrrs" ir)numbersrrLtypesrnumpyrB matplotlibrrmatplotlib.gridspecr axes_dividerrr r mpl_axesr r r _make_class_factoryrr$rAxesGridrrrrs"