fume-manage-python.git

U  
T±dc@ã@sndZddlZddlZddlmZddlmZddlmZm    Z    ddl
mZGdd    d    eZGd
ddeZ Gdd d eZddZddZd-ddZddZddZGdddeZddZGddde    j    ZGdd d e    jZGd!d"d"eZGd#d$d$eZGd%d&d&e    jZd.d'd(Ze ej ee¡e !ej d)¡e "d*e¡e "d+e¡e #ej e¡e $d,e¡dS)/aÑ
Blizzard Mipmap Format (.blp)
Jerome Leclanche <jerome@leclan.ch>
 
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.
éN)ÚIntEnum)ÚBytesIOé)ÚImageÚ    ImageFile)Ú    deprecatec@seZdZdZdS)ÚFormatrN)Ú__name__Ú
__module__Ú__qualname__ÚJPEG©r r úId:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\PIL/BlpImagePlugin.pyr)src@seZdZdZdZdZdS)ÚEncodingrééN)r    r
rÚUNCOMPRESSEDÚDXTZUNCOMPRESSED_RAW_BGRAr r r rr-src@seZdZdZdZdZdS)Ú AlphaEncodingrréN)r    r
rÚDXT1ÚDXT3ÚDXT5r r r rr3srcCstdtdtdi ¡D]X\}}| |¡r|t|d}||jkrt||d|jd|||Sqdtd|d}t    |dS)    NZBLP_FORMAT_Z BLP_ENCODING_ZBLP_ALPHA_ENCODING_é
Ú.zmodule 'z' has no attribute 'ú')
rrrÚitemsÚ
startswithÚlenÚ__members__rr    ÚAttributeError)ÚnameÚenumÚprefixÚmsgr r rÚ__getattr__9sý
 
 r%cCs*|d?d@d>|d?d@d>|d@d>fS)Nééréé?rr )Úir r rÚ
unpack_565Hsr+Fc    CsÄt|d}ttttf}t|D]}|d}t d||¡\}}}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}nd    \}}}}|r¦|| ||||g¡q||| |||g¡q|qnq(|S)
úE
    input: one "row" of data (i.e. will produce 4*width pixels)
    éz<HHIérréÿrr)rrrr©rÚ    bytearrayÚrangeÚstructÚunpack_fromr+Úextend)ÚdataÚalphaÚblocksÚretÚblockÚidxÚcolor0Úcolor1ÚbitsÚr0Úg0Úb0Úr1Úg1Úb1Újr*ÚcontrolÚaÚrÚgÚbr r rÚdecode_dxt1LsD
 
 
 
rKc    CsÔt|d}ttttf}t|D]¤}|d}|||d}t d|¡}t d|d¡\}}t d|d¡\}t|\}    }
}t|\}} }tdD]2}d}tdD]}d||d    }||}|rÚd}|dL}nd
}|dM}|d9}|d    d||?d @}|dkr"|    |
|}}}n|dkr>|| |}}}nv|d    krzd    |    |d }d    |
| d }d    ||d }n:|d kr´d    ||    d }d    | |
d }d    ||d }|| ||||g¡qªqq(|S)r,éz<8Bú<HHr-ú<Iér.FrTéérrrr0)r6r8r9r:r;r>r<r=Úcoder?r@rArBrCrDrEÚhighr*Úalphacode_indexrGÚ
color_coderHrIrJr r rÚdecode_dxt3sH
 
 
 
 
rVc    CsÀt|d}ttttf}t|D]}|d}|||d}t d|¡\}}t d|d¡}|d|dd>B|dd>B|dd    >B}|d
|dd>B}    t d|d¡\}
}t d |d¡\}t|
\} }}t|\}}}tdD]Ì}tdD]¼}dd||}|dkr&|    |?d@}n0|dkrF|    d?|d>d@B}n||d?d@}|d
krf|}nt|dkrv|}nd||krd|||d|d}n<|dkr®d
}n,|dkr¾d}nd|||d|d}|dd||?d@}|d
kr| ||}}}n|dkr*|||}}}nv|dkrfd| |d}d||d}d||d}n:|dkr d|| d}d||d}d||d}|| ||||g¡qøqêq(|S)zG
    input: one "row" of data (i.e. will produce 4 * width pixels)
    rLz<BBz<6Brrr-r.r(érrrMrNrOrrPér/r0)r6r8r9r:r;Za0Za1r>Z
alphacode1Z
alphacode2r<r=rRr?r@rArBrCrDrEr*rTZ    alphacoderGrUrHrIrJr r rÚdecode_dxt5¸s^,
 
 
 
 
 
 
 
 
 
 
rYc@seZdZdS)ÚBLPFormatErrorN)r    r
rr r r rrZþsrZcCs|dddkS)Nr.©óBLP1óBLP2r )r#r r rÚ_acceptsr^c@s eZdZdZdZdZddZdS)ÚBlpImageFilez 
    Blizzard Mipmap Format
    ÚBLPzBlizzard Mipmap FormatcCs¼|j d¡|_|j dtj¡t d|j d¡¡\|_|j dtj¡t d|j d¡¡|_    |jdkrr|j 
¡}nd    t|j}t||jrd
nd|_ |d|jd |j d dffg|_dS)Nr.r(ú<brrú<IIr-r[zBad BLP magic ÚRGBAÚRGB©rrr)ÚfpÚreadÚmagicÚseekÚosÚSEEK_CURr3ÚunpackÚ_blp_alpha_depthÚ_sizeÚdecodeÚreprrZÚmodeÚsizeÚtile)ÚselfÚdecoderr$r r rÚ_opens
zBlpImageFile._openN)r    r
rÚ__doc__ÚformatÚformat_descriptionrvr r r rr_sr_c@s8eZdZdZddZddZddZdd    Zd
dZdS) Ú_BLPBaseDecoderTc
CsLz| ¡| ¡Wn2tjk
rF}zd}t||W5d}~XYnXdS)NzTruncated BLP file)éÿÿÿÿr)Ú_read_blp_headerÚ_loadr3ÚerrorÚOSError)rtÚbufferÚer$r r rro$sz_BLPBaseDecoder.decodecCsä|j d¡t d| d¡¡\|_t d| d¡¡\|_t d| d¡¡\|_t d| d¡¡\|_|j dt    j
¡t d| d¡¡|_t|t r¸t d| d¡¡\|_|j dt    j
¡t d| d¡¡|_t d| d¡¡|_dS)    Nr.ú<irarrbr-ú<16Ié@)Úfdrir3rlÚ
_safe_readÚ_blp_compressionÚ _blp_encodingrmÚ_blp_alpha_encodingrjrkrrÚ
isinstanceÚBLP1DecoderÚ_blp_offsetsÚ_blp_lengths)rtr r rr|-s
z _BLPBaseDecoder._read_blp_headercCst |j|¡S)N)rrr)rtÚlengthr r rr@sz_BLPBaseDecoder._safe_readc    Csdg}tdD]R}zt d| d¡¡\}}}}Wntjk
rJYq`YnX| ||||f¡q|S)Néú<4Br.)r2r3rlrr~Úappend)rtr9r*rJrIrHrGr r rÚ _read_paletteCsz_BLPBaseDecoder._read_palettec
Cst}t| |jd¡}zt d| d¡¡\}Wntjk
rLYqYnX||\}}}}|||f}    |jrx|    |f7}    |     |    ¡q|S)Nrú<Br)
r1rrrr3rlrgr~rmr5)
rtÚpaletter6Ú_dataÚoffsetrJrIrHrGÚdr r rÚ
_read_bgraMs
 
 
z_BLPBaseDecoder._read_bgraN)    r    r
rZ    _pulls_fdror|rrrr r r rrz!s    
rzc@seZdZddZddZdS)rcCs|jtjkr| ¡nh|jdkrf|jdkrL| ¡}| |¡}| t|¡q~dt    |j}t
|ndt    |j}t
|dS)Nr)r.r(zUnsupported BLP encoding zUnsupported BLP compression )rrrÚ_decode_jpeg_streamrrrÚ
set_as_rawÚbytesrprZ)rtrr6r$r r rr}^s
 
 
 
 
zBLP1Decoder._loadc Csäddlm}t d| d¡¡\}| |¡}| |jd|j ¡¡| |jd¡}||}t    |}||}t
 |j¡|j dkr¬|jd\}}}}    ||||    ddffg|_| d¡ ¡\}
}}t
 d|||
f¡}| | ¡¡dS)Nr)Ú JpegImageFilerNr.rZCMYKrd)ZJpegImagePluginrr3rlrrrÚtellrrrZ_decompression_bomb_checkrrrqrsÚconvertÚsplitÚmergerÚtobytes) rtrZjpeg_header_sizeZjpeg_headerr6ÚimageZdecoder_nameZextentsrÚargsrHrIrJr r rrns
 
zBLP1Decoder._decode_jpeg_streamN)r    r
rr}rr r r rr]src@seZdZddZdS)ÚBLP2DecodercCsÚ| ¡}|j |jd¡|jdkr°|jtjkr@| |¡}qÈ|jtj    krt
}|jtj krº|jdddd}t|jdddD]*}t| |¡t|jdD]}||7}q¨qnÚ|jtjkr|jdddd}t|jdddD]"}t| |¡D]}||7}qqônz|jtjkr||jdddd}t|jdddD]$}t| |¡D]}||7}qfqTndt|j}t|nd    t|j}t|nd
t|j}t|| t|¡dS)Nrrrr.r-)r7rLzUnsupported alpha encoding zUnknown BLP encoding zUnknown BLP compression )rrrirrrrrrrr1rrrrrr2rKrÚboolrmrrVrrYrprZrr)rtrr6ZlinesizeZybrr$r r rr}s@ÿ
 
 
zBLP2Decoder._loadN)r    r
rr}r r r rr¤sr¤c@s eZdZdZddZddZdS)Ú
BLPEncoderTc    CsZd}|j dd¡}tdD]:}||d|dd\}}}}|t d||||¡7}q|S)Nórcrr.rr)ÚimZ
getpaletter2r3Úpack)rtr6rr*rHrIrJrGr r rÚ_write_palette³s zBLPEncoder._write_palettec    
Cs| ¡}dt|}tjd|fd}|jj\}}|tjd||fd7}||7}t|D].}t|D] }|t d|j ||f¡¡7}qhq\t|d|fS)Nér)rrrrrrrrrrrrrrrrr)rªrr3r©r¨rrr2Zgetpixel)    rtÚbufsizeZpalette_datarr6ÚwÚhÚyÚxr r rÚencode»s zBLPEncoder.encodeN)r    r
rZ
_pushes_fdrªr±r r r rr¦°sr¦cCs
|jdkrd}t||j d¡dkr*dnd}| |¡| t dd¡¡| t d    tj¡¡| t d    |j    jd
krxdnd¡¡| t d    d¡¡| t d    d¡¡| tjd|j
¡|dkræ| t dd ¡¡| t dd¡¡t ||dd|j
d|jfg¡dS)NÚPzUnsupported BLP image modeZblp_versionÚBLP1r\r]rrrarcrrbr(r`re)rb) rqÚ
ValueErrorZencoderinfoÚgetÚwriter3r©rrrrrrÚ_save)r¨rfÚfilenameZsave_allr$rhr r rr·Ís
 
"r·z.blpr³ZBLP2r`)F)F)%rwrjr3r"rÚiorÚrrZ
_deprecaterrrrr%r+rKrVrYÚNotImplementedErrorrZr^r_Z    PyDecoderrzrr¤Z    PyEncoderr¦r·Z register_openrxZregister_extensionZregister_decoderZ register_saveZregister_encoderr r r rÚ<module>s8
75F<$/