o Fh<@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 K/var/www/html/loop/nvenv/lib/python3.10/site-packages/PIL/BlpImagePlugin.pyr*src@eZdZdZdZdZdS)EncodingrN)r r r UNCOMPRESSEDDXTUNCOMPRESSED_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       "rBc 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<8B color_coder?r@rAr r r decode_dxt3usL      rMc 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) rCzrLr?r@rAr r r decode_dxt5sb ,     *rUc@s eZdZdS)BLPFormatErrorN)r r r r r r rrVsrVcCs|dddvS)Nr$BLP1BLP2r )prefixr r r_acceptsr[c@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 _read_palette5s z_BLPBaseDecoder._read_palettec Cst}t||jd} z td|d\}Wn tjy%Y|Sw||\}}}}|||f} |jr;| |f7} | | q)NrTdr r r _read_bgra?s   z_BLPBaseDecoder._read_bgraN) r r r _pulls_fdrlr{rrrr r r rrys  ryc@seZdZddZddZdS)rcCs||jtjkr |dS|jdkr2|jdvr&|}||}||dSdt|j}t |dt|j}t |)Nr)r$rzUnsupported BLP encoding zUnsupported BLP compression ) rrr _decode_jpeg_streamrrr set_as_rawrmrV)rrrr-rtr r rr|Ps     zBLP1Decoder._loadc Csddlm}td|d\}||}||jd|j||jd}||}t |}||}t |j |j dkrV|jd\}}}} |||| ddffg|_|d\} } } t d| | | f}||dS)Nr) JpegImageFilerEr$rCMYKra)JpegImagePluginrr*rirrrtellrrr_decompression_bomb_checkrorprqconvertsplitmergertobytes) rrrjpeg_header_size jpeg_headerr-image decoder_nameextentsrargsr?r@rAr r rr`s    zBLP1Decoder._decode_jpeg_streamN)r r r r|rr 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.rCzUnsupported alpha encoding zUnknown BLP encoding zUnknown BLP compression )rrrfrrrrrrrr(rrrror)rBrboolrjrrMrrUrmrVr)rrrr-linesizeybrrtr r rr|tsJ          zBLP2Decoder._loadN)r r r r|r r r rrss rc@s eZdZdZddZddZdS) BLPEncoderTc Csd}|jdd}tt|dD]}||d|dd\}}}}|td||||7}qt|dkr?|d7}t|dks5|S)Nr`r$rris)im getpaletter)r'r*pack)rrr-rr r?r@rAr>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)Nr)rrrrrrrrrrrrrrrrr)rr'r*rrror)getpixel) rrbufsize palette_datarr-whyxr r rencodes    zBLPEncoder.encodeN)r r r _pushes_fdrrr r r rrs rcCs|jdkr d}t||jddkrdnd}|||tdd|td tj|td |j jd krs8    75F<$/