U
    ڲg                     @  s   d dl mZ d dlZd dlZd dlmZ ddlmZmZ ddl	m
Z ddl	mZ dd	d
ddZG dd dejZG dd dejZede dddddddZeejee eeje eejd dS )    )annotationsN)IO   )Image	ImageFile)i16le)o16lebytesbool)prefixreturnc                 C  s   | d d dkS )N   )   DanMs   LinS )r   r   r   6/tmp/pip-unpacked-wheel-xela2va7/PIL/MspImagePlugin.py_accept'   s    r   c                   @  s"   e Zd ZdZdZddddZdS )MspImageFileMSPzWindows PaintNone)r   c                 C  s   | j d k	st| j d}t|s.d}t|d}tdddD ]}|t||A }q>|dkrfd}t|d| _t|dt|df| _|d d d	krd
d| j	 ddfg| _
ndd| j	 dd fg| _
d S )N    znot an MSP filer      zbad MSP checksum1r      r   rawr   r   r   r   r   r   )fpAssertionErrorreadr   SyntaxErrorrangei16_mode_sizesizeZtile)selfsmsgchecksumir   r   r   _open4   s     zMspImageFile._openN)__name__
__module____qualname__formatformat_descriptionr*   r   r   r   r   r   0   s   r   c                   @  s    e Zd ZdZdddddZdS )
MspDecoderTr	   ztuple[int, int])bufferr   c                 C  s  | j d k	stt }td| jjd d  }z8| j d t	d| jj
 d| j | jj
d }W n2 tjk
r } zd}t||W 5 d }~X Y nX t|D ]\}}z|d	kr|| W q| j |}	t|	|krd
| d| }t|d	}
|
|k rj|	|
 }|
d7 }
|d	krFt	d|	|
\}}|||  |
d7 }
q|}||	|
|
|   |
|7 }
qW q tjk
r } zd| }t||W 5 d }~X Y qX q| | d dS )N)         r   <Hr   zTruncated MSP file in row mapr   zTruncated MSP file, expected z bytes on row r   ZBczCorrupted MSP file in row r   )r   )fdr   ioBytesIO	bytearraystateZxsizeseekstructunpack_fromZysizer   errorOSError	enumeratewritelenZ
set_as_rawgetvalue)r%   r1   imgZ
blank_lineZrowmaper'   xZrowlenrowidxZruntypeZruncountZrunvalr   r   r   decodes   sJ     




zMspDecoder.decodeN)r+   r,   r-   Z	_pulls_fdrK   r   r   r   r   r0   N   s   #r0   r   zImage.Imagez	IO[bytes]zstr | bytesr   )imr   filenamer   c                 C  s   | j dkr d| j  d}t|dgd }tdtd |d< |d< | j\|d	< |d
< d\|d< |d< d\|d< |d< | j\|d< |d< d}|D ]}||A }q||d< |D ]}|t| qt| |dd| j ddfg d S )Nr   zcannot write mode z as MSPr      s   Das   nMr   r      )r   r   r      r   r3   r4   	      r   r   r   r   )moderA   r!   r$   rC   o16r   _save)rL   r   rM   r'   headerr(   hr   r   r   rU      s     


rU   z.msp)
__future__r   r9   r>   typingr    r   r   _binaryr   r!   r   rT   r   r   Z	PyDecoderr0   Zregister_decoderrU   Zregister_openr.   Zregister_saveZregister_extensionr   r   r   r   <module>   s   	R