gnLddlZddlmZmZddlmZd dZGddZdS) N)make_axes_locatableSize)Axes{Gz?c Nt|}|tj|z}dd|zz dz tj|z}dd|zz dz tj|z}|tj|||g||||||g||dddg}|t|}dD] } || | d f||d |} |d| } | | | j | j zD]} | d  | jD]} | jd n#t$$rYnwxYw||  | }|D]} || |S)a Parameters ---------- ax : `~matplotlib.axes.Axes` Axes instance to create the RGB Axes in. pad : float, optional Fraction of the Axes height to pad. axes_class : `matplotlib.axes.Axes` or None, optional Axes class to use for the R, G, and B Axes. If None, use the same class as *ax*. **kwargs : Forwarded to *axes_class* init for the R, G, and B Axes. rr)ny1N)r rT)original)sharexsharey)nxnyF)rrAxesYAxesXset_horizontal set_verticalset_axes_locator new_locatortype get_figure get_positionyaxisget_ticklabelsxaxis set_visibleaxisvaluesmajor_ticklabelsAttributeErrorappendadd_axes)axpad axes_classkwargsdividerpad_sizexsizeysizeax_rgbrax1locatortr figs V/granary/edev/quantum/lib/python3.11/site-packages/mpl_toolkits/axes_grid1/axes_rgb.py make_rgb_axesr4sN""%%GTZ^^#H#gq[DJrNN *E#gq[DJrNN *E DJrNNHe<=== %5(EBCCC++Aqb+99::: F"XX   j"//4/*H*H9 "2991799%%r%22 W%%%))++ci.F.F.H.HH ! !A MM%  )) 9 9%11%8888 9    D   c --//C S Ms6G GGc(eZdZdZeZdddZdZdS)RGBAxesu 4-panel `~.Axes.imshow` (RGB, R, G, B). Layout:: ┌───────────────┬─────┐ │ │ R │ │ ├─────┤ │ RGB │ G │ │ ├─────┤ │ │ B │ └───────────────┴─────┘ Subclasses can override the ``_defaultAxesClass`` attribute. By default RGBAxes uses `.mpl_axes.Axes`. Attributes ---------- RGB : ``_defaultAxesClass`` The Axes object for the three-channel `~.Axes.imshow`. R : ``_defaultAxesClass`` The Axes object for the red channel `~.Axes.imshow`. G : ``_defaultAxesClass`` The Axes object for the green channel `~.Axes.imshow`. B : ``_defaultAxesClass`` The Axes object for the blue channel `~.Axes.imshow`. r)r'c|d|j}||i|x|_}||t |f||d|\|_|_|_|j|j|j|jfD]P}|j ddj d|j ddj dQdS)a Parameters ---------- pad : float, default: 0 Fraction of the Axes height to put as padding. axes_class : `~matplotlib.axes.Axes` Axes class to use. If not provided, ``_defaultAxesClass`` is used. *args Forwarded to *axes_class* init for the RGB Axes **kwargs Forwarded to *axes_class* init for the RGB, R, G, and B Axes r()r'r(Nw)pop_defaultAxesClassRGBrr%r4RGBr line set_color major_ticksset_markeredgecolor)selfr'argsr)r(r&r/s r3__init__zRGBAxes.__init__[sZZ d.DEE " D3F3332   $$$!. ":J":":28":":Hdfdfdf5 = =C HQQQK  & &s + + + HQQQK # 7 7 < < < < = =c |j|jcxkr |jks*ntd|jd|jd|jdtj|||g}tj|}||dddddf<tj|}||dddddf<tj|}||dddddf<|jj|fi|} |jj|fi|} |jj|fi|} |j j|fi|} | | | | fS)a Create the four images {rgb, r, g, b}. Parameters ---------- r, g, b : array-like The red, green, and blue arrays. **kwargs : Forwarded to `~.Axes.imshow` calls for the four images. Returns ------- rgb : `~matplotlib.image.AxesImage` r : `~matplotlib.image.AxesImage` g : `~matplotlib.image.AxesImage` b : `~matplotlib.image.AxesImage` zInput shapes (z, z) do not matchNrrr ) shape ValueErrornpdstack zeros_liker;imshowr<r=r>) rCrgbr)r;r<r=r>im_rgbim_rim_gim_bs r3 imshow_rgbzRGBAxes.imshow_rgbrsd$17----ag----NNNAGNNqwNNNPP PiAq "" M#  !!!QQQ' M#  !!!QQQ' M#  !!!QQQ'  ////tv}Q))&))tv}Q))&))tv}Q))&))tT4''rFN)__name__ __module__ __qualname____doc__rr:rErUrFr3r6r6<sQ8"#=====. ( ( ( ( (rFr6)rN) numpyrJ axes_dividerrrmpl_axesrr4r6rZrFr3r^s333333332222jV(V(V(V(V(V(V(V(V(V(rF