o :ήcy @sddgZddlZddlmZddlmZmZddl m Z m Z e ee dkr4dd l mZed d d dd dZdddZddZdddZdddZdS)imreadimsaveN)version)Image __version__) img_as_ubyte img_as_uintz8.1.2)warnzYour installed pillow version is < 8.1.2. Several security issues (CVE-2021-27921, CVE-2021-25290, CVE-2021-25291, CVE-2021-25293, and more) have been fixed in pillow 8.1.2 or higher. We recommend to upgrade this library.) stacklevelcKsht|tr(t|d}t|}t|||dWdS1s!wYdSt|}t|||dS)aLoad an image from file. Parameters ---------- fname : str or file File name or file-like-object. dtype : numpy dtype object or string specifier Specifies data type of array elements. img_num : int, optional Specifies which image to read in a file with multiple images (zero-indexed). kwargs : keyword pairs, optional Addition keyword arguments to pass through. Notes ----- Files are read using the Python Imaging Library. See PIL docs [1]_ for a list of supported formats. References ---------- .. [1] http://pillow.readthedocs.org/en/latest/handbook/image-file-formats.html rb)dtypeimg_numN) isinstancestropenrpil_to_ndarray)fnamerrkwargsfimrE/tmp/pip-target-vg8gfxp4/lib/python/skimage/io/_plugins/pil_plugin.pyrs    $ c CsLz|dWnty&}zd}t|}d|j||f}t|d}~wwg}d}d} z|| Wn ty>Ynw|} |durT|| krT|d| d7} q-|jdkrd|jdkrd|durdd}|jd kr|durqt |}|ry| d } n6|jdkrd |j vr| d } n&| d } n |jdkr| d } nd|jvr| d } n |jdkr| d } |j dr|j } |jdrdnd}d|jvr|dd}t| |} | ddd| _ntj| |d} || | d7} |durnq.t|dr|jr|j|durt|dkrt|S|r|dS|r$td|dS)zqImport a PIL Image object to an ndarray, in memory. Parameters ---------- Refer to ``imread``. rzLhttp://pillow.readthedocs.org/en/latest/installation.html#external-librarieszACould not load "%s" Reason: "%s" Please see documentation at: %sNPNGIuint16PL transparencyRGBARGB1ACMYKI;16Bz>u2z|d kr>|t j }d }}n t|}nt|}d }d }z| }Wn t y^| }Ynw|jdkrt||jj}z ||d |W|St y||d |Y|Sw|jd|jd f}z t|||}W|St yt|||}Y|Sw)z`Export an ndarray to a PIL object. Parameters ---------- Refer to ``imsave``. rr"r!rr )pngrr&rrrrrawr)ndimrr?rkindr maxminastyper<uint8r>AttributeErrortostringrnewT frombytesr=)arr format_strr4 mode_base array_bufferr image_shaperrrndarray_to_pilsJ            rncKst|ts |dur d}t|tr|drd}t|}|jjdkr*|tj }|j dvr7t d|j f|j dkrG|j dd vrGt d t ||d }|j|fd |i|dS) a;Save an image to disk. Parameters ---------- fname : str or file-like object Name of destination file. arr : ndarray of uint8 or float Array (image) to save. Arrays of data-type uint8 should have values in [0, 255], whereas floating-point arrays must be in [0, 1]. format_str: str Format to save as, this is defaulted to PNG if using a file-like object; this will be derived from the extension if fname is a string kwargs: dict Keyword arguments to the Pillow save function (or tifffile save function, for Tiff files). These are format dependent. For example, Pillow's JPEG save function supports an integer ``quality`` argument with values in [1, 95], while TIFFFile supports a ``compress`` integer argument with values in [0, 9]. Notes ----- Use the Python Imaging Library. See PIL docs [1]_ for a list of other supported formats. All images besides single channel PNGs are converted using `img_as_uint8`. Single Channel PNGs have the following behavior: - Integer values in [0, 255] and Boolean types -> img_as_uint8 - Floating point and other integers -> img_as_uint16 References ---------- .. [1] http://pillow.readthedocs.org/en/latest/handbook/image-file-formats.html Nrz.pngb)r rz!Invalid shape for image array: %srr rYz*Invalid number of channels in image array.)rjr3)rrlowerr:r< asanyarrayrr_rbrcr^r0r?rnsave)rrirjrimgrrrrs #        )NN)N)__all__numpyr< packagingrPILrr pil_versionutilrr parsewarningsr rrr5rnrrrrrs    !W 3