U
    Evf                     @  s   d dl mZ d dlZd dlmZ ddlmZmZmZ ddl	m
Z ddl	mZ ddl	mZ d	d
ddddddZG dd dejZdddddddZddddddd Zeeje eeje eejd!d"d#d$g eejd% dS )&    )annotationsN)IO   )Image	ImageFileImagePalette)i16le)o8)o16leP1LLAzBGR;5BGRBGRA))r      )   r   )r   r   )r      )   r   )r      )r       c                   @  s0   e Zd ZdZdZddddZddddZd	S )
TgaImageFileZTGAZTargaNone)returnc                 C  s  | j d k	st| j d}|d }|d }|d }|d }|d }t|dt|df| _|d	ks| jd dks| jd dks|d
krd}t||dkrd| _|dkrd| _n|dkrd| _nB|dkr|rdnd| _n*|dkrd| _|dkrd| _nd}t||d@ }|dk| _|dkr"d}n|dkr2d}nd}t||| j	d< |d@ r\d| j	d < |rt| j || j	d!< |r0t|d"t|d#|d$   }	}
}|dkrt
d%d&|	 | j d|
  | _nh|d'krt
d(d)|	 | j d"|
  | _n:|dkr$t
d*d+|	 | j d,|
  | _nd-}t|zdt|d$@ |f }|d@ rpdd.| j | j  |||ffg| _n"d/d.| j | j  |d|ffg| _W n tk
r   Y nX d S )0N   r   r   r   r            )r   r   )r   r   r   r   r   znot a TGA file)r      r   r   r   )r   	   r   )r   
   RGBr   RGBAzunknown TGA mode0   )r   r#   )r   r#   )r   r   zunknown TGA orientationorientationr   tga_rlecompression
id_sectionr         zBGR;15s     r   r   s      r   s          zunknown TGA map depthr   r   raw)fpAssertionErrorreadi16_sizesizeSyntaxError_mode_flip_horizontallyinfor   r-   paletteMODEStellZtileKeyError)selfsid_lencolormaptype	imagetypedepthflagsmsgr%   startr3   Zmapdepthrawmode rF   6/tmp/pip-unpacked-wheel-ciywl2yd/PIL/TgaImagePlugin.py_open5   s    





 
 
 
 

zTgaImageFile._openc                 C  s*   | j r&| jd k	st| jtjj| _d S )N)r6   imr/   Z	transposer   Z	TransposeZFLIP_LEFT_RIGHT)r<   rF   rF   rG   load_end   s    zTgaImageFile.load_endN)__name__
__module____qualname__formatformat_descriptionrH   rJ   rF   rF   rF   rG   r   1   s   kr   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   r!   r"   zImage.Imagez	IO[bytes]strr   )rI   r.   filenamer   c              
   C  s@  zt | j \}}}}W n: tk
rP } zd| j d}t||W 5 d }~X Y nX d| jkrh| jd }	n| jd| jd}
|
dk}	|	r|d7 }| jd| jdd}t|}|d	krd	}|d d	 }t	d
 |r
| j
d k	st| j
dd}t|d d }}nd\}}| jdkr$d}nd}| jd| jdd}|dkrR|dB }|t|t| t| td t| t| td td t| jd  t| jd  t| t|  |r|| |r|| |	rt| |dd| j d||ffg n$t| |dd| j d|d|ffg |d d S )Nzcannot write mode z as TGArler'   r&   r   r(       z-id_section has been trimmed to 255 charactersr!   r   r   r   r,   )r   r"   r   r%   r$   r   r   r-   s           TRUEVISION-XFILE. )SAVEmoder;   OSErrorZencoderinfogetr7   lenwarningswarnrI   r/   Z
getpalettewriter	   o16r3   r   _save)rI   r.   rQ   rE   bitsr?   r@   erC   rR   r'   r(   r>   r8   ZcolormaplengthZcolormapentryrB   r%   rF   rF   rG   r^      s    


	


    r^   z.tgaz.icbz.vdaz.vstzimage/x-tga)
__future__r   rZ   typingr   rS   r   r   r   _binaryr   r1   r	   r
   r]   r9   r   rU   r^   Zregister_openrN   Zregister_saveZregister_extensionsZregister_mimerF   rF   rF   rG   <module>   s6   	{
M