o -ñ hkã@sPddlmZddlZddlmZmZddlmZm Z d dd„Z Gd d „d ƒZ dS) é)Ú MethodTypeNé)Úmake_axes_locatableÚSize)ÚAxesÚSimpleAxisArtistç{®Gáz„?c Ksft|ƒ}|t |¡}dd|dt |¡}dd|dt |¡}| t |¡||g¡| |||||g¡| |jdddd¡g}|durOt|ƒ}dD]Q} ||  ¡|j d d f||d œ|¤Ž} |jd| d } |  | ¡| j   ¡| j   ¡D]} |  d ¡q{z| j ¡D]} | j d ¡q‰Wn tyœYnw| | ¡qQ|  ¡}|D]} | | ¡q©|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. rééréÿÿÿÿ)Úny1N)ér rT)Úoriginal)ÚsharexÚsharey)ÚnxÚnyF)rrÚAxesYÚAxesXÚset_horizontalÚ set_verticalÚset_axes_locatorÚ new_locatorÚtypeÚ get_figureÚ get_positionÚyaxisÚget_ticklabelsÚxaxisÚ set_visibleÚaxisÚvaluesÚmajor_ticklabelsÚAttributeErrorÚappendÚadd_axes)ÚaxÚpadÚ axes_classÚkwargsÚdividerÚpad_sizeÚxsizeÚysizeÚax_rgbrÚax1ÚlocatorÚtr Úfig©r3úT/var/www/vscode/kcb/lib/python3.10/site-packages/mpl_toolkits/axes_grid1/axes_rgb.pyÚ make_rgb_axes s@ÿÿ  ÿ ÿ  r5c@s*eZdZdZeZddœdd„Zdd„ZdS) Ú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 Os| d|j¡}||i|¤Ž|_}| ¡ |¡t|f||dœ|¤Ž\|_|_|_|j|j|j|jfD]N}t |j t ƒrjt   |¡}|jt|jd|jdƒt|jd|jdƒt|jd|jdƒt|jd|jdƒd n|j }|d d …j d ¡|d d …j d ¡q3d S) 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(rÚbottomr ÚtopÚleftÚright)r7r8r9r:NÚw)ÚpopÚ_defaultAxesClassÚRGBrr%r5ÚRÚGÚBÚ isinstancer rrÚAxisDictÚupdaterrÚspinesrÚlineÚ set_colorÚ major_ticksÚset_markeredgecolor)Úselfr'Úargsr)r(r&r/Úadr3r3r4Ú__init__]s. ÿÿÿ  üõzRGBAxes.__init__c Ks |j|jkr |jksntd|j›d|j›d|j›dƒ‚t |||g¡}t |¡}||dd…dd…df<t |¡}||dd…dd…df<t |¡}||dd…dd…df<|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Ú ValueErrorÚnpÚdstackÚ zeros_liker>Úimshowr?r@rA) rJÚrÚgÚbr)r>r?r@rAÚim_rgbÚim_rÚim_gÚim_br3r3r4Ú imshow_rgb}s ÿ    zRGBAxes.imshow_rgbN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr=rMr[r3r3r3r4r6>s  r6)rN) ÚtypesrÚnumpyrPÚ axes_dividerrrÚmpl_axesrrr5r6r3r3r3r4Ús  5