a
    ׶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   b/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/PIL/BmpImagePlugin.py_accept1   s    r   c                 C   s   t | dv S )N)   (   @   l   |   )i32r   r   r   r   _dib_accept5   s    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   s  | j j| j j }}|r|| i }t|d|d< d|d< t| j |d d }|d dkrt|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 rdnd|d< t|d|d< |d st|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rRt|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rz|d n
d|d > |d< |d.|d  kr|d dkr|d|d  7 }t|d d/\| _}	| jd0u rt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| _nB|d dAv r|d* |
|d  v r||d |d* f }	ntdBntdBnH|d | jkr|d d?kr |dCkr dD\}	| _ntdE|d  d-| jdFkr(d|d   k rHdGks`n tdH|d  d-n|d }|||d  }dI}|d dkrdJntt|d }t|D ]8\}}||| || d  }|t|d krdK}q|r|d dkrdLndM| _| j}	n"dF| _t|dkrd7nd;|| _|d | j	d< dNdd|d |d	 f|pX| j  |	|d |d  dO d? dP@ |d ffg| _d0S )Qz Read relevant info about the BMPr   header_size	directionr   r$   widthr   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       z'BmpImageFile._bitmap.<locals>.<genexpr>dpi4   )r_maskg_maskb_maska_mask$   rB   )r?   r@   rA   r?   r@   rA   Zrgb_maskZ	rgba_maskzUnsupported BMP header type ()   )NNNzUnsupported BMP pixel depth ()      r2   r$   rG   rH   r2         ~ r2   rH   rG   rJ   r$   r$   r$   r$   rJ   rG   rH   r$   rG   rH   r2   i   i     i |  i  rP   )r   r   r   r   ZXBGRRGBABGRAr   zBGR;16r   ))r   rF   )r   rM   )r   rK   )r   rI   )r   rL   )r   rN   )r   rO   )r   rQ   r   A)r   r   z Unsupported BMP bitfields layout   )rS   rR   zUnsupported BMP compression (r
   i   zUnsupported BMP Palette size (Tr$   r2   F1LrawrP   )fpreadseekr    r   
_safe_readi16r'   tupleinfor(   len	enumerateOSError_sizegetBIT2MODEmodelistranger   r   rY   palettetellZtile)selfheaderoffsetr\   r]   	file_infoZheader_dataidxmaskZraw_modeZ	SUPPORTEDZ
MASK_MODESpaddingrk   Z	greyscaleindices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 headerrE   zNot a BMP filer3   )ro   N)r[   r\   r   SyntaxErrorr    rw   )rm   Z	head_dataro   r   r   r   _open  s
    
zBmpImageFile._openN)r$   r$   )__name__
__module____qualname____doc__format_descriptionformatZCOMPRESSIONSitemskvvarsrw   ry   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)rw   )rm   r   r   r   ry     s    zDibImageFile._openN)rz   r{   r|   r   r~   ry   r   r   r   r   r     s   r   )rW   r   r   )rX   r      )r
   r   r   )r   r   r$   )rS   r   r$   )rW   rX   r
   r   rR   c                 C   s   t | ||d d S )NF)_save)imr[   filenamer   r   r   	_dib_save,  s    r   Tc              
   C   s  zt | j \}}}W n8 tyL } z td| j d|W Y d }~n
d }~0 0 | j}|dd}	ttdd |	}
| jd | d d	 d
 d@ }d}|| jd  }|rd| |d  }|| }|dkrt	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  qnP| 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 )Nr8   g      ?)int)r:   r   r   r   <lambda>;  r<   z_save.<locals>.<lambda>r$   r1   r   r%   rZ   r   r   rE   r   l    z)File size is too large for the BMP formatr       rW   rV   rX   r   r
   r   r   rY   )r$   r$   r*   )SAVErh   KeyErrorrd   Zencoderinforf   r`   mapsize
ValueErrorwriteo32o16r   rj   r   Z
getpaletter   r   )r   r[   r   Zbitmap_headerrawmoder/   r7   era   r=   ppmZstridern   imagero   	file_sizeir   r   r   r   0  sr    *

	
r   z.bmpz	image/bmpz.dibN)T) r   r   r   _binaryr   r_   r   r    r   r   r   r	   r   rg   r   r!   r"   r   r   r   r   Zregister_openr   Zregister_saveZregister_extensionZregister_mimer   r   r   r   <module>   sB    Z	
@