o
    g1                     @   st   d dl Z ddlmZmZ ddlmZ ddddd	d
ddZdd ZG dd dejZ	e
e	je	e ee	jd dS )    N   )Image	ImageFile)i32le)AL)r   r   )RGBALA)RGBzYCC;P)r   zYCCA;P)r
   r
   )r   r   ))i  )i   )i  i )i   i  i  )i  i i i )i   i  i  )i  i i i c                 C   s   | d d t jkS )N   )olefileMAGIC)prefix r   P/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/PIL/FpxImagePlugin.py_accept*   s   r   c                       sT   e Zd ZdZdZdd ZdddZdd	d
Zdd Z fddZ	 fddZ
  ZS )FpxImageFileFPXFlashPixc              
   C   sb   z	t | j| _W n ty } zd}t||d }~ww | jjjdkr*d}t|| d d S )Nz!not an FPX file; invalid OLE filez$56616700-C154-11CE-8553-00AA00A1F95Bznot an FPX file; bad root CLSIDr   )	r   	OleFileIOfpoleOSErrorSyntaxErrorrootclsid_open_index)selfemsgr   r   r   _open6   s   
zFpxImageFile._openr   c           
      C   s   | j d|ddg}|d |d f| _t| j}d}|dkr,|d }|d7 }|dks |d | _| jd	> }|d
|B  }g }t|d}|dkrMd}	t|	t|D ]}|	t|d|d  d@  qQt
t| \| _| _i | _tdD ]}d|d	> B }||v r|| | j|< qs| d| j d S )NData Object Store 06dzImage Contentsi  i  r   @         i     zInvalid number of bandsr   i   i  )r   getproperties_sizemaxsizemaxidi32r   rangeappendMODEStuplemoderawmodejpeg_open_subimage)
r   indexpropr+   iidscolorsbandsr   r   r   r   r   G   s8   



zFpxImageFile._open_indexr   c              
   C   sB  d|dd|ddg}| j |}|d |d}t|dt|d	f}t|d
t|df}t|d}t|d}	|| jkrFd}
t|
|d|  |t|d|	 }d }}|\}}|\}}g | _tdt	||	D ]}t
||| }t
||| }t||d	 }|dkr| jd||||ft||d | jff nb|dkr| jd||||ft||d | j|dd
 ff nC|dkr|d }|d }| j}|r|dkrd\}}nd }n|}| jd||||ft||d ||ff |r| j| | _nd}
t|
|| }||krd|| }}||kr nqn|| _d | _d S )Nr!   r"   zResolution 04dzSubimage 0000 Header   $   r&   r   r%          zsubimage mismatch   r   rawr   fillr$         r   )YCbCrKCMYKr4   zunknown/invalid compression)r   
openstreamreadr-   r+   r   seektiler.   lenminr/   r3   r4   tile_prefixstreamr   )r   r6   subimagerP   r   r:   r+   tilesizeoffsetlengthr   xyxsizeysizextileytiler8   x1y1compressioninternal_color_conversionjpeg_tablesr3   jpegmoder   r   r   r5   x   s   







	
	




zFpxImageFile._open_subimagec                 C   s0   | j s| j| jd d dg | _ tj| S )Nr$   zSubimage 0000 Data)r   r   rI   rP   r   loadr   r   r   r   ra      s   zFpxImageFile.loadc                    s   | j   t   d S N)r   closesuperrb   	__class__r   r   rd         
zFpxImageFile.closec                    s   | j   t   d S rc   )r   rd   re   __exit__)r   argsrf   r   r   ri      rh   zFpxImageFile.__exit__)r   )r   r   )__name__
__module____qualname__formatformat_descriptionr    r   r5   ra   rd   ri   __classcell__r   r   rf   r   r   2   s    

1pr   z.fpx)r    r   r   _binaryr   r-   r0   r   r   register_openrn   register_extensionr   r   r   r   <module>   s     J