;c]c @`sdZddlmZmZmZddlZddlZddlZddl m Z ddl m Z m Z ddlmZmZejeZd Zd e fd YZd efd YZdefdYZdefdYZdZdZdZdddddZdedZddl m!Z!m"Z"dZ#e$dedede!de"deZ%dZ&e&dS(s+ Plugin that wraps the the Pillow library. i(tabsolute_importtprint_functiontdivisionNi(tformats(tFormatt image_as_uinti(tpillow_formatst pillow_docss Parameters for reading ---------------------- pilmode : str From the Pillow documentation: * 'L' (8-bit pixels, grayscale) * '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 grayscale (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 as_gray : bool If True, the image is converted using mode 'F'. When `mode` is not None and `as_gray` is True, the image is first converted according to `mode`, and the result is then "flattened" using mode 'F'. t PillowFormatcB`seZdZeZd ZdZdZdZ e dZ dZ dZ dZdejfd YZd ejfd YZRS( s/ Base format class for Pillow formats. titcO`s,tt|j||tj|_dS(N(tsuperRt__init__t threadingtRLockt_lock(tselftargstkwargs((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyR EscC`s|jS(s The PIL plugin id. (t _plugin_id(R((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt plugin_idJscC`s|j|jsbt|_ddl}t|dsFtdnddlm}||_n|jdkrt dn|j}WdQX|j d kr|j n |j |S( Nit __version__s/Imageio Pillow plugin requires Pillow, not PIL!(tImages*Imageio Pillow plugin requires Pillow lib.tPNGtJPEGtBMPtGIFtPPM(RRRRR( Rt_pillow_importedtTruetPILthasattrt ImportErrorRt_ImagetNonet RuntimeErrorRtpreinittinit(RRR((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt _init_pillowPs         cC`s}|j}|jd|jdkry|j|jkry|j|j\}}|rv|jrs||jrstSqvqyndS(Nit?(R&tmodetmodesRtOPENt firstbytesR(RtrequestRtfactorytaccept((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt _can_readgs cC`sX|j}|jd|jdkrT|j|jkrT|j|jkrQtSqTndS(NiR'(R&R(R)t extensiont extensionsRtSAVER(RR,R((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt _can_writeps  tReadercB`sMeZdedZdZdZdZdZdZ dZ RS(cC`sP|jj}y|j|jj\}}Wn'tk rUtd|jjnX|j|_||jd|_ t |dr|j |j j n|j j r|j j jr|j j j|j j _nt|j td|dt|j |_|dk r||jdRLR=tclose(R((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt_closes  cC`s|jS(N(RH(R((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt _get_lengthscC`s<y|jj|Wn!tk r7td|nXdS(NsCould not seek to index %i(R>tseektEOFErrort IndexError(Rtindex((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt_seeks cC`s||jkr+td||jfn|jj}||krV|j|n*x'||kr|d7}|j|qYW|jjr|jjjr|jjj|jj_n|jj dt |j|j }||jj fS(NsImage index %i > %iii( RHRTR>ttellRVR@RARBRCtgetdatat pil_get_frameRGtinfo(RRUR tim((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt _get_datas  cC`s.|dkp|dks$tn|jjS(Ni(R"RTR>RZ(RRU((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt_get_meta_datas N( t__name__t __module__R"RKRJR<RPRQRVR\R](((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyR4ws     tWritercB`s,eZdZdZdZdZRS(cC`s{|jj}y|j|jj|_Wn'tk rRtd|jjnX|jj |_ i|_ t |_ dS(NsFormat %s cannot write images.(R9R&R2Rt _save_funcR:R#R;R,RMR=t_metaRKt_written(RR((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyRJs  cC`sdS(N((R((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyRPscC`s|jr"td|jjn|jdkrf|jddkrf|dddddf}nt|_|jj|t ||jj |jj dt}d|jkr|j }n|j |jd|jj |jt|dS( Ns&Format %s only supports single images.iiiit prefer_uint8tbitsR9(RcR#R9R;tndimtshapeRRbtupdatetndarray_to_pilRtpoptquantizetsaveR=RN(RR[tmetatimg((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt _append_datas "" $"cC`s|jj|dS(N(RbRh(RRm((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyt set_meta_datas(R^R_RJRPRoRp(((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyR`s  N(R^R_t__doc__RKRR"R!t_modest _descriptionR tpropertyRR&R/R3RR4R`(((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyR;s   Gt PNGFormatcB`s@eZdZdejfdYZdejfdYZRS(s A PNG format based on Pillow. This format supports grayscale, RGB and RGBA images. Parameters for reading ---------------------- ignoregamma : bool Avoid gamma correction. Default True. pilmode : str From the Pillow documentation: * 'L' (8-bit pixels, grayscale) * '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 grayscale (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 as_gray : bool If True, the image is converted using mode 'F'. When `mode` is not None and `as_gray` is True, the image is first converted according to `mode`, and the result is then "flattened" using mode 'F'. Parameters for saving --------------------- optimize : bool If present and true, instructs the PNG writer to make the output file as small as possible. This includes extra processing in order to find optimal encoder settings. transparency: This option controls what color image to mark as transparent. dpi: tuple of two scalars The desired dpi in each direction. pnginfo: PIL.PngImagePlugin.PngInfo Object containing text tags. compress_level: int ZLIB compression level, a number between 0 and 9: 1 gives best speed, 9 gives best compression, 0 gives no compression at all. Default is 9. When ``optimize`` option is True ``compress_level`` has no effect (it is set to 9 regardless of a value passed). compression: int Compatibility with the freeimage PNG format. If given, it overrides compress_level. icc_profile: The ICC Profile to include in the saved file. bits (experimental): int This option controls how many bits to store. If omitted, the PNG writer uses 8 bits (256 colors). quantize: Compatibility with the freeimage PNG format. If given, it overrides bits. In this case, given as a number between 1-256. dictionary (experimental): dict Set the ZLIB encoder dictionary. prefer_uint8: bool Let the PNG writer truncate uint16 image arrays to uint8 if their values fall within the range [0, 255]. Defaults to true for legacy compatibility, however it is recommended to set this to false to avoid unexpected behavior when saving e.g. weakly saturated images. R4cB`s#eZdeedZdZRS(cC`stjj|d|d|S(NRIR6(RR4RJ(RRIR6t ignoregamma((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyRJ+scC`stjj||\}}|jjjdtsyt|d}Wntt fk r`qXt|j t j kr|dnd}d}|||||d|(n||fS(NRvtgammaiig?gt_getexiftitemsRxt_rotate(RRUR[RZRttagtvaluetdecoded((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyR\s  "cC`s|jjjdtry|dd}Wntk r=qX|d krMn|d krntj|d}n|dkrtj|d}n|dkrtj|}n|dkrtj|}qn|S(s Use Orientation information from EXIF meta data to orient the image correctly. Similar code as in FreeImage plugin. RRt Orientationiiiiiiii(ii(ii(ii(ii(iiii(R,RRxRR:R|trot90tfliplr(RR[Rmtori((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyRs       N( R^R_R"RKRRJR<R\R(((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyR4s R`cB`s#eZdeedZdZRS(iKcK`st|}|dks$|dkr3tdntdtd|}||dRRRxR(RRUR[RZRRRR((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyR\Cs  "cC`s|jjjdtry|dd}Wntk r=qX|d krMn|d krntj|d}n|dkrtj|d}n|dkrtj|}n|dkrtj|}qn|S(s Use Orientation information from EXIF meta data to orient the image correctly. Similar code as in FreeImage plugin. RRRiiiiiiii(ii(ii(ii(ii(iiii(R,RRxRR:R|RR(RR[RmR((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyRRs       N(R^R_R"RKRJR<R\R(((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyR44s R`cB`s eZdddZdZRS(tratesicK`s|ddhkr!tdnt|}|dkri|dksQ|dkritdj|n<|dkr|dks|dkrtd j|n||d <|g|d t|dr:tt|dddr:|jq:ndS(NROtfp(RtgetattrR"RO(R[((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyRNscC`sjy|jdWnQtk re}d}|d7}t|}d|j||f}t|nXdS(Nis8http://pillow.readthedocs.io/en/latest/installation.htmls#external-librariessACould not load "%s" Reason: "%s" Please see documentation at: %s(RXRRRRz(R[tetsitetpillow_error_messaget error_message((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyRDs  cC`s|jdkrtS|jjddr,tStj|jjd}|j \}}|||d!}tj tj |dS(NtPRiiii(ii( R(RKRZRxR"R|tasarrayt getpalettetreshapet getextrematallclosetdiff(t pil_imageR@tstarttstopt valid_palette((s5lib/python2.7/site-packages/imageio/plugins/pillow.pyRFsc C`s|d krt|}n|}|d k rQ||jkr|j|}qn|rZn|jdkr|r|jd}no|jdkr|jjdd d k r|jd}q|jjd krtj|jj dtj }t |jdr|jj |j_ n|jj r0|jj n |jj}t|}d|f|_|jdd ks|jdd kr|dd krtj|d d d d fd tj|jd|jf}n|jdr;|jdd kr|d d dddgfn|d d dddd gf}ntj|tj }y||}Wqtk r|jd}qXqtr|jd}q|jd}nBd|jkr|jd}n!|jdkr|jd}n|r|jd}n4t|tj r<|jdkr<|jd}n|jjdr|j} |jjdrodnd}d|jkr|jdd}ntj|j|j}| d d d|_nH|jdkr|jdkr|d krd}ntj|d|}|S(!s is_gray: Whether the image *is* gray (by inspecting its palette). as_gray: Whether the resulting image must be converted to gaey. mode: The mode to convert to. RtLRtRGBAtRGBiRCiiitXNiitBGRitAtCMYKtFt1sI;16tBs>u2ss8   #   a.