e<dZddlmZddlZddlZddlmZddlmZddl m Z m Z Gdd eZ Gd d eZ Gd d eZdZd$dZdZdZGddeZdZGdde jZGdde j.ZGddeZGddeZGdde j6ZdZe j<ej>eee j@ej>d e jBd!ee jBd"ee jDej>ee jFd#ey)%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 ImageFileceZdZdZy)FormatrN)__name__ __module__ __qualname__JPEG4/usr/lib/python3/dist-packages/PIL/BlpImagePlugin.pyr r )s Drr ceZdZdZdZdZy)EncodingrN)r r r UNCOMPRESSEDDXTUNCOMPRESSED_RAW_BGRArrrrr-sL CrrceZdZdZdZdZy) AlphaEncodingrrN)r r r DXT1DXT3DXT5rrrrr3s D D Drrc<|dz dzdz|dz dzdz|dzdzfS)N r?rr)is r unpack_565r%9s2 "W  "a1f_$:QX!O KKrc t|dz}ttttf}t|D]/}|dz}tjd||\}}}t |\} } } t |\} } }tdD]}tdD]}|dz}|dz }d}|dk(r| | | }}}n|dk(r| | |}}}nx|dk(r@||kDr"d| z| zdz}d| z| zdz}d| z|zdz}nL| | zdz}| | zdz}| |zdz}n3|dk(r.||kDr"d| z| zdz}d| z| zdz}d|z| zdz}nd \}}}}|r||j |g||j g2|S) E input: one "row" of data (i.e. will produce 4*width pixels) zCv*-ai%11&$D' B' Bq -A1X -(qya< "b!qA\ "b!qA\Vb[Q.Vb[Q.Vb[Q."WN"WN"WN\Vb[Q.Vb[Q.Vb[Q.%/ 1aFMM1aA,/FMM1a),? - -*-X Jrc &t|dz}ttttf}t|D]M}|dz}|||dz}tjd|}tjd|d\}}tjd|d\}t |\} } } t |\} } }tdD]}d}tdD]}d|z|zd z}||}|rd}|dz}nd }|d z}|d z}|d d|z|zzz d z}|dk(r| | | }}}nY|dk(r| | |}}}nM|d k(r"d | z| zd z}d | z| zd z}d | z|zd z}n&|d k(r!d | z| zd z}d | z| zd z}d |z| zd z}||j |gP|S)r'z<8Br?r@rAhighr$alphacode_indexrC color_coderDrErFs r decode_dxt3rSts Y"_F ; Y[)+ >Cv(,bjS38$!!%/++E5!<$$T5"5' B' Bq ,AD1X ,#$q519"2) D!GADHAR"a1q519o5= ? "b!qA1_ "b!qA1_R"*AR"*AR"*A1_R"*AR"*AR"*AA q!Ql+5 , ,(,T Jrc zt|dz}ttttf}t|D]}|dz}|||dz}tjd|\}}tjd|d}|d|ddzz|ddzz|dd zz}|d |d dzz} tjd |d\} } tjd |d\} t | \} }}t | \}}}tdD])}tdD]}dd|z|zz}|dkr | |z dz}n|dk(r| dz |d zdzz}n ||dz z dz}|d k(r|}nF|d k(r|}n>||kDrd|z |z|d z |zzdz}n$|dk(rd }n|dk(rd}nd|z |z|d z |zzdz}| dd|z|zzz dz}|d k(r| ||}}}nY|d k(r|||}}}nM|dk(r"d| z|zdz}d|z|zdz}d|z|zdz}n&|dk(r!d|z| zdz}d|z|zdz}d|z|zdz}||j |g,|S)zG input: one "row" of data (i.e. will produce 4 * width pixels) rIzr?r@rAr$rQ alphacoderCrRrDrErFs r decode_dxt5r\s Y"_F ; Y[)+ >Cv9,bjS38$##E51B!!%2!WQ1 -aB?47b=Q !WQ1 - ++E5!<$$T5"5' B' Bq( ,A1X' ,"#q1uqy/"b(!+!>$ FI$*!+r!1zQ$6N OI!+"0D!E MI>A!^A"Wi-2-Q"0DDJA!^A!^Ai-2-Q"0DDJA"a1q519o5= ? "b!qA1_ "b!qA1_R"*AR"*AR"*A1_R"*AR"*AR"*AA q!Ql+O' ,( ,#9,v Jrc eZdZy)BLPFormatErrorN)r r r rrrr^r^srr^c|dddvS)Nr)BLP1BLP2r)prefixs r_acceptrds "1:+ ++rceZdZdZdZdZdZy) BlpImageFilez Blizzard Mipmap Format BLPzBlizzard Mipmap Formatc|jjd|_|jjdtj t jd|jjd\|_|jjdtj t jd|jjd|_ |jdvr|jj}n#d t|j}t||jrd nd |_ |d |jzd |jd dffg|_y)Nr)r"s{(9(9!(<=> "MM$ 1 > !JAq!QAq A$$aT  KKN<<    s'BB%$B%N) r r r _pulls_fdrwrrrrrrrrrs!IK&5 rrceZdZdZdZy)rc|jtjk(r|jy|jdk(rm|jdvr<|j }|j |}|jt|ydt|j}t|dt|j}t|)Nr)r)r"zUnsupported BLP encoding zUnsupported BLP compression ) rr r_decode_jpeg_streamrrr set_as_rawbytesrxr^)r}rr2rs rrzBLP1Decoder._loadOs  FKK /  $ $ &  " "a '!!V+,,.w/d ,1$t7I7I2J1KL$S))0d6H6H1I0JKC % %rcddlm}tjd|j d\}|j |}|j |j d|j jz |j |jd}||z}t|}||}tj|j|jdk(r%|jd\}}}} |||| ddffg|_|jdj!\} } } tj"d| | | f}|j%|j'y)Nr) JpegImageFilerKr)rCMYKrl)JpegImagePluginrr/rtrrrtellrrr_decompression_bomb_checkrzr{r|convertsplitmergertobytes) r}rjpeg_header_size jpeg_headerr2image decoder_nameextentsrargsrDrErFs rrzBLP1Decoder._decode_jpeg_stream_s"2$mmD$//!2DE oo&67  ))!,tww||~=>t0034T!t}d# '' 3 :: 27**Q- /L'64'&47F:KLMEJ--&,,.1a EAq!9-  (rN)r r r rrrrrrrNs & )rrceZdZdZy) BLP2Decodercd|j}|jj|jd|jdk(r+|j t jk(r|j|}n|j t jk(rt}|jtjk(rv|jddzdzdz}t|jddzdzD];}t!|j#|t%|j&D]}||z } =nc|jtj(k(r`|jddzdzdz}t|jddzdzD]&}t+|j#|D]}||z } (n|jtj,k(r`|jddzdzdz}t|jddzdzD]&}t/|j#|D]}||z } (nidt1|j}t3|d t1|j }t3|d t1|j}t3||j5t7|y) Nrrrr)r()r3rIzUnsupported alpha encoding zUnknown BLP encoding zUnknown BLP compression )rrrqrrrrrrrr-rrrrzr.rGrboolrurrSrr\rxr^rr)r}rr2linesizeybrrs rrzBLP2Decoder._loadss}$$&  T&&q)*  A %!!X%:%::w/##x||3 {++}/A/AA $ ! q 0Q6:H#TYYq\A%5!$;<&!, OOH5T$BWBW=X"&A!AID&& --1C1CC $ ! q 0Q6;H#TYYq\A%5!$;<&!,T__X-F!G&A AID&&--1C1CC $ ! q 0Q6;H#TYYq\A%5!$;<&!,T__X-F!G&A AID&&8T=U=U8V7WXC(---d43E3E.F-GH$S))-T$2G2G-H,IJC % % d $rN)r r r rrrrrrrs+%rrceZdZdZdZdZy) BLPEncoderTc d}|jjdd}tt|dzD]1}||dz|dzdz\}}}}|t j d||||z }3t|dkr|dz }t|dkr|S)Nrrkr)rris)im getpaletter.r,r/pack)r}r2rr$rDrErFrCs r_write_palettezBLPEncoder._write_palettes''$$VV4s7|q() 3A Q!a%15JAq!Q FKKq!Q2 2D 3$i'! K D$i'! rc |j}dt|z}tjd|gd}|jj \}}|tjd||zgdz }||z }t |D]F}t |D]6}|tjd|jj||fz }8Ht|d|fS)Nr)rrrrrrrrrrrrrrrrr)rr,r/rrrzr.getpixel) r}bufsize palette_datarr2whyxs rencodezBLPEncoder.encodes**,  3|#44{{668Y8ww||1  FAE8Y88 q DA1X D D$''*:*:Aq6*BCC D D4y!T!!rN)r r r _pushes_fdrrrrrrrsJ"rrc|jdk7r d}t||jjddk(rdnd}|j ||j t j dd|j t j d tj|j t j d |jjd k(rdnd |j t j d d |j t j d d |j t j d g|j|dk(rJ|j t j dd |j t j dd tj||dd|jzd |jfgy)NPzUnsupported BLP image mode blp_versionBLP1rarbrrrirkrrjr"rgrm) r{ ValueError encoderinfogetwriter/rrrrrzr_save)rrnfilenamerrps rrrsK ww#~*o~~))-8FBGEHHUOHHV[[q !"HHV[[x44 56HHV[[BJJOOv$=q1 EFHHV[[q !"HHV[[q !"HHV[[ ) )*  T1%& T1%& OOBeVbgg%5q"''BCDrz.blprBLP2rg)F)$r __future__rrrr/enumriorrrr rrr%rGrSr\NotImplementedErrorr^rdrf PyDecoderrrr PyEncoderrr register_openrregister_extensionregister_decoder register_saveregister_encoderrrrrsH<#  W w  G L4n2jCL ( ,J9&&J69i))9x!)/!)H,%/,%^"$$">E*L''w?,,f5v{+v{+L''/uj)r