U ڲg .@sddlmZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZddlmZdd lmZdd lmZdd lmZdd lmZd ZdddddddZdddddZGdddZGddde j Ze ejeee ejee !ejde "ejddS) ) annotationsN)BytesIO)ceillog)IO)BmpImagePluginImage ImageFilePngImagePlugin)i16le)i32le)o8)o16le)o32les Image.Imagez IO[bytes]z str | bytesNone)imfpfilenamereturnc Cs|t|jddk}|jddddddd d g}g}|g|jd g}|j\}}tt|D]} | d |ksd| d |ksd| d dksd| d dkrqd|D]|} | j| krq|| |rtj | j d } | g} |D]<} | j| krqtj | j d } | | kr|| | | qqdq| }|j | t jjdd||qd|tt||t|d}|D]}|j\}}|t|dkr|nd |t|dkr|nd |rtj |j d dnd\} }|t||d|d|t| t}|rT||d| dkr`t d| }t||dd| d dfgn ||d|d |}|r|ddt|d|dd}t|}|t||t||}||||||}||qndS)NZ bitmap_formatbmpsizes)r)r) r)0r)@r)r)rZ append_imagesrrr)Z reducing_gapr)rrsZdibr1rawrr)r!rZpng )write_MAGICZ encoderinfogetsizesortedsetappendrZSAVEmodecopyZ thumbnailr Z ResamplingZLANCZOSo16lentellrrsavenewr _saveseekreado32)rrrrrframesZ provided_imswidthheightr+Z provided_imbitsZ bits_usedZother_imframeoffsetcolorsZimage_ioZand_maskZ image_bytesZ bytes_lencurrentrB6/tmp/pip-unpacked-wheel-xela2va7/PIL/IcoImagePlugin.pyr6+s  0        "       $  r6bytesbool)prefixrcCs|ddtkS)N)r))rFrBrBrC_acceptvsrHc@s@eZdZddZddZdddZddd Zd d d d dZdS)IcoFilec Cs8|d}t|sd}t|||_g|_t|d|_t|jD]}|d}|d|d|d|dt|dt|dt|d t|d d }d D]}||sd ||<q|dp|ddkrt t |ddpd |d<|d|df|d<|d|d|d<|j |q@t |jddd|_t |jdddd|_dS)zL Parse image from file-like object containing ico file data znot an ICO filerGrrrr&r%r')r;r<nb_colorreservedZplanesbppr+r?r;r<rrNrL color_depthr;r<dimsquarecSs|dS)NrPrBxrBrBrCz"IcoFile.__init__..)keycSs|dS)NrRrBrSrBrBrCrUrVT)rWreverseN) r8rH SyntaxErrorbufentryi16Znb_itemsrangei32rrr.r,)selfrZsmsgiZ icon_headerjrBrBrC__init__{s@       zIcoFile.__init__cCsdd|jDS)zJ Get a list of all available icon sizes and color depths. cSsh|]}|d|dfqSrOrB).0hrBrBrC sz IcoFile.sizes..)r[r_rBrBrCrsz IcoFile.sizesFcCs@t|jD]0\}}||dkr |dks2||dkr |Sq dS)NrQFrPr) enumerater[)r_r+rNrbrfrBrBrC getentryindexs  zIcoFile.getentryindexcCs||||S)z, Get an image from the icon )r>rj)r_r+rNrBrBrCgetimageszIcoFile.getimageintr)idxrc Cs|j|}|j|d|jd}|j|d|ddtjkrdt|j}t|j ndt |j}t|j |j dt |j ddf|_ |jd\}}}}|d|j ||f|jd<|d} d | kr"|j||j|j d|j dd d dd } td |j | d d} n|j d} | d dkrP| d |j dd 7} t | |j dd} |d|d| }|j||j| }td|j |d dt | ddf} |d}|| |S)z- Get an image from frame idx r?r%Nrrr&r#rNrrGrKLr")rnrr$r+r!z1;Ir$ZRGBA)r[rZr7r8r r)Z PngImageFiler Z_decompression_bomb_checkr+rZ DibImageFilerl_sizeZtileZ frombufferconvertZputalpha)r_rmheaderdatardeoarNZ alpha_bytesmaskw total_bytesZand_mask_offsetZ mask_datarBrBrCr>sP       *     z IcoFile.frameN)F)F)__name__ __module__ __qualname__rdrrjrkr>rBrBrBrCrIzs 9  rIc@sXeZdZdZdZdZddddZedd Zej d d Zd d Z d ddddZ dS) 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. If you are unable to view the icon in Windows, convert the image to "RGBA" mode before saving. This plugin is a refactored version of Win32IconImagePlugin by Bryan Davis . https://code.google.com/archive/p/casadebender/wikis/Win32IconImagePlugin.wiki ZICOz Windows Iconr)rcCs:t|j|_|j|jd<|jjdd|_|dS)NrrrQ)rIricorinfor[r+loadrhrBrBrC_open3s zIcoImageFile._opencCs|jSN)rorhrBrBrCr+9szIcoImageFile.sizecCs$||jdkrd}t|||_dS)Nrz2This is not one of the allowed sizes of this image)r ValueErrorro)r_valuerarBrBrCr+=scCs|jdk r$|jj|jkr$tj|S|j|j}||j|_d|_|j|_|j r^|j |_ |j|jkrt d|j |j}t |jd}|j||<t||jd<|j|_dS)NzImage was not the expected sizer)rr+r rr~rkZpyaccessr/_modeZpalettewarningswarnrjlistrr-)r_rindexrrBrBrCrDs     zIcoImageFile.loadrl)posrcCsdSrrB)r_rrBrBrC load_seekZszIcoImageFile.load_seekN) rzr{r|__doc__formatformat_descriptionrpropertyr+setterrrrBrBrBrCr}s  r}z.icoz image/x-icon)# __future__rriormathrrtypingrrr r r _binaryr r\r r^rrr1rr9r)r6rHrIr}Z register_openrZ register_saveZregister_extensionZ register_mimerBrBrBrCs(        K"I