a
    i=icØ  ã                	   @   sÐ   d dl Z d dlZddlmZmZ ddlmZ ddlmZ dd„ Z	ddd	d	d
ddddœZ
G dd„ dejƒZdd„ ZG dd„ dejƒZe de¡ e ejee	¡ e eje¡ e ejd¡ e ejg d¢¡ dS )é    Né   )ÚImageÚ	ImageFile)Úi16be)Úo8c                 C   s   t | ƒdkot| ƒdkS )Né   éÚ  )ÚlenÚi16)Úprefix© r   úc/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/PIL/SgiImagePlugin.pyÚ_accept!   s    r   ÚLúL;16BÚRGBzRGB;16BÚRGBAzRGBA;16B))r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   é   r   )r   r   r   )r   r   é   )r   r   r   c                   @   s   e Zd ZdZdZdd„ ZdS )ÚSgiImageFileZSGIzSGI Image File Formatc              	   C   sp  d}| j  |¡}t|ƒs tdƒ‚|d }|d }t|dƒ}t|dƒ}t|dƒ}t|dƒ}|||f}	d	}
zt|	 }
W n ty„   Y n0 |
d	kr–td
ƒ‚||f| _|
 d¡d | _	| j	dkrÀd| _
d}|dkrF|| | }|dkrdd| j || j	d|ffg| _n@g | _|}| j	D ].}| j dd| j ||d|ff¡ ||7 }qn&|dkrldd| j ||
||ffg| _d S )Né   zNot an SGI image filer   r   r   é   é   é
   Ú úUnsupported SGI image modeú;r   r   z	image/rgbéÿÿÿÿÚSGI16)r   r   Úrawr   Zsgi_rle)ÚfpÚreadr   Ú
ValueErrorr
   ÚMODESÚKeyErrorÚ_sizeÚsplitÚmodeZcustom_mimetypeÚsizeZtileÚappend)ÚselfZheadlenÚsÚcompressionÚbpcÚ	dimensionÚxsizeÚysizeÚzsizeZlayoutÚrawmodeÚorientationÚpagesizeÚoffsetÚlayerr   r   r   Ú_open8   sL    








ÿ
ÿ
ÿzSgiImageFile._openN)Ú__name__Ú
__module__Ú__qualname__ÚformatÚformat_descriptionr7   r   r   r   r   r   3   s   r   c              	   C   sF  | j dkr&| j dkr&| j dkr&tdƒ‚| j}| dd¡}|dvrHtdƒ‚d	}d
}d}d}| j\}	}
| j dkrz|
dkrzd}n| j dkrˆd}t| j ƒ}|dks¢|dkr¦d}t|  ¡ ƒ|krÒtd|› dt|  ¡ ƒ› ƒ‚d}d}tj 	tj 
|¡¡d }| dd¡}d}| t d|¡¡ | t|ƒ¡ | t|ƒ¡ | t d|¡¡ | t d|	¡¡ | t d|
¡¡ | t d|¡¡ | t d|¡¡ | t d|¡¡ | t dd¡¡ | t d|¡¡ | t dd¡¡ | t d|¡¡ | t dd¡¡ d}|dkrd}|  ¡ D ]}| | d|d|¡¡ qt|dƒrB| ¡  d S )Nr   r   r   r   r-   r   )r   r   z%Unsupported number of bytes per pixelr   r   r   r   r   z(incorrect number of bands in SGI write: z vs éÿ   ÚasciiÚignorez>hz>Hz>lZ4só    Z79sr+   Z404sr   r   Úflush)r'   r"   ZencoderinfoÚgetr(   r	   ZgetbandsÚosÚpathÚsplitextÚbasenameÚencodeÚwriteÚstructÚpackr   r&   ÚtobytesÚhasattrrA   )Úimr    ÚfilenameÚinfor-   r3   Zmagic_numberZrleÚdimÚxÚyÚzZpinminZpinmaxZimg_nameZcolormapr2   Úchannelr   r   r   Ú_save}   s`    


ÿ
rU   c                   @   s   e Zd ZdZdd„ ZdS )ÚSGI16DecoderTc           	      C   sŠ   | j \}}}| jj| jj }t| jƒ}| j d¡ t|ƒD ]J}t	 
d| jj| jjf¡}| | j d| ¡dd||¡ | j |j|¡ q:dS )Nr   r   r   r   r   )r   r   )ÚargsÚstater/   r0   r	   r'   ÚfdÚseekÚranger   ÚnewÚ	frombytesr!   rM   Zputband)	r*   Úbufferr2   Zstrider3   r4   r1   ZbandrT   r   r   r   ÚdecodeË   s    
ÿzSGI16Decoder.decodeN)r8   r9   r:   Z	_pulls_fdr_   r   r   r   r   rV   È   s   rV   r   z	image/sgi)z.bwz.rgbz.rgbaz.sgi)rC   rI   r   r   r   Ú_binaryr   r
   r   r   r#   r   rU   Z	PyDecoderrV   Zregister_decoderZregister_openr;   Zregister_saveZregister_mimeZregister_extensionsr   r   r   r   Ú<module>   s,   øJK