U &^'@sddlZddlZddlmZddlmZmZddlmZm Z m Z m Z ddl m Z mZmZdZdd Zd d ZGd d d ZGddde j Ze ejeee ejee ejde ejddS)N)BytesIO)ceillog)BmpImagePluginImage ImageFilePngImagePlugin)i8i16lei32lesc s|t|jddddddddg}|j\tfd d |}t|}|td t || t |d }|D]}|\|td dkrnd|td dkrnd|d|d|d|td dt }| }|j |tjdd||d|d|}t |} |td| |td|| } |||||| }|| qzdS)Nsizes)r)r) r)0r)@r)r)rcs8|dks0|dks0|ddks0|ddkr4dSdS)NrrrFTxheightwidthr6/tmp/pip-install-a1j0c_p1/Pillow/PIL/IcoImagePlugin.py/s    z_save..z.)keycSs|dS)NrCrrrrrrrD)r,r6 SyntaxErrorbufentryi16Znb_itemsranger i32rrappendsortedreverse)selfrGsiZ icon_headerjrrr__init__Ws@          zIcoFile.__init__cCsdd|jDS)zJ Get a list of all available icon sizes and color depths. cSsh|]}|d|dfqSr@r).0hrrr sz IcoFile.sizes..)rHrOrrrr sz IcoFile.sizesFcCs@t|jD]0\}}||dkr |dks2||dkr |Sq dS)NrBFrAr) enumeraterH)rOr"r?rQrUrrr getentryindexs  zIcoFile.getentryindexcCs||||S)z, Get an image from the icon )framerY)rOr"r?rrrgetimageszIcoFile.getimagec Cs|j|}|j|d|jd}|j|d|ddtjkrXt|j}nt|j}t |j |j dt |j ddf|_ |jd\}}}}|d|j ||f|jd<|d} d} tjD]\} } | | dkr| } qqd| krB|j||j|j d|j dd d dd } t d |j | d d }n|j d}|ddkrp|d|j dd7}|t |j d|j d| d}t ||j dd}|j||j|}t d|j |d dt |ddf}|d}|||S)z- Get an image from frame idx r0r;Nrrr9)rrrr4r:Lraw)r\rg @1z1;Ir^ZRGBA)rHrGr+r,r r Z PngImageFilerZ DibImageFilerZ_decompression_bomb_checkr"int_sizeZtileZBIT2MODEitemsZ frombufferconvertZputalpha)rOidxheaderdatar-deoamoder?kvZ alpha_bytesmaskwZand_mask_offset total_bytesZ mask_datarrrrZsX       * $    z IcoFile.frameN)F)F)__name__ __module__ __qualname__rSr rYr[rZrrrrr7Vs :  r7c@sJeZdZdZdZdZddZeddZej ddZd d Z d d Z d S) IcoImageFilea PIL read-only image support for Microsoft Windows .ico files. By default the largest resolution image in the file will be loaded. This can be changed by altering the 'size' attribute before calling 'load'. The info dictionary has a key 'sizes' that is a list of the sizes available in the icon file. Handles classic, XP and Vista icon formats. When saving, PNG compression is used. Support for this was only added in Windows Vista. This plugin is a refactored version of Win32IconImagePlugin by Bryan Davis . https://code.google.com/archive/p/casadebender/wikis/Win32IconImagePlugin.wiki ZICOz Windows IconcCs:t|j|_|j|jd<|jjdd|_|dS)Nr rrB)r7r.icor inforHr"loadrWrrr_opens zIcoImageFile._opencCs|jSN)rarWrrrr"szIcoImageFile.sizecCs ||jdkrtd||_dS)Nr z2This is not one of the allowed sizes of this image)rv ValueErrorra)rOvaluerrrr"scCs|jr|jj|jkrdS|j|j}||j|_|j|_|j|jkrtd|j|j}t |j d}|j||<t ||j d<|j|_dS)NzImage was not the expected sizer ) r-r"rur[rwrkwarningswarnrYr$rvset)rOr-indexr rrrrw#s   zIcoImageFile.loadcCsdSryrrWrrr load_seek6szIcoImageFile.load_seekN) rqrrrs__doc__formatformat_descriptionrxpropertyr"setterrwrrrrrrts  rtz.icoz image/x-icon)r%r|iormathrrrrrr _binaryr r rIr rKr r3r6r7rtZ register_openrZ register_saveZregister_extensionZ register_mimerrrrs +'D