IR-e1dZddlZddlmZdgZddZGddZGd d eZGd d eZ Gd de Z ddZ dS)z Combine 3 images to produce a properly-scaled RGB image following Lupton et al. (2004). The three images must be aligned and have the same pixel scale and size. For details, see : https://ui.adsabs.harvard.edu/abs/2004PASP..116..133L N)ZScaleIntervalmake_lupton_rgbc||||td|S||z|zdz }tj||jS)aT Return a naive total intensity from the red, blue, and green intensities. Parameters ---------- image_r : ndarray Intensity of image to be mapped to red; or total intensity if ``image_g`` and ``image_b`` are None. image_g : ndarray, optional Intensity of image to be mapped to green. image_b : ndarray, optional Intensity of image to be mapped to blue. Returns ------- intensity : ndarray Total intensity from the red, blue and green intensities, or ``image_r`` if green and blue images are not provided. NzDplease specify either a single image or red, green, and blue images.g@)dtype) ValueErrornpasarrayr)image_rimage_gimage_b intensitys @lib/python3.11/site-packages/astropy/visualization/lupton_rgb.pycompute_intensityrs_('/GOV 7"W,3I :iw} 5 5 55c2eZdZdZddZdZdZdZdZdS) MappingaA Baseclass to map red, blue, green intensities into uint8 values. Parameters ---------- minimum : float or sequence(3) Intensity that should be mapped to black (a scalar or array for R, G, B). image : ndarray, optional An image used to calculate some parameters of some mappings. NcBttjtjj|_ t |n#t$r d|gz}YnwxYwt |dkrtd||_ tj ||_ dS)Nz)please provide 1 or 3 values for minimum.) floatr iinfouint8max _uint8Maxlen TypeErrorrminimumr _image)selfrimages r__init__zMapping.__init__>srx11566 $ LLLL $ $ $7)mGGG $ w<<1  HII I j'' sAAActj|}tj|}tj|}|j|jks|j|jkr5d}t||j|j|jtj||||tjS)a Convert 3 arrays, image_r, image_g, and image_b into an 8-bit RGB image. Parameters ---------- image_r : ndarray Image to map to red. image_g : ndarray Image to map to green. image_b : ndarray Image to map to blue. Returns ------- RGBimage : ndarray RGB (integer, 8-bits per channel) color image as an NxNx3 numpy array. z/The image shapes must match. r: {}, g: {} b: {}) r r shaperformatdstack_convert_images_to_uint8astyper)rr r r msgs rmake_rgb_imagezMapping.make_rgb_imageKs$*W%%*W%%*W%% MW] * * 0N0NCCSZZ w}gmTTUU Uy  ) )'7G D D  &   rc$t|||S)a Return the total intensity from the red, blue, and green intensities. This is a naive computation, and may be overridden by subclasses. Parameters ---------- image_r : ndarray Intensity of image to be mapped to red; or total intensity if ``image_g`` and ``image_b`` are None. image_g : ndarray, optional Intensity of image to be mapped to green. image_b : ndarray, optional Intensity of image to be mapped to blue. Returns ------- intensity : ndarray Total intensity from the red, blue and green intensities, or ``image_r`` if green and blue images are not provided. )r)rr r r s rrzMapping.intensityis*!'7;;;rctjdd5tj|d|jcdddS#1swxYwYdS)a Return an array which, when multiplied by an image, returns that image mapped to the range of a uint8, [0, 255] (but not converted to uint8). The intensity is assumed to have had minimum subtracted (as that can be done per-band). Parameters ---------- I : ndarray Intensity to be mapped. Returns ------- mapped_I : ndarray ``I`` mapped to uint8 ignoreinvaliddividerN)r errstatecliprrIs rmap_intensity_to_uint8zMapping.map_intensity_to_uint8s$[( ; ; ; 1 171a00 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s?AAc||jdz }||jdz }||jdz }|||||}|||g}|D]=}||z}tjd5d||dk<dddn #1swxYwY>|j}|\}} } tjdd5t |D]\} }tj|| ktj|| ktj||k||z|z |tj| |k||z| z |tj| | ktj| |k||z| z |tj| |k||z| z |tj }||||k<||| < dddn #1swxYwY|S)zo Use the mapping to convert images image_r, image_g, and image_b to a triplet of uint8 images. rrr,)r.Nr-) rr4rr r0r enumeratewherer'r) rr r r fac image_rgbcpixmaxr0g0b0is rr&z Mapping._convert_images_to_uint8sDLO+DLO+DLO+))$..'7*S*STTgw/   A HAX...  !a%                B[( ; ; ; ! !!),, ! !1HGHRvq6zBBBvq6zBBB HRvq6zBBBvq6zBBB  &""!'!f*  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !&s%> BB B DGGGNN) __name__ __module__ __qualname____doc__r!r)rr4r&rrrr2sn   ( ( ( (<<<<.111*(((((rrc eZdZdZddZdZdS) LinearMappinga A linear map map of red, blue, green intensities into uint8 values. A linear stretch from [minimum, maximum]. If one or both are omitted use image min and/or max to set them. Parameters ---------- minimum : float Intensity that should be mapped to black (a scalar or array for R, G, B). maximum : float Intensity that should be mapped to white (a scalar). Nc>||=|td||}||}t|||||_| d|_dS||krtdt||z |_dS)NzCyou must provide an image if you don't set both minimum and maximum)rr z,minimum and maximum values must not be equal)rminrrr!maximum_ranger)rrrKr s rr!zLinearMapping.__init__s ?go} 3))++))++we<<< ?DKKK'!! !OPPP' 122DKKKrc tjdd5tj|dkdtj||jk|j|z |j|jz cdddS#1swxYwYdSNr,r-r)r r0r8rLrr2s rr4z$LinearMapping.map_intensity_to_uint8s [( ; ; ;  8Q$dnq&8$.4;:V                  sA A00A47A4)NNNrBrCrDrEr!r4rFrrrHrHsA  3333,     rrHc eZdZdZddZdZdS) AsinhMappinga A mapping for an asinh stretch (preserving colours independent of brightness). x = asinh(Q (I - minimum)/stretch)/Q This reduces to a linear stretch if Q == 0 See https://ui.adsabs.harvard.edu/abs/2004PASP..116..133L Parameters ---------- minimum : float Intensity that should be mapped to black (a scalar or array for R, G, B). stretch : float The linear stretch of the image. Q : float The asinh softening parameter. ct||d}t||krd}n d}||kr|}d}||jzt j||zz |_|t|z |_dS)Ng>g?g _B) rr!absrr arcsinh_sloper_soften)rrstretchQepsilonQmaxfracs rr!zAsinhMapping.__init__sw''' q66G  AAD4xxT^+bj.B.BB 5>>) rc tjdd5tj|dkdtj||jz|jz|z cdddS#1swxYwYdSrN)r r0r8rUrWrVr2s rr4z#AsinhMapping.map_intensity_to_uint8s [( ; ; ; W W8AFArz!dl2B'C'Cdk'QTU'UVV W W W W W W W W W W W W W W W W W Ws?A##A'*A'N)rRrOrFrrrQrQsF&****"WWWWWrrQceZdZdZddZdS)AsinhZScaleMappingaO A mapping for an asinh stretch, estimating the linear stretch by zscale. x = asinh(Q (I - z1)/(z2 - z1))/Q Parameters ---------- image1 : ndarray or a list of arrays The image to analyse, or a list of 3 images to be converted to an intensity image. image2 : ndarray, optional the second image to analyse (must be specified with image3). image3 : ndarray, optional the third image to analyse (must be specified with image2). Q : float, optional The asinh softening parameter. Default is 8. pedestal : float or sequence(3), optional The value, or array of 3 values, to subtract from the images; or None. Notes ----- pedestal, if not None, is removed from the images when calculating the zscale stretch, and added back into Mapping.minimum[] NrRc||||td|g}n|||g}| t|n#t$r d|gz}YnwxYwt|dkrtdt|}t |D]\}}||dkr|||z ||< nt|dgz}t |}t |} t| d|i} | j | j dz } | j } t |D]\}} | |xx| z cc<t || | |||_ dS)Nz5please specify either a single image or three images.rz please provide 1 or 3 pedestals.gr r)rrrlistr7rr get_limitsrHrKrrQr!r)rimage1image2image3rYpedestalr r@im zscale_limitszscalerXrlevels rr!zAsinhZScaleMapping.__init__6s >V^Nv~ KHEEVV,E   *H  * * *z> *8}}!! !CDDDKKE"5)) 0 02A;#%%!HQK/E!H 05zzSE)H!5)&((33E::  ;U;;.6>!#44.!(++  HAu AJJJ% JJJJdGWa888 s4AA)NNrRN)rBrCrDrEr!rFrrr_r_s22&&&&&&rr_rRct|||}||||}|r!ddl} | j||d|S)a Return a Red/Green/Blue color image from up to 3 images using an asinh stretch. The input images can be int or float, and in any range or bit-depth. For a more detailed look at the use of this method, see the document :ref:`astropy:astropy-visualization-rgb`. Parameters ---------- image_r : ndarray Image to map to red. image_g : ndarray Image to map to green. image_b : ndarray Image to map to blue. minimum : float Intensity that should be mapped to black (a scalar or array for R, G, B). stretch : float The linear stretch of the image. Q : float The asinh softening parameter. filename : str Write the resulting RGB image to a file (file type determined from extension). Returns ------- rgb : ndarray RGB (integer, 8-bits per channel) color image as an NxNx3 numpy array. rNlower)origin)rQr)matplotlib.imager imsave) r r r rrXrYfilenameasinhMaprgb matplotlibs rrr_sdBGWa00H  ! !'7G < >> JrrA)rrkrRN) rEnumpyr r__all__rrrHrQr_rrFrrrxs3  6666BKKKKKKKK\.....G...b(W(W(W(W(W7(W(W(WV@@@@@@@@HDH))))))r