o
    'j6hL                  	   @  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ZdddddddddZG dd dejZd&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 )'    )annotationsN)IO   )Image	ImageFile)i16be)o8prefixbytesreturnboolc                 C  s   t | dkot| dkS )N     )leni16)r	    r   a/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/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dZdS )	SgiImageFileSGIzSGI Image File Formatr   Nonec                 C  s  | j d usJ d}| j |}t|sd}t||d }|d }t|d}t|d}t|d}t|d}	|||	f}
d	}zt|
 }W n	 tyK   Y nw |d	krVd
}t|||f| _|dd | _	| j
dkrkd| _d}|dkr|| | }|dkrtdd| j || j
d|fg| _d S g | _|}| j
D ]}| jtdd| j ||d|f ||7 }qd S |dkrtdd| j ||||fg| _d S d S )N   zNot an SGI image filer   r   r         
    Unsupported SGI image mode;r   r   z	image/rgbSGI16)r   r   rawr   sgi_rle)fpreadr   
ValueErrorr   MODESKeyError_sizesplit_modemodecustom_mimetyper   _Tilesizetileappend)selfheadlensmsgcompressionbpc	dimensionxsizeysizezsizelayoutrawmodeorientationpagesizeoffsetlayerr   r   r   _open8   sj   








	


zSgiImageFile._openN)r   r   )__name__
__module____qualname__formatformat_descriptionrF   r   r   r   r   r   4   s    r   imImage.Imager(   	IO[bytes]filenamestr | bytesr   c              	   C  sD  | j dvrd}t|| j}|dd}|dvrd}t|d}d}d	}d
}	| j\}
}| j dkr7|dkr7d}	n| j dkr>d}	t| j }|	dv rId}t|  |krad| dt|   }t|d	}d}tj	tj
|d	 }t|tr||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r |  d S d S )N>   r   r   r   r"   r;   r   )r   r   z%Unsupported number of bytes per pixelr$   r   r   r   r   r   >   r   r   z(incorrect number of bands in SGI write: z vs    asciiignorez>hz>Hz>l4s    79sr8   404sr   r&   flush)r0   r*   encoderinfogetr3   r   getbandsospathsplitextbasename
isinstancestrencodewritestructpackr   r.   tobyteshasattrrX   )rL   r(   rO   r9   infor;   rB   magic_numberrledimxyzpinminpinmaximg_namecolormaprA   channelr   r   r   _save   sf   





rt   c                   @  s   e Zd ZdZd	ddZdS )
SGI16DecoderTbuffer$bytes | Image.SupportsArrayInterfacer   tuple[int, int]c           	      C  s   | j d usJ | jd usJ | j\}}}| jj| jj }t| j}| j d t	|D ]%}t
d| jj| jjf}|| j d| dd|| | j|j| q+dS )Nr   r   r   r&   r   )r$   r   )fdrL   argsstater=   r>   r   r0   seekranger   new	frombytesr)   putband)	r6   rv   rA   striderB   rC   r?   bandrs   r   r   r   decode   s   
zSGI16Decoder.decodeN)rv   rw   r   rx   )rG   rH   rI   	_pulls_fdr   r   r   r   r   ru      s    ru   r%   z	image/sgi)z.bwz.rgbz.rgbaz.sgi)r	   r
   r   r   )rL   rM   r(   rN   rO   rP   r   r   )
__future__r   r\   rd   typingr   r!   r   r   _binaryr   r   r   r   r+   r   rt   	PyDecoderru   register_decoderregister_openrJ   register_saveregister_mimeregister_extensionsr   r   r   r   <module>   s2   

UM