o
    Z#a7                     @   s.  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lm
Z d dlmZ ddd	d
dddZdd Zdd ZG dd dejZG dd deZddddddZdd Zd%dd Zeejee eeje eejd! eejd" eejee eeje eejd# eejd" d$S )&   )Image	ImageFileImagePalette)i16le)i32le)o8)o16le)o32le)PzP;1)r
   zP;4)r
   r
   )RGBBGR;15)r   BGR)r   BGRX)r                   c                 C   s   | d d dkS )N      BM prefixr   r   4/usr/lib/python3/dist-packages/PIL/BmpImagePlugin.py_accept1   s   r   c                 C   s   t | dv S )N)   (   @   l   |   )i32r   r   r   r   _dib_accept5      r!   c                   @   sX   e Zd ZdZdZdZdddddd	d
Ze D ]	\ZZ	e	e
 e< qdddZdd ZdS )BmpImageFilez0Image plugin for the Windows Bitmap format (BMP)Windows BitmapZBMP    r   r      r      )RAWZRLE8ZRLE4	BITFIELDSZJPEGZPNGc              	   C   sz  | j j| j j}}|r|| i }t|d|d< d|d< t| j |d d }|d dkrTt|d|d< t|d|d	< t|d|d
< t|d|d< | j|d< d|d< n|d dv r|d dk|d< |d ridnd|d< t|d|d< |d s}t|dndt|d |d	< t|d|d
< t|d|d< t|d|d< t|d|d< t|dt|df|d< t|d|d< d|d< tdd  |d D | j	d!< |d | j
krt|d"krtg d#D ]\}}t|d$|d  ||< qnd|d%< d&D ]
}t|d||< q|d' |d( |d) f|d*< |d' |d( |d) |d% f|d+< n
td,|d  d-|d |d	 f| _|ddr9|d nd|d > |d< |d.|d  krY|d dkrY|d|d  7 }t|d d/\| _}	| jd0u rttd1|d  d-|d | j
krg d2d3gd4d5gd6}
d7d8d9d:d:d;d<d=d>}|d |
v r|d d?kr|d+ |
|d  v r||d |d+ f }	d@|	v rd9n| j| _nI|d dAv r|d* |
|d  v r||d |d* f }	n,tdBtdB|d | jkr |d d?kr|dCkrdD\}	| _n
tdE|d  d-| jdFkrd|d   k rdGks)n tdH|d  d-|d }|||d  }dI}|d dkr@dJntt|d }t|D ]\}}||| || d  }|t|d krgdK}qL|r||d dkrudLndM| _| j}	ndF| _t|dkrd7nd;|| _|d | j	d< dNdd|d |d	 f|p| j  |	|d |d  dO d? dP@ |d ffg| _d0S )Qz Read relevant info about the BMPr   header_sizeZ	directionr   r%   widthr   ZheightZplanes   bitscompressionr&   Zpalette_padding)r   r   r   r         Zy_flipr   l        r   
   r   	data_size   r   Zpixels_per_meter   colorsc                 s   s    | ]}|d  V  qdS )o_C@Nr   ).0xr   r   r   	<genexpr>v   s    z'BmpImageFile._bitmap.<locals>.<genexpr>dpi4   )r_maskg_maskb_maska_mask$   r@   )r=   r>   r?   r=   r>   r?   Zrgb_maskZ	rgba_maskzUnsupported BMP header type ()   )NNNzUnsupported BMP pixel depth ()      r1   r%   rE   rF   r1         ~ r1   rF   rE   rH   r%   r%   r%   r%   rH   rE   rF   r%   rE   rF   r1   i   i     i |  i  rN   )r   r   r   r   ZXBGRRGBABGRAr   zBGR;16r   ))r   rD   )r   rK   )r   rI   )r   rG   )r   rJ   )r   rL   )r   rM   )r   rO   r   A)r   r   z Unsupported BMP bitfields layout   )rQ   rP   zUnsupported BMP compression (r
   i   zUnsupported BMP Palette size (Tr%   r1   F1LrawrN   )fpreadseekr    r   Z
_safe_readi16r(   tupleinfor)   len	enumerateOSError_sizegetBIT2MODEmodelistranger   r   rW   palettetellZtile)selfheaderoffsetrZ   r[   	file_infoZheader_dataidxmaskZraw_modeZ	SUPPORTEDZ
MASK_MODESZpaddingrh   Z	greyscaleindicesZindvalZrgbr   r   r   _bitmapH   s   



 





zBmpImageFile._bitmapc                 C   s6   | j d}t|stdt|d}| j|d dS )z-Open file, check magic number and read headerrC   zNot a BMP filer2   )rl   N)rY   rZ   r   SyntaxErrorr    rr   )rj   Z	head_datarl   r   r   r   _open  s
   
zBmpImageFile._openNr%   r%   )__name__
__module____qualname____doc__format_descriptionformatZCOMPRESSIONSitemskvvarsrr   rt   r   r   r   r   r#   <   s    
 >r#   c                   @   s   e Zd ZdZdZdd ZdS )DibImageFileZDIBr$   c                 C   s   |    d S )N)rr   )rj   r   r   r   rt     r"   zDibImageFile._openN)rv   rw   rx   r{   rz   rt   r   r   r   r   r     s    r   )rU   r   r   )rV   r      )r
   r   r   )r   r   r%   )rQ   r   r%   )rU   rV   r
   r   rP   c                 C   s   t | ||d d S )NF)_save)imrY   filenamer   r   r   	_dib_save,  s   r   Tc              
   C   s  z
t | j \}}}W n ty  } z
td| j d|d }~ww | j}|dd}	ttdd |	}
| jd | d d	 d
 d@ }d}|| jd  }|rrd| |d  }|| }|dkrat	d|
dt| td t|  |
t|t| jd  t| jd  td t| td t| t|
d  t|
d  t| t|  |
d|d   | jdkrdD ]}|
t|d  qn%| jdkrtdD ]}|
t|d  qn| jdkr|
| jdd t| |dd| j d||dffg d S )Nzcannot write mode z as BMPr;   )`   r   c                 S   s   t | d d S )Nr7   g      ?)int)r9   r   r   r   <lambda>;  s    z_save.<locals>.<lambda>r%   r0   r   r&   rX   r   r   rC   r   l    z)File size is too large for the BMP formatr       rU   rT   rV   r   r
   r   r   rW   ru   r+   )SAVEre   KeyErrorra   Zencoderinforc   r]   mapsize
ValueErrorwriteo32o16r   rg   r   Z
getpaletter   r   )r   rY   r   Zbitmap_headerZrawmoder.   r6   er^   r;   ZppmZstriderk   Zimagerl   	file_sizeir   r   r   r   0  sz   

	



(r   z.bmpz	image/bmpz.dibN)T) r   r   r   Z_binaryr   r\   r   r    r   r   r   r	   r   rd   r   r!   r#   r   r   r   r   Zregister_openr{   Zregister_saveZregister_extensionZregister_mimer   r   r   r   <module>   sD    Z	
@