File: //lib/python3/dist-packages/PIL/__pycache__/MspImagePlugin.cpython-38.pyc
U
g�
^� � @ s� d dl Z d dlZddlmZmZ ddlmZmZm Z
dd� Z G dd� dej�Z G d d
� d
ej
�Ze�d e� d d
� Ze�e je e � e�e je� e�e jd� dS )� N� )�Image� ImageFile)�i8�i16le�o16lec C s | d d� dkS )N� �� DanMs LinS� )�prefixr r �4/usr/lib/python3/dist-packages/PIL/MspImagePlugin.py�_accept$ s r c @ s e Zd ZdZdZdd� ZdS )� MspImageFile�MSPz
Windows Paintc C s� | j �d�}|d d� dkr$td��d}tddd�D ] }|t|||d � �A }q4|dkrbtd��d| _t|dd � �t|d d � �f| _|d d� d
kr�d d | j dd
fg| _ndd | j dd fg| _d S )N� r r znot an MSP filer � zbad MSP checksum�1� r
�raw�r r �r r r r ) �fp�read� SyntaxError�range�i16�modeZ_size�sizeZtile)�self�s�checksum�ir r r
�_open2 s "zMspImageFile._openN)�__name__�
__module__� __qualname__�formatZformat_descriptionr# r r r r
r - s r c @ s e Zd ZdZdd� ZdS )�
MspDecoderTc C sz t �� }td| jjd d �}z4| j�d� t�d| jj | j�
| jj d ��}W n tj k
rr t d��Y nX t
|�D ]�\}}z�|dkr�|�|� W q|| j�
|�}t|�|kr�t d ||f��d}||k �r<t|| �} |d
7 }| dk�rt�d ||�\}
} |�| |
� |d7 }q�| }
|�||||
� � ||
7 }q�W q| tj k
�rb t d | ��Y q|X q|| �|�� d
� dS )N)� � � r z<%dHr z Truncated MSP file in row mapr z/Truncated MSP file, expected %d bytes on row %sr ZBcz Corrupted MSP file in row %dr r )�io�BytesIO� bytearray�stateZxsize�fd�seek�structZ unpack_fromZysizer �error�OSError� enumerate�write�lenr Z
set_as_raw�getvalue) r �bufferZimgZ
blank_lineZrowmap�xZrowlen�row�idxZruntypeZruncountZrunvalr r r
�decoden sH
�
�
zMspDecoder.decodeN)r$ r% r& Z _pulls_fdr= r r r r
r( I s #r( r c C s� | j dkrtd| j ��dgd }td�td� |d< |d< | j\|d< |d < d
\|d < |d < d
\|d
< |d< | j\|d< |d< d}|D ] }||A }q�||d< |D ]}|�t|�� q�t�| |dd| j ddfg� d S )Nr zcannot write mode %s as MSPr � s Das nMr r � )r r r � r r* r+ � � r r r r )r r4 r r r6 �o16r �_save)Zimr �filename�headerr! �hr r r
rD � s
rD z.msp)r, r2 � r r Z_binaryr r r r rC r r Z PyDecoderr( Zregister_decoderrD Z
register_openr' Z
register_saveZregister_extensionr r r r
�<module> s P