o
    tBh|?                     @   sT  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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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d'e eejee eeje eejd( eejd& dS )*    N   )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   i/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/PIL/BmpImagePlugin.py_accept3   s   r   c                 C   s   t | dv S )N)   (   @   l   |   )i32r   r   r   r   _dib_accept7      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 BitmapBMPr   r   r      r      )RAWRLE8RLE4	BITFIELDSJPEGPNGc              	   C   s  | 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-d2}
|d | j
krg d3d4gd5d6gd7}d8d9d:d;d;d<d=d>d?}|d |v r|d d@kr|d+ ||d  v r||d |d+ f }	dA|	v rd:n| j| _nT|d dBv r|d* ||d  v r||d |d* f }	n7tdCtdC|d | jkr|d d@kr|dDkrdE\}	| _n|d | jkrdF}
n
tdG|d  d-| jdHkrd|d   k r+dIks6n tdJ|d  d-|d }|||d  }dK}|d dkrMdLntt|d }t|D ]\}}||| || d  }|t|d krtdM}qY|r|d dkrdNndO| _| j}	ndH| _t|dkrd8nd<|| _|d | j	d< |
dd|d |d	 f|p| j  |	|d |d  dP d? dQ@ |d ffg| _d0S )Rz Read relevant info about the BMPr   header_size	directionr   r   widthr   heightplanes   bitscompressionr'   palette_padding)r   r   r   r          y_flipr   l        r   
   r   	data_size   r   pixels_per_meter   colorsc                 s   s    | ]}|d  V  qdS )o_C@Nr   ).0xr   r   r   	<genexpr>x   s    z'BmpImageFile._bitmap.<locals>.<genexpr>dpi4   )r_maskg_maskb_maska_mask$   rK   )rH   rI   rJ   rH   rI   rJ   rgb_mask	rgba_maskzUnsupported BMP header type ()   )NNNzUnsupported BMP pixel depth (raw)      r:   r   rS   rT   r:         ~ r:   rT   rS   rV   r   r   r   r   rV   rS   rT   r   rS   rT   r:   i   i     i |  i  r\   )r   r   r   r   XBGRRGBABGRAr   zBGR;16r   ))r   rR   )r   rY   )r   rW   )r   rU   )r   rX   )r   rZ   )r   r[   )r   r]   r   A)r   r   z Unsupported BMP bitfields layout   )r`   r_   bmp_rlezUnsupported BMP compression (r   i   zUnsupported BMP Palette size (Tr   r:   F1Lr\   )fpreadseekr!   r   
_safe_readi16r)   tupleinfor,   len	enumerateOSError_sizegetBIT2MODEmoder*   listranger   r   rQ   palettetelltile)selfheaderoffsetri   rj   	file_infoheader_dataidxmaskraw_modedecoder_name	SUPPORTED
MASK_MODESpaddingrx   	greyscaleindicesindvalrgbr   r   r   _bitmapJ   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 headerrP   zNot a BMP filer<   )r}   N)rh   ri   r   SyntaxErrorr!   r   )r{   	head_datar}   r   r   r   _open
  s
   
zBmpImageFile._openNr   r   )__name__
__module____qualname____doc__format_descriptionformatCOMPRESSIONSitemskvvarsr   r   r   r   r   r   r$   >   s    
 Ar$   c                   @   s   e Zd ZdZdd ZdS )BmpRleDecoderTc           
      C   s  t  }d}t|| jj| jj k r| jd}| jd}|r!|s"n|d }|rD|| | jjkr9td| jj| }||| 7 }||7 }n|d dkret|| jj dkrb|d7 }t|| jj dksTd}nc|d dkrlnh|d dkr| jd}t|dk rnU| jd\}}	|d||	| jj   7 }t|| jj }n,| j|d }||7 }t||d k rn#||d 7 }| j d dkr| j	dt
j t|| jj| jj k s| t|dd| jd f dS )Nr   r       r   r   r0   )r0   r   )	bytearrayro   statexsizeysizefdri   maxry   rj   osSEEK_CUR
set_as_rawbytesargs)
r{   bufferdatarD   pixelsbyte
num_pixels
bytes_readrightupr   r   r   decode  sJ   
)zBmpRleDecoder.decodeN)r   r   r   	_pulls_fdr   r   r   r   r   r     s    r   c                   @   s   e Zd ZdZdZdd ZdS )DibImageFileDIBr%   c                 C   s   |    d S )N)r   )r{   r   r   r   r   R  r#   zDibImageFile._openN)r   r   r   r   r   r   r   r   r   r   r   M  s    r   )re   r   r   )rf   r      )r   r   r   )r   r   r   )r`   r   r   )re   rf   r   r   r_   c                 C   s   t | ||d d S )NF)_save)imrh   filenamer   r   r   	_dib_saved  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 BMPrF   )`   r   c                 S   s   t | d d S )NrB   g      ?)int)rD   r   r   r   <lambda>s  s    z_save.<locals>.<lambda>r   r9   r   r'   rg   r   r   rP   r   l    z)File size is too large for the BMP formatr   r   re   rd   rf   r   r   r   r   rQ   r   r0   )SAVEru   KeyErrorrq   encoderinfors   rm   mapsize
ValueErrorwriteo32o16r   rw   r   
getpaletter   r   )r   rh   r   bitmap_headerrawmoder6   rA   ern   rF   ppmstrider|   imager}   	file_sizeir   r   r   r   h  sz   

	



(r   z.bmpz	image/bmprc   z.dib)T)r    r   r   r   _binaryr   rl   r   r!   r   r	   r   r
   r   rt   r   r"   r$   	PyDecoderr   r   r   r   r   register_openr   register_saveregister_extensionregister_mimeregister_decoderr   r   r   r   <module>   sJ    Z6	
@