U Ev f<@sddZddlmZddlZddlZddlmZddlmZddl m Z m Z Gdd d eZ Gd d d eZ Gd d d eZddZd+ddZddZddZGdddeZddZGddde j ZGddde jZGdd d eZGd!d"d"eZGd#d$d$e jZd%d&Ze ejeee ejd'e !d(ee !d)ee "ejee #d*edS),a Blizzard Mipmap Format (.blp) Jerome Leclanche The contents of this file are hereby released in the public domain (CC0) Full text of the CC0 license: https://creativecommons.org/publicdomain/zero/1.0/ BLP1 files, used mostly in Warcraft III, are not fully supported. All types of BLP2 files used in World of Warcraft are supported. The BLP file structure consists of a header, up to 16 mipmaps of the texture Texture sizes must be powers of two, though the two dimensions do not have to be equal; 512x256 is valid, but 512x200 is not. The first mipmap (mipmap #0) is the full size image; each subsequent mipmap halves both dimensions. The final mipmap should be 1x1. BLP files come in many different flavours: * JPEG-compressed (type == 0) - only supported for BLP1. * RAW images (type == 1, encoding == 1). Each mipmap is stored as an array of 8-bit values, one per pixel, left to right, top to bottom. Each value is an index to the palette. * DXT-compressed (type == 1, encoding == 2): - DXT1 compression is used if alpha_encoding == 0. - An additional alpha bit is used if alpha_depth == 1. - DXT3 compression is used if alpha_encoding == 1. - DXT5 compression is used if alpha_encoding == 7. ) annotationsN)IntEnum)BytesIO)Image ImageFilec@seZdZdZdS)FormatrN)__name__ __module__ __qualname__JPEGr r 6/tmp/pip-unpacked-wheel-ciywl2yd/PIL/BlpImagePlugin.pyr*src@seZdZdZdZdZdS)EncodingrN)r r r UNCOMPRESSEDDXTZUNCOMPRESSED_RAW_BGRAr r r rr.src@seZdZdZdZdZdS) AlphaEncodingrrN)r r r DXT1DXT3DXT5r r r rr4srcCs*|d?d@d>|d?d@d>|d@d>fS)N r?rr )ir r r unpack_565:srFc Cst|d}ttttf}t|D]}|d}td||\}}}t|\} } } t|\} } }tdD]L}tdD]<}|d@}|d?}d}|dkr| | | }}}n|dkr| | |}}}n|dkr6||krd| | d}d| | d}d| |d}n$| | d}| | d}| |d}nR|dkr||kr|d| | d}d| | d}d|| d}n d \}}}}|r||||||gq||||||gq|qnq(|S) E input: one "row" of data (i.e. will produce 4*width pixels) zsD        r?c Cst|d}ttttf}t|D]}|d}|||d}td|}td|d\}}td|d\}t|\} } } t|\} } }tdD]2}d}tdD]}d||d }||}|rd}|dL}n d }|d M}|d 9}|d d||?d @}|dkr"| | | }}}n|dkr>| | |}}}nv|d krzd | | d }d | | d }d | |d }n:|d krd | | d }d | | d }d || d }||||||gqqq(|S)rz<8Br r r decode_dxt3usH       rJc Cst|d}ttttf}t|D]}|d}|||d}td|\}}td|d}|d|dd>B|dd>B|dd >B}|d |d d>B} td |d\} } td |d\} t| \} }}t| \}}}tdD]}tdD]}dd||}|dkr&| |?d@}n0|dkrF| d?|d >d@B}n||d?d@}|d krf|}nt|d krv|}nd||krd|||d |d}n<|dkrd }n,|dkrd}nd|||d |d}| dd||?d@}|d kr| ||}}}n|d kr*|||}}}nv|dkrfd| |d}d||d}d||d}n:|dkrd|| d}d||d}d||d}||||||gqqq(|S)zG input: one "row" of data (i.e. will produce 4 * width pixels) r@zr r r decode_dxt5s^ ,           rMc@s eZdZdS)BLPFormatErrorN)r r r r r r rrNsrNcCs|dddkS)Nr!BLP1BLP2r )prefixr r r_acceptsrSc@s eZdZdZdZdZddZdS) BlpImageFilez Blizzard Mipmap Format BLPzBlizzard Mipmap FormatcCs|jd|_|jdtjtd|jd\|_|jdtjtd|jd|_ |jdkrr|j }nd t |j}t ||jrd nd |_ |d |jd |jd dffg|_dS)Nr!rr=r<r;r r r _read_palette5s  z_BLPBaseDecoder._read_palettec Cst}t||jd}ztd|d\}Wntjk rLYqYnX||\}}}}|||f} |jrx| |f7} | | q|S)Nrr=r<r;dr r r _read_bgra?s    z_BLPBaseDecoder._read_bgraN) r r r Z _pulls_fdrdrsr}rrr r r rrqs   rqc@seZdZddZddZdS)rcCs~|jtjkr|nd|jdkrb|jdkrH|}||}||qzdt|j}t |ndt|j}t |dS)Nr)r!rzUnsupported BLP encoding zUnsupported BLP compression ) r~rr _decode_jpeg_streamrrr set_as_rawrerN)rjrr*rlr r rrtPs       zBLP1Decoder._loadc Csddlm}td|d\}||}||jd|j||jd}||}t |}||}t |j |j dkr|jd\}}}} |||| ddffg|_|d\} } } t d| | | f}||dS)Nr) JpegImageFilerBr!rZCMYKrY)ZJpegImagePluginrr'rar}rr|tellrrrZ_decompression_bomb_checkrgrhriconvertsplitmergertobytes) rjrZjpeg_header_sizeZ jpeg_headerr*imageZ decoder_nameZextentsrargsr<r=r>r r rr`s    zBLP1Decoder._decode_jpeg_streamN)r r r rtrr r r rrOsrc@seZdZddZdS) BLP2DecodercCs|}|j|jd|jdkr|jtjkr@||}q|jtj krt }|j t j kr|jdddd}t|jdddD]*}t||t|jdD] }||7}qqn|j t jkr|jdddd}t|jdddD]"}t||D]}||7}qqnz|j t jkr||jdddd}t|jdddD]$}t||D]}||7}qfqTndt|j }t|nd t|j}t|nd t|j}t|||dS) Nrrrr!r )r+r@zUnsupported alpha encoding zUnknown BLP encoding zUnknown BLP compression )rr|r^rr~rrrrrr%rrrrgr&r?r}boolrbrrJrrMrerNr)rjrr*ZlinesizeZybrrlr r rrtts@      zBLP2Decoder._loadN)r r r rtr r r rrssrc@s eZdZdZddZddZdS) BLPEncoderTc Csxd}|jdd}tt|dD]:}||d|dd\}}}}|td||||7}q"t|dkrt|d7}q^|S)NrXr!rris)imZ getpaletter&r$r'pack)rjr*rrr<r=r>r;r r r_write_palettes   zBLPEncoder._write_palettec Cs|}dt|}tjd|fd}|jj\}}|tjd||fd7}||7}t|D].}t|D] }|td|j||f7}qhq\t|d|fS)Nrz)rrrrrrrrrrrrrrrrr)rr$r'rrrgr&Zgetpixel) rjbufsizeZ palette_datarr*whyxr r rencodes     zBLPEncoder.encodeN)r r r Z _pushes_fdrrr r r rrs rcCs |jdkrd}t||jddkr*dnd}|||tdd|td tj|td |j jd krxdnd |td d |td d |tjd|j |dkr|tdd |tdd t ||dd|j d |jfgdS)NPzUnsupported BLP image modeZ blp_versionBLP1rPrQryrrVrXrrWrrUrZ)rW) rh ValueErrorZ encoderinfogetwriter'rrrrrgr_save)rr[filenamerlr]r r rrs  "rz.blprZBLP2rU)F)$rn __future__rr_r'enumriorrrrrrrr?rJrMNotImplementedErrorrNrSrTZ PyDecoderrqrrZ PyEncoderrrZ register_openroZregister_extensionZregister_decoderZ register_saveZregister_encoderr r r rs6    75F<$/