gVddlmZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z ddlmZdZGd d Zejed ZGd d ZGddeZeZdS))NumberN)_apicbook) SubplotSpec)SizeSubplotDividerDivider)Axesc| }| }|jd|||jd||dS)Nbottom ticklabelslabelleft)axistoggle)ax bottom_onleft_on bottom_offleft_offs W/granary/edev/quantum/lib/python3.11/site-packages/mpl_toolkits/axes_grid1/axes_grid.py _tick_onlyr sUJ{HGH *EEEGFOhh?????c:eZdZfdZdddZdZfdZxZS) CbarAxesBasecH||_tj|i|dSN) orientationsuper__init__)selfr argskwargs __class__s rr"zCbarAxesBase.__init__s-&$)&)))))rN)ticksc N|jdvrdnd}|jj|f|||d|}|S)Ntopr horizontalvertical)caxr r')r figurecolorbar)r#mappabler'r%r cbs rr/zCbarAxesBase.colorbarsU ,0AAALL  "T[ !(9+(-991799 rcX|j|j}|||dS)Nr)rr r)r#brs r toggle_labelzCbarAxesBase.toggle_label!s-y)* q *****rcd|j}t||_dSr)r r!cla)r#r r&s rr6zCbarAxesBase.cla%s*&   &r)__name__ __module__ __qualname__r"r/r4r6 __classcell__r&s@rrrs~*****+/+++'''''''''rrzCbar{}ceZdZdZeZ ddddddddd Zd Zd Zd Z d Z dZ dZ dZ dZdZdZdZdZdZdS)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. Nrow{Gz?FTL) share_allshare_xshare_y label_mode axes_classaspectc|\|_|_||j|jz}n+d|cxkr|j|jzksntd||_t t jtj|d\|_ |_ tj ddg|||_ | |j} n3t| t t"fr| \} }t%j| fi|} t)gg| }t|t*t,frt/||fi||_n_t3|d krt/|g|Ri||_n6t3|d krt5||fi||_nt7d |j}tj|j|jfdt< }t?|jD]W}| |\}}|r |d x}}n|r |d|fnd}| r ||dfnd}| |||||||f<X|!|j dkrdnd"|_#|j$"|_%|"|_&|j%dd|_'|(|j#D]}|)||*| 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: None 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)r+r,rFzIncorrect 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)r#figrect nrows_ncolsrWrJaxes_padrArBrCrDrErFclsr%kw axes_arrayicolr>rNrOrs rr"z Grid.__init__;sYp$/  T[ >[4;.FFv::::t{!::::: MOOO 47 J!4456561d1 He, BBBB#  /JJ  T5M 2 2 :$KC"*399&99J R"V < < < dV[1 2 2 5*3;;;;DMM YY!^^*3<<<<<locators rrvzGrid._init_locatorss t{##  A /-... OOCFF # # #QB HHRLLLL t{##  A .,--- OOCFF # # #QB HHRLLLLt{## 7 7A((++HCm//C=XdkAo.C%D0FFG M!  - -g 6 6 6 6 $$Q''' ""1%%%%%rc|jdkrt||j\}}nt||j\}}||fS)NrI)r_divmodrTrU)r#nrr>s rrnzGrid._get_col_rowsD ?h & &a--HCa--HCCxrc*t|jSr)rhrqr#s r__len__z Grid.__len__s4=!!!rc|j|Sr)rq)r#rs r __getitem__zGrid.__getitem__s}Qrc|j|jfS)zV Return the number of rows and columns of the grid as (nrows, ncols). )rTrUrs r get_geometryzGrid.get_geometrys{DK''rcN|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. rrNr\ fixed_sizer])r#r|s r set_axes_padzGrid.set_axes_pads(+31+')1!&&&rc2|jj|jjfS)z Return the axes padding. Returns ------- hpad, vpad Padding (horizontal pad, vertical pad) in inches. rrs r get_axes_padzGrid.get_axes_pads $/#.0 0rc:|j|dS)z%Set the aspect of the SubplotDivider.N)rg set_aspect)r#rFs rrzGrid.set_aspects   (((((rc4|jS)z(Return the aspect of the SubplotDivider.)rg get_aspectrs rrzGrid.get_aspects}'')))rcH|dkr|jD]}t|dddS|dkr|jdddD]}t|dd|jdd}t|dd|jd dD];}|ddD]}t|dd|d}t|dd >2u=====!!$R(B rUE : : : :'+ > >crc(AABrT4@@@@@W2===== > >S[[m = =2t<<<<<B rUE : : : : : :v~~$ 1FGGGGGG~rc|jSr)rgrs r get_dividerzGrid.get_divider3s }rc:|j|dSr)rg set_locator)r#rs rrzGrid.set_axes_locator6s !!'*****rc4|jSr)rg get_locatorrs rget_axes_locatorzGrid.get_axes_locator9s}((***rNr>r?)r7r8r9__doc__r r`r"rvrnrrrrrrrrxrrrrrr=r=.s1   p(! p(p(p(p(p(d&&&:"""   ((( 5 5 5 0 0 0)))***2G2G2Gh++++++++rr=c BeZdZ d dddddddddd fd Zd ZxZS) ImageGridNr>r?FTr@rightz5%) rArFrD cbar_mode cbar_locationcbar_pad cbar_size cbar_set_caxrEc tjgd| tjgd| | |_| |_| |_| |_t |||||||dd|| | |jD]}| ||r|jdkr|j D]}|jd|_ d S|jd krbt|j D]K\}}| |\}}|jd vr|j||_ 9|j||_ Ld St|j |jD]\}}||_ d 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. 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 )eachsingleedgeN)r)rrr r*)rT)rJr|rArBrCrFrDrErrrrrN)rr^_colorbar_mode_colorbar_location _colorbar_pad_colorbar_sizer!r" cbar_axesrwrqr- enumeraternzip)r#ryrzr{rWrJr|rArFrDrrrrrrErindexrr>r-r&s rr"zImageGrid.__init__@s@ ;;;%. 0 0 0 0 ===)6 8 8 8 8'"/%'  {F(tF!j  : : : .  B LL      !"h..-//B!^A.BFF//$..!*4=!9!955IE2#0077HC.2CCC!%!4!%!4 55 #4=$.AA!!GB BFF ! !!!rcj,jdvrjj_njj_fdt jD_j}j}g}g}g}g}|dkr|dvr|dkrj tj j z}| tjj|| tjj|jddd}n|d krjtjj z}| tjj|| tjj|jddd }t jD]"} j| d #jd|jdd t-jdD]\} } |r| j| r#tj | d jd}n-tj jdd jd}|dkr|dks |dkr| dkr|| t3|| tjj|| tjj|| t3|| ||dkr|dks|dkr| jdz kr|| tjj|| t3|| tjj|g} g} t-jddddD]\}} |r| j| r#tj| d jd}n-tjjdd jd}|d kr|dks |dkr|dkr|| t3|| tjj|| tjj|| t3|| ||dkr|dks|dkr|j dz kr|| tjj|| t3|| tjj|t jD]} | \} }j|| | j dz |z }j| ||dkr|dvr4j|| | j dz |z }n7|dvr3j|| | j dz |z }j| ||dkr|dkr| dks|dkrc| jdz krUj|d| j dz |z }j|||d kr|j dz ks |dkrN|dkrHj|| | d}j| ||dkr|dkrj tj j z}| tjj|| tjj|jddd}n|dkrjtjj z}| tjj|| tjj|jddd }|dvrwt jD]"} j| d #jd|jdd n |dkr8t jD]"} j| d #n|dkr|dvrj }nj}t |D]"} j| d #t | dzjD]"}j|d #n[t jD]F} j| d j| gddGj|j|dS)Nrcg|]N}tjjdjjjOS)r)r )_cbaraxes_class_factoryr`rqr.rgrjr).0rr#s r z,ImageGrid._init_locators..sk))) < #D$: ; ; a ')C)C)E)E 3 5 5 5)))rr)rr rrrS)rrny1r )rnx1rFTaxes)rFref_axrrrrr*r)rrr))rr*)?rMbP?ractive)which)rrr\rr]rmrWrrrTrAxesXrurfrom_anyrrgrrUAxesY set_visiblerrrtrqrhrsrn set_positionrr)r#cb_mode cb_locationrrrh_cb_posrrrrrrv_cb_posr>countjs` rrvzImageGrid._init_locatorss    %&*;;;%)%9%D""%)%8%C"))))4;'' ))) %-   h  ;2D#D#Df$$[4:dm#<#<<t':B??@@@t'92>>???-33qQB3GG(([4:dm#<#<<t':B??@@@t'92>>???-33qbQ3GG4;'' 5 5q!--e4444 N1  . .w 7 7 7 N1  ) )$ / / / q!122 A AGC /-... HZ6$-:JKKKZ a 0'-dmA6FHHHv%% F**$..3!88A'''t':B??@@@t'92>>??? OOCFF # # # HHRLLLw&& F**$..3$+/3I3It'92>>???A'''t':B??@@@ !1!!4TTrT!:;; A AGC .,--- HZ6$-:JKKKZ a 0'-dmA6FHHHx'' F**$..3!88A'''t':B??@@@t'92>>??? OOCFF # # # HHRLLLu$$ F**$..3$+/3I3It'92>>???A'''t':B??@@@t{## B BA((++HCm//8C=3;DKM#>???t':B??@@@-33raR3HH%%[4:dm#<#<<t'92>>???t':B??@@@-33qbR3HH...t{++99AN1%11%8888q!227;;;q!--d333   4;'' 4 4q!--d3333 4   ///  5\\ 4 4q!--d33331q5$+.. 5 5q!--e4444 54;'' ? ?q!--e444q!../E/E/E5=/???? $$Q''' ""1%%%%%rr)r7r8r9r"rvr:r;s@rrr=s   `!!&" `!`!`!`!`!`!`!D^&^&^&^&^&^&^&rr)numbersrrdnumpyrZ matplotlibrrmatplotlib.gridspecr axes_dividerrr r mpl_axesr rr_make_class_factoryrr=rAxesGridrrrrsI""""""""++++++7777777777@@@''''''''.4%3L(KKL+L+L+L+L+L+L+L+^C&C&C&C&C&C&C&C&L r