σ ›ίΘ[c@`sΝdZddlmZmZmZmZddlZddlm Z dgZ ddd„Z de fd „ƒYZd efd „ƒYZd efd „ƒYZdefd„ƒYZddddd„ZdS(uι 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 i(tabsolute_importtdivisiontprint_functiontunicode_literalsNi(tZScaleIntervalumake_lupton_rgbcC`sk|dks|dkrC|dko-|dks?tdƒ‚n|S|||d}tj|d|jƒS(ux 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. uDplease specify either a single image or red, green, and blue images.g@tdtypeN(tNonet ValueErrortnptasarrayR(timage_rtimage_gtimage_bt intensity((s?lib/python2.7/site-packages/astropy/visualization/lupton_rgb.pytcompute_intensitys tMappingcB`sAeZdZddd„Zd„Zd„Zd„Zd„ZRS(uJ 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. cC`sttjtjƒjƒ|_yt|ƒWntk rLd|g}nXt|ƒdkrntdƒ‚n||_ tj |ƒ|_ dS(Niu)please provide 1 or 3 values for minimum.( tfloatRtiinfotuint8tmaxt _uint8Maxtlent TypeErrorRtminimumR t_image(tselfRtimage((s?lib/python2.7/site-packages/astropy/visualization/lupton_rgb.pyt__init__>s  cC`s¦tj|ƒ}tj|ƒ}tj|ƒ}|j|jksQ|j|jkr~d}t|j|j|j|jƒƒ‚ntj|j|||ƒƒjtjƒS(uθ 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. u/The image shapes must match. r: {}, g: {} b: {}( RR tshapeRtformattdstackt_convert_images_to_uint8tastypeR(RR R R tmsg((s?lib/python2.7/site-packages/astropy/visualization/lupton_rgb.pytmake_rgb_imageKs$'cC`st|||ƒS(uώ 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 ((s?lib/python2.7/site-packages/astropy/visualization/lupton_rgb.pyR gscC`s9tjddddƒtj|d|jƒSWdQXdS(uί 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 tinvaliduignoretdivideiN(RterrstatetclipR(RtI((s?lib/python2.7/site-packages/astropy/visualization/lupton_rgb.pytmap_intensity_to_uint8~sc C`sΒ||jd}||jd}||jd}|j|j|||ƒƒ}|||g}x(|D] }||9}d||dkR=(RR'((s?lib/python2.7/site-packages/astropy/visualization/lupton_rgb.pyR(s(R3R4R5RR((((s?lib/python2.7/site-packages/astropy/visualization/lupton_rgb.pyR:άs tAsinhZScaleMappingcB`s#eZdZddddd„ZRS(uj 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[] icC`s·|dks|dkrK|dko-|dks?tdƒ‚n|g}n|||g}|dk ryt|ƒWntk r”d|g}nXt|ƒdkrΆtdƒ‚nt|ƒ}xUt|ƒD]1\}}||dkrΟ|||||s"   )*D