U e{j1@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|dks|dkr,|dkr |dks(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 _/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/visualization/lupton_rgb.pycompute_intensitysrc@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. NcCsjtttjj|_z 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 : 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_imageKs    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 iszMapping.intensityc Cs6tjdddt|d|jW5QRSQRXdS)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)rerrstateZcliprrIr r rmap_intensity_to_uint8szMapping.map_intensity_to_uint8c Cs`||jd}||jd}||jd}|||||}|||g}|D]0}||9}tjddd||dk<W5QRXqL|j}|\}} } tjdddt|D]\} }t|| kt|| kt||k||||t| |k||| |t| | kt| |k||| |t| |k||| |tj }||||k<||| <qW5QRX|S)zn Use the mapping to convert images image_r, image_g, and image_b to a triplet of uint8 images rrr#)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 rr2s  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)rrr6rr r rrs zLinearMapping.__init__c CsVtjddd>t|dkdt||jk|j||j|jW5QRSQRXdSNr#r$r)rr'r-r7rr(r r rr*s z$LinearMapping.map_intensity_to_uint8)NNNr0r1r2r3rr*r r r rr4s r4c@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)rrstretchQepsilonZQmaxfracr r rrs  zAsinhMapping.__init__c CsNtjddd6t|dkdt||j|j|W5QRSQRXdSr8)rr'r-r=r?r>r(r r rr*sz#AsinhMapping.map_intensity_to_uint8N)r;r9r r r rr:s r:c@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>|dks|dkr0|dkr |dks(td|g}n |||g}|dk rz t|Wntk rld|g}YnXt|dkrtdt|}t|D]$\}}||dkr|||||<qnt|dg}t|}t|} t| d|i} | j | j d} | j } t|D]\}} | || 7<qt || | |||_ dS)Nz5please specify either a single image or three images.rz please provide 1 or 3 pedestals.grr)rrrlistr,rrZ get_limitsr4r6rr:rr)rZimage1Zimage2Zimage3rAZpedestalrr/ZimZ zscale_limitsZzscaler@rlevelr r rr7s8     zAsinhZScaleMapping.__init__)NNr;N)r0r1r2r3rr r r rrDsrDr;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)r:r"Zmatplotlib.imagerZimsave) r r r rr@rAfilenameZasinhMaprgbZ matplotlibr r rr`s ! )NN)rrGr;N) r3numpyrr__all__rrr4r:rDrr r r rs  !1,D