U
    0[cU                     @   s   d dl 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 Zeeje eeje eejddddg eejd dS )    N   )Image	ImageFileImagePalette)i16le)o8)o16leP1LLAzBGR;5BGRBGRA))r      )   r   )r   r   )r      )   r   )r      )r       c                   @   s$   e Zd ZdZdZdd Zdd ZdS )TgaImageFileZTGAZTargac                 C   s  | j d}|d }|d }|d }|d }|d }t|dt|df| _|d	ksv| jd dksv| jd dksv|d
kr~td|dkrd| _|dkrd| _q|dkrd| _n6|dkrd| _n&|dkrd| _|dkrd| _ntd|d@ }|dk| _|dkrd}n|dkrd}ntd|| jd< |d@ r:d| jd < |rR| j || jd!< |r t|d"t|d#|d$   }}	}
|
dkrt	
d%d&| | j d|	  | _nZ|
d'krt	
d(d)| | j d"|	  | _n,|
dkr t	
d*d+| | j d,|	  | _zdt|d$@ |f }|d@ r@dd-| j | j  |||ffg| _n"d.d-| j | j  |d|ffg| _W n tk
rz   Y nX d S )/N   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          r   r   raw)fpreadi16Z_sizesizeSyntaxErrormode_flip_horizontallyinfor   r)   ZpaletteMODEStellZtileKeyError)selfsid_lencolormaptype	imagetypeZdepthflagsr!   startr-   Zmapdepthrawmode r=   1lib/python3.8/site-packages/PIL/TgaImagePlugin.py_open5   s    





 
 
 
 

zTgaImageFile._openc                 C   s   | j r| jtjj| _d S )N)r0   imZ	transposer   Z	TransposeZFLIP_LEFT_RIGHT)r5   r=   r=   r>   load_end   s    zTgaImageFile.load_endN)__name__
__module____qualname__formatZformat_descriptionr?   rA   r=   r=   r=   r>   r   0   s   dr   )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   c              
   C   s   zt | j \}}}}W n6 tk
rL } ztd| j d|W 5 d }~X Y nX d| jkrd| 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
 |rd\}}}n
d\}}}| jdkrd}nd}| jd| jdd}|dkr(|dB }|
t|t| t| t| t| t| td td t| jd  t| jd  t| t|  |
r|
|
 |r|
| jdd |rt| |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 characters)r      r   )r   r   r   )r   r   r   r!   r    r   r   r   r   r(   r)   s           TRUEVISION-XFILE. )SAVEr/   r4   OSErrorZencoderinfogetr1   lenwarningswarnwriter   o16r-   r@   Z
getpaletter   _save)r@   r*   filenamer<   bitsr8   r9   erF   r#   r$   r7   ZcolormapfirstZcolormaplengthZcolormapentryr:   r!   r=   r=   r>   rR      s    &




	

    rR   z.tgaz.icbz.vdaz.vstzimage/x-tga)rN   rG   r   r   r   Z_binaryr   r,   r   r   rQ   r2   r   rJ   rR   Zregister_openrE   Zregister_saveZregister_extensionsZregister_mimer=   r=   r=   r>   <module>   s2   	t
K