o
    f-                  	   @  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   K/home/ubuntu/webapp/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rdd| j || j
d|ffg| _d S g | _|}| j
D ]}| jdd| j ||d|ff ||7 }qd S |dkrdd| j ||||f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_mimetypesizetileappend)selfheadlensmsgcompressionbpc	dimensionxsizeysizezsizelayoutrawmodeorientationpagesizeoffsetlayerr   r   r   _open8   sX   










zSgiImageFile._openN)r   r   )__name__
__module____qualname__formatformat_descriptionrE   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    79sr7   404sr   r&   flush)r0   r*   encoderinfogetr2   r   getbandsospathsplitextbasename
isinstancestrencodewritestructpackr   r.   tobyteshasattrrW   )rK   r(   rN   r8   infor:   rA   magic_numberrledimxyzpinminpinmaximg_namecolormapr@   channelr   r   r   _save   sf   





rs   c                   @  s   e Zd ZdZd	ddZdS )
SGI16DecoderTbufferr
   r   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   )fdrK   argsstater<   r=   r   r0   seekranger   new	frombytesr)   putband)	r5   ru   r@   striderA   rB   r>   bandrr   r   r   r   decode   s   
zSGI16Decoder.decodeN)ru   r
   r   rv   )rF   rG   rH   	_pulls_fdr   r   r   r   r   rt      s    rt   r%   z	image/sgi)z.bwz.rgbz.rgbaz.sgi)r	   r
   r   r   )rK   rL   r(   rM   rN   rO   r   r   )
__future__r   r[   rc   typingr   r!   r   r   _binaryr   r   r   r   r+   r   rs   	PyDecoderrt   register_decoderregister_openrI   register_saveregister_mimeregister_extensionsr   r   r   r   <module>   s2   

LM