o Ԧ/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 2lib/python3.10/site-packages/PIL/BlpImagePlugin.pyr*src@eZdZdZdZdZdS)EncodingrN)r r r UNCOMPRESSEDDXTZUNCOMPRESSED_RAW_BGRAr r r rr.rc@r) AlphaEncodingrrN)r r r DXT1DXT3DXT5r r r rr4rrcCs*|d?d@d>|d?d@d>|d@d>fS)N r?rr )ir r r unpack_565:s*r Fc Cst|d}ttttf}t|D]}|d}td||\}}}t|\} } } t|\} } }tdD]}tdD]}|d@}|d?}d}|dkrU| | | }}}nh|dkrb| | |}}}n[|dkr||krd| | d}d| | d}d| |d}n:| | d}| | d}| |d}n'|dkr||krd| | d}d| | d}d|| d}nd \}}}}|r||||||gq<|||||gqsH       "r?c Cst|d}ttttf}t|D]}|d}|||d}td|}td|d\}}td|d\}t|\} } } t|\} } }tdD]}d}tdD]}d||d }||}|rjd}|dL}nd }|d M}|d 9}|d d||?d @}|dkr| | | }}}nF|dkr| | |}}}n9|d krd | | d }d | | d }d | |d }n|d krd | | d }d | | d }d || d }||||||gqSqKq|S)r!z<8Br r r decode_dxt3usL      rIc 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@}n|dkr| d?|d >d@B}n||d?d@}|d kr|}n6|d kr|}n/||krd|||d |d}n|dkrd }n|dkrd}nd|||d |d}| dd||?d@}|d kr| ||}}}nI|d kr |||}}}n;|dkr)d| |d}d||d}d||d}n|dkrFd|| d}d||d}d||d}||||||gq{quq|S)zG input: one "row" of data (i.e. will produce 4 * width pixels) r@zr r r decode_dxt5sb ,     *rLc@s eZdZdS)BLPFormatErrorN)r r r r r r rrMsrMcCs|dddvS)Nr#BLP1BLP2r )prefixr r r_acceptsrRc@s eZdZdZdZdZddZdS) BlpImageFilez Blizzard Mipmap Format BLPzBlizzard Mipmap FormatcCs|jd|_|jdtjtd|jd\|_|jdtjtd|jd|_ |jdvr9|j }n d t |j}t ||jrJd nd |_ |d |jd |jd dffg|_dS)Nr#rr=r<r;r r r _read_palette5s z_BLPBaseDecoder._read_palettec Cst}t||jd} z td|d\}Wn tjy%Y|Sw||\}}}}|||f} |jr;| |f7} | | q)NrTr=r<r;dr r r _read_bgra?s   z_BLPBaseDecoder._read_bgraN) r r r Z _pulls_fdrcrqr{rrr r r rros  roc@seZdZddZddZdS)rcCs||jtjkr |dS|jdkr2|jdvr&|}||}||dSdt|j}t |dt|j}t |)Nr)r#rzUnsupported BLP encoding zUnsupported BLP compression ) r|rr _decode_jpeg_streamr}rr set_as_rawrdrM)rirr+rkr r rrrPs     zBLP1Decoder._loadc Csddlm}td|d\}||}||jd|j||jd}||}t |}||}t |j |j dkrV|jd\}}}} |||| ddffg|_|d\} } } t d| | | f}||dS)Nr) JpegImageFilerBr#rZCMYKrX)ZJpegImagePluginrr)r`r{rrztellrrrZ_decompression_bomb_checkrfrgrhZconvertsplitZmergertobytes) rirZjpeg_header_sizeZ jpeg_headerr+ZimageZ decoder_nameZextentsrargsr<r=r>r r rr`s    zBLP1Decoder._decode_jpeg_streamN)r r r rrrr r r rrOs rc@seZdZddZdS) BLP2DecodercCs|}|j|jd|jdkr|jtjkr||}n|jtj krt }|j t j krZ|jdddd}t|jdddD]}t||t|jdD]}||7}qQqCn|j t jkr|jdddd}t|jdddD]}t||D]}||7}qqvnR|j t jkr|jdddd}t|jdddD]}t||D]}||7}qqn$dt|j }t|d t|j}t|d t|j}t|||dS) Nrrrr#r")r,r@zUnsupported alpha encoding zUnknown BLP encoding zUnknown BLP compression )rrzr]rr|r}rrrrr'r~rrrfr(r?r{boolrarrIrrLrdrMr)rirr+ZlinesizeZybrrkr r rrrtsJ          zBLP2Decoder._loadN)r r r rrr r r rrss rc@s eZdZdZddZddZdS) BLPEncoderTc Csd}|jdd}tt|dD]}||d|dd\}}}}|td||||7}qt|dkr?|d7}t|dks5|S)NrWr#rris)imZ getpaletter(r&r)pack)rir+rrr<r=r>r;r r r_write_palettes   zBLPEncoder._write_palettec Cs|}dt|}tjd|gdR}|jj\}}|tjd||gdR7}||7}t|D]}t|D]}|td|j||f7}q6q0t|d|fS)Nrx)rrrrrrrrrrrrrrrrr)rr&r)rrrfr(Zgetpixel) ribufsizeZ palette_datarr+whyxr r rencodes    zBLPEncoder.encodeN)r r r Z _pushes_fdrrr r r rrs rcCs|jdkr d}t||jddkrdnd}|||tdd|td tj|td |j jd krs8    75F<$/