U mdn@s@ddlZddlmZmZddlmZd ddZGdd d ZdS) N)make_axes_locatableSize)Axes{Gz?c Ksnt|}|t|}dd|dt|}dd|dt|}|t|||g||||||g||jddddg}|dkrt|}dD]} || |j d d f||d |} |jd| d } | | | j | j D]} | d qz"| jD]} | jd qWntk r>YnX|| q| }|D]} || qX|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)Zny1N)rrT)original)ZsharexZsharey)nxnyF)rrZAxesYZAxesXZset_horizontalZ set_verticalZset_axes_locatorZ new_locatortype get_figureZ get_positionZyaxisZget_ticklabelsZxaxisZ set_visibleaxisvaluesZmajor_ticklabelsAttributeErrorappendadd_axes)axpad axes_classkwargsdividerZpad_sizeZxsizeZysizeZax_rgbr ax1locatortrZfigrY/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/mpl_toolkits/axes_grid1/axes_rgb.py make_rgb_axess>   rc@s*eZdZdZeZddddZddZdS) 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)rcOs|d|j}||||_}||t|f||d|\|_|_|_|j|j|j|jfD]0}|j ddj d|j ddj dqbdS)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)rrNw)pop_defaultAxesClassRGBrrrRGBrlineZ set_colorZ major_ticksZset_markeredgecolor)selfrargsrrrrrrr__init__[s zRGBAxes.__init__c 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 : 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 ValueErrornpZdstackZ zeros_liker$Zimshowr%r&r') r)rgbrr$r%r&r'Zim_rgbZim_rZim_gZim_brrr imshow_rgbrs    zRGBAxes.imshow_rgbN)__name__ __module__ __qualname____doc__rr#r+r2rrrrr <sr )rN) numpyr.Z axes_dividerrrZmpl_axesrrr rrrrs  5