G8cT ddlmZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z ddlmZdZGd d Zej$d Gd d eeZej(edZGddZGddeZeZy))NumberN)_apicbook) SubplotSpec)SizeSubplotDividerDivider)Axesc| }| }|jdj|||jdj||y)Nbottom ticklabelslabelleft)axistoggle)ax bottom_onleft_on bottom_offleft_offs C/usr/lib/python3/dist-packages/mpl_toolkits/axes_grid1/axes_grid.py _tick_onlyr sGJ{HGGH *EGGFOhh?c:eZdZfdZdddZdZfdZxZS) CbarAxesBasec@||_d|_t| |i|yN) orientation_locatorsuper__init__)selfr argskwargs __class__s rr#zCbarAxesBase.__init__s$&  $)&)rN)ticksc l|jdvrdnd}|jj|f|||d|}|S)Ntopr horizontalvertical)caxr r()r figurecolorbar)r$mappabler(r&r cbs rr0zCbarAxesBase.colorbarsN ,,0AAL  "T[[ ! !(9+(-9179 rc\|j|j}|j||y)Nr)rr r)r$brs r toggle_labelzCbarAxesBase.toggle_label"s&yy))* q *rcH|j}t| ||_yr)r r"cla)r$r r's rr7zCbarAxesBase.cla&s&&   &r)__name__ __module__ __qualname__r#r0r5r7 __classcell__r's@rrrs!* +/+''rr3.5c eZdZy)CbarAxesN)r8r9r:rrr?r?,srr?zCbar{}ceZdZdZeZ dddddddddZdZd Zd Z d Z d Z d Z dZ dZdZdZdZdZdZej*ddZy)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. 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,t.frt1||fi||_nOt5|d k(rt1|g|i||_n,t5|d k(rt7||fi||_n t9d |j2j;}tj<|j|jfdt> }tA|jD]G}|jC|\}}|r|d x}}n|r|d|fnd}| r||dfnd}| |||||||f<I|jE|jdk(rdndjG|_$|jJjG|_&|jG|_'|jLdd|_(|jS|jHD]}|jU||jW| y)a3 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.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"}, 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. Nrz7ngrids must be positive and not larger than nrows*ncolscolumnrow) direction)r,r-rIzIncorrect rect format)dtype)rr)sharexshareyCF)order),_nrows_ncols ValueErrorngridsmaprFixednp broadcast_to_horiz_pad_size_vert_pad_sizer check_in_list _direction_defaultAxesClass isinstancelisttuple functoolspartialdictstrrrr _dividerlenr TypeError get_positionfullobjectrange _get_col_rowraveltolistaxes_allT axes_columnaxes_rowaxes_llc_init_locatorsadd_axesset_label_mode)r$figrect nrows_ncolsr[rNaxes_padrDrErFrGrHrIclsr&kw axes_arrayicolrMrRrSrs rr#z Grid.__init__Asj$/  T[ >[[4;;.Fv:t{{!:: MOO; MOO 47 JJ!4561d1 He, B#  //J  T5M 2$KC"**39&9J R"V < dS&+6 7*3;;DM Y!^*3<<2u= >!!!$R(B rUE :''+ >cr(ABrT4@AW2= >S[mm =2t< =B rUE : rc|jSr)rlrs r get_dividerzGrid.get_divider+s }}rc:|jj|yr)rl set_locator)r$rs rrzGrid.set_axes_locator.s !!'*rc6|jjSr)rl get_locatorrs rget_axes_locatorzGrid.get_axes_locator1s}}((**rr=c6|jjSr)rlget_vsize_hsizers rrzGrid.get_vsize_hsize4s}},,..rNrMg{Gz?)r8r9r:__doc__r rdr#r{rsrrrrrrrr}rrrr deprecatedrr@rrrBrB4s   m(! m(^&:" ( 5 0)*';R++T__U//rrBc BeZdZ d dddddddddd fd Zd ZxZS) ImageGridNFTrCrightz5%) rDrIrG cbar_mode cbar_locationcbar_pad cbar_size cbar_set_caxrHc 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. 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)rNrrDrErFrIrGrHrrrrrN)rrb_colorbar_mode_colorbar_location _colorbar_pad_colorbar_sizer"r# cbar_axesr|rvr. enumeraterszip)r$r~rrr[rNrrDrIrGrrrrrrHrindexrrMr.r's rr#zImageGrid.__init__<sr@ ;%. 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]Y}t|j|jdj|jj|j[c}|_ |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-ddd } t |j D] } |j| j3d "|jdj5 |jdj3d t7|j8dD]\} } |r|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(sH| |j.dz k(s\|j't!j(|j||j't;||j't!j(|j*|g} g}t7|j<ddddD]\}} |r|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(sH||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-ddd } |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)Nrr)r r)rr rrW)rrny1r )rnx1rFTaxes)rIref_axrrrrr+r)rrr*)rr+)?rMbP?ractive)which)#rrr`rrarrr[_cbaraxes_class_factoryrdrvr/rlrorrrXrAxesXrzrfrom_anyrrrYAxesY set_visiblerrryrmrxrs set_positionrr)r$rcb_mode cb_locationrrrh_cb_posrrrrrrv_cb_posrMcountjs rr{zImageGrid._init_locatorss     %&&*;;%)%9%9%D%D"%)%8%8%C%C" 4;;' ) < #D$:$: ; a '')C)C)E 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 AGC--.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!:; AGC,,-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)Aor)r8r9r:r#r{r;r<s@rrr9s>   `!!&" `!D^&rr)numbersrrhnumpyr^ matplotlibrrmatplotlib.gridspecr axes_dividerrr r mpl_axesr rrrr?_make_class_factoryrrBrAxesGridr@rrrs"+77@''0 |T  4%33L(KB/B/JC&C&L r