B
    zb4                 @   s   d dl Z 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 eeZdd ZG d	d
 d
ejZdddddZdd Zeejee eeje eejd eejd dS )    N   )Image	ImageFileImagePalette)i16le)o8)o16lec             C   s   | d dko| d dkS )Nr   
   r   )r             )prefixr   r   1lib/python3.7/site-packages/PIL/PcxImagePlugin.py_accept'   s    r   c               @   s   e Zd ZdZdZdd ZdS )PcxImageFileZPCXZ
Paintbrushc             C   s  | j d}t|stdt|dt|dt|dd t|dd f}|d |d	 ksh|d
 |d krptdtjd%|  |d }|d
 }|d }t|d}td|||| t|dt|df| jd< |dkr|dkrd }}n>|dkr"|dkr"d}d| }t	d|dd | _
n|dkr|dkr|dkrd }}| j dtj | j d}t|dkr|d	 dkrxFtdD ]:}	||	d
 d |	d
 d  t|	d
 krd }}P qW |dkrt	d|dd  | _
| j d n0|dkr|dkr|d
krd}d}ntd || _|d |d	  |d
 |d  f| _| jd	 | d! d }
||
krz|
|
d 7 }
d"| j }tjd&| j  d$|| j  |||
 ffg| _d S )'N   znot a PCX file         r   r	   r
   r   r   zbad PCX image sizeBBox: %s %s %s %sA   B   z-PCX version %s, bits %s, planes %s, stride %s      dpi1)r
   r   PzP;%dLRGB   @   r   Lii     zRGB;Lzunknown PCX mode   )r   r   size: %sx%spcx)r   )r$   )fpreadr   SyntaxErrori16loggerdebuginfor   rawpaletteseekioSEEK_ENDlenranger   OSErrormodeZ_sizesizetellZtile)selfsZbboxversionbitsplanesZprovided_strider5   rawmodeistrider   r   r   _open4   s`    , 
*
"

zPcxImageFile._openN)__name__
__module____qualname__formatZformat_descriptionr@   r   r   r   r   r   /   s   r   )r
   r   r   r   )r   r   r   r!   )r   r   r   r   )r   r   r   zRGB;L)r   r!   r   r   c          
   C   s  yt | j \}}}}W n6 tk
rL } ztd| j d|W d d }~X Y nX | jd | d d }||d 7 }td| jd || | j}	d}
|td	t| td
 t| t	d t	d t	| jd d
  t	| jd
 d
  t	|
d  t	|
d
  d d d t| t	| t	d
 t	|	d  t	|	d
  d  |
 dksTtt| |dd| j d||| ffg | jdkr|td | jdd}|ddt|  7 }|| n@| jdkr|td x$tdD ]}|t|d  qW d S )NzCannot save z images as PCXr   r#   r   r
   z6PcxImagePlugin._save: xwidth: %d, bits: %d, stride: %d)d   rE   r	   r   s                           s       s6                                                         r   r%   )r   r   r   r   r   i   r!   r"   r   )SAVEr5   KeyError
ValueErrorr6   r*   r+   writer   o16r7   AssertionErrorr   _saveimZ
getpaletter2   r3   )rN   r&   filenamer:   r;   r<   r=   er?   Zscreenr   r.   r>   r   r   r   rM      s4    &&rM   z.pcxzimage/x-pcx)r0   Zlogging r   r   r   Z_binaryr   r)   r   r   rK   Z	getLoggerrA   r*   r   r   rG   rM   Zregister_openrD   Zregister_saveZregister_extensionZregister_mimer   r   r   r   <module>   s"   
XI