ó ˽÷Xc @`s«dZddlmZmZmZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZyddlmZmZWn)ek r×ddlZddlZnXeedƒsöeje_nddd d d d d ddg Zddddd„Zedd„Zdd„Z edd„Z!dZ"dddddddd„Z#dd„Z$d„Z%ddd„Z&d„Z'dS(sÁ A collection of image utilities using the Python Imaging Library (PIL). Note that PIL is not a dependency of SciPy and this module is not available on systems that don't have PIL installed. i(tdivisiontprint_functiontabsolute_importN( tamintamaxtraveltasarraytcasttarangetonestnewaxist transposet iscomplexobjtuint8t issubdtypetarray(tImaget ImageFiltert frombytest fromimagettoimagetimsavetimreadt bytescaletimrotatetimresizetimshowtimfilteriÿcC`s|jtkr|S|dkr.tdƒ‚n|dkrItdƒ‚n||krdtdƒ‚n|d kr|jƒ}n|d krš|jƒ}n||}|dkr¿tdƒ‚n|dkrÔd}nt||ƒ|}||||}|j||ƒdjtƒS( s€ Byte scales an array (image). Byte scaling means converting the input image to uint8 dtype and scaling the range to ``(low, high)`` (default 0-255). If the input image already has dtype uint8, no scaling is done. Parameters ---------- data : ndarray PIL image data array. cmin : scalar, optional Bias scaling of small values. Default is ``data.min()``. cmax : scalar, optional Bias scaling of large values. Default is ``data.max()``. high : scalar, optional Scale max value to `high`. Default is 255. low : scalar, optional Scale min value to `low`. Default is 0. Returns ------- img_array : uint8 ndarray The byte-scaled array. Examples -------- >>> from scipy.misc import bytescale >>> img = np.array([[ 91.06794177, 3.39058326, 84.4221549 ], ... [ 73.88003259, 80.91433048, 4.88878881], ... [ 51.53875334, 34.45808177, 27.5873488 ]]) >>> bytescale(img) array([[255, 0, 236], [205, 225, 4], [140, 90, 70]], dtype=uint8) >>> bytescale(img, high=200, low=100) array([[200, 100, 192], [180, 188, 102], [155, 135, 128]], dtype=uint8) >>> bytescale(img, cmin=0, cmax=255) array([[91, 3, 84], [74, 81, 5], [52, 34, 28]], dtype=uint8) iÿs+`high` should be less than or equal to 255.is+`low` should be greater than or equal to 0.s0`high` should be greater than or equal to `low`.s$`cmax` should be larger than `cmin`.igà?N( tdtypeR t ValueErrortNonetmintmaxtfloattcliptastype(tdatatcmintcmaxthightlowtcscaletscaletbytedata((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyR!s(.         cC`s%tj|ƒ}t|d|d|ƒS(s‹ Read an image from a file as an array. Parameters ---------- name : str or file object The file name or file object to be read. flatten : bool, optional If True, flattens the color layers into a single gray-scale layer. mode : str, optional Mode to convert image to, e.g. ``'RGB'``. See the Notes for more details. Returns ------- imread : ndarray The array obtained by reading the image. Notes ----- `imread` uses the Python Imaging Library (PIL) to read an image. The following notes are from the PIL documentation. `mode` can be one of the following strings: * 'L' (8-bit pixels, black and white) * 'P' (8-bit pixels, mapped to any other mode using a color palette) * 'RGB' (3x8-bit pixels, true color) * 'RGBA' (4x8-bit pixels, true color with transparency mask) * 'CMYK' (4x8-bit pixels, color separation) * 'YCbCr' (3x8-bit pixels, color video format) * 'I' (32-bit signed integer pixels) * 'F' (32-bit floating point pixels) PIL also provides limited support for a few special modes, including 'LA' ('L' with alpha), 'RGBX' (true color with padding) and 'RGBa' (true color with premultiplied alpha). When translating a color image to black and white (mode 'L', 'I' or 'F'), the library uses the ITU-R 601-2 luma transform:: L = R * 299/1000 + G * 587/1000 + B * 114/1000 When `flatten` is True, the image is converted using mode 'F'. When `mode` is not None and `flatten` is True, the image is first converted according to `mode`, and the result is then flattened using mode 'F'. tflattentmode(RtopenR(tnameR,R-tim((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyRis3cC`sBt|ddƒ}|dkr.|j|ƒn|j||ƒdS(sæ Save an array as an image. Parameters ---------- name : str or file object Output file name or file object. arr : ndarray, MxN or MxNx3 or MxNx4 Array containing image values. If the shape is ``MxN``, the array represents a grey-level image. Shape ``MxNx3`` stores the red, green and blue bands along the last dimension. An alpha layer may be included, specified as the last colour band of an ``MxNx4`` array. format : str Image format. If omitted, the format to use is determined from the file name extension. If a file object was used instead of a file name, this parameter should always be used. Examples -------- Construct an array of gradient intensity values and save to file: >>> from scipy.misc import imsave >>> x = np.zeros((255, 255)) >>> x = np.zeros((255, 255), dtype=np.uint8) >>> x[:] = np.arange(255) >>> imsave('gradient.png', x) Construct an array with three colour bands (R, G, B) and store to file: >>> rgb = np.zeros((255, 255, 3), dtype=np.uint8) >>> rgb[..., 0] = np.arange(255) >>> rgb[..., 1] = 55 >>> rgb[..., 2] = 1 - np.arange(255) >>> imsave('rgb_gradient.png', rgb) t channel_axisiN(RRtsave(R/tarrtformatR0((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyR s % cC`sÙtj|ƒstdƒ‚n|d k rN||jkr|j|ƒ}qnB|jdkrd|jkr~|jdƒ}q|jdƒ}n|r¨|jdƒ}n!|jdkrÉ|jdƒ}nt|ƒ}|S( s/ Return a copy of a PIL image as a numpy array. Parameters ---------- im : PIL image Input image. flatten : bool If true, convert the output to grey-scale. mode : str, optional Mode to convert image to, e.g. ``'RGB'``. See the Notes of the `imread` docstring for more details. Returns ------- fromimage : ndarray The different colour bands/channels are stored in the third dimension, such that a grey-image is MxN, an RGB-image MxNx3 and an RGBA-image MxNx4. sInput is not a PIL image.tPt transparencytRGBAtRGBtFt1tLN(Rt isImageTypet TypeErrorRR-tconverttinfoR(R0R,R-ta((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyRÍs  s7Mode is unknown or incompatible with input array shape.c C`sát|ƒ}t|ƒr'tdƒ‚nt|jƒ} t| ƒdkpot| ƒdkood| kpod| k} | s‡tdƒ‚nt| ƒdkrÄ| d| df} |dkrê|jtjƒ} t j || | j ƒƒ} | S|dkrÝt |d |d |d |d|ƒ} t j d | | j ƒƒ} |dk rc| jt|dtƒj ƒƒnv|d krÙtddddtƒdd…tftddtƒtdd…f}| jt|dtƒj ƒƒn| S|dkr||k} t j d| | j ƒƒ} | S|dkr5tt|ƒƒ}n|dkrVtt|ƒƒ}n|d||||||}|dkr´|jtjƒ} t j || | j ƒƒ} n ttƒ‚| S|dkrAd| krþtjt| ƒdkƒd}qGtjt| ƒdkƒ}t|ƒr2|d}qGtdƒ‚n|}| |}|dkrltdƒ‚nt |d |d |d |d|ƒ} |dkr¿| j ƒ}| d| df} np|dkr÷t| dƒj ƒ}| d| df} n8|dkr/t| d ƒj ƒ}| d| df} n|dkrY|dkrPd}qYd}n|d!krtttƒ‚n|d"krž|dkržtdƒ‚qžn|d#krÈ|dkrÈtdƒ‚qÈnt j || |ƒ} | S($s±Takes a numpy array and returns a PIL image. The mode of the PIL image depends on the array shape and the `pal` and `mode` keywords. For 2-D arrays, if `pal` is a valid (N,3) byte-array giving the RGB values (from 0 to 255) then ``mode='P'``, otherwise ``mode='L'``, unless mode is given as 'F' or 'I' in which case a float and/or integer array is made. Notes ----- For 3-D arrays, the `channel_axis` argument tells which dimension of the array holds the channel data. For 3-D arrays if one of the dimensions is 3, the mode is 'RGB' by default or 'YCbCr' if selected. The numpy array must be either 2 dimensional or 3 dimensional. s&Cannot convert a complex-valued array.iiis8'arr' does not have a suitable array shape for any mode.iiR9R;R5R'R(R%R&RiNR:gð?tIs!Could not find channel dimension.s$Channel axis dimension is not valid.R8R7tYCbCrtCMYKsInvalid array shape for mode.(NR;R5(i(ii(iii(iii(sRGBsRGBARBRC(sRGBRB(sRGBARC(RR RtlisttshapetlenR#tnumpytfloat32RRttostringRRt putpaletteR RR R RRRtuint32t_errstrt flatnonzeroR (R3R'R(R%R&tpalR-R1R$REtvalidtdata32timageR+tcatnumchtstrdata((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyRs”  $   " %#"    "    "    $             tbilinearcC`sdt|ƒ}idd6dd6dd6dd6dd 6}t|ƒ}|j|d ||ƒ}t|ƒS( s Rotate an image counter-clockwise by angle degrees. Parameters ---------- arr : ndarray Input array of image to be rotated. angle : float The angle of rotation. interp : str, optional Interpolation - 'nearest' : for nearest neighbor - 'bilinear' : for bilinear - 'lanczos' : for lanczos - 'cubic' : for bicubic - 'bicubic' : for bicubic Returns ------- imrotate : ndarray The rotated array of image. itnearestitlanczosiRUitbicubictcubictresample(RRtrotateR(R3tangletinterptfuncR0((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyRvs  ) cC`s»t|ƒ}tjdƒ\}}y|j|ƒWntdƒ‚nXddl}|j|ƒ|jjddƒ}|j d||fƒ}|j |ƒ|dkr·tdƒ‚ndS( sò Simple showing of an image through an external viewer. Uses the image viewer specified by the environment variable SCIPY_PIL_IMAGE_VIEWER, or if that is not defined then `see`, to view a temporary file generated from array data. Parameters ---------- arr : ndarray Array of image data to show. Returns ------- None Examples -------- >>> a = np.tile(np.arange(255), (255,1)) >>> from scipy import misc >>> misc.imshow(a) s.pngs"Error saving temporary image data.iNtSCIPY_PIL_IMAGE_VIEWERtsees%s %ssCould not execute image viewer.( RttempfiletmkstempR2t RuntimeErrortostclosetenvirontgettsystemtunlink(R3R0tfnumtfnameRdtcmdtstatus((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyR–s     c C`söt|d|ƒ}t|ƒ}t|tƒr\|d}tt|jƒ|jtƒƒ}nNtt|ƒtƒr–tt|jƒ|jtƒƒ}n|d|df}idd6dd6dd6d d 6d d 6}|j |d ||ƒ}t |ƒS( s  Resize an image. Parameters ---------- arr : ndarray The array of image to be resized. size : int, float or tuple * int - Percentage of current size. * float - Fraction of current size. * tuple - Size of the output image. interp : str, optional Interpolation to use for re-sizing ('nearest', 'lanczos', 'bilinear', 'bicubic' or 'cubic'). mode : str, optional The PIL image mode ('P', 'L', etc.) to convert `arr` before resizing. Returns ------- imresize : ndarray The resized array of image. See Also -------- toimage : Implicitly used to convert `arr` according to `mode`. scipy.ndimage.zoom : More generic implementation that does not use PIL. R-gY@iiRVRWiRUiRXRYRZ( RttypeRtintttupleRtsizeR#R!tresizeR( R3RqR]R-R0ttstpercentR^timnew((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyRÀs   %%)cC`s¨i tjd6tjd6tjd6tjd6tjd6tjd6tjd6tjd6tj d 6tj d 6}t |ƒ}||kr‘t d ƒ‚nt |j||ƒƒS( sP Simple filtering of an image. Parameters ---------- arr : ndarray The array of Image in which the filter is to be applied. ftype : str The filter that has to be applied. Legal values are: 'blur', 'contour', 'detail', 'edge_enhance', 'edge_enhance_more', 'emboss', 'find_edges', 'smooth', 'smooth_more', 'sharpen'. Returns ------- imfilter : ndarray The array with filter applied. Raises ------ ValueError *Unknown filter type.* If the filter you are trying to apply is unsupported. tblurtcontourtdetailt edge_enhancetedge_enhance_moretembosst find_edgestsmootht smooth_moretsharpensUnknown filter type.(RtBLURtCONTOURtDETAILt EDGE_ENHANCEtEDGE_ENHANCE_MOREtEMBOSSt FIND_EDGEStSMOOTHt SMOOTH_MOREtSHARPENRRRtfilter(R3tftypet_tdictR0((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyRîs            ((t__doc__t __future__RRRRGRaRRRRRRR R R R R RRtPILRRt ImportErrorthasattrt fromstringRt__all__RRtFalseRRRRLRRRRR(((s1/tmp/pip-build-7oUkmx/scipy/scipy/misc/pilutil.pyts0  X  H7 -4q *.