o i2\@sddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZddlmZdd ZGd d d ZGd d d eeZGdddZGdddeZeZdS))NumberN)cbook) SubplotSpec)SizeSubplotDividerDivider)AxescCs8| }| }|jdj||d|jdj||ddS)Nbottom ticklabelslabelleft)axistoggle)ax bottom_onleft_on bottom_offleft_offr\/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/axes_grid.py _tick_onlysrcspeZdZfddZeddddddd Zejd d d Zejd d d Z ddZ ddZ fddZ Z S) CbarAxesBasecs(||_d|_d|_tj|i|dS)NT) orientation_default_label_on_locatorsuper__init__)selfrargskwargs __class__rrrszCbarAxesBase.__init__3.2locatorticksN)r&cKs|jdvrd}nd}tjdrDtjddd|durtd}d d lm}|||f||d |}|j d |j |_ ||_|j |_ntjj||f||d |}|j|_ |j|_||S) Ntopr horizontalverticalzmpl_toolkits.legacy_colorbarr$zSince %(since)s, mpl_toolkits's own colorbar implementation is deprecated; it will be removed %(removal)s. Set the 'mpl_toolkits.legacy_colorbar' rcParam to False to use Matplotlib's default colorbar implementation and suppress this deprecation warning.)messager)Colorbar)rr&changed)rmplrcParamsrwarn_deprecatedticker MaxNLocatorcolorbarr- callbacksSMconnect update_normal_cbid cbar_axisget_major_locatorrcolorbar_factory colorbar_cidr% _config_axes)rmappabler&r!rr-cbrrrr4sB    zCbarAxesBase.colorbar3.3zmappable.colorbar_cid) alternativez.colorbar().locatorcCsB|}|d|jddjdd|j}|j|jj|ddS)zMake an axes patch and outline.FN)all) set_navigaterrrr)rrbrrrr=Cs  zCbarAxesBase._config_axescCs$||_|j|j}|j||ddS)Nr )rrrr)rrDrrrr toggle_labelKs zCbarAxesBase.toggle_labelcst|dSN)rclar=rr"rrrGPs  zCbarAxesBase.cla)__name__ __module__ __qualname__rr_rename_parameterr4_deprecate_privatize_attributecbidr%r=rErG __classcell__rrr"rrs   rc@s eZdZdS)CbarAxesN)rIrJrKrrrrrPUsrPc@seZdZdZeZedd        d)dd d d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(ZdS)*Gridaq 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. r@add_allNrow{Gz?TFL)aspectc Cs>|\|_|_|dur|j|j}nd|kr"|j|jks'tdtd||_ttjt|d\|_ |_ t j ddg|d||_ | durL|j} nt| ttfr`| \}}tj|fi|} tgg| d}t|tttfrzt||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%dd|_'|(|r|j#D]}|)|q|*| dS)ai 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"). axes_pad : float or (float, float), default: 0.02 Padding or (horizontal padding, vertical padding) between axes, in inches. add_all : bool, default: True Whether to add the axes to the figure using `.Figure.add_axes`. This parameter is deprecated. 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"}, 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. axes_class : subclass of `matplotlib.axes.Axes`, default: None aspect : bool, default: False Whether the axes aspect ratio follows the aspect ratio of the data limits. NrcolumnrS) direction)r)r*rV)dtype)rr)sharexsharey)+_nrows_ncols ExceptionngridsmaprFixednp broadcast_to_horiz_pad_size_vert_pad_sizer_check_in_list _direction_defaultAxesClass isinstancelisttuple functoolspartialdictstrrrr_dividerlenr get_positionfullobjectrange _get_col_rowraveltolistaxes_allT axes_columnaxes_rowaxes_llc_init_locatorsadd_axesset_label_mode)rfigrect nrows_ncolsrdrZaxes_padrR share_allshare_xshare_y label_mode axes_classrVclsr!kw axes_arrayicolrSr^r_rrrrrfs\ 9       z Grid.__init__c Csg}g}t|jD]}|r||j|t|td}||q g}g}t|jD]}|r8||j|t|td}||q.t|j D]"}| |\}} |j j ||||jd| d} |j || qO|j ||j |dS)Nrnxny)rzrbappendrirvrScaledrarjrdr{ru new_locatorr~set_axes_locatorset_horizontal set_vertical) rhh_ax_pos_szvv_ax_posrrrSr%rrrrs0       zGrid._init_locatorscCs:|jdkrt||j\}}||fSt||j\}}||fS)NrY)rldivmodrarb)rnrrSrrrr{s zGrid._get_col_rowcCs t|jSrF)rvr~rHrrr__len__ z Grid.__len__cCs |j|SrF)r~)rrrrr __getitem__rzGrid.__getitem__cCs |j|jfS)zV Return the number of rows and columns of the grid as (nrows, ncols). )rarbrHrrr 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. rrNri fixed_sizerj)rrrrr set_axes_pad s zGrid.set_axes_padcCs|jj|jjfS)z Return the axes padding. Returns ------- hpad, vpad Padding (horizontal pad, vertical pad) in inches. rrHrrr get_axes_pads zGrid.get_axes_padcCs|j|dS)z%Set the aspect of the SubplotDivider.N)ru set_aspect)rrVrrrr"szGrid.set_aspectcC |jS)z(Return the aspect of the SubplotDivider.)ru get_aspectrHrrrr&s zGrid.get_aspectcCs|dkr|jD]}t|ddqdS|dkr^|jdddD] }t|dddq|jdd}t|ddd|jd dD]}|ddD] }t|dddqF|d}t|dddq>dS|d kr{|jD] }t|dddqe|j}t|ddddSdS) a Define which axes have tick labels. Parameters ---------- mode : {"L", "1", "all"} 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. rBFrUrNr`T)rrr1)r~rrr)rmoderrrrrr*s*  zGrid.set_label_modecCs|jSrF)rurHrrr get_dividerSszGrid.get_dividercCs|j|dSrF)ru set_locator)rr%rrrrVszGrid.set_axes_locatorcCrrF)ru get_locatorrHrrrget_axes_locatorYrzGrid.get_axes_locatorcCrrF)ruget_vsize_hsizerHrrrr\rzGrid.get_vsize_hsize) NrSrTTFTTrUN)rIrJrK__doc__r rmr_delete_parameterrrr{rrrrrrrrrrrrrrrrrQYs<  q   ) rQcsPeZdZeZedd           dfd d Zd dZZ S) ImageGridr@rRNrSrTTFrUright5%cs&| |_| |_| |_||_|r!tj|||||||dd| | |d ntj||||||||dd| | |d |rA|jD]}||q9|r|jdkrV|jD]}|jd|_ qKdS|jdkrt |jD]\}}| |\}}|jdvrw|j||_ q`|j||_ q`dSt |j|jD] \}}||_ qdSdS) 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. add_all : bool, default: True Whether to add the axes to the figure using `.Figure.add_axes`. This parameter is deprecated. share_all : bool, default: False Whether all axes share their x- and 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. 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 T)rZrrrrrVrr) rZrrRrrrrVrrsingleredgerrN) _colorbar_mode_colorbar_location _colorbar_pad_colorbar_sizerr cbar_axesrr~cax enumerater{zip)rrrrrdrZrrRrrVr cbar_mode cbar_locationcbar_pad cbar_size cbar_set_caxrrindexrrSrr"rrresHD       zImageGrid.__init__cshjdurjdvrjj_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]\} } |r| j| rt j | djdd}nt j jddjdd}|dkr |dks|dkr | dkr | t|| t j|| t j|| t|| ||dkrH|dks-|dkrH| jdkrH| t j|| t|| t j|qg} g} tjddddD]\}} |rf| j| rut 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|qYtjD]} | \} }jj|| | j d|d}j| ||dkrU|dvr5jj|| | j d|d}n|dvrKjj|| | j d|d}j| |q|dkr|dkrd| dksq|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 qjd|jdd nj|dkrStjD] } j| d qFnS|dkr|dvraj }nj}t|D] } j| d qht| djD] }j|d q|ntjD]} j| d j| jgdddqj|j|dS)Nrcs,g|]}jjdjjjdqS)r)r)_defaultCbarAxesClassr~figurerurwr).0rrHrr sz,ImageGrid._init_locators..r)rr rrr`)rrny1r )rnx1rFTaxes)rVref_axeachrrrr(r)rrr')rr()?rMbP?ractive)which)rrrirrjrzrdrrrarAxesXrrfrom_anyrrurrbAxesY set_visiblerrrr~rvrr{ set_positionrr)rcb_mode cb_locationrrrh_cb_posrr%rrrrv_cb_posrScountjrrHrrs                                    zImageGrid._init_locators) NrSrTTFTrUNrNrTN) rIrJrKrPrrrrrrOrrr"rr`s$ ir)numbersrrqnumpyrg matplotlibr/rmatplotlib.tickerr2matplotlib.gridspecr axes_dividerrrrmpl_axesr rrrPrQrAxesGridrrrrs&     ?