a
    #Df.                     @   sp   d Z ddlZddlZddlmZmZmZ ee	Z
dZG dd deZG dd deZd	d
 ZG dd dZdS )z"
PIL formats for multiple images.
    N   )PillowFormatimage_as_uintndarray_to_pilc                   @   s   e Zd ZdZdZdS )
TIFFFormatizTIFF format (Pillow)N)__name__
__module____qualname___modes_description r   r   h/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/imageio/plugins/pillowmulti.pyr      s   r   c                   @   s*   e Zd ZdZdZdZG dd dejZdS )	GIFFormatz(See :mod:`imageio.plugins.pillow_legacy`ZiIz Static and animated gif (Pillow)c                   @   s&   e Zd ZdddZdd	 Zd
d ZdS )zGIFFormat.Writerr   N
      Fc                 C   s<  ddl m} tdd |dD \}}	}
|dkrJ|	dkrJtd| d	t|}|d
k sb|dkrjtd|dvrd
ttdd  }t	
d|  |d u rdt| | _n*t|ttfrdd |D | _n
t|| _t|}|dks|tdkrd}t|}t|}|rdnd
| _| j }t||||t|| _d S )Nr   )__version__c                 s   s   | ]}t |V  qd S N)int).0xr   r   r   	<genexpr>)       z)GIFFormat.Writer._open.<locals>.<genexpr>.r   r   zPillow vz is not supported by ImageIO's legacy pillow plugin when writing GIFs. Consider switching to the new plugin or downgrading to `pillow<10.1.0`.   r   z!GIF quantize param must be 2..256)r                @      r   r    g+?zDWarning: palettesize (%r) modified to a factor of two between 2-256.g      ?c                 S   s   g | ]}t |qS r   )float)r   dr   r   r   
<listcomp>?   r   z*GIFFormat.Writer._open.<locals>.<listcomp>inf)ZPILr   tuplesplitImportErrorr   
ValueErrornplog2loggerwarningr!   	_duration
isinstancelistbool_disposerequestZget_file	GifWriter_writer)selfloopdurationZfpsZpalettesize	quantizerZsubrectanglesZpillow_versionmajorminorpatchfpr   r   r   _open   s>    	


zGIFFormat.Writer._openc                 C   s   | j   d S r   )r4   closer5   r   r   r   _closeQ   s    zGIFFormat.Writer._closec                 C   s~   t |dd}|jdkr:|jd dkr:|d d d d df }| j}t|trd|tt|d | jj	 }| j
}| j||| d S )Nr   )Zbitdepth   r   r   )r   ndimshaper-   r.   r/   minlenr4   _countr1   	add_image)r5   immetar7   disposer   r   r   _append_dataT   s    
zGIFFormat.Writer._append_data)r   Nr   r   r   F)r   r	   r
   r=   r@   rL   r   r   r   r   Writer   s         
3rM   N)r   r	   r
   __doc__r   r   r   rM   r   r   r   r   r      s   r   c                 C   s   | j dddS )Nr   little)	byteorder)to_bytes)r   r   r   r   intToBina   s    rR   c                   @   sp   e Zd ZdZdddZdd Zd	d
 Zdd Zdd Zdd Z	dddZ
dd Zd ddZdd Zd!ddZdS )"r3   zClass that for helping write the animated GIF file. This is based on
    code from images2gif.py (part of visvis). The version here is modified
    to allow streamed writing.
    Tr   r   c                 C   sF   || _ || _|| _|| _|| _d | _d | _d| _ddlm	} || _	d S )Nr   )getdata)
r<   opt_subrectangleopt_loopopt_quantizeropt_palette_size_previous_image_global_paletterG   PIL.GifImagePluginrS   )r5   filerT   rU   rV   rW   rS   r   r   r   __init__k   s    zGifWriter.__init__c           	      C   s   |d }}| j r| |\}}| || j| j}ddlm} ||d d }| jdkrl| ||| j	 || _
| ||||| || _|  jd7  _d S )Nr   r   r   )	getheaderrA   r   )rT   getSubRectangleconverToPILrV   rW   rZ   r^   rG   write_headerrU   rY   write_imagerX   )	r5   rI   r7   rK   Zim_rectrectim_pilr^   paletter   r   r   rH      s    

zGifWriter.add_imagec                 C   s<   |  |}| |}| j| | j| | j| d S r   )getheaderAnim	getAppExtr<   write)r5   rI   ZglobalPaletter6   headerZappextr   r   r   ra      s
    

zGifWriter.write_headerc                 C   s   | j dd d S )N;utf-8)r<   rh   encoder?   r   r   r   r>      s    zGifWriter.closec                 C   s   | j }| |}d}|r4t|dk r4||d7 }qt|dksDJ | ||}	| ||}
|| jksn|dkr||
 ||	 || |d n||
 || |D ]}|| qd S )Nr      r   r      )r<   rS   rF   popgetImageDescriptorgetGraphicsControlExtrY   rh   )r5   rI   re   rc   r7   rK   r<   dataZimdesZlidZgraphextr"   r   r   r   rb      s"    





zGifWriter.write_imagec                 C   s4   d}|t |jd 7 }|t |jd 7 }|d7 }|S )z4Get animation header. To replace PILs getheader()[0]s   GIF89ar   r   s     rR   size)r5   rI   bbr   r   r   rf      s
    zGifWriter.getheaderAnimNc                 C   s`   |du rd}d}|t |d 7 }|t |d 7 }|t |jd 7 }|t |jd 7 }|d7 }|S )a  Used for the local color table properties per image.
        Otherwise global color table applies to all frames irrespective of
        whether additional colors comes in play that require a redefined
        palette. Still a maximum of 256 color per frame, obviously.

        Written by Ant1 on 2010-08-22
        Modified by Alex Robinson in Janurari 2011 to implement subrectangles.
        Nr]      ,r   r      rs   )r5   rI   Zxyru   r   r   r   rp      s    zGifWriter.getImageDescriptorc                 C   sP   |dkrdS |dkrd}d}|dkrLd}|d7 }|d7 }|t |7 }|d7 }|S )	z|Application extension. This part specifies the amount of loops.
        If loop is 0 or inf, it goes on infinitely.
        r   r   r   i  s   !s   NETSCAPE2.0s       )rR   )r5   r6   ru   r   r   r   rg      s    zGifWriter.getAppExt皙?r   c                 C   sJ   d}|t |d@ d> d7 }|tt|d d 7 }|d7 }|d7 }|S )ab  Graphics Control Extension. A sort of header at the start of
        each image. Specifies duration and transparancy.

        Dispose
        -------
          * 0 - No disposal specified.
          * 1 - Do not dispose. The graphic is to be left in place.
          * 2 - Restore to background color. The area used by the graphic
            must be restored to the background color.
          * 3 - Restore to previous. The decoder is required to restore the
            area overwritten by the graphic with what was there prior to
            rendering the graphic.
          * 4-7 -To be defined.
        s   !rA   r   rk   d   g      ?rx   )chrrl   rR   r   )r5   r7   rK   ru   r   r   r   rq      s    zGifWriter.getGraphicsControlExtc           
      C   s   | j dkr|dfS | j}t|| }|jdkr:|d}t|d}t|d}|jr|jrt|d t|d d  }}t|d t|d d  }}	nd\}}d\}}	|||	||f ||ffS )ar  Calculate the minimal rectangle that need updating. Returns
        a two-element tuple containing the cropped image and an x-y tuple.

        Calculating the subrectangles takes extra time, obviously. However,
        if the image sizes were reduced, the actual writing of the GIF
        goes faster. In some cases applying this method produces a GIF faster.
        r   r]   rA   r   r   rB   )r   r   )	rG   rX   r)   absrC   sumZargwherert   r   )
r5   rI   prevdiffXYZx0x1Zy0y1r   r   r   r_     s    



 zGifWriter.getSubRectanglec                 C   s   t |d}|dv r:d}|d}t||}|j||d}nB|dv rp|dkrV|d}n
|d}|j||d	}ntd
| |S )zConvert image to Paletted PIL image.

        PIL used to not do a very good job at quantization, but I guess
        this has improved a lot (at least in Pillow). I don't think we need
        neuqant (and we can add it later if we really want).
        Zgif)ZnqZneuquantr   ZRGBA)colors)r   r   r   r   ZRGB)r   methodzInvalid value for quantizer: %r)r   convertNeuQuantquantizer(   )r5   rI   r8   Zpalette_sizerd   Znq_samplefacZ
nqInstancer   r   r   r`   9  s    



zGifWriter.converToPIL)Tr   r   r   )N)ry   r   )r   )r   r	   r
   rN   r\   rH   ra   r>   rb   rf   rp   rg   rq   r_   r`   r   r   r   r   r3   e   s        
	!

 r3   )rN   loggingnumpyr)   Zpillow_legacyr   r   r   	getLoggerr   r+   r   r   r   rR   r3   r   r   r   r   <module>   s   
L