a ߙfb0@stdZddlZddlmZdgZdddZGdd d ZGd d d eZGd d d eZ Gddde Z dddZ 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_rgbcCsL|dus|dur,|dur |dus(td|S|||d}tj||jdS)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 intensityr ?lib/python3.9/site-packages/astropy/visualization/lupton_rgb.pycompute_intensitys rc@s:eZdZdZd ddZddZddZd d Zd d ZdS)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. NcCshtttjj|_z t|Wnty<d|g}Yn0t|dkrRtd||_ t ||_ dS)Nz)please provide 1 or 3 values for minimum.) floatrZiinfouint8max _uint8Maxlen TypeErrorrminimumr_image)selfrimager r r__init__=s   zMapping.__init__cCsnt|}t|}t|}|j|jks6|j|jkrRd}t||j|j|jt||||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: {}) rrshaperformatZdstack_convert_images_to_uint8astyper)rr r r msgr r rmake_rgb_imageJs   zMapping.make_rgb_imagecCs 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 r r rr fszMapping.intensitycCsBtjddd t|d|jWdS1s40YdS)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 ignoreinvalidZdividerN)rerrstateZcliprrIr r rmap_intensity_to_uint8}szMapping.map_intensity_to_uint8c Cs||jd}||jd}||jd}|||||}|||g}|D]D}||9}tjddd||dk<WdqL1s0YqL|j}|\}} } tjdddt|D]\} }t|| kt|| kt||k||||t| |k||| |t| | kt| |k||| |t| |k||| |tj }||||k<||| <qWdn1s|0Y|S)z\Use the mapping to convert images image_r, image_g, and image_b to a triplet of uint8 imagesrrr#)r%Nr$) rr)r rr&r enumeratewherer r) rr r r ZfacZ image_rgbcZpixmaxZr0Zg0Zb0ir r rrs8 ,     *z Mapping._convert_images_to_uint8)NN) __name__ __module__ __qualname____doc__rr"r r)rr r r rr1s  rc@s"eZdZdZdddZddZdS) 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). NcCs|dus|dur@|dur td|dur0|}|dur@|}tj|||d||_|durfd|_n||krvtdt|||_dS)NzCyou must provide an image if you don't set both minimum and maximum)rrz,minimum and maximum values must not be equal)rminrrrmaximum_ranger)rrr5rr r rrszLinearMapping.__init__c Csbtjddd@t|dkdt||jk|j||j|jWdS1sT0YdSNr#r$r)rr&r,r6rr'r r rr)s  z$LinearMapping.map_intensity_to_uint8)NNNr/r0r1r2rr)r r r rr3s r3c@s"eZdZdZdddZddZdS) 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. cCsbt||d}t||kr"d}nd}||kr2|}d}||jt|||_|t||_dS)Ng>g?g _B) rrabsrrarcsinh_sloper_soften)rrstretchQepsilonZQmaxZfracr r rrs  zAsinhMapping.__init__c CsZtjddd8t|dkdt||j|j|WdS1sL0YdSr7)rr&r,r<r>r=r'r r rr)sz#AsinhMapping.map_intensity_to_uint8N)r:r8r r r rr9s r9c@seZdZdZdddZdS)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[] Nr:cCs<|dus|dur0|dur |dus(td|g}n |||g}|durz t|Wntyjd|g}Yn0t|dkrtdt|}t|D]$\}}||dkr|||||<qnt|dg}t|}t|} t| d|i} | j | j d} | j } t|D]\}} | || 7<qt || | |||_ dS)z Nz5please specify either a single image or three images.rz please provide 1 or 3 pedestals.grr)rrrlistr+rrZ get_limitsr3r5rr9rr)rZimage1Zimage2Zimage3r@Zpedestalrr.ZimZ zscale_limitsZzscaler?rlevelr r rr s4      zAsinhZScaleMapping.__init__)NNr:N)r/r0r1r2rr r r rrBsrBr:c Cs<t|||}||||}|r8ddl} | jj||dd|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)r9r"Zmatplotlib.imagerZimsave) r r r rr?r@filenameZasinhMapZrgbZ matplotlibr r rrJs )NN)rrEr:N) r2Znumpyrr__all__rrr3r9rBrr r r rs  )*D