U \ a2X@sddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z ddlmZdd ZGd d d ZGd d d eeZGdddZGdddeZeZdS))NumberN)_api) SubplotSpec)SizeSubplotDividerDivider)AxescCs8| }| }|jdj||d|jdj||ddS)NbottomZ ticklabelslabelleft)axistoggle)ax bottom_onleft_onZ bottom_offZleft_offrE/tmp/pip-unpacked-wheel-wjyw_3jo/mpl_toolkits/axes_grid1/axes_grid.py _tick_onlysrcsbeZdZfddZddddZejddd Zejdd d Zd d Z d dZ fddZ Z S) CbarAxesBasecs$||_d|_d|_tj||dS)NT) orientation_default_label_on_locatorsuper__init__)selfrargskwargs __class__rrrszCbarAxesBase.__init__N)tickscKsL|jdkrd}nd}tjj||f||d|}|j|_|j|_||S)Ntopr horizontalvertical)rr!) rmplcolorbarZColorbarZ colorbar_cidZ_cbidlocatorr _config_axes)rZmappabler!rrcbrrrr's  zCbarAxesBase.colorbar3.3zmappable.colorbar_cid) alternativez.colorbar().locatorcCsB|}|d|jddjdd|j}|j|jj|ddS)zMake an axes patch and outline.FN)all)Z set_navigaterrrr)rrbrrrr)0s  zCbarAxesBase._config_axescCs$||_|j|j}|j||ddS)Nr )rrrr)rr.rrrr toggle_label8s zCbarAxesBase.toggle_labelcst|dSN)rclar)rrrrr1=s zCbarAxesBase.cla) __name__ __module__ __qualname__rr'rZdeprecate_privatize_attributeZcbidr(r)r/r1 __classcell__rrrrrs rc@s eZdZdS)CbarAxesN)r3r4r5rrrrr7Bsr7c @seZdZdZeZeddd)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 Cs2|\|_|_|dkr"|j|j}n&d|kr>|j|jksHntd||_ttjt|d\|_ |_ t j ddg|d||_ | dkr|j} n$t| ttfr| \}}tj|f|} tgg| d}t|tttfrt||f||_nJt|d krt|f|||_n(t|d kr(t||f||_ntd|j}tj|j|jfdtd }t|jD]l}| |\}}|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. Nrcolumnr:) direction)r$r%r=)Zdtype)rr)sharexsharey)+_nrows_ncols ExceptionngridsmaprZFixednpZ broadcast_to_horiz_pad_size_vert_pad_sizerZ check_in_list _direction_defaultAxesClass isinstancelisttuple functoolspartialdictstrrrr_dividerlenr get_positionfullobjectrange _get_col_rowZraveltolistaxes_allT axes_columnaxes_rowaxes_llc_init_locatorsadd_axesset_label_mode)rfigrect nrows_ncolsrJrAaxes_padr9 share_allshare_xshare_y label_mode axes_classr=clsrkwZ axes_arrayicolr:rDrErrrrrSs`9      z Grid.__init__c Csg}g}t|jD]6}|r&||j|t|td}||qg}g}t|jD]6}|rp||j|t|td}||q\t|j D]D}| |\}} |j j ||||jd| d} |j || q|j ||j |dS)Nrnxny)r]rHappendrMrYrZScaledrGrNrJr^rX new_locatorr`set_axes_locatorset_horizontal set_vertical) rhh_ax_pos_szvv_ax_posrsrtr:r(rrrres2       zGrid._init_locatorscCs4|jdkrt||j\}}nt||j\}}||fS)Nr@)rOdivmodrGrH)rnrtr:rrrr^s zGrid._get_col_rowcCs t|jSr0)rYr`r2rrr__len__sz Grid.__len__cCs |j|Sr0)r`)rrsrrr __getitem__szGrid.__getitem__cCs |j|jfS)zV Return the number of rows and columns of the grid as (nrows, ncols). )rGrHr2rrr get_geometryszGrid.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. rrNrM fixed_sizerN)rrkrrr set_axes_pads zGrid.set_axes_padcCs|jj|jjfS)z Return the axes padding. Returns ------- hpad, vpad Padding (horizontal pad, vertical pad) in inches. rr2rrr get_axes_pads zGrid.get_axes_padcCs|j|dS)z%Set the aspect of the SubplotDivider.N)rX set_aspect)rr=rrrrszGrid.set_aspectcCs |jS)z(Return the aspect of the SubplotDivider.)rX get_aspectr2rrrrszGrid.get_aspectcCs|dkr"|jD]}t|ddqn|dkr|jdddD]}t|dddq<|jdd}t|ddd|jd dD]:}|ddD]}t|dddq|d}t|dddqzn6|d kr|jD]}t|dddq|j}t|ddddS) 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. r-Fr<rNrFT)rrr1)r`rrbrd)rmoderrtrrrrgs$  zGrid.set_label_modecCs|jSr0)rXr2rrr get_divider@szGrid.get_dividercCs|j|dSr0)rXZ set_locator)rr(rrrrzCszGrid.set_axes_locatorcCs |jSr0)rXZ get_locatorr2rrrget_axes_locatorFszGrid.get_axes_locatorcCs |jSr0)rXget_vsize_hsizer2rrrrIszGrid.get_vsize_hsize) Nr:r;TFTTr<N)r3r4r5__doc__r rPrdelete_parameterrrer^rrrrrrrrgrrzrrrrrrr8Fs:   q   )r8cs6eZdZeZedddfd d Zd dZZ S) ImageGridr+r9Nr:r;TFr<right5%cs"| |_| |_| |_||_|rBtj|||||||dd| | |d n&tj||||||||dd| | |d |r|jD]}||qr|r|jdkr|jD]}|jd|_ qnr|jdkrt |jD]:\}}| |\}}|jdkr|j||_ q|j||_ qn t |j|jD]\}}||_ q dS) 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)rArkrlrmrnr=rorp) rArkr9rlrmrnr=rorpsingleredger rN) _colorbar_mode_colorbar_location _colorbar_pad_colorbar_sizerr cbar_axesrfr`cax enumerater^zip)rrhrirjrJrArkr9rlr=roZ cbar_modeZ cbar_locationZcbar_padZ cbar_sizeZ cbar_set_caxrprindexrtr:rrrrrRsbD      zImageGrid.__init__cs~jdkr*jdkr jj_n jj_fddtjD_j}j}g}g}g}g}|dkrb|dkrb|dkrʈj t j }| t j|| t j|jjdddd }nV|d kr jt j }| t j|| t j|jjdddd }tjD]} j| d q*jd|jdd tjdD]0\} } |r| j| rt j | djdd}nt j jddjdd}|dkr$|dks|dkr$| dkr$| t|| t j|| t j|| t|| ||dkrp|dksj|dkrp| jdkrp| t j|| t|| t j|qpg} g} tjddddD]0\}} |r| j| rt j| djdd}nt jjddjdd}|d krx|dksB|dkrx|dkrx| t|| t j|| t j|| t|| ||dkr|dks|dkr|j dkr| t j|| t|| t j|qtjD]~} | \} }jj|| | j d|d}j| ||dkr|dkrjj|| | j d|d}n,|dkrjj|| | j d|d}j| |n|dkr|dkr| dks|dkr*| jdkr*jj|d| j d|d}j||nV|d krD|j dksX|dkr|dkrjj|| | dd}j| |q|dkr|dkrj t j }| t j|| t j|jjdddd }nV|dkr<jt j }| t j|| t j|jjdddd }|dkrbtjD]} j| d qPjd|jdd n|dkrtjD]} j| d qn|dkr$|dkrԈj }nj}t|D]} j| d qt| djD]}j|d q n>tjD]2} j| d j| jddddgddq.j|j|dS)Nrcs,g|]$}jjdjjjdqS)r)r)_defaultCbarAxesClassr`figurerXrZr).0rr2rr s  z,ImageGrid._init_locators..r)r r r rrF)rvrwZny1r )rvZnx1rwFTZaxes)r=Zref_axZeachrrrr#ru)rr r")rr#g?gMbP?active)which)rrrMrrNr]rJrrrGrZAxesXrdrxZfrom_anyrrXryrHZAxesYZ set_visiblerzrrcr`rYrbr^Z set_positionr{r|)rZcb_modeZ cb_locationr}rr~Zh_cb_posrr(rsrtrrZv_cb_posr:countjrr2rresH            "                       zImageGrid._init_locators) Nr:r;TFTr<NrNrTN) r3r4r5r7rrrrrer6rrrrrMs" ir)ZnumbersrrTZnumpyrLZ matplotlibr&rZmatplotlib.gridspecrZ axes_dividerrrrZmpl_axesr rrr7r8rZAxesGridrrrrs"    -