o
    'h{>                     @   sn  d Z ddlZddlZddlmZ ddlmZ ddlmZm	Z	 ddl
mZ G dd	 d	eZG d
d deZG dd deZdd Zdd Zd-ddZdd Zdd ZG dd deZdd ZG dd de	j	ZG dd  d e	jZG d!d" d"eZG d#d$ d$eZG d%d& d&e	jZd-d'd(Zeej ee e!ej d) e"d*e e"d+e e#ej e e$d,e dS ).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                   @   s   e Zd ZdZdS )Formatr   N)__name__
__module____qualname__JPEG r   r   M/var/www/html/scripts/venv/lib/python3.10/site-packages/PIL/BlpImagePlugin.pyr   )   s    r   c                   @      e Zd ZdZdZdZdS )Encodingr         N)r	   r
   r   UNCOMPRESSEDDXTUNCOMPRESSED_RAW_BGRAr   r   r   r   r   -       r   c                   @   r   )AlphaEncodingr   r      N)r	   r
   r   DXT1DXT3DXT5r   r   r   r   r   3   r   r   c                 C   s   t dtdtdi D ],\}}| |r6| t|d  } | |jv r6t| |  d|j d|   ||    S q
dt d|  d}t	|)	NBLP_FORMAT_BLP_ENCODING_BLP_ALPHA_ENCODING_
   .zmodule 'z' has no attribute '')
r   r   r   items
startswithlen__members__r   r	   AttributeError)nameenumprefixmsgr   r   r   __getattr__9   s   


 r+   c                 C   s*   | d? d@ d> | d? d@ d> | d@ d> fS )N      r      ?   r   r   )ir   r   r   
unpack_565H   s   *r1   Fc              	   C   s  t | d }t t t t f}t|D ]}|d }td| |\}}}t|\}	}
}t|\}}}tdD ]}tdD ]}|d@ }|d? }d}|dkrU|	|
|}}}nh|dkrb|||}}}n[|dkr||krd|	 | d }d|
 | d }d| | d }n:|	| d }|
| d }|| d }n'|dkr||krd| |	 d }d| |
 d }d| | d }nd	\}}}}|r|| ||||g q<|| |||g q<q6q|S )
E
    input: one "row" of data (i.e. will produce 4*width pixels)
       z<HHI   r   r      r   r   )r   r   r   r   r$   	bytearrayrangestructunpack_fromr1   extend)dataalphablocksretblockidxcolor0color1bitsr0g0b0r1g1b1jr0   controlargbr   r   r   decode_dxt1L   sH   "rQ   c              	   C   s  t | d }t t t t f}t|D ]}|d }| ||d  }td|}td|d\}}td|d\}t|\}	}
}t|\}}}tdD ]}d}tdD ]}d| | d	 }|| }|rjd}|dL }nd
}|dM }|d9 }|d	d| |  ? d@ }|dkr|	|
|}}}nF|dkr|||}}}n9|d	krd	|	 | d }d	|
 | d }d	| | d }n|dkrd	| |	 d }d	| |
 d }d	| | d }|| ||||g qSqKq|S )r2      z<8B<HHr3   <I   r4   Fr   T      r   r   r   r6   )r<   r>   r?   r@   rA   rD   rB   rC   coderE   rF   rG   rH   rI   rJ   rK   highr0   alphacode_indexrM   
color_coderN   rO   rP   r   r   r   decode_dxt3   sL   
r\   c              	   C   s  t | d }t t t t f}t|D ]>}|d }| ||d  }td|\}}td|d}|d |d d> B |d d> B |d d	> B }|d
 |d d> B }	td|d\}
}td|d\}t|
\}}}t|\}}}tdD ]}tdD ]}dd| |  }|dkr|	|? d@ }n|dkr|	d? |d> d@ B }n||d ? d@ }|d
kr|}n6|dkr|}n/||krd| | |d |  d }n|dkrd
}n|dkrd}nd| | |d |  d }|dd| |  ? d@ }|d
kr|||}}}nI|dkr|||}}}n;|dkr)d| | d }d| | d }d| | d }n|dkrFd| | d }d| | d }d| | d }|| ||||g q{quq|S )zG
    input: one "row" of data (i.e. will produce 4 * width pixels)
    rR   z<BBz<6Br   r   r3   r4   r.      r   r   rS   rT   rU   r   rV      r5   r6   )r<   r>   r?   r@   rA   a0a1rD   
alphacode1
alphacode2rB   rC   rX   rE   rF   rG   rH   rI   rJ   rK   r0   rZ   	alphacoderM   r[   rN   rO   rP   r   r   r   decode_dxt5   sb   ,


*rd   c                   @   s   e Zd ZdS )BLPFormatErrorN)r	   r
   r   r   r   r   r   re      s    re   c                 C   s   | d d dv S )Nr4      BLP1   BLP2r   )r)   r   r   r   _accept  s   ri   c                   @   s    e Zd ZdZdZdZdd ZdS )BlpImageFilez 
    Blizzard Mipmap Format
    BLPzBlizzard Mipmap Formatc                 C   s   | j d| _| j dtj td| j d\| _| j dtj td| j d| _	| jdv r9| j
 }nd	t| j }t|| jrJd
nd| _|d| j d| jddffg| _d S )Nr4   r.   <br   r   <IIr3   rf   zBad BLP magic RGBARGBr   r   r   )fpreadmagicseekosSEEK_CURr9   unpack_blp_alpha_depth_sizedecodereprre   modesizetile)selfdecoderr*   r   r   r   _open  s   
"zBlpImageFile._openN)r	   r
   r   __doc__formatformat_descriptionr   r   r   r   r   rj     s
    rj   c                   @   s8   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d ZdS )_BLPBaseDecoderTc              
   C   s@   z|    |   W dS  tjy } zd}t||d }~ww )NzTruncated BLP file)r   )_read_blp_header_loadr9   errorOSError)r   bufferer*   r   r   r   rz   $  s   

z_BLPBaseDecoder.decodec                 C   s   | j d td| d\| _td| d\| _td| d\| _td| d\| _| j dt	j
 td| d| _t| tr\td| d\| _| j dt	j
 td| d| _td| d| _d S )	Nr4   <irl   r   rm   r3   <16I@   )fdrt   r9   rw   
_safe_read_blp_compression_blp_encodingrx   _blp_alpha_encodingru   rv   r}   
isinstanceBLP1Decoder_blp_offsets_blp_lengths)r   r   r   r   r   -  s   
z _BLPBaseDecoder._read_blp_headerc                 C   s   t | j|S )N)r   r   r   )r   lengthr   r   r   r   @  s   z_BLPBaseDecoder._safe_readc              	   C   s`   g }t dD ]'}ztd| d\}}}}W n tjy#   Y  |S w |||||f q|S )N   <4Br4   )r8   r9   rw   r   r   append)r   r?   r0   rP   rO   rN   rM   r   r   r   _read_paletteC  s   z_BLPBaseDecoder._read_palettec           
      C   s   t  }t| | jd }	 ztd|d\}W n tjy%   Y |S w || \}}}}|||f}	| jr;|	|f7 }	|	|	 q)Nr   T<Br   )
r7   r   r   r   r9   rw   rr   r   rx   r;   )
r   paletter<   _dataoffsetrP   rO   rN   rM   dr   r   r   
_read_bgraM  s   


z_BLPBaseDecoder._read_bgraN)	r	   r
   r   	_pulls_fdrz   r   r   r   r   r   r   r   r   r   !  s    	
r   c                   @   s   e Zd Zdd Zdd ZdS )r   c                 C   s   | j tjkr|   d S | j dkr4| jdv r(|  }| |}| t| d S dt	| j }t
|dt	| j }t
|)Nr   )r4   r.   zUnsupported BLP encoding zUnsupported BLP compression )r   r   r   _decode_jpeg_streamr   r   r   
set_as_rawbytesr{   re   )r   r   r<   r*   r   r   r   r   ^  s   


zBLP1Decoder._loadc                 C   s   ddl m} td| d\}| |}| | jd | j   | | jd }|| }t	|}||}t
|j |jdkrV|jd \}}}}	||||	d dffg|_|d \}
}}t
d|||
f}| |  d S )Nr   )JpegImageFilerT   r4   r   CMYKro   )JpegImagePluginr   r9   rw   r   r   r   tellr   r   r   _decompression_bomb_checkr}   r|   r~   convertsplitmerger   tobytes)r   r   jpeg_header_sizejpeg_headerr<   imagedecoder_nameextentsr   argsrN   rO   rP   r   r   r   r   n  s   

zBLP1Decoder._decode_jpeg_streamN)r	   r
   r   r   r   r   r   r   r   r   ]  s    r   c                   @   s   e Zd Zdd ZdS )BLP2Decoderc                 C   s  |   }| j| jd  | jdkr| jtjkr| |}n| jtj	krt
 }| jtjkrZ| jd d d d }t| jd d d D ]}t| |t| jdD ]}||7 }qQqCn| jtjkr| jd d d d }t| jd d d D ]}t| |D ]}||7 }qqvnR| jtjkr| jd d d d }t| jd d d D ]}t| |D ]}||7 }qqn$dt| j }t|d	t| j }t|d
t| j }t|| t| d S )Nr   r   r   r4   r3   )r=   rR   zUnsupported alpha encoding zUnknown BLP encoding zUnknown BLP compression )r   r   rt   r   r   r   r   r   r   r   r7   r   r   r   r}   r8   rQ   r   boolrx   r   r\   r   rd   r{   re   r   r   )r   r   r<   linesizeybr   r*   r   r   r   r     sJ   




zBLP2Decoder._loadN)r	   r
   r   r   r   r   r   r   r     s    r   c                   @   s    e Zd ZdZdd Zdd ZdS )
BLPEncoderTc              	   C   sZ   d}| j dd}tdD ]}||d |d d  \}}}}|td||||7 }q|S )N    rn   r   r4   r   r   )im
getpaletter8   r9   pack)r   r<   r   r0   rN   rO   rP   rM   r   r   r   _write_palette  s    zBLPEncoder._write_palettec           	   
   C   s   |   }dt| }tjd|gdR  }| jj\}}|tjd|| gdR  7 }||7 }t|D ]}t|D ]}|td| j||f7 }q6q0t|d|fS )N   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   )r   r$   r9   r   r   r}   r8   getpixel)	r   bufsizepalette_datar   r<   whyxr   r   r   encode  s   zBLPEncoder.encodeN)r	   r
   r   
_pushes_fdr   r   r   r   r   r   r     s    r   c                 C   s  | j dkrd}t|| jddkrdnd}|| |tdd |td	tj |td	| j	j d
kr<dnd |td	d |td	d |tjdg| j
R   |dkru|tdd |tdd t| |dd| j
 d| j fg d S )NPzUnsupported BLP image modeblp_versionBLP1rg   rh   r   r   rl   rn   r   rm   r.   rk   rp   )r|   
ValueErrorencoderinfogetwriter9   r   r   r   r   r}   r   _save)r   rq   filenamesave_allr*   rs   r   r   r   r     s   

"$r   z.blpr   BLP2rk   )F)%r   ru   r9   r(   r   ior    r   r   
_deprecater   r   r   r   r+   r1   rQ   r\   rd   NotImplementedErrorre   ri   rj   	PyDecoderr   r   r   	PyEncoderr   r   register_openr   register_extensionregister_decoderregister_saveregister_encoderr   r   r   r   <module>   s:    
75F<$/
