o i0@sddlZddlmZddlmZmZddlmZe dddd d Z ej dd d ddZ GdddZ ej ddd Gddde ZdS)N)cbook)make_axes_locatableSize)Axes3.3add_all{Gz?Tc Kst|}|t|}dd|dt|}dd|dt|}|t|||g||||||g||jddddg} |dur^z|j}Wn t y]t |}YnwdD]Q} || |j d d f||d |} |jd| d } | | | j | jD]} | d qz| jD]}|jd qWn t yYnw| | q`|r| }| D]} || q| S)zX Parameters ---------- pad : float Fraction of the axes height. rr)ny1N)r rT)original)sharexsharey)nxnyF)rrAxesYAxesXset_horizontal set_verticalset_axes_locator new_locator _axes_classAttributeErrortype get_figure get_positionyaxisget_ticklabelsxaxis set_visibleaxisvaluesmajor_ticklabelsappendadd_axes)axpad axes_classrkwargsdividerpad_sizexsizeysizeax_rgbrax1locatortr#figr5[/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/axes_rgb.py make_rgb_axessJ         r7zax.imshow(np.dstack([r, g, b]))) alternativecKs|jt|||gfi|S)N)imshownpdstack)r(rgbr+r5r5r6 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)r)rcOs|d|j}||i||_}|r||n||d<t|f||d|\|_|_|_|j|j|j|jfD]}|j ddj d|j ddj dq:dS)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 r*r)r)r*Nw)pop_defaultAxesClassRGBrr'r7RGBr#line set_color major_ticksset_markeredgecolor)selfr)rargsr+r*r(r1r5r5r6__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)rDrr'rErFrG)rLr5r5r6add_RGB_to_figure{szRGBAxes.add_RGB_to_figurec Ks |j|jkr |jksntd|jd|jd|jdt|||g}t|}||dddddf<t|}||dddddf<t|}||dddddf<|jj|fi|} |jj|fi|} |jj|fi|} |j j|fi|} | | | | fS)a Create the four images {rgb, r, g, b}. Parameters ---------- r : array-like The red array g : array-like The green array b : array-like The blue array 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; zeros_likerDr9rErFrG) rLr<r=r>r+rDrErFrGim_rgbim_rim_gim_br5r5r6r?s     zRGBAxes.imshow_rgbN) __name__ __module__ __qualname____doc__rrCr_delete_parameterrN deprecatedrOr?r5r5r5r6r@@s   r@c@s eZdZdS) RGBAxesBaseN)rWrXrYr5r5r5r6r]sr])r NT)numpyr: matplotlibr axes_dividerrrmpl_axesrr[r7r\r?r@r]r5r5r5r6s     2  i