U
    Žb                     @   s   d dl mZmZ d dlmZ d dlmZ d dlmZ dZ	dddd	d
dd	dZ
dd ZG dd dejZG dd dejZdd Zeejee eeje ede eejddddg eejd dS )   )Image	ImageFile)i16be)o8)o32les    	
1LRGBZCMYKPRGBA)   P4   P5   P6s   P0CMYKs   PyPs   PyRGBAs   PyCMYKc                 C   s   | dd dko| d dkS )N    r      Ps   0456y )prefixr   r   6/tmp/pip-unpacked-wheel-_wxctax1/PIL/PpmImagePlugin.py_accept*   s    r   c                   @   s,   e Zd ZdZdZdd Zdd Zdd Zd	S )
PpmImageFileZPPMzPbmplus imagec                 C   s:   d}t dD ](}| jd}|r(|tkr, q6||7 }q|S )N       r   )rangefpreadb_whitespace)selfmagic_cr   r   r   _read_magic7   s    
zPpmImageFile._read_magicc                 C   s   d}t |dkr\| jd}|s$q\n.|tkr6|s\qq\n|dkrR| jddkrq>q||7 }q|sjtdnt |dkrtd| |S )Nr   
   r      #s   
z Reached EOF while reading headerzToken too long in file header: )lenr   r   r   
ValueError)r   tokenr   r   r   r   _read_tokenA   s&    

zPpmImageFile._read_tokenc                 C   s6  |   }zt| }W n tk
r0   tdY nX dddd|| _|dkrZd| _d}n
| | _}d}td	D ]|}t| 	 }|d
kr|}qp|dkr|}|dkr qqp|dkrp|}	|	dkr|dkrd| _|	dkr|dkrd}qp|	dkrpd}qp|dkr|d
dfn||	f}
||f| _
|d
d
||f| j |
fg| _d S )Nznot a PPM filezimage/x-portable-bitmapzimage/x-portable-graymapzimage/x-portable-pixmap)r   r   r   r   1;Iraw   r   r         r   I  I;16Bppm)r    MODESKeyErrorSyntaxErrorgetZcustom_mimetypemoder   intr&   _sizer   tellZtile)r   Zmagic_numberr4   rawmodeZdecoder_nameixr%   xsizeysizemaxvalargsr   r   r   _openY   sF    

zPpmImageFile._openN)__name__
__module____qualname__formatformat_descriptionr    r&   r>   r   r   r   r   r   2   s
   
r   c                   @   s   e Zd ZdZdd ZdS )
PpmDecoderTc                 C   s*  t  }t| jd d}|dk r"dnd}| jdkr4dnd}| jdkrFdnd}t| j}t|| jj| jj	 | | k r| j
|| }t||| k rqt|D ]V}	|dkr||	 nt||	| }
t|t|
| | }
|| jdkrt|
nt|
7 }qqV| jdkr
d	n| j}| t||d
df dS )Nr-      r   r*   r,      r+   zI;32r   )rE   r   )	bytearrayminr=   r4   r   Zgetmodebandsr#   stater:   r;   fdr   r   i16roundo32r   Z
set_as_rawbytes)r   bufferdatar<   Zin_byte_countZout_byte_countZout_maxZbandsZpixelsbvaluer8   r   r   r   decode   s$     "zPpmDecoder.decodeN)r?   r@   rA   Z	_pulls_fdrT   r   r   r   r   rD      s   rD   c              
   C   s   | j dkrd\}}nN| j dkr(d\}}n:| j dkr<d\}}n&| j dkrPd\}}ntd	| j  d
||d| j   |dkr|d n&|dkr|dkr|d n
|d t| |dd| j d|ddffg d S )Nr   )r'   r   r   )r   r   r,   )r.   r   )r	   r   )r	   r   zcannot write mode z as PPMs   
%d %d
r   s   255
r   s   65535
r(   )r   r   r   r   )r4   OSErrorwritesizer   _save)Zimr   filenamer8   headr   r   r   rX      s"    








rX   r/   z.pbmz.pgmz.ppmz.pnmzimage/x-portable-anymapN) r   r   _binaryr   rL   r   r   rN   r   r0   r   r   Z	PyDecoderrD   rX   Zregister_openrB   Zregister_saveZregister_decoderZregister_extensionsZregister_mimer   r   r   r   <module>   s*   S