B q\1@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 : http://adsabs.harvard.edu/abs/2004PASP..116..133L N)ZScaleIntervalmake_lupton_rgbcCsL|dks|dkr,|dkr |dks(td|S|||d}tj||jdS)ax Return a naive total intensity from the red, blue, and green intensities. Parameters ---------- image_r : `~numpy.ndarray` Intensity of image to be mapped to red; or total intensity if ``image_g`` and ``image_b`` are None. image_g : `~numpy.ndarray`, optional Intensity of image to be mapped to green. image_b : `~numpy.ndarray`, optional Intensity of image to be mapped to blue. Returns ------- intensity : `~numpy.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.7/site-packages/astropy/visualization/lupton_rgb.pycompute_intensitys rc@s:eZdZdZd ddZddZddZd d Zd d ZdS)MappingaJ 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 : `~numpy.ndarray`, optional An image used to calculate some parameters of some mappings. NcCsjtttjj|_y t|Wntk r>d|g}YnXt|dkrTtd||_ 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 : `~numpy.ndarray` Image to map to red. image_g : `~numpy.ndarray` Image to map to green. image_b : `~numpy.ndarray` Image to map to blue. Returns ------- RGBimage : `~numpy.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 : `~numpy.ndarray` Intensity of image to be mapped to red; or total intensity if ``image_g`` and ``image_b`` are None. image_g : `~numpy.ndarray`, optional Intensity of image to be mapped to green. image_b : `~numpy.ndarray`, optional Intensity of image to be mapped to blue. Returns ------- intensity : `~numpy.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.intensityc Cs*tjdddt|d|jSQRXdS)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 : `~numpy.ndarray` Intensity to be mapped. Returns ------- mapped_I : `~numpy.ndarray` ``I`` mapped to uint8 ignore)invaliddividerN)rerrstateZclipr)rIr r rmap_intensity_to_uint8}szMapping.map_intensity_to_uint8c Csh||jd}||jd}||jd}|||||}|||g}x8|D]0}||9}tjddd||dk<WdQRXqNW|j}|\}} } tjdddxt|D]\} }t|| kt|| kt||k||||t| |k||| |t| | kt| |k||| |t| |k||| |tj }||||k<||| <qWWdQRX|S)z\Use the mapping to convert images image_r, image_g, and image_b to a triplet of uint8 imagesrrr#)r$N)r$r%) rr(r rr&r enumeratewherer r) rr r r ZfacZ image_rgbcZpixmaxZr0Zg0Zb0ir r rrs.      & 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|dks|dkr@|dkr td|dkr0|}|dkr@|}tj|||d||_|dkrfd|_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)rrr4rr r rrszLinearMapping.__init__c CsJtjddd2t|dkdt||jk|j||j|jSQRXdS)Nr#)r$r%r)rr&r+r5r)rr'r r rr(s z$LinearMapping.map_intensity_to_uint8)NNN)r.r/r0r1rr(r r r rr2s  r2c@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 http://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 CsBtjddd*t|dkdt||j|j|SQRXdS)Nr#)r$r%r)rr&r+r9r;r:)rr'r r rr(sz#AsinhMapping.map_intensity_to_uint8N)r7)r.r/r0r1rr(r r r rr6s r6c@seZdZdZdddZdS)AsinhZScaleMappingaj A mapping for an asinh stretch, estimating the linear stretch by zscale. x = asinh(Q (I - z1)/(z2 - z1))/Q Parameters ---------- image1 : `~numpy.ndarray` or a list of arrays The image to analyse, or a list of 3 images to be converted to an intensity image. image2 : `~numpy.ndarray`, optional the second image to analyse (must be specified with image3). image3 : `~numpy.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[] Nr7cCsF|dks|dkr0|dkr |dks(td|g}n |||g}|dk ry t|Wntk rld|g}YnXt|dkrtdt|}x@t|D]$\}}||dkr|||||<qWnt|dg}t|}t|} t| d|i} | j | j d} | j } x&t|D]\}} | || 7<qWt || | |||_ dS)z Nz5please specify either a single image or three images.rz please provide 1 or 3 pedestals.grr)rrrlistr*rrZ get_limitsr2r4rr6rr)rZimage1Zimage2Zimage3r=Zpedestalrr-ZimZ zscale_limitsZzscaler<rlevelr r rr s4     zAsinhZScaleMapping.__init__)NNr7N)r.r/r0r1rr r r rr?sr?r7c 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-visualization-rgb`. Parameters ---------- image_r : `~numpy.ndarray` Image to map to red. image_g : `~numpy.ndarray` Image to map to green. image_b : `~numpy.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 : `~numpy.ndarray` RGB (integer, 8-bits per channel) color image as an NxNx3 numpy array. rNlower)origin)r6r"Zmatplotlib.imagerZimsave) r r r rr<r=filenameZasinhMapZrgbZ matplotlibr r rrJs ! )NN)rrBr7N) r1Znumpyrr__all__rrr2r6r?rr r r rs  )*D