o
    e<                     @  sd  d Z ddlm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 G dd	 d	eZG d
d deZG dd de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& 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.
    )annotationsN)IntEnum)BytesIO   )Image	ImageFilec                   @  s   e Zd ZdZdS )Formatr   N)__name__
__module____qualname__JPEG r   r   E/var/www/html/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*   | d? d@ d> | d? d@ d> | d@ d> fS )N      r      ?   r   r   )ir   r   r   
unpack_5659   s   *r!   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   len	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_dxt1=   sH   "rB   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 )r"      z<8B<HHr#   <I   r$   Fr   T      r   r   r   r&   )r-   r/   r0   r1   r2   r5   r3   r4   coder6   r7   r8   r9   r:   r;   r<   highr    alphacode_indexr>   
color_coder?   r@   rA   r   r   r   decode_dxt3t   sL   
rM   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)
    rC   z<BBz<6Br   r   r#   r$   r      r   r   rD   rE   rF   r   rG      r%   r&   )r-   r/   r0   r1   r2   a0a1r5   
alphacode1
alphacode2r3   r4   rI   r6   r7   r8   r9   r:   r;   r<   r    rK   	alphacoder>   rL   r?   r@   rA   r   r   r   decode_dxt5   sb   ,


*rU   c                   @  s   e Zd ZdS )BLPFormatErrorN)r	   r
   r   r   r   r   r   rV      s    rV   c                 C  s   | d d dv S )Nr$      BLP1   BLP2r   )prefixr   r   r   _accept   s   r[   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 )Nr$   r   <br   r   <IIr#   rW   zBad BLP magic RGBARGBr   r   r   )fpreadmagicseekosSEEK_CURr*   unpack_blp_alpha_depth_sizedecodereprrV   _modesizemodetile)selfdecodermsgr   r   r   _open   s   
"zBlpImageFile._openN)r	   r
   r   __doc__formatformat_descriptionru   r   r   r   r   r\      s
    r\   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_loadr*   errorOSError)rr   bufferert   r   r   r   rl     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 )	Nr$   <ir^   r   r_   r#   <16I@   )fdrf   r*   ri   
_safe_read_blp_compression_blp_encodingrj   _blp_alpha_encodingrg   rh   ro   
isinstanceBLP1Decoder_blp_offsets_blp_lengths)rr   r   r   r   r{     s   
z _BLPBaseDecoder._read_blp_headerc                 C  s   t | j|S )N)r   r   r   )rr   lengthr   r   r   r   1  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   <4Br$   )r)   r*   ri   r   r}   append)rr   r0   r    rA   r@   r?   r>   r   r   r   _read_palette4  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   )
r(   r   r   r   r*   ri   rd   r}   rj   r,   )
rr   paletter-   _dataoffsetrA   r@   r?   r>   dr   r   r   
_read_bgra>  s   


z_BLPBaseDecoder._read_bgraN)	r	   r
   r   	_pulls_fdrl   r{   r   r   r   r   r   r   r   ry     s    	
ry   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   )r$   r   zUnsupported BLP encoding zUnsupported BLP compression )r   r   r   _decode_jpeg_streamr   r   r   
set_as_rawbytesrm   rV   )rr   r   r-   rt   r   r   r   r|   O  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   )JpegImageFilerE   r$   r   CMYKra   )JpegImagePluginr   r*   ri   r   r   r   tellr   r   r   _decompression_bomb_checkro   rp   rq   convertsplitmerger   tobytes)rr   r   jpeg_header_sizejpeg_headerr-   imagedecoder_nameextentsr   argsr?   r@   rA   r   r   r   r   _  s   

zBLP1Decoder._decode_jpeg_streamN)r	   r
   r   r|   r   r   r   r   r   r   N  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   r$   r#   )r.   rC   zUnsupported alpha encoding zUnknown BLP encoding zUnknown BLP compression )r   r   rf   r   r   r   r   r   r   r   r(   r   r   r   ro   r)   rB   r   boolrj   r   rM   r   rU   rm   rV   r   r   )rr   r   r-   linesizeybr   rt   r   r   r   r|   s  sJ   




zBLP2Decoder._loadN)r	   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  s   d}| j dd}tt|d D ]}||d |d d  \}}}}|td||||7 }qt|dk r?|d7 }t|dk s5|S )N    r`   r$   r   r   i   s       )im
getpaletter)   r'   r*   pack)rr   r-   r   r    r?   r@   rA   r>   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'   r*   r   r   ro   r)   getpixel)	rr   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BLP1rX   rY   r   r   r^   r`   r   r_   r   r]   rb   )rp   
ValueErrorencoderinfogetwriter*   r   r   r   r   ro   r   _save)r   rc   filenamert   re   r   r   r   r     s   

"$r   z.blpr   BLP2r]   )F)$rv   
__future__r   rg   r*   enumr   ior    r   r   r   r   r   r!   rB   rM   rU   NotImplementedErrorrV   r[   r\   	PyDecoderry   r   r   	PyEncoderr   r   register_openrw   register_extensionregister_decoderregister_saveregister_encoderr   r   r   r   <module>   s8    
75F<$/