OwlCyberSecurity - MANAGER
Edit File: BlpImagePlugin.cpython-38.pyc
U tCf�8 � @ s� d Z ddlZddlmZ ddlmZmZ dZdZdZ dZ dZ dZ dZ d d � Zd!d d �Zdd� Zdd� ZG dd� de�ZG dd� dej�ZG dd� dej�ZG dd� de�ZG dd� de�Ze�ejed d � � e�ejd � e� d 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)�BytesIO� )�Image� ImageFile� � � c C s* | d? d@ d> | d? d@ d> | d@ d> fS )N� � r � �? r � )�ir r �4/usr/lib/python3/dist-packages/PIL/BlpImagePlugin.py� unpack_5650 s r Fc C s� t | �d }t� t� t� t� f}t|�D �]�}|d }t�d| |�\}}}t|�\} } } t|�\} } }td�D �]L}td�D �]<}|d@ }|d? }d}|dkr�| | | }}}n�|dkr�| | | }}}n�|dk�r6||k�rd| | d }d| | d }d| | d }n$| | d }| | d }| | d }nR|dk�r�||k�r|d| | d }d| | d }d| | d }n d \}}}}|�r�|| �||||g� q||| �|||g� q|qnq(|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�structZ unpack_fromr �extend)�data�alpha�blocks�ret�block�idx�color0�color1�bits�r0�g0�b0�r1�g1�b1�jr Zcontrol�a�r�g�br r r � decode_dxt14 sD r/ c C s� t | �d }t� t� t� t� f}t|�D �]�}|d }| ||d � }t�d|�}t�d|d�\}}t�d|d�\}t|�\} } } t|�\} } }td�D �]2}d}td�D �] }d| | d }|| }|r�d}|dL }n d }|d M }|d 9 }|d d| | ? d @ }|dk�r"| | | }}}n�|dk�r>| | | }}}nv|d k�rzd | | d }d | | d }d | | d }n:|d k�r�d | | d }d | | d }d | | d }|| �||||g� q�q�q(|S )r � z<8B�<HHr �<I� r Fr T� � r r r r )r r r r r r# r! r" �coder$ r% r& r' r( r) r* Zhighr �alphacode_indexr+ � color_coder, r- r. r r r � decode_dxt3k sH r9 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| �\}}}td�D �]�}td�D �]�}dd| | }|dk�r&| |? d@ }n0|dk�rF| d? |d > d@ B }n||d ? d@ }|d k�rf|}nt|d k�rv|}nd||k�r�d| | |d | d }n<|dk�r�d }n,|dk�r�d}n d| | |d | d }| dd| | ? d@ }|d k�r| || }}}n�|d k�r*||| }}}nv|dk�rfd| | d }d| | d }d| | d }n:|dk�r�d| | d }d| | d }d| | d }|| �||||g� q�q�q(|S )zG input: one "row" of data (i.e. will produce 4 * width pixels) r0 z<BBz<6Br r r r r � r r r1 r2 r3 r r4 � r r ) r r r r r Za0Za1r# Z alphacode1Z alphacode2r! r" r6 r$ r% r& r' r( r) r* r r7 Z alphacoder+ r8 r, r- r. r r r � decode_dxt5� s^ , r< c @ s e Zd ZdS )�BLPFormatErrorN)�__name__� __module__� __qualname__r r r r r= � s r= c @ s( e Zd ZdZdZdZdd� Zdd� ZdS ) � BlpImageFilez Blizzard Mipmap Format ZBLPzBlizzard Mipmap Formatc C s| | j �d�| _| �� | jdkr,d}d| _n.| jdkrLd}| jrDdnd| _ntd| j ��|d | j d | jd d ffg| _d S ) Nr � BLP1�BLP1ZRGB� BLP2�BLP2ZRGBAzBad BLP magic %r�r r r r ) �fp�read�magic�_read_blp_header�mode�_blp_alpha_depthr= �size�tile)�self�decoderr r r �_open� s zBlpImageFile._openc C s� t �d| j�d��\| _t �d| j�d��\| _t �d| j�d��\| _t �d| j�d��\| _t �d| j�d��\| _t �d| j�d��| _ | j dkr�t �d| j�d��\| _t �d| j�d��\| _ t �d| j�d ��| _ t �d| j�d ��| _ d S � Nz<ir z<br z<IIr rB z<16I�@ )r �unpackrG rH �_blp_compression� _blp_encodingrL �_blp_alpha_encoding� _blp_mipsZ_sizerI � _blp_subtype� _blp_offsets� _blp_lengths�rO r r r rJ s z BlpImageFile._read_blp_headerN)r> r? r@ �__doc__�formatZformat_descriptionrQ rJ r r r r rA � s rA c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �_BLPBaseDecoderTc C sR z.| j �d� | j �d�| _| �� | �� W n tjk rL td��Y nX dS )Nr r zTruncated Blp filerF ) �fd�seekrH rI rJ �_loadr �error�OSError)rO �bufferr r r �decode s z_BLPBaseDecoder.decodec C s t �| j|�S )N)r � _safe_readr` )rO Zlengthr r r rg ! s z_BLPBaseDecoder._safe_readc C sd g }t d�D ]R}z t�d| �d��\}}}}W n tjk rJ Y q`Y nX |�||||f� q |S )N� z<4Br )r r rT rg rc �append)rO r r r. r- r, r+ r r r � _read_palette$ s z _BLPBaseDecoder._read_palettec C s� t �d| �d��\| _t �d| �d��\| _t �d| �d��\| _t �d| �d��\| _t �d| �d��\| _t �d| �d��| _| j dkr�t �d| �d��\| _t �d| �d��\| _ t �d| �d ��| _ t �d| �d ��| _ d S rR ) r rT rg rU rV rL rW rX rM rI rY rZ r[ r\ r r r rJ . s z _BLPBaseDecoder._read_blp_headerN)r> r? r@ Z _pulls_fdrf rg rj rJ r r r r r_ s r_ c @ s e Zd Zdd� Zdd� ZdS )� BLP1Decoderc C s� | j tkr| �� n�| j dkr�| jdkr�t� }| �� }t| �| jd ��}zt � d|� d��\}W n t j k r| Y q�Y nX || \}}}}|� |||g� qJ| �t|�� q�td| j ��ntd| j ��d S )Nr )r r r �<BzUnsupported BLP encoding %rz Unsupported BLP compression %r)rU �BLP_FORMAT_JPEG�_decode_jpeg_streamrV r rj r rg r[ r rT rH rc r � set_as_raw�bytesr= ) rO r �palette�_data�offsetr. r- r, r+ r r r rb B s( ��zBLP1Decoder._loadc C s� ddl m} t�d| �d��\}| �|�}| �| jd | j�� � | �| jd �}|| }t |�}||�}t � |j � |j | _ |j| _|j| _d S )Nr )� JpegImageFiler2 r )ZPIL.JpegImagePluginrt r rT rg rZ r` �tellr[ r r Z_decompression_bomb_checkrM rN rG rK )rO rt Zjpeg_header_sizeZ jpeg_headerr Zimager r r rn ] s z BLP1Decoder._decode_jpeg_streamN)r> r? r@ rb rn r r r r rk A s rk c @ s e Zd Zdd� ZdS )� BLP2Decoderc C s | � � }t� }| j�| jd � | jdk�r�| jtkr�t| � | j d ��}zt � d|� d��\}W n t jk r| Y q�Y nX || \}}}}|�|||f� qJ�q| jtk�r�| jtk�r| jd d d d } t| jd d d �D ],} t| � | �t| j�d�D ]} || 7 }�qq�n�| jtk�rx| jd d d d } t| jd d d �D ]$} t| � | ��D ]} || 7 }�qb�qPnn| jtk�r�| jd d d d } t| jd d d �D ]$} t| � | ��D ]} || 7 }�qq�nt d | j ��nt d | j ��nt d | j ��| � t |�� d S ) Nr r rl r r r )r r0 z Unsupported alpha encoding %rzUnknown BLP encoding %rzUnknown BLP compression %r) rj r r` ra rZ rU rV �BLP_ENCODING_UNCOMPRESSEDr rg r[ r rT rH rc r �BLP_ENCODING_DXTrW �BLP_ALPHA_ENCODING_DXT1rM r r/ �boolrL �BLP_ALPHA_ENCODING_DXT3r9 �BLP_ALPHA_ENCODING_DXT5r< r= ro rp ) rO rq r rr rs r. r- r, r+ ZlinesizeZyb�dr r r rb n sJ � �zBLP2Decoder._loadN)r> r? r@ rb r r r r rv m s rv c C s | d d� dkS )Nr )rB rD r )�pr r r �<lambda>� � r z.blprC rE )F) r] r �ior � r r rm rw rx Z"BLP_ENCODING_UNCOMPRESSED_RAW_BGRAry r{ r| r r/ r9 r< �NotImplementedErrorr= rA Z PyDecoderr_ rk rv Z register_openr^ Zregister_extensionZregister_decoderr r r r �<module> s6 75F*-,5 �