GifImagePlugin.pyc
1 o 2 Y��cЏ � @ s� d dl Z d dlZd dlZd dlZd dlmZ ddlmZmZm Z m 3 Z 4 m Z ddl m Z ddl mZ ddl mZ G dd � d e�ZejZd 5 d � ZG d d � d e j �Zdddd�Zdd� Zdd� Zdd� Zdd� Zdd� Z dd � Z d;d d �Z d d!� Z d"d#� Z d$d%� Z!d Z"d&d'� Z#d(d)� Z$d*d+� Z%d,d-� Z&d.d/� Z'd0d1� Z(d2d3� Z)d<d4d5�Z*d=d7d8�Z+e�,ej-ee� e�.ej-e � e�/ej-e � e�0ej-d9� e�1ej-d:� dS )>� N)�IntEnum� )�Image� 6 ImageChops� ImageFile� ImagePalette� ImageSequence)�i16le)�o8)�o16lec @ s e Zd ZdZdZdZdZdS )�LoadingStrategyz.. versionadded:: 9.1.0r r � N)�__name__� 7 __module__� __qualname__�__doc__�RGB_AFTER_FIRST� RGB_AFTER_DIFFERENT_PALETTE_ONLY� 8 RGB_ALWAYS� r r ��C:\Users\Jacks.GUTTSPC\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\LocalCache\local-packages\Python310\site-packages\PIL\GifImagePlugin.pyr ' s 9 r c C s | d d� dv S )N� )s GIF87as GIF89ar )�prefixr r r �_accept6 � r c s~ e Zd ZdZdZdZdZdd� Zdd� Zd d 10 � Z e 11 d d � �Z e 12 d d� �Z dd� Z ddd�Z� fdd�Zdd� Zdd� Z� ZS )� GifImageFileZGIFzCompuserve GIFFNc C s, | j �d�}|r|d r| j �|d �S d S )Nr r )�fp�read)�self�sr r r �dataG s zGifImageFile.datac C s\ t dt|�d�D ]#}|d || kr'||d kr'||d ks+ dS dS qdS )Nr � r r TF)�range�len)r �p�ir r r �_is_palette_neededM s 4��z GifImageFile._is_palette_neededc C s� | j �d�}t|�sd}t|��|d d� | jd< t|d�t|d�f| _g | _|d }|d@ d }|d @ rU|d 13 | jd < | j �d |> �}| �|�rUt � 14 d |�}| | _ | _ | j | _ | j �� | _d | _d | _| �d� d S )N� znot a GIF filer �version� � 15 � r � � � 16 backgroundr! �RGBr )r r r � SyntaxError�info�i16�_size�tiler& r �raw�global_palette�palette�_fp�tell�_GifImageFile__rewind� _n_frames� _is_animated�_seek)r r �msg�flags�bitsr$ r r r �_openS s( 17 zGifImageFile._openc C s\ | j d u r+| �� }z | �| �� d d� q ty% | �� d | _ Y nw | �|� | j S )NTr F)r; r9 r= �EOFError�seek�r �currentr r r �n_framesp s 18 � � 19 zGifImageFile.n_framesc C s~ | j d u r<| jd ur| jdk| _ | j S | �� }|r d| _ | j S z | �dd� d| _ W n ty6 d| _ Y nw | �|� | j S )Nr TF)r< r; r9 r= rB rC rD r r r � is_animated| s 20 21 � � 22 23 � 24 zGifImageFile.is_animatedc C s� | � |�sd S || jk rd | _| �d� | j}t| jd |d �D ]!}z| �|� W q! tyB } z | �|� d}t|�|�d }~ww d S )Nr r zno more images in GIF file)� _seek_check�_GifImageFile__frame�imr= r" rB rC )r �frame� 25 last_frame�f�er> r r r rC � s 26 27 28 29 30 ���zGifImageFile.seekTc s� |dkr!d� _ d � _d� _� j�� j� d� _d� jv r � jd= n � jr*|r*� � � |� jd kr:d|� �}t 31 |��� j� _ � j rT� j �� j � � � � rQ � � � sLd� _ � j � d�}|r`|dkrbt�d }i }d }d }d } |su� j � d�}|r{|dkr}�n:|dk�r#� j � d�}� � � } 32 |d d kr�| 33 d } | d@ r�| 34 d 35 }t| 36 d�d |d < d | @ } | d? } | r�| � _nb|d dkr�d} | 37 r�| | 38 7 } � � � } 39 | 40 s�d|v r�|d d| 7 < n| |d< d }ql|d dk�r|dk�r| 41 � j �� f|d< | 42 d d� dk�r� � � } 43 t| 44 �d 45 k�r| 46 d dk�rt| 47 d�� jd< � � � �r" � � � �s n�|dk�r�� j � d�}t|d�t|d�}}|t|d� |t|d� }}|� jd k�sX|� jd k�rr|�rrt|� jd �t|� jd �f� _t�� j� ||||f} |d } | d @ dk}| d @ �r�| d @ d }� j � d 48 |> �}� �|��r�t�d |�}nd }� j � d�d }� j �� � _ n d }qm|d u �r�t�|� _|�s�d S g � _� j�rֈ j�� j� j � |d u�r�|n� j � _ |� _ |dk�r� j �rt t!j"k�r�|d u�r�d!nd � _#nd"� _#nd#� _#|�s� j �rdd$l$m$} |� j �}|� _%nA� j#d"k�r\t t!j&k�s*|�r\d � _'d%� jv �rO� j�(� jd% d� � j�)d!tj*j+�� _d!� _#� jd%= n d � _#� j�)d tj*j+�� _� fd&d'�}| � _ z�� jdk �rpd � _n�� jdk�rĈ j \}}}}|| || f}t�|� d"}� j�,d%|�}|d u�r�� j#d(v �r�d!}||�d) }n� j�,d*d�}� j#d(v �r�d }||�}tj-�.|||�� _nD� jd u�rԈ �/� j� j �� _n4|d u�r� j \}}}}|| || f}t�|� d"}|}� j#d(v �r�d!}||�d) }tj-�.|||�� _W n 49 t0�y Y nw |d u�rId}|d u�r9|dk�r1t t!j"k�r0|� jd%< n� j#d(v�r9|}d+||||f� j |||ffg� _|�,d��rV|d � jd< d,D ]}||v �rh|| � j|< �qX|� jv �rr� j|= �qXd S )-Nr ������commentr zcannot seek to frame � ;T� !�� r! r* �duration� r �� � � 50 �� � extensionr- � NETSCAPE2.0�loop� ,� � r r) �@ r, r+ r/ F�RGBA�P�L)�copy� transparencyc s8 � j rt� j j| d | d d � �} | S | | | f} | S )Nr! )�_frame_palette�tupler7 )�color�r r r �_rgbR s 51 52 �z GifImageFile._seek.<locals>._rgb�r/ ra )r r. �gif)rT rZ )1Z_GifImageFile__offset�disposerI r8 rC r: Zdisposal_methodr1 r4 �load� 53 ValueErrorr r r rB r2 r9 r# �size�maxr3 r �_decompression_bomb_checkr& r r5 rJ �paste�dispose_extentr6 rf �_frame_transparency�LOADING_STRATEGYr r �moderd r7 r �pyaccess�putpalettealpha�convert�Dither�FLOYDSTEINBERG�get�core�fill�_crop�AttributeError)r rK Z update_imager> r r7 r1 Zframe_transparency� interlaceZframe_dispose_extent�blockr? Z dispose_bitsrP �x0�y0�x1�y1r@ r$ rd rj Z dispose_sizeZ dispose_moderh re �kr ri r r= � sn 54 � 55 56 � 57 � � 58 59 �� 60 &" 61 � 62 ` 63 64 � 65 66 67 � 68 69 �� 70 71 72 73 � 74 �� 75 ��zGifImageFile._seekc s� | j rdnd}d | _| jdkr | jd ur tj�|| j| j�| _n&| j dv rF| j| _| j rCtj�d| j| jp6d�| _| jj 76 | j � � � nd | _|| _ d | _ t � � � d S )Nrb rc r rk )rf �_prev_imrI ru r r~ r rp rJ rw � 77 putpalette�getdata�super� load_prepare)r Z temp_mode�� __class__r r r� � s" 78 79 80 �� 81 zGifImageFile.load_preparec C s� | j dkr0| jdkr.ttjkr.| jd ur | j�| jd� d| _nd| _| j�| jt j 82 j �| _d S | j s5d S | jd urI| j�| jd� | j�d�}n| j�d�}| � || j�}| j | _| jj| _|jdkro| j�|| j|� d S | j�|| j� d S )Nr rb ra r/ )rI rw rv r r ru rJ ry rz r r{ r| r� r� rt rs )r Zframe_imr r r �load_end� s( 83 84 85 86 87 zGifImageFile.load_endc C s | j S �N)rI ri r r r r9 � s zGifImageFile.tell)T)r r r �format�format_description�!_close_exclusive_fp_after_loadingr6 r r& rA �propertyrF rG rC r= r� r� r9 � __classcell__r r r� r r ? s& 88 89 90 r rc rb )�1rc rb c C s� | j tv r | �� | S t�| j �dkr?| jdtjjd�} | jj dkr=| jj � 91 � D ]}|d dkr<| jj | | j d< | S q(| S | �d�S ) a 92 Takes an image (or frame), returns an image in a mode that is appropriate 93 for saving in a Gif. 94 95 It may return the original image, or it may return an image converted to 96 palette or 'L' mode. 97 98 :param im: Image object 99 :returns: Image object 100 r/ rb �r7 ra r! r re rc ) rw �RAWMODErn r � getmodebaserz �Palette�ADAPTIVEr7 �colors�keysr1 )rJ �rgbar r r �_normalize_mode� s 101 � 102 r� c C sX d}|r t |tttf�rt|dd� �}t |tj�r t|j�}| jdkr1|s0| j�d�dd� }n|s>tdd� t d�D ��}tjd|d�| _|r�g }t dt 103 |�d �D ] }t |||d � �}| jj � |�}||v rkd}|�|� qRt|�D ]\}}|du r�t t 104 |��D ] }||vr�|||< nq�qu| �|�} nt| |�}|dur�| �||�S || j_| S ) 105 at 106 Normalizes the palette for image. 107 - Sets the palette to the incoming palette, if provided. 108 - Ensures that there's a palette for L mode images 109 - Optimizes the palette if necessary/desired. 110 111 :param im: Image object 112 :param palette: bytes object containing the source palette, or .... 113 :param info: encoderinfo 114 :returns: Image object 115 Ni rb r/ c s s � | ]}|d V qdS )r! Nr )�.0r% r r r � <genexpr> s � z%_normalize_palette.<locals>.<genexpr>r� r r! )� 116 isinstance�bytes� bytearray�listr r7 rw rJ � 117 getpaletter" r# rg r� r} �append� enumerate� remap_palette� _get_optimize) rJ r7 r1 �source_palette�used_palette_colorsr% Z source_color�index�jr r r �_normalize_palette� sF 118 119 � �� 120 r� c C s� t | �}|j�� D ] \}}| j�||� q t||| j�}t|| j�D ]}|�|� q"d}t| �r4|dB }t || d|� dt| �f|_ 121 t � ||dd| j dt|j fg� |�d� d S )Nr r` �r r r) rl � )r� r1 �items� encoderinfo� 122 setdefaultr� �_get_global_header�write� get_interlace�_write_local_header� encoderconfigr �_saverp r� rw )rJ r r7 Zim_outr� �vr r? r r r �_write_single_frame& s $r� c C sD t |�t | �krt�|| �}|�� S t�|�d�| �d��}|�� S )Nr/ )�_get_palette_bytesr �subtract_modulorz �getbbox)�base_im�im_frame�deltar r r �_getbbox; s ��r� c C s� | j �d�}| j �d| j�d��}g }d}d }t�| g| j �dg ��D ]�}t�|�D ]�} t| �� �} |dkrL| j� � D ]\} 123 } | 124 dkrDq;| j � 125 | 126 | � q;| j �� } t | || �} d| jv re| � 127 d| jd � t |t tf�rs|| | d< n|d u r�d| jv r�| jd | d< t |t tf�r�|| | d< |d7 }|r�|d } t| d | �}|s�| �d�r�| d d | d 7 < q*| �d�d 128 kr�|d u r�| j �d| j�dd ��}t| |�}t�d | j|�}|�|d d j� t|| �}nd }|�| || d �� q*q#t|�dk�rA|D ]?}|d } |d �s t| |d �D ]}|�|� �qd}n|�s&d|d d< | �|d �} |d d d 129 � }t|| ||d � q�dS d| j v �r^t | j d t tf��r`t | j d �| j d< d S d S d S )NrT �disposalr � append_imagesre r rO rJ r� r )r r r rb )rJ �bboxr� r� r� T�include_color_table) r� r} r1 � itertools�chainr �Iteratorr� rd r� r� r� r� r� rg r� �_get_backgroundr �newrp r� r7 r� r# r� r� �crop�_write_frame_data�sum)rJ r r7 rT r� � im_frames� frame_countZ background_im� 130 imSequencer� r� r� r� �previousr� rh r. � 131 frame_datar �offsetr r r �_write_multiple_framesE s| 132 133 134 � 135 136 ��+ 137 ��r� c C s t | ||dd� d S )NT)�save_all)r� )rJ r �filenamer r r � _save_all� s r� Fc C s� d| j v s 138 d| jv r| j �d| j�d��}n d }| j �dd�| j d< |r*t| ||�s0t| ||� |�d� t|d�r@|�� d S d S )Nr7 �optimizeTrQ �flush)r� r1 r} r� r� r� �hasattrr� )rJ r r� r� r7 r r r r� � s 139 140 �r� c C s$ | j �dd�}t| j�dk rd}|S )Nr� r � r )r� r} �minrp )rJ r� r r r r� � s r� c C s� d}zd|j v r|j d }n|jd }t|�}W n ttfy# Y n w d}t||j �}|d urCz|�|�}W n tyB d}Y nw d|j v rRt|j d d �}nd}t|j �dd��}|se|dkse|r�|ridnd} | |d > O } |sud}| �d 141 t d � t d � t | � t 142 |� t |� t d� � |j �d �} 143 | 144 r�t |�} t | �} | r�|dB }|| B }| �dt 145 |d � t 146 |d � t 147 |j d � t 148 |j d � t |� � | 149 r�| r�| �t| �� | �t d�� d S )NFre TrT r* r r� r r rR rS r_ r� r, r] r) )r� r1 �int�KeyErrorro r� r� r} r� r 150 �o16r� �_get_color_table_sizerp �_get_header_palette) r rJ r� r? Ztransparent_color_existsre r� rT r� Z packed_flagr� � palette_bytes�color_table_sizer r r r� � s� 151 152 � � 153 ������� 154 155 � 156 � � ���r� c 157 C s | � � }ztt|d��S}| jdkr tjd|g|tjd� n:dd|g}dg}tj|tjtjd�}tj||j|tjd�}|j� � |� 158 � } | rJt� | |��|� 159 � } | rVt� | |��W d � n1 s`w Y W zt � |� W d S tyx Y d S w zt � |� W w ty� Y w w )N�wbr/ Zppmtogif)�stdout�stderrZppmquant�256)�stdinr� r� )�_dump�openrw � 160 subprocess� 161 check_call�DEVNULL�Popen�PIPEr� �close�wait�CalledProcessError�os�unlink�OSError) 162 rJ r r� �tempfilerM Z quant_cmdZ togif_cmdZ 163 quant_procZ 164 togif_proc�retcoder r r � _save_netpbm� sL 165 � 166 167 �� 168 � �� �� �r� c C s� | j dv ra|rc|�dd�retp| j dk}|s | j| j dk rgg }t| �� �D ] \}}|r1|�|� q&|s<t|�t |�kr>|S t | j 169 j 170 �t � | j 171 j � }d|d � � > }t |�|d kri|dkrk|S dS dS dS dS dS dS ) aL 172 Palette optimization is a potentially expensive operation. 173 174 This function determines if the palette should be optimized using 175 some heuristics, then returns the list of palette entries in use. 176 177 :param im: Image object 178 :param info: encoderinfo 179 :returns: list of indexes of palette entries in use, or None 180 )rb rc r� r rc i r r N)rw r} �_FORCE_OPTIMIZE�width�heightr� � histogramr� rq r# r7 r � getmodebands� 181 bit_length)rJ r1 Zoptimiser� r% �countZnum_palette_colorsZcurrent_palette_sizer r r r� ' s* 182 �� � r� c C s6 | sdS t | �dk r dS t�t�t | �d d��d S )Nr r^ r r! r )r# �math�ceil�log)r� r r r r� U s 183 r� c C s< t | �}d|> t| �d }|dkr | td�d | 7 } | S )z� 184 Returns the palette, null padded to the next power of 2 (*3) bytes 185 suitable for direct inclusion in the GIF header 186 187 :param palette_bytes: Unpadded palette bytes, in RGBRGB form 188 :returns: Null padded palette 189 r r! r )r� r# r 190 )r� r� Zactual_target_size_diffr r r r� _ s 191 r� c C s | j j S )z� 192 Gets the palette for inclusion in the gif header 193 194 :param im: Image object 195 :returns: Bytes, len<=768 suitable for inclusion in gif header 196 r� )rJ r r r r� q s r� c 197 C sd d}|r0t |t�r.z 198 | j�|| �}W |S ty- } zt|�dvr"� W Y d }~|S d }~ww |}|S )Nr )z$cannot allocate more than 256 colorsz/cannot add non-opaque RGBA color to RGB palette)r� rg r7 �getcolorro �str)rJ Zinfo_backgroundr. rN r r r r� { s 199 � 200 � 201 �� 202 r� c C sr d}| j �d�dks |r d|v s d|v s |�d�s |�d�r d}t| |�d��}t| �}t|�}d | t| jd 203 � t| jd � t|d �t|�td 204 � t|�g}d|v rw|� d td� td� d td� td � t|d � td 205 � � |�d�r�d td� }|d }t 206 |t �r�|� � }t d 207 t|�d�D ]} || | d � } 208 |tt| 209 ��| 210 7 }q�|td 211 �7 }|� |� |S )z2Return a list of strings representing a GIF headers 87ar( s 89are r\ rT rP r. s GIFr r r, rR rY r- r[ r! rV )r1 r} r� r� r� r� rp r 212 r� r� r� r �encoder" r# ) rJ r1 r( r. r� r� �headerZ comment_blockrP r% Zsubblockr r r r� � sl ���� � � 213 � ����� 214 ��� 215 216 217 218 r� c C sT z&||_ t| ||d� t�|| dd|j dt|j fg� | �d� W |` d S |` w )Nr rl r� r� )r� r� r r� rp r� rw r� )r r� r� �paramsr r r r� � s � r� c C sd t | |�}|du r i }d|vrd| jv r| jd |d< t| ||�}|j| _|j| _t| |�}||fS )a 219 Legacy Method to get Gif data from image. 220 221 Warning:: May modify image data. 222 223 :param im: Image object 224 :param palette: bytes object containing the source palette, or .... 225 :param info: encoderinfo 226 :returns: tuple of(list of header items, optimized palette) 227 228 Nr. )r� r1 r� r7 rJ r� )rJ r7 r1 r� Zim_modr r r r � getheader� s 229 230 r r� c K s0 G dd� d�}| � � |� }t|| ||� |jS )a� 231 Legacy Method 232 233 Return a list of strings representing this image. 234 The first string is a local image header, the rest contains 235 encoded image data. 236 237 To specify duration, add the time in milliseconds, 238 e.g. ``getdata(im_frame, duration=1000)`` 239 240 :param im: Image object 241 :param offset: Tuple of (x, y) pixels. Defaults to (0, 0) 242 :param \**params: e.g. duration or other encoder info parameters 243 :returns: List of bytes containing GIF encoded frame data 244 245 c @ s e Zd Zg Zdd� ZdS )zgetdata.<locals>.Collectorc S s | j �|� d S r� )r r� )r r r r r r� r z getdata.<locals>.Collector.writeN)r r r r r� r r r r � Collector s r )rn r� r )rJ r� r 246 r r r r r r� s 247 r� z.gifz image/gif)F)NN)r� )2r� r r� r� �enumr � r r r r r �_binaryr r2 r 248 r r� r r rv r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� � register_openr� � register_save�register_save_all�register_extension� register_mimer r r r �<module> sT 4 249 L 250 251 B2. 252 253 > 254 255 $