U
    ^+d                     @   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rd}t||dkrd| _|dkrd| _q|dkrd| _n:|dkrd| _n*|dkrd| _|dkrd| _nd}t||d@ }|dk| _|dkrd}n|dkrd}nd}t||| jd< |d@ rFd| jd < |r^| 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@ rLd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 )/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i16_sizesizeSyntaxErrormode_flip_horizontallyinfor   r)   paletteMODEStellZtileKeyError)selfsid_lencolormaptype	imagetypedepthflagsmsgr!   startr.   Zmapdepthrawmode rA   6/tmp/pip-unpacked-wheel-wujwo0tk/PIL/TgaImagePlugin.py_open5   s    





 
 
 
 

zTgaImageFile._openc                 C   s   | j r| jtjj| _d S )N)r1   imZ	transposer   Z	TransposeZFLIP_LEFT_RIGHT)r7   rA   rA   rB   load_end   s    zTgaImageFile.load_endN)__name__
__module____qualname__formatformat_descriptionrC   rE   rA   rA   rA   rB   r   0   s   gr   )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   s0  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d}t|d d }}nd\}}| jdkrd}nd}| jd| jdd}|dkrB|dB }|t|t| t| td t| t| td td t| jd  t| jd  t| t|  |r|| |r|| |	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 charactersr   r   r   r   r(   )r   r   r   r!   r    r   r   r)   s           TRUEVISION-XFILE. )SAVEr0   r6   OSErrorZencoderinfogetr2   lenwarningswarnrD   Z
getpalettewriter   o16r.   r   _save)rD   r*   filenamer@   bitsr:   r;   er>   rK   r#   r$   r9   r3   ZcolormaplengthZcolormapentryr=   r!   rA   rA   rB   rV      s    


	


    rV   z.tgaz.icbz.vdaz.vstzimage/x-tga)rR   rL   r   r   r   _binaryr   r,   r   r   rU   r4   r   rN   rV   Zregister_openrI   Zregister_saveZregister_extensionsZregister_mimerA   rA   rA   rB   <module>   s2   	w
M