;c]c@sdZddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z mZmZmZmZmZmZmZmZe jeZeZidd6dd6d d 6d d 6d d6dd6ZeedZdZdZe j dddde j!Z"de#fdYZ$de#fdYZ%de#fdYZ&de#fdYZ'de#fd YZ(d!e#fd"YZ)d#e)fd$YZ*d%e)fd&YZ+e(Z,dS('s Module imageio/freeimage.py This module contains the wrapper code for the freeimage library. The functions defined in this module are relatively thin; just thin enough so that arguments and results are native Python/numpy data types. i(tabsolute_importtprint_functiontwith_statementNi( tget_remote_filetload_libtDictt resource_dirst string_typest binary_typetIS_PYPYt get_platformtInternetNotAllowedErrortNeedDownloadErrors!libfreeimage-3.16.0-osx10.6.dylibtosx32tosx64sFreeImage-3.15.4-win32.dlltwin32sFreeImage-3.15.1-win64.dlltwin64slibfreeimage-3.16.0-linux32.sotlinux32slibfreeimage-3.16.0-linux64.sotlinux64cCsRt}|rN|tkrNdt|}td|d|d|dt_ndS(s Download the FreeImage library to your computer. Parameters ---------- directory : str | None The directory where the file will be cached if a download was required to obtain the file. By default, the appdata directory is used. This is also the first directory that is checked for a local version of the file. force_download : bool | str If True, the file will be downloaded even if a local copy exists (and this copy will be overwritten). Can also be a YYYY-MM-DD date to ensure a file is up-to-date (modified date of a file on disk, if present, is checked). s freeimage/tfnamet directorytforce_downloadN(R tFNAME_PER_PLATFORMRtNonetfit_lib(RRtplatR((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytdownload3s  cCstjdd}|r|St}|r|tkrytdt|dtSWqtk rfqtk rtdqt k r}t j t |qXndS(s= Ensure we have our version of the binary freeimage lib. tIMAGEIO_FREEIMAGE_LIBs freeimage/tautosNeed FreeImage library. You can obtain it with either: - download using the command: imageio_download_bin freeimage - download by calling (in Python): imageio.plugins.freeimage.download() N( tostgetenvRR RRtFalseR R t RuntimeErrortloggertwarningtstr(tlibRte((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytget_freeimage_libJs    cCs|jtjS(N(tencodetsystgetfilesystemencoding(tx((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pythtiitdtypetFI_TYPEScBseZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zi eje6eje6eje6eje6eje6eje6eje 6eje 6eje 6eje 6eje 6eje6Zieejdf6eejdf6eejdf6eejdf6eejdf6eejdf6eejdf6eejdf6e ejdf6e ejdf6e ejdf6e ejdf6e ejdf6eejdf6Zi ge6ge6ge6ge6ge6ge 6ge 6dge 6dge 6dge 6dge6ZRS( iiiiiiiiii i i i (t__name__t __module__t FIT_UNKNOWNt FIT_BITMAPt FIT_UINT16t FIT_INT16t FIT_UINT32t FIT_INT32t FIT_FLOATt FIT_DOUBLEt FIT_COMPLEXt FIT_RGB16t FIT_RGBA16tFIT_RGBFt FIT_RGBAFtnumpytuint8tuint16tint16tuint32tint32tfloat32tfloat64t complex128tdtypestfi_typest extra_dims(((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyR/nsj               tIO_FLAGScBseZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZd Zd Zd Z d Z!d Z"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6dZ7dZ8d Z9d Z:dZ;dZ<dZ=dZ>dZ?dZ@dZAdZBdZCdZDdZEdZFdZGdZHdZIdZJd ZKd ZLd ZMd ZNdZOdZPdZQdZRdZSdZTdZUdZVRS(iiiiiiii i@iiiiii ii@iiiiii (WR0R1tFIF_LOAD_NOPIXELSt BMP_DEFAULTt BMP_SAVE_RLEt CUT_DEFAULTt DDS_DEFAULTt EXR_DEFAULTt EXR_FLOATtEXR_NONEtEXR_ZIPtEXR_PIZt EXR_PXR24tEXR_B44tEXR_LCt FAXG3_DEFAULTt GIF_DEFAULTt GIF_LOAD256t GIF_PLAYBACKt HDR_DEFAULTt ICO_DEFAULTt ICO_MAKEALPHAt IFF_DEFAULTt J2K_DEFAULTt JP2_DEFAULTt JPEG_DEFAULTt JPEG_FASTt JPEG_ACCURATEt JPEG_CMYKtJPEG_EXIFROTATEtJPEG_QUALITYSUPERBtJPEG_QUALITYGOODtJPEG_QUALITYNORMALtJPEG_QUALITYAVERAGEtJPEG_QUALITYBADtJPEG_PROGRESSIVEtJPEG_SUBSAMPLING_411tJPEG_SUBSAMPLING_420tJPEG_SUBSAMPLING_422tJPEG_SUBSAMPLING_444t JPEG_OPTIMIZEt JPEG_BASELINEt KOALA_DEFAULTt LBM_DEFAULTt MNG_DEFAULTt PCD_DEFAULTtPCD_BASEt PCD_BASEDIV4t PCD_BASEDIV16t PCX_DEFAULTt PFM_DEFAULTt PICT_DEFAULTt PNG_DEFAULTtPNG_IGNOREGAMMAtPNG_Z_BEST_SPEEDtPNG_Z_DEFAULT_COMPRESSIONtPNG_Z_BEST_COMPRESSIONtPNG_Z_NO_COMPRESSIONtPNG_INTERLACEDt PNM_DEFAULTt PNM_SAVE_RAWtPNM_SAVE_ASCIIt PSD_DEFAULTtPSD_CMYKtPSD_LABt RAS_DEFAULTt RAW_DEFAULTt RAW_PREVIEWt RAW_DISPLAYt SGI_DEFAULTt TARGA_DEFAULTtTARGA_LOAD_RGB888tTARGA_SAVE_RLEt TIFF_DEFAULTt TIFF_CMYKt TIFF_PACKBITSt TIFF_DEFLATEtTIFF_ADOBE_DEFLATEt TIFF_NONEtTIFF_CCITTFAX3tTIFF_CCITTFAX4tTIFF_LZWt TIFF_JPEGt TIFF_LOGLUVt WBMP_DEFAULTt XBM_DEFAULTt XPM_DEFAULT(((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRKstMETADATA_MODELScBsDeZdZdZdZdZdZdZdZdZ dZ d Z RS( iiiiiiiiii ( R0R1t FIMD_COMMENTStFIMD_EXIF_MAINtFIMD_EXIF_EXIFt FIMD_EXIF_GPStFIMD_EXIF_MAKERNOTEtFIMD_EXIF_INTEROPt FIMD_IPTCtFIMD_XMPt FIMD_GEOTIFFtFIMD_ANIMATION(((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRstMETADATA_DATATYPEcBskeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZieje6eje6eje6dejfdejfge6eje6eje6eje6eje6eje6eje 6eje 6dejfdejfge 6eje 6eje 6eje6dejfdejfdejfdejfge6ZRS(iiiiiiiii i i i i iiiit numeratort denominatortRtGtBtA(R0R1t FIDT_BYTEt FIDT_ASCIIt FIDT_SHORTt FIDT_LONGt FIDT_RATIONALt FIDT_SBYTEtFIDT_UNDEFINEDt FIDT_SSHORTt FIDT_SLONGtFIDT_SRATIONALt FIDT_FLOATt FIDT_DOUBLEtFIDT_IFDt FIDT_PALETTEt FIDT_LONG8t FIDT_SLONG8t FIDT_IFD8R?R@RARCtuint64tint64tint8RBRDRERFRH(((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyR&sJ                t FreeimagecBs:eZdZi)ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd 6ejd:fd 6ejd:fd 6ejd:fd 6ejd:fd 6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd6ejd:fd 6ejd:fd!6ejd:fd"6ejd:fd#6ejd:fd$6ejd:fd%6ejd:fd&6ejd:fd'6ejd:fd(6ejd:fd)6Zd*Z e d+Z d,Z d-Z d.Zd/Zd0Zd1Zd2Zd3Zd4Zd5Zd:d6Zd7d8Zd7d9ZRS(;s Class to represent an interface to the FreeImage library. This class is relatively thin. It provides a Pythonic API that converts Freeimage objects to Python objects, but that's about it. The actual implementation should be provided by the plugins. The recommended way to call into the Freeimage library (so that errors and warnings show up in the right moment) is to use this object as a context manager: with imageio.fi as lib: lib.FreeImage_GetPalette() tFreeImage_AllocateTtFreeImage_FindFirstMetadatatFreeImage_GetBitstFreeImage_GetPalettetFreeImage_GetTagKeytFreeImage_GetTagValuetFreeImage_CreateTagtFreeImage_SavetFreeImage_LoadtFreeImage_LoadFromMemorytFreeImage_OpenMultiBitmapt#FreeImage_LoadMultiBitmapFromMemorytFreeImage_LockPagetFreeImage_OpenMemorytFreeImage_GetVersiontFreeImage_GetFIFExtensionListtFreeImage_GetFormatFromFIFtFreeImage_GetFIFDescriptiontFreeImage_ColorQuantizeExtFreeImage_IsLittleEndiantFreeImage_SetOutputMessagetFreeImage_GetFIFCounttFreeImage_IsPluginEnabledtFreeImage_GetFileTypetFreeImage_GetTagTypetFreeImage_GetTagLengthtFreeImage_FindNextMetadatatFreeImage_FindCloseMetadatatFreeImage_GetFIFFromFilenametFreeImage_FIFSupportsReadingtFreeImage_FIFSupportsWritingtFreeImage_FIFSupportsExportTypetFreeImage_FIFSupportsExportBPPtFreeImage_GetHeighttFreeImage_GetWidthtFreeImage_GetImageTypetFreeImage_GetBPPtFreeImage_GetColorsUsedtFreeImage_ConvertTo32BitstFreeImage_GetPitchtFreeImage_Unloadcs|d_tj_g_tjjdr?t j }n t j }|dt j t j fd}|_dS(NtwincsO|jd}jj|x)tjdkrJjjdq"WdS(Nsutf-8ii(tdecodet _messagestappendtlentpop(tfiftmessage(tself(s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt error_handlers(RRt threadingtRLockt_lockRR)tplatformt startswithtctypest WINFUNCTYPEt CFUNCTYPEtc_inttc_char_pt_error_handler(RtfunctypeR((Rs9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt__init__s    'cCs|jdkrTy|jWqTtk rP}d|_|jt|7_qTXnt|jtrxt|jn|jS(Ns+The freeimage library could not be loaded: (RRtload_freeimagetOSErrorR$t isinstanceR!(Rterr((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyR%s cCs$y |jWntk rtSXtS(N(R%t ExceptionR tTrue(R((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pythas_libs   cCst}y?|j|j|jjjddkrDt}nWntk rXnX|s}t|j|jn|jj |j |jjjd|_ dS(s Try to load the freeimage lib from the system. If not successful, try to download the imageio version and try again. sutf-8s3.15N( R t_load_freeimaget _register_apiR%RRR R R'RRt lib_version(Rtsuccess((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRs      c Cs ddg}ddddg}t}t}|rvt|}x3|D](}|jdtjj|d|qGWntjdd}|dk r|jd|nyt |||\}}Wn/t k r}t |d } t | nX||_ ||_ dS( Nt freeimaget libfreeimaget FreeImageslibfreeimage.dylibslibfreeimage.soslibfreeimage.so.3iRs& Please install the FreeImage library.(RR RtinsertRtpathtjoinRRRR R$Rt lib_fname( Rt lib_namestexact_lib_namestres_dirsRRtdirR%R terr_msg((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRs*      )  cCsNxG|jjD]6\}\}}t|j|}||_||_qWdS(N(t_APItitemstgetattrR%trestypetargtypes(RtfR"R#tfunc((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRs" cCs|jj|jS(N(RtacquireR%(R((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt __enter__s cGs|j|jjdS(N(t_show_any_warningsRtrelease(Rtargs((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt__exit__s cCs g|_dS(sy Reset the list of output messages. Call this before loading or saving an image with the FreeImage API. N(R(R((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt _reset_logscCs1|jr)dj|j}|j|SdSdS(s Get the output messages produced since the last reset as one string. Returns 'No known reason.' if there are no messages. Also resets the log. t sNo known reason.N(RRR,(Rtres((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt_get_error_messages   cCs1|jr-tjd|j|jndS(s If there were any messages since the last reset, show them as a warning. Otherwise do nothing. Also resets the messages. simageio.freeimage warning: N(RR"R#R/R,(R((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyR(+s cCsg|jD] }|^q S(sy Return a list of the last 256 output messages (warnings and errors) produced by the FreeImage library. (R(Rtm((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytget_output_log3scCs||}d}|dkr*tdn|dkr|d k r|jtj|t|}|jtj|t|}|jtj|n|dkrt j j |r|j t |d}qn|dkr|jt |}n|dkrtd|n^|dkrK|j| rKtd|n/|dkrz|j| rztd |n|SWd QXd S( s Get the freeimage Format (FIF) from a given filename. If mode is 'r', will try to determine the format by reading the file, otherwise only the filename is used. This function also tests whether the format supports reading/writing. itrws"Invalid mode (must be "r" or "w").tris$Cannot determine format of file "%s"tws$Cannot write the format of file "%s"s#Cannot read the format of file "%s"N(t ValueErrorRRRRRtFreeImage_GetFileTypeFromMemorytc_void_ptFreeImage_CloseMemoryRRtisfileRtefnRRR(RtfilenametmodetbytesR%tftypetfimemory((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytgetFIF:s,      icCst||||S(sY create_bitmap(filename, ftype, flags=0) Create a wrapped bitmap object. (tFIBitmap(RR;R>tflags((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt create_bitmapcscCst||||S(sm create_multipage_bitmap(filename, ftype, flags=0) Create a wrapped multipage bitmap object. (tFIMultipageBitmap(RR;R>RB((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytcreate_multipage_bitmapisN(R0R1t__doc__RR7RRRRRtpropertyR%RRRRR'R+R,R/R(R1R@RCRE(((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRRst      "      ) t FIBaseBitmapcBsAeZdZdZdZddZdZdZRS(cCs:||_||_||_||_d|_g|_dS(N(t_fit _filenamet_ftypet_flagsRt_bitmapt _close_funcs(RRR;R>RB((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRqs      cCs|jdS(N(tclose(R((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt__del__yscCs|jdk r|jrxQ|jD]F}y,|j|d}||dWdQXWq"tk rgq"Xq"Wg|_d|_ndS(Nii(RMRRNRIR (Rt close_functfun((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRO|s    cCs\|jdk rn|dkr6|jjj|f}n||_|rX|jj|ndS(sK Function to set the bitmap and specify the function to unload it. N(RMRRIR%RRNR(RtbitmapRQ((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt _set_bitmaps  c Cs@gtjjD]+\}}|jdr|d|f^q}t}tj}|j}x|D]\}}|j||j tj |}tj|}|rit } xj| r|j |j d} |j|} |j|} tj| } | j|j|}tt|}|}| tjkrQ|j dd}n| tjkrtj| }trt|ttfrqy>tj|d|j}t|dkr|d}nWqt k rqXn|j!|t}||| <|j"|tj |} qW|j#|qiqiW|SWdQXdS(NtFIMD_isutf-8treplaceR.ii($Rt__dict__R RRRR7RIRRMtbyrefR RRRRtc_chart from_addressRRt bytearrayRRRHR R tlistttupleR?t frombuffertcopyRR t setdefaultRR(RtnametnumbertmodelstmetadatattagR%t model_nametmdhandletmorettag_namettag_typet byte_sizetchar_ptrtdatat tag_bytesttag_valR.tsubdict((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt get_meta_datasN%        c Csi}x=tjjD],\}}|jdr|||dimageio.freeimage warning: Could not determine tag type of %r.sutf-8s6imagio.freeimage warning: Could not set tag %r: %s, %s('RRWR RRItgetRRRR7R R RR(R t UnicodeErrorRRRthasattrR?tarrayR.R"R#ttostringtsizetFreeImage_SetTagKeytFreeImage_SetTagTypetFreeImage_SetTagCounttFreeImage_SetTagLengthtFreeImage_SetTagValueRtFreeImage_SetMetadataRMR R/R$tFreeImage_DeleteTag(RRdRcRaRbRsR%RfRpRiRoRetis_asciiRnRjt tag_countttag_keyR ((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyt set_meta_datas^          (N( R0R1RRPRORRTRqR(((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRHps    GRAcBskeZdZdZd dZd dZdZdZdZ dZ dZ d d d Z RS( s' Wrapper for the FI bitmap object. c CsSt|tjst|j}|j}|d \}}t|dkrUd}n)t|dkrt|d}n |d}y#tj|j |f}||_ Wnt k rt dnX|j }d|j|} |j|||| ddd} tj| } | s0td|j jn|j| |j| fWdQXdS(Niiiis,Cannot write arrays of given type and shape.is)Could not allocate bitmap for storage: %s(R R?tndarraytAssertionErrortshapeR.RR/RIttypet_fi_typetKeyErrorR5RItitemsizeRRR7R!R/RTR( RRxRR.R3tct n_channelstfi_typeR%tbppRS((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytallocate*s.        !cCs|dkr|j}n|j{}|j|jt||j}tj|}|st d|j|jj fn|j ||j |fWdQXdS(NsCould not load bitmap "%s": %s( RRJRIRRKR:RLRR7R5R/RTR(RR;R%RS((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytload_from_filenameOs   !c Cs|dkr|j}n|j}|j}|j}|j}|tjkrl|j||j |}n|j ||}|st dn|j ||t ||j}|std|j|jjfnWdQXdS(Ns2Cannot save image of this format to this file typesCould not save file "%s": %s(RRJRKRMRRIR/R3RRRt TypeErrorRR:RLR!R/(RR;R>RSRR%t can_writeR.((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytsave_to_filenameqs"      !c Cs3|j\}}}|j||t}|j}|j}WdQXd}t|dkr|r|jtjkr||d}||d} ||d} |ddkrtj | | |fS|ddkr ||d} tj | | || fSt d|n||j } | S(NcSs|jdkr#|dddjS|jdkrR|dddddfjS|jdkr|dddddddfjS|jdkr|dddddddddfjSdS(Niiiii(tndimtT(tarr((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytns )iiiiis Cannot handle images of shape %s( t_get_type_and_shapet_wrap_bitmap_bits_in_arrayR RIRRRR?R@tdstackR5R_( RR.RRRxR%tisleRtbtgR3ta((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pytget_image_datas"  *cCst|tjst|j}|j}|j}|j}WdQX|d \}}t|dkrd}||f} n8t|dkr|d}|||f} n |d}d} |j | |t } t|dkr|r|j tj kr|dddddf} |dddddf} |dddddf}| || d<| | | d<| | | d<|ddkr|dddddf}| || dt |_ t rt nt j|d|d| d| SWngt k rH|rt|_ t j|d|Stt| } t j| d|j} t|dkr|d| d|d|df| _| d|dd|d d|df}t j|d| j} xt|dD],}|dddd|fj| ||d | df| _| d|d d|dfj} | SnXdS( s%Return an ndarray view on the data in a FreeImage bitmap. Only valid for as long as the bitmap is loaded (if single page) / locked in memory (if multipage). This is used in loading data, but also during saving, to prepare a strided numpy array buffer. NiiiR.tbuffertstridesii(RIRRMRRRRRYRZR RtTEST_NUMPY_NO_STRIDEStNotImplementedErrorR?RR tzerosRR[R^R_RR.trangeR(RRR.tsaveR%tpitchtbitstheightRkRRRmR=Rxtarray2ti((s9lib/python2.7/site-packages/imageio/plugins/_freeimage.pyRs<       #  &1-'c Cs|j=}|j|j}|j|j}|j|j}WdQX|d|j}||}||jd}d|kodknst|jd||f} tj | |j } |dkr|j | ddd|jddf s@      F    >m ,y