U \ a@sddlZddlmZddlmZmZddlmZe dddd d Z ej dd d ddZ GdddZ ej ddd Gddde ZdS)N)_api)make_axes_locatableSize)Axes3.3add_all{Gz?Tc Kst|}|t|}dd|dt|}dd|dt|}|t|||g||||||g||jddddg} |dkrz |j}Wnt k rt |}YnXdD]} || |j d d f||d |} |jd| d } | | | j | jD]} | d qz"| jD]}|jd q4Wnt k r`YnX| | q|r| }| D]} || q| S)zX Parameters ---------- pad : float Fraction of the axes height. rr)Zny1N)r rT)original)ZsharexZsharey)ZnxnyF)rrZAxesYZAxesXZset_horizontalZ set_verticalZset_axes_locatorZ new_locatorZ _axes_classAttributeErrortype get_figure get_positionZyaxisZget_ticklabelsZxaxisZ set_visibleaxisvaluesZmajor_ticklabelsappendadd_axes)axpad axes_classrkwargsdividerZpad_sizeZxsizeZysizeZax_rgbrax1locatortrZfigr D/tmp/pip-unpacked-wheel-wjyw_3jo/mpl_toolkits/axes_grid1/axes_rgb.py make_rgb_axessF    r"zax.imshow(np.dstack([r, g, b]))) alternativecKs|jt|||gf|S)N)imshownpdstack)rrgbrr r r! imshow_rgb;sr*c@sJeZdZdZeZedddddddZe dd d Z d d Z d S)RGBAxesa 4-panel imshow (RGB, R, G, B). Layout: +---------------+-----+ | | R | + +-----+ | RGB | G | + +-----+ | | B | +---------------+-----+ Subclasses can override the ``_defaultAxesClass`` attribute. Attributes ---------- RGB : ``_defaultAxesClass`` The axes object for the three-channel imshow. R : ``_defaultAxesClass`` The axes object for the red channel imshow. G : ``_defaultAxesClass`` The axes object for the green channel imshow. B : ``_defaultAxesClass`` The axes object for the blue channel imshow. rrrT)rrcOs|d|j}||||_}|r2||n||d<t|f||d|\|_|_|_|j|j|j|jfD]0}|j ddj d|j ddj dqpdS)a Parameters ---------- pad : float, default: 0 fraction of the axes height to put as padding. add_all : bool, default: True Whether to add the {rgb, r, g, b} axes to the figure. This parameter is deprecated. axes_class : matplotlib.axes.Axes *args Unpacked into axes_class() init for RGB **kwargs Unpacked into axes_class() init for RGB, R, G, B axes rr)rrNw)pop_defaultAxesClassRGBrrr"RGBrlineZ set_colorZ major_ticksZset_markeredgecolor)selfrrargsrrrrr r r!__init__^s zRGBAxes.__init__cCs:|j|j|j|j|j|jdS)z@Add red, green and blue axes to the RGB composite's axes figure.N)r/rrr0r1r2)r4r r r!add_RGB_to_figure|szRGBAxes.add_RGB_to_figurec Ks|j|jkr|jks>ntd|jd|jd|jdt|||g}t|}||dddddf<t|}||dddddf<t|}||dddddf<|jj|f|} |jj|f|} |jj|f|} |j j|f|} | | | | fS)a Create the four images {rgb, r, g, b}. Parameters ---------- r, g, b : array-like The red, green, and blue arrays. kwargs : imshow kwargs kwargs get unpacked into the 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 ValueErrorr%r&Z zeros_liker/r$r0r1r2) r4r'r(r)rr/r0r1r2Zim_rgbZim_rZim_gZim_br r r!r*s    zRGBAxes.imshow_rgbN) __name__ __module__ __qualname____doc__rr.rdelete_parameterr6 deprecatedr7r*r r r r!r+@s  r+c@s eZdZdS) RGBAxesBaseN)r:r;r<r r r r!r@sr@)r NT)Znumpyr%Z matplotlibrZ axes_dividerrrZmpl_axesrr>r"r?r*r+r@r r r r!s    2  f