OwlCyberSecurity - MANAGER
Edit File: animation.cpython-38.pyc
U ��]~ � @ s� d dl Z d dlZd dlZd dlmZmZ d dlZd dlZd dlZd dl m Z d dl Z d dl Z d dl Z d dlZd dlmZ d dlZd dlZd dlZd dlmZmZmZmZ d dlmZm Z m Z m Z e� e �Z!e �"� dkr�d Z#Z$nd Z#d d � Z%G d d � d e&�Z'e'� Z(G d d� de j)�Z*G dd� de*�Z+G dd� de+�Z,e(�-d�G dd� de+��Z.G dd� de&�Z/e(�-d�G dd� de/e+��Z0e(�-d�G d d � d e/e,��Z1G d d � d e/�Z2e(�-d �G d!d"� d"e2e0��Z3e(�-d#�G d$d%� d%e2e1��Z4G d&d'� d'e&�Z5e(�-d(�G d)d*� d*e5e+��Z6e(�-d+�G d,d-� d-e5e,��Z7d.d/� Z8d0d1� Z9e(�-d2�G d3d4� d4e,��Z:G d5d6� d6e&�Z;G d7d8� d8e;�Z<G d9d:� d:e<�Z=G d;d<� d<e<�Z>dS )=� N)�BytesIO� TextIOWrapper)�Path)�TemporaryDirectory)�DISPLAY_TEMPLATE�INCLUDED_FRAMES� JS_INCLUDE� STYLE_INCLUDE)�cbook�rcParams�rcParamsDefault� rc_contextZWindowsi c C sP dd� }t | | | �| | }t || | �| | }||||�||||�fS )a# Compute figure size so that pixels are a multiple of n Parameters ---------- w, h : float Size in inches dpi : float The dpi n : int The target multiple Returns ------- wnew, hnew : float The new figure size in inches. c S sv t | | �| dkrrt t�| tj�| �| dkrBt�| tj�} n0t t�| tj �| �| dkrrt�| tj �} | S �Nr )�int�npZ nextafter�inf)�x�dpi�n� r �6/usr/lib/python3/dist-packages/matplotlib/animation.py�correct_roundoffT s z*adjusted_figsize.<locals>.correct_roundoff)r )�w�hr r r ZwnewZhnewr r r �adjusted_figsize> s r c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�MovieWriterRegistryz<Registry of available writer classes by human readable name.c C s t � | _t � | _d| _d S �NF)�dict�avail� _registered�_dirty��selfr r r �__init__d s z MovieWriterRegistry.__init__c C s d| _ dS )z$Sets a flag to re-setup the writers.TN)r r! r r r � set_dirtyi s z MovieWriterRegistry.set_dirtyc s � �fdd�}|S )z�Decorator for registering a class under a name. Example use:: @registry.register(name) class Foo: pass c s | �j � < | �� r | �j� < | S �N)r � isAvailabler )� writerClass��namer" r r �wrapperv s z-MovieWriterRegistry.register.<locals>.wrapperr )r" r) r* r r( r �registerm s z MovieWriterRegistry.registerc C s | j r| �� dS )z2If dirty, reasks the writers if they are availableN)r �reset_available_writersr! r r r �ensure_not_dirty} s z$MovieWriterRegistry.ensure_not_dirtyc C s dd� | j �� D �| _d| _dS )z3Reset the available state of all registered writersc S s i | ]\}}|� � r||�qS r )r& )�.0r) r' r r r � <dictcomp>� s � z?MovieWriterRegistry.reset_available_writers.<locals>.<dictcomp>FN)r �itemsr r r! r r r r, � s �z+MovieWriterRegistry.reset_available_writersc C s | � � t| j�S )z%Get a list of available MovieWriters.)r- �listr r! r r r r1 � s zMovieWriterRegistry.listc C s | � � || jkS )z�Check if given writer is available by name. Parameters ---------- name : str Returns ------- available : bool )r- r �r" r) r r r � is_available� s z MovieWriterRegistry.is_availablec C sJ | � � | jstd��z | j| W S tk rD td�|���Y nX d S )NzNo MovieWriters available!z(Requested MovieWriter ({}) not available)r- r � RuntimeError�KeyError�formatr2 r r r � __getitem__� s �z MovieWriterRegistry.__getitem__N) �__name__� __module__� __qualname__�__doc__r# r$ r+ r- r, r1 r3 r7 r r r r r b s r c @ sJ e Zd ZdZejd dd��Zejdd� �Zejdd� �Ze j d d � �Z dS ) �AbstractMovieWritera� Abstract base class for writing movies. Fundamentally, what a MovieWriter does is provide is a way to grab frames by calling grab_frame(). setup() is called to start the process and finish() is called afterwards. This class is set up to provide for writing movie frame data to a pipe. saving() is provided as a context manager to facilitate this process as:: with moviewriter.saving(fig, outfile='myfile.mp4', dpi=100): # Iterate over frames moviewriter.grab_frame(**savefig_kwargs) The use of the context manager ensures that setup() and finish() are performed as necessary. An instance of a concrete subclass of this class can be given as the ``writer`` argument of `Animation.save()`. Nc C s dS )a� Perform setup for writing the movie file. Parameters ---------- fig : `~matplotlib.figure.Figure` The figure object that contains the information for frames outfile : string The filename of the resulting movie file dpi : int, optional The DPI (or resolution) for the file. This controls the size in pixels of the resulting movie file. Default is ``fig.dpi``. Nr �r" �fig�outfiler r r r �setup� s zAbstractMovieWriter.setupc K s dS )�� Grab the image information from the figure and save as a movie frame. All keyword arguments in savefig_kwargs are passed on to the `savefig` command that saves the figure. Nr �r" �savefig_kwargsr r r � grab_frame� s z AbstractMovieWriter.grab_framec C s dS �z,Finish any processing for writing the movie.Nr r! r r r �finish� s zAbstractMovieWriter.finishc o s0 | j |||f|�|� z | V W 5 | �� X dS )z� Context manager to facilitate writing the movie file. ``*args, **kw`` are any parameters that should be passed to `setup`. N)r@ rF )r" r> r? r �args�kwargsr r r �saving� s zAbstractMovieWriter.saving)N) r8 r9 r: r; �abc�abstractmethodr@ rD rF � contextlib�contextmanagerrI r r r r r< � s r< c @ s� e Zd ZdZd dd�Zedd� �Zdd � Zd d d �Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zedd� �Zedd� �ZdS ) � MovieWritera� Base class for writing movies. This is a base class for MovieWriter subclasses that write a movie frame data to a pipe. You cannot instantiate this class directly. See examples for how to use its subclasses. Attributes ---------- frame_format : str The format used in writing frame data, defaults to 'rgba' fig : `~matplotlib.figure.Figure` The figure to capture data from. This must be provided by the sub-classes. � Nc C s� | j tkrtd��|| _d| _|dkr2td | _n|| _|dkrLtd | _n|| _|dkrltt| j �| _ n|| _ |dkr�t � | _ n|| _ dS )a� MovieWriter Parameters ---------- fps : int Framerate for movie. codec : string or None, optional The codec to use. If ``None`` (the default) the ``animation.codec`` rcParam is used. bitrate : int or None, optional The bitrate for the saved movie file, which is one way to control the output file size and quality. The default value is ``None``, which uses the ``animation.bitrate`` rcParam. A value of -1 implies that the bitrate should be determined automatically by the underlying utility. extra_args : list of strings or None, optional A list of extra string arguments to be passed to the underlying movie utility. The default is ``None``, which passes the additional arguments in the ``animation.extra_args`` rcParam. metadata : Dict[str, str] or None A dictionary of keys and values for metadata to include in the output file. Some keys that may be of use include: title, artist, genre, subject, copyright, srcform, comment. zNMovieWriter cannot be instantiated directly. Please use one of its subclasses.�rgbaN�animation.codec�animation.bitrate) � __class__rN � TypeError�fps� frame_formatr �codec�bitrater1 �args_key� extra_argsr �metadata)r" rU rW rX rZ r[ r r r r# � s$ � zMovieWriter.__init__c C s* | j �� \}}t|| j �t|| j �fS )z7A tuple ``(width, height)`` in pixels of a movie frame.)r> �get_size_inchesr r )r" r r r r r � frame_size6 s zMovieWriter.frame_sizec C s� | j dkrb| j�� \}}t||| jd�\}}||f||fksp| jj||dd� t�d||||� n| j�� \}}tjd| j �� ||fS )N�h264� T)Zforwardz>figure size (inches) has been adjusted from %s x %s to %s x %s� frame size in pixels is %s x %s)r` ) rW r> r\ r r �set_size_inches�_log�info�debugr] )r" ZwoZhor r r r r �_adjust_frame_size< s �z MovieWriter._adjust_frame_sizec C s> || _ || _|dkr | jj}|| _| �� \| _| _| �� dS )a� Perform setup for writing the movie file. Parameters ---------- fig : `~matplotlib.figure.Figure` The figure object that contains the information for frames outfile : str The filename of the resulting movie file dpi : int, optional The DPI (or resolution) for the file. This controls the size in pixels of the resulting movie file. Default is fig.dpi. N)r? r> r re �_w�_h�_runr= r r r r@ I s zMovieWriter.setupc C s4 | � � }t�d|� tj}tj||||td�| _d S )Nz$MovieWriter.run: running command: %s)�stdin�stdout�stderr� creationflags)�_argsrb rc � subprocess�PIPE�Popen�subprocess_creation_flags�_proc)r" Zcommandro r r r rh b s �zMovieWriter._runc C s | � � dS rE )�cleanupr! r r r rF m s zMovieWriter.finishc K sB t �d� | j�| j| j� | jj| �� f| j| j d�|�� dS )rA �'MovieWriter.grab_frame: Grabbing frame.�r6 r N) rb rd r> ra rf rg �savefig� _frame_sinkrV r rB r r r rD q s ��zMovieWriter.grab_framec C s | j jS )z3Return the place to which frames should be written.)rr ri r! r r r rw � s zMovieWriter._frame_sinkc C s t d�S )z9Assemble list of utility-specific command-line arguments.z)args needs to be implemented by subclass.��NotImplementedErrorr! r r r rm � s zMovieWriter._argsc C s� | j �� \}}| �� �� tt|���� }tt|���� }|r\t�| j j rPt j nt j d|� |r~t�| j j rrt j nt j d|� | j j r�t �| j j | j j||��dS )z>Clean-up and collect the process used to write the movie file.zMovieWriter stdout: %szMovieWriter stderr: %sN)rr Z communicaterw �closer r �readrb �log� returncode�loggingZWARNING�DEBUGrn ZCalledProcessErrorrG )r" �out�errr r r rs � s. � � �zMovieWriter.cleanupc C s t t| j �S )z� Return the binary path to the commandline tool used by a specific subclass. This is a class method so that the tool can be looked for before making a particular MovieWriter subclass available. )�strr �exec_key��clsr r r �bin_path� s zMovieWriter.bin_pathc C s t �| �� �dk S )zO Check to see if a MovieWriter subclass is actually available. N)�shutilZwhichr� r� r r r r& � s zMovieWriter.isAvailable)rO NNNN)N)r8 r9 r: r; r# �propertyr] re r@ rh rF rD rw rm rs � classmethodr� r& r r r r rN � s$ � : rN c @ sd e Zd ZdZdd� Zddd�Zed d � �Zejd d � �Zd d � Z dd� Z dd� Z dd� Z dd� Z dS )�FileMovieWriterzx`MovieWriter` for writing to individual files and stitching at the end. This must be sub-classed to be useful. c O s t j| f|�|� td | _d S )Nzanimation.frame_format)rN r# r rV �r" rG rH r r r r# � s zFileMovieWriter.__init__N�_tmpTc C sN || _ || _|dkr | j j}|| _| �� || _|| _d| _t� | _d| _ dS )a9 Perform setup for writing the movie file. Parameters ---------- fig : `~matplotlib.figure.Figure` The figure to grab the rendered frames from. outfile : str The filename of the resulting movie file. dpi : number, optional The dpi of the output file. This, with the figure size, controls the size in pixels of the resulting movie file. Default is fig.dpi. frame_prefix : str, optional The filename prefix to use for temporary files. Defaults to ``'_tmp'``. clear_temp : bool, optional If the temporary files should be deleted after stitching the final result. Setting this to ``False`` can be useful for debugging. Defaults to ``True``. Nr z %s%%07d.%s) r> r? r re � clear_temp� temp_prefix�_frame_counterr1 � _temp_names�fname_format_str)r" r> r? r � frame_prefixr� r r r r@ � s zFileMovieWriter.setupc C s | j S )z� Format (png, jpeg, etc.) to use for saving the frames, which can be decided by the individual subclasses. )� _frame_formatr! r r r rV � s z FileMovieWriter.frame_formatc C s" || j kr|| _n | j d | _d S r )�supported_formatsr� )r" rV r r r rV � s c C s | j | j| jf S r% )r� r� rV r! r r r �_base_temp_name� s z FileMovieWriter._base_temp_namec C sB | � � | j }| j�|� t�d| j|� | jd7 _t|d�S )Nz7FileMovieWriter.frame_sink: saving frame %d to fname=%s� �wb)r� r� r� �appendrb rd �open�r" �fnamer r r rw � s �zFileMovieWriter._frame_sinkc K s@ t �d� | �� �$}| jj|f| j| jd�|�� W 5 Q R X dS )z� Grab the image information from the figure and save as a movie frame. All keyword arguments in savefig_kwargs are passed on to the `savefig` command that saves the figure. rt ru N)rb rd rw r> rv rV r )r" rC Z myframesinkr r r rD � s ��zFileMovieWriter.grab_framec C s | � � t�| � d S r% )rh rN rF r! r r r rF s zFileMovieWriter.finishc C s8 t �| � | jr4t�d| j� | jD ]}t�|� q$d S )Nz)MovieWriter: clearing temporary fnames=%s)rN rs r� rb rd r� �os�remover� r r r rs s � zFileMovieWriter.cleanup)Nr� T)r8 r9 r: r; r# r@ r� rV �setterr� rw rD rF rs r r r r r� � s � $ r� Zpillowc sB e Zd Zedd� �Z� fdd�Zd dd�Zdd � Zd d � Z� Z S ) � PillowWriterc C s( z dd l }W n tk r" Y dS X dS )Nr FT)�PIL� ImportError)r� r� r r r r& s zPillowWriter.isAvailablec s( |� d�d krd|d< t� j||� d S )NrZ r )�get�superr# r� �rS r r r# ' s zPillowWriter.__init__Nc C s g | _ || _|| _|| _d S r% )�_frames�_outfileZ_dpi�_figr= r r r r@ , s zPillowWriter.setupc K sd ddl m} t� }| jj|ft|dd�� | jj�� }| j� |� dt |j �t |j �f|�� �� d S )Nr )�ImagerP )r6 ZRGBA)r� r� r r� rv r �canvasZ get_rendererr� r� Z frombytesr �widthZheight�getvalue)r" rC r� ZbufZrendererr r r rD 2 s �zPillowWriter.grab_framec C s4 | j d j| jd| j dd � td| j �dd� d S )Nr Tr� �� )Zsave_allZ append_imagesZduration�loop)r� �saver� r rU r! r r r rF > s �zPillowWriter.finish)N) r8 r9 r: r� r& r# r@ rD rF � __classcell__r r r� r r� s r� c s8 e Zd ZdZdZdZedd� �Ze� fdd��Z � Z S )� FFMpegBasez�Mixin class for FFMpeg output. To be useful this must be multiply-inherited from with a `MovieWriterBase` sub-class. zanimation.ffmpeg_pathzanimation.ffmpeg_argsc C s� d| j g}| j dkr,d| jkr,|�ddg� | jdkrJ|�dd| j g� | jr\|�| j� | j�� D ] \}}|�dd ||f g� qf|d | jg S ) N�-vcodecr^ �-pix_fmtZyuv420pr z-bz%dkz -metadataz%s=%sz-y)rW rZ �extendrX r[ r0 r? )r" rG �k�vr r r � output_argsP s zFFMpegBase.output_argsc s, t � �� o*dtj| �� gttjtjd�jkS )Ns LibAv)rl rj rk ) r� r& rn �runr� rq ZDEVNULLro rk r� r� r r r& d s ��zFFMpegBase.isAvailable) r8 r9 r: r; r� rY r� r� r� r& r� r r r� r r� F s r� Zffmpegc @ s e Zd ZdZdd� ZdS )� FFMpegWriterzwPipe-based ffmpeg writer. Frames are streamed directly to ffmpeg via a pipe and written in a single pass. c C s\ | � � dddddd| j d| jdt| j�g }t�� tjkrF|dd g7 }|d d g| j 7 }|S ) Nz-fZrawvideor� z-sz%dx%dr� �-rz -loglevel�error�-izpipe:) r� r] rV r� rU rb ZgetEffectiveLevelr~ r r� �r" rG r r r rm w s � zFFMpegWriter._argsN�r8 r9 r: r; rm r r r r r� p s r� Z ffmpeg_filec @ s. e Zd ZdZdddddddd d g Zd d � Zd S )�FFMpegFileWriterz}File-based ffmpeg writer. Frames are written to temporary files on disk and then stitched together at the end. �png�jpeg�ppm�tiff�sgi�bmp�pbm�rawrP c C s, | � � dt| j�d| �� dt| j�g| j S )Nr� r� z-vframes)r� r� rU r� r� r� r! r r r rm � s ��zFFMpegFileWriter._argsN�r8 r9 r: r; r� rm r r r r r� � s �r� c @ s$ e Zd ZdZdZdZeejj �ZdS )� AVConvBasez�Mixin class for avconv output. To be useful this must be multiply-inherited from with a `MovieWriterBase` sub-class. zanimation.avconv_pathzanimation.avconv_argsN) r8 r9 r: r; r� rY r� rN r& �__func__r r r r r� � s r� Zavconvc @ s e Zd ZdZdS )� AVConvWriterzwPipe-based avconv writer. Frames are streamed directly to avconv via a pipe and written in a single pass. N�r8 r9 r: r; r r r r r� � s r� Z avconv_filec @ s e Zd ZdZdS )�AVConvFileWriterz|File-based avconv writer. Frames are written to temporary files on disk and then stitched together at the end. Nr� r r r r r� � s r� c sT e Zd ZdZdZdZedd� �Zedd� �Ze � fdd ��Z e � fd d ��Z � Z S ) �ImageMagickBasez�Mixin class for ImageMagick output. To be useful this must be multiply-inherited from with a `MovieWriterBase` sub-class. zanimation.convert_pathzanimation.convert_argsc C s d| j S )Ng Y@)rU r! r r r �delay� s zImageMagickBase.delayc C s | j gS r% )r? r! r r r r� � s zImageMagickBase.output_argsc s" t � �� }|dkr t�d�j}|S )NZconvertZmagick)r� r� �mplZ_get_executable_info� executable)r� Zbinpathr� r r r� � s zImageMagickBase.bin_pathc s( z t � �� W S tk r" Y dS X d S r )r� r& �FileNotFoundErrorr� r� r r r& � s zImageMagickBase.isAvailable) r8 r9 r: r; r� rY r� r� r� r� r� r& r� r r r� r r� � s r� Z imagemagickc @ s e Zd ZdZdd� ZdS )�ImageMagickWriterz|Pipe-based animated gif. Frames are streamed directly to ImageMagick via a pipe and written in a single pass. c C s4 | � � dd| j dddt| j�ddd| j g | j S ) Nz-sizez%ix%iz-depth�8�-delay�-loop�0z%s:-)r� r] r� r� rV r� r! r r r rm � s ��zImageMagickWriter._argsNr� r r r r r� � s r� Zimagemagick_filec @ s. e Zd ZdZdddddddd d g Zd d � Zd S )�ImageMagickFileWriterz�File-based animated gif writer. Frames are written to temporary files on disk and then stitched together at the end. r� r� r� r� r� r� r� r� rP c C s, | � � dt| j�ddd| j| jf g| j S )Nr� r� r� z%s*.%s)r� r� r� r� rV r� r! r r r rm � s ��zImageMagickFileWriter._argsNr� r r r r r� � s �r� c C s t jt| �tj�| d �|d�S )z(frame_list should be a list of filenamesr )�Nframes� frame_dirrV )r r6 �lenr� �path�dirname�Z frame_listrV r r r �_included_frames s �r� c s& d�dd� � �fdd�t| �D �� S )z7frame_list should be a list of base64-encoded png filesz, frames[{0}] = "data:image/{1};base64,{2}" � � c 3 s( | ] \}}�� |� |�d d��V qdS )r� z\ N)r6 �replace)r. �iZ frame_data�rV �templater r � <genexpr> s �z#_embedded_frames.<locals>.<genexpr>)�join� enumerater� r r� r �_embedded_frames s �r� Zhtmlc s\ e Zd ZddddgZdZedd� �Zd� fd d � Zd� fdd� Z� fdd�Z dd� Z � Z S )� HTMLWriterr� r� r� Zsvgzanimation.html_argsc C s dS �NTr r� r r r r& s zHTMLWriter.isAvailable� NFr� c sj || _ |�� | _|d kr$td | _n|| _| jd9 _| jdkrRtd�| j���t� �|||||� d S )N�animation.embed_limit� )r� �onceZreflectz unrecognized default_mode {!r}) � embed_frames�lower� default_moder � _bytes_limit� ValueErrorr6 r� r# ) r" rU rW rX rZ r[ r� r� � embed_limitr� r r r# s �zHTMLWriter.__init__c s� t j�|�\}}|dkr td��g | _d| _d| _| jsn|d krH|d }t j�|�s^t � |� t j� |d�}nd }t � j ||||dd� d S )N)z.htmlz.htmz outfile must be *.htm or *.htmlr Fr� �frame)r� ) r� r� �splitextr� � _saved_frames� _total_bytes� _hit_limitr� �exists�makedirsr� r� r@ )r" r> r? r r� �rootZextr� r� r r r@ 2 s zHTMLWriter.setupc s� | j r�| jrd S t� }| jj|f| j| jd�|�� t�|� � �� d�}| j t |�7 _ | j | j kr�t�d| j | j � d| _q�| j�|� nt� jf |�S d S )Nru �asciiz�Animation size has reached %s bytes, exceeding the limit of %s. If you're sure you want a larger animation embedded, set the animation.embed_limit rc parameter to a larger value (in MB). This and further frames will be dropped.T)r� r� r r> rv rV r �base64� encodebytesr� �decoder� r� r� rb �warningr� r� r� rD )r" rC �fZ imgdata64r� r r rD F s( �� �zHTMLWriter.grab_framec C s� | j r t| j| j�}t| j�}nt| j| j�}t| j�}tdddd�}d|| jd < d| j }t | j d��:}|� t t � |� tjf t�� j|||d�|��� W 5 Q R X d S )Nr� )Z once_checkedZ loop_checkedZreflect_checked�checkedZ_checkedr� r )�idr� � fill_frames�interval)r� r� r� rV r� r� r� r r� rU r� r? �writer r r r6 �uuidZuuid4�hex)r" r r� Z mode_dictr Zofr r r rF ] s0 � � � ��zHTMLWriter.finish)r� NNNNFr� N)N) r8 r9 r: r� rY r� r& r# r@ rD rF r� r r r� r r� s � r� c @ s� e Zd ZdZd.dd�Zdd� Zdd � Zd/dd �d d �Zd d� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd0d'd(�Zd1d*d+�Zd,d-� ZdS )2� Animationa% This class wraps the creation of an animation using matplotlib. It is only a base class which should be subclassed to provide needed behavior. This class is not typically used directly. Parameters ---------- fig : `~matplotlib.figure.Figure` The figure object that is used to get draw, resize, and any other needed events. event_source : object, optional A class that can run a callback when desired events are generated, as well as be stopped and started. Examples include timers (see :class:`TimedAnimation`) and file system notifications. blit : bool, optional controls whether blitting is used to optimize drawing. Defaults to ``False``. See Also -------- FuncAnimation, ArtistAnimation NFc C s\ || _ |o|jj| _| �� | _|| _|j�d| j�| _ | j j�d| j �| _ | jrX| � � d S )N� draw_eventZ close_event) r� r� Z supports_blit�_blit� new_frame_seq� frame_seq� event_source� mpl_connect�_start�_first_draw_id�_stop� _close_id� _setup_blit)r" r> r �blitr r r r# � s �zAnimation.__init__c G s: | j j�| j� d| _| �� | j�| j� | j�� dS )z� Starts interactive animation. Adds the draw frame command to the GUI handler, calls show to start the event loop. N) r� r� �mpl_disconnectr � _init_drawr � add_callback�_step�startr� r r r r � s zAnimation._startc G s> | j r| jj�| j� | jj�| j� | j�| j� d | _d S r% ) r r� r� r � _resize_idr r �remove_callbackr r� r r r r � s zAnimation._stop)�progress_callbackc s� |dkrt d }n.t|t�s@tdd� |||||fD ��r@td��| dkrLi } � jdk rl� jj�� j� d} nd} |dkr�t � d�r�d � j }|dkr�t d }|d kr�� jj }|dkr�t d }|dkr�t d }� g} | dk r�| � � fdd�| D �� t|t��rf|t jk�r"t | |||||d�}nDt �� �r^t t �� d }t�d||� ||||||d�}ntd��t�dt|�� d| k�r�t�d� | �d� t� �� t d dk�r�t�d� dt d< |�� j||��� | D ]}|�� �q�d}dd� | D �}d|k�rd}nt|�}tdd� | D �� D ]R}t| |�D ]4\}}|j|dd � | dk �r,| ||� |d 7 }�q,|j f | � �q W 5 Q R X W 5 Q R X | �r�� jj� d � j �� _dS ) a� Save the animation as a movie file by drawing every frame. Parameters ---------- filename : str The output filename, e.g., :file:`mymovie.mp4`. writer : :class:`MovieWriter` or str, optional A `MovieWriter` instance to use or a key that identifies a class to use, such as 'ffmpeg'. If ``None``, defaults to :rc:`animation.writer` = 'ffmpeg'. fps : number, optional Frames per second in the movie. Defaults to ``None``, which will use the animation's specified interval to set the frames per second. dpi : number, optional Controls the dots per inch for the movie frames. This combined with the figure's size in inches controls the size of the movie. If ``None``, defaults to :rc:`savefig.dpi`. codec : str, optional The video codec to be used. Not all codecs are supported by a given :class:`MovieWriter`. If ``None``, default to :rc:`animation.codec` = 'h264'. bitrate : number, optional Specifies the number of bits used per second in the compressed movie, in kilobits per second. A higher number means a higher quality movie, but at the cost of increased file size. If ``None``, defaults to :rc:`animation.bitrate` = -1. extra_args : list, optional List of extra string arguments to be passed to the underlying movie utility. If ``None``, defaults to :rc:`animation.extra_args`. metadata : Dict[str, str], optional Dictionary of keys and values for metadata to include in the output file. Some keys that may be of use include: title, artist, genre, subject, copyright, srcform, comment. extra_anim : list, optional Additional `Animation` objects that should be included in the saved movie file. These need to be from the same `matplotlib.figure.Figure` instance. Also, animation frames will just be simply combined, so there should be a 1:1 correspondence between the frames from the different animations. savefig_kwargs : dict, optional Is a dictionary containing keyword arguments to be passed on to the `savefig` command which is called repeatedly to save the individual frames. progress_callback : function, optional A callback function that will be called for every frame to notify the saving progress. It must have the signature :: def func(current_frame: int, total_frames: int) -> Any where *current_frame* is the current frame number and *total_frames* is the total number of frames to be saved. *total_frames* is set to None, if the total number of frames can not be determined. Return values may exist but are ignored. Example code to write the progress to stdout:: progress_callback = lambda i, n: print(f'Saving frame {i} of {n}') Notes ----- *fps*, *codec*, *bitrate*, *extra_args* and *metadata* are used to construct a `.MovieWriter` instance and can only be passed if *writer* is a string. If they are passed as non-*None* and *writer* is a `.MovieWriter`, a `RuntimeError` will be raised. N�animation.writerc s s | ]}|d k V qd S r% r )r. �argr r r r� s �z!Animation.save.<locals>.<genexpr>z�Passing in values for arguments fps, codec, bitrate, extra_args, or metadata is not supported when writer is an existing MovieWriter instance. These should instead be passed as arguments when creating the MovieWriter instance.TF� _interval� @�@z savefig.dpi�figurerQ rR c 3 s | ]}|j � j kr|V qd S r% )r� )r. �animr! r r r� E s �)rZ r[ r z5MovieWriter %s unavailable; trying to use %s instead.zZCannot save animation: no writers are available. Please install ffmpeg to save animations.zAnimation.save using %sZ bbox_inchesz�Warning: discarding the 'bbox_inches' argument in 'savefig_kwargs' as it may cause frame size to vary, which is inappropriate for animation.z savefig.bboxZtightzkDisabling savefig.bbox = 'tight', as it may cause frame size to vary, which is inappropriate for animation.c S s g | ]}t |d d��qS )� save_countN)�getattr�r. �ar r r � <listcomp>w s �z"Animation.save.<locals>.<listcomp>c S s g | ] }|� � �qS r )�new_saved_frame_seqr' r r r r) } s )r r� r ) r � isinstancer� �anyr4 r r� r� r �hasattrr! r r� �writersr r1 rb r r� rc �type�popr rI r �sum�zip�_draw_next_framerD r r )r" �filename�writerrU r rW rX rZ r[ Z extra_animrC r Zreconnect_first_drawZall_animZ alt_writerr$ Z frame_numberZsave_count_listZ total_frames�data�dr r! r r� � s� T �� � � � � � $ �zAnimation.savec G s: z t | j�}| �|| j� W dS tk r4 Y dS X dS )z� Handler for getting events. By default, gets the next frame in the sequence and hands the data off to be drawn. TFN)�nextr r3 r � StopIteration)r" rG � framedatar r r r � s zAnimation._stepc C s t | j�S )z+Return a new sequence of frame information.)�iter� _framedatar! r r r r � s zAnimation.new_frame_seqc C s | � � S )z8Return a new sequence of saved/cached frame information.)r r! r r r r* � s z Animation.new_saved_frame_seqc C s& | � ||� | �|� | �||� d S r% )� _pre_draw� _draw_frame� _post_draw�r" r: r r r r r3 � s zAnimation._draw_next_framec C s d S r% r r! r r r r � s zAnimation._init_drawc C s |r| � | j| j� d S r% )� _blit_clear�_drawn_artists� _blit_cacher@ r r r r= � s zAnimation._pre_drawc C s t d��d S )NzDNeeds to be implemented by subclasses to actually make an animation.rx )r" r: r r r r> � s zAnimation._draw_framec C s, |r | j r | �| j | j� n | jj�� d S r% )rB � _blit_drawrC r� r� � draw_idler@ r r r r? � s zAnimation._post_drawc C sj g }|D ]>}|j |kr.|jj�|j j�||j < |j �|� |�|j � qt|�D ]}|jj�|j� qPd S r% ) �axesr# r� Zcopy_from_bboxZbboxZ draw_artistr� �setr )r" �artists�bg_cacheZ updated_axr( �axr r r rD � s zAnimation._blit_drawc C s6 dd� |D �}|D ] }||kr|j j�|| � qd S )Nc S s h | ] }|j �qS r )rF r' r r r � <setcomp>� s z(Animation._blit_clear.<locals>.<setcomp>)r# r� Zrestore_region)r" rH rI rF r( r r r rA � s zAnimation._blit_clearc sn t � � _g � _� jjD ]0}|j�d� fdd�� |j�d� fdd�� q� jj�d� j �� _ � � d � j � d S )NZ xlim_changedc s � j �| d �S r% �rC r0 �rJ r! r r �<lambda>� � z'Animation._setup_blit.<locals>.<lambda>Z ylim_changedc s � j �| d �S r% rL rM r! r r rN � rO � resize_event) r rC rB r� rF Z callbacksZconnectr� r �_handle_resizer r? r )r" rJ r r! r r � s � � �zAnimation._setup_blitc G sD | j j�| j� | j�� | j�� | �� | j j� d| j �| _d S )Nr ) r� r� r r r �stoprC �clearr r � _end_redrawr� r r r rQ � s �zAnimation._handle_resizec C s> | � d d� | j�� | jj�| j� | jj�d| j�| _d S )NFrP ) r? r r r� r� r r r rQ )r" Zevtr r r rT � s �zAnimation._end_redrawc C s d}t | d�s�|dkr td }|d9 }t� �T}t|d�}ttd }|dtd d | j d �}| jt|�|d � t� |� � �}W 5 Q R X t |�}||kr�t � d ||� n|�d�| _dj|j� | _t | d��rddg} t | d�r�| jr�| �d� |j| j| jd�| �d�S dS dS )a� Convert the animation to an HTML5 ``<video>`` tag. This saves the animation as an h264 video, encoded in base64 directly into the HTML5 video tag. This respects the rc parameters for the writer as well as the bitrate. This also makes use of the ``interval`` to control the speed, and uses the ``repeat`` parameter to decide whether to loop. Parameters ---------- embed_limit : float, optional Limit, in MB, of the returned animation. No animation is created if the limit is exceeded. Defaults to :rc:`animation.embed_limit` = 20.0. Returns ------- video_tag : str An HTML5 video tag with the animation embedded as base64 encoded h264 video. If the *embed_limit* is exceeded, this returns the string "Video too large to embed." z�<video {size} {options}> <source type="video/mp4" src="data:video/mp4;base64,{video}"> Your browser does not support the video tag. </video>� _base64_videoNr� r� ztemp.m4vr r^ rR r" )rW rX rU �r5 z�Animation movie is %s bytes, exceeding the limit of %s. If you're sure you want a large animation embedded, set the animation.embed_limit rc parameter to a larger value (in MB).r� zwidth="{}" height="{}"ZcontrolsZautoplay�repeatr� � )Zvideo�size�optionszVideo too large to embed.)r- r r r r. r! r� r� r� r� Z read_bytesr� rb r r rU r6 r] Z _video_sizerW r� r� ) r" r� Z VIDEO_TAG�tmpdirr� �Writerr5 Zvid64Zvid_lenrZ r r r �to_html5_video sD � � � �zAnimation.to_html5_videoTc C s� |dkr t | d�r d| j }|dkr2| jr.dnd}t | d�s�t� �:}t|d�}t|||d�}| jt|�|d � |�� | _ W 5 Q R X | j S ) z-Generate HTML representation of the animationNr! r� r� r� �_html_representationz temp.html)rU r� r� rV ) r- r! rW r r r� r� r� Z read_textr^ )r" rU r� r� r[ r� r5 r r r � to_jshtmlQ s �zAnimation.to_jshtmlc C s, t d }|dkr| �� S |dkr(| �� S dS )z#IPython display hook for rendering.zanimation.htmlZhtml5ZjshtmlN)r r] r_ )r" Zfmtr r r � _repr_html_i s zAnimation._repr_html_)NF) NNNNNNNNN)N)NTN)r8 r9 r: r; r# r r r� r r r* r3 r r= r> r? rD rA r rQ rT r] r_ r` r r r r r x s< �� D O r c @ s2 e Zd ZdZd dd�Zdd� Zd d � Zd d � ZdS )�TimedAnimationa4 :class:`Animation` subclass for time-based animation. A new frame is drawn every *interval* milliseconds. Parameters ---------- fig : `~matplotlib.figure.Figure` The figure object that is used to get draw, resize, and any other needed events. interval : number, optional Delay between frames in milliseconds. Defaults to 200. repeat_delay : number, optional If the animation in repeated, adds a delay in milliseconds before repeating the animation. Defaults to ``None``. repeat : bool, optional Controls whether the animation should repeat when the sequence of frames is completed. Defaults to ``True``. blit : bool, optional Controls whether blitting is used to optimize drawing. Defaults to ``False``. �� NTc O sL || _ || _|| _|d kr,|j�� }| j |_tj| |f|�d|i|�� d S )Nr )r! � _repeat_delayrW r� Z new_timerr r r# )r" r> r Z repeat_delayrW r rG rH r r r r# � s ��zTimedAnimation.__init__c G sv t j| f|�� }|sn| jrn| �� | �� | _| jr\| j�| j� | j� | j � | j| j_ dS t j| f|�� S n|S dS )z- Handler for getting events. TN) r r rW r r r rc r r r � _loop_delayr )r" rG Z still_goingr r r r � s zTimedAnimation._stepc G s | j �| j� t�| � d S r% )r r rd r r r� r r r r � s zTimedAnimation._stopc G s4 | j �| j� | j| j _| j �| j� t�| � d S r% )r r rd r! r r r r r� r r r rd � s zTimedAnimation._loop_delay)rb NTN)r8 r9 r: r; r# r r rd r r r r ra r s � ra c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �ArtistAnimationa Animation using a fixed set of `Artist` objects. Before creating an instance, all plotting should have taken place and the relevant artists saved. Parameters ---------- fig : `~matplotlib.figure.Figure` The figure object that is used to get draw, resize, and any other needed events. artists : list Each list entry a collection of artists that represent what needs to be enabled on each frame. These will be disabled for other frames. interval : number, optional Delay between frames in milliseconds. Defaults to 200. repeat_delay : number, optional If the animation in repeated, adds a delay in milliseconds before repeating the animation. Defaults to ``None``. repeat : bool, optional Controls whether the animation should repeat when the sequence of frames is completed. Defaults to ``True``. blit : bool, optional Controls whether blitting is used to optimize drawing. Defaults to ``False``. c O s$ g | _ || _tj| |f|�|� d S r% )rB r< ra r# )r" r> rH rG rH r r r r# � s zArtistAnimation.__init__c C sf t � }| �� D ]>}|D ]4}|�d� |�| j� |�� |kr|�|�� � qq|D ]}|j�� qRd S r ) rG r � set_visible� set_animatedr Z get_figure�addr� rE )r" Zfigsr �artistr> r r r r � s zArtistAnimation._init_drawc C s0 |r| � | j| j� n| jD ]}|�d� q dS )z5 Clears artists from the last frame. FN)rA rB rC rf )r" r: r ri r r r r= � s zArtistAnimation._pre_drawc C s || _ |D ]}|�d� q d S r� )rB rf )r" rH ri r r r r> s zArtistAnimation._draw_frameN)r8 r9 r: r; r# r r= r> r r r r re � s re c @ s@ e Zd ZdZddd�dd�Zdd� Zd d � Zd d � Zd d� ZdS )� FuncAnimationa� Makes an animation by repeatedly calling a function *func*. Parameters ---------- fig : `~matplotlib.figure.Figure` The figure object that is used to get draw, resize, and any other needed events. func : callable The function to call at each frame. The first argument will be the next value in *frames*. Any additional positional arguments can be supplied via the *fargs* parameter. The required signature is:: def func(frame, *fargs) -> iterable_of_artists If ``blit == True``, *func* must return an iterable of all artists that were modified or created. This information is used by the blitting algorithm to determine which parts of the figure have to be updated. The return value is unused if ``blit == False`` and may be omitted in that case. frames : iterable, int, generator function, or None, optional Source of data to pass *func* and each frame of the animation - If an iterable, then simply use the values provided. If the iterable has a length, it will override the *save_count* kwarg. - If an integer, then equivalent to passing ``range(frames)`` - If a generator function, then must have the signature:: def gen_function() -> obj - If *None*, then equivalent to passing ``itertools.count``. In all of these cases, the values in *frames* is simply passed through to the user-supplied *func* and thus can be of any type. init_func : callable, optional A function used to draw a clear frame. If not given, the results of drawing from the first item in the frames sequence will be used. This function will be called once before the first frame. The required signature is:: def init_func() -> iterable_of_artists If ``blit == True``, *init_func* must return an iterable of artists to be re-drawn. This information is used by the blitting algorithm to determine which parts of the figure have to be updated. The return value is unused if ``blit == False`` and may be omitted in that case. fargs : tuple or None, optional Additional arguments to pass to each call to *func*. save_count : int, optional The number of values from *frames* to cache. interval : number, optional Delay between frames in milliseconds. Defaults to 200. repeat_delay : number, optional If the animation in repeated, adds a delay in milliseconds before repeating the animation. Defaults to *None*. repeat : bool, optional Controls whether the animation should repeat when the sequence of frames is completed. Defaults to *True*. blit : bool, optional Controls whether blitting is used to optimize drawing. Note: when using blitting any animated artists will be drawn according to their zorder. However, they will be drawn on top of any previous artists, regardless of their zorder. Defaults to *False*. cache_frame_data : bool, optional Controls whether frame data is cached. Defaults to *True*. Disabling cache might be helpful when frames contain large objects. NT)�cache_frame_datac s� |r || _ nd| _ || _|| _|| _� d kr6tj| _nRt� �rF� | _nBt� � �rt� fdd�| _t � d�r�t � �| _n� fdd�| _� | _| jd kr�d| _n t | j�| _|| _ g | _tj| |f|� g | _d S )Nr c s t � �S r% )r; r ��framesr r rN � rO z(FuncAnimation.__init__.<locals>.<lambda>�__len__c s t t� ��S r% )r; �ranger rl r r rN � rO �d )rm �_func� _init_funcr% � itertools�count� _iter_gen�callabler �iterabler- r� r �_cache_frame_data� _save_seqra r# ) r" r> �funcrm Z init_funcZfargsr% rk rH r rl r r# i s. zFuncAnimation.__init__c C s | � � S r% )ru r! r r r r � s zFuncAnimation.new_frame_seqc sV | j r t| j �| _t| j�S | jd k r8t�| �� | j�S | �� � � fdd�}|� S d S )Nc 3 sD z t d�D ]} t� �V q W n tk r0 Y nX tjddd� d S )Nrp z2.2z�FuncAnimation.save has truncated your animation to 100 frames. In the future, no such truncation will occur; please pass 'save_count' accordingly.)�message)ro r8 r9 r Zwarn_deprecated)�_�r r r �gen� s �z.FuncAnimation.new_saved_frame_seq.<locals>.gen)ry r1 Z_old_saved_seqr; r% rs �islicer )r" r~ r r} r r* � s z!FuncAnimation.new_saved_frame_seqc C sb | j d kr | �t| �� �� n:| � � | _| jrX| jd kr@td��| jD ]}|�| j� qFg | _d S )Nz7The init_func must return a sequence of Artist objects.) rr r> r8 r rB r r4 rg ry )r" r( r r r r � s zFuncAnimation._init_drawc C s� | j r| j�|� | j| j d � | _| j|f| j�� | _| jr~| jd krRtd��t | jdd� d�| _| jD ]}|� | j� qld S )Nz@The animation function must return a sequence of Artist objects.c S s | � � S r% )Z get_zorder)r r r r rN � rO z+FuncAnimation._draw_frame.<locals>.<lambda>)�key) rx ry r� r% rq rm rB r r4 �sortedrg )r" r: r( r r r r> � s � zFuncAnimation._draw_frame)NNNN) r8 r9 r: r; r# r r* r r> r r r r rj s U �� 2 rj )?rJ r� rL �ior r rs r~ r� �pathlibr �platformr� rn �sysZtempfiler r Znumpyr Z matplotlibr� Zmatplotlib._animation_datar r r r r r r r Z getLoggerr8 rb �systemrq ZCREATE_NO_WINDOWr �objectr r. �ABCr< rN r� r+ r� r� r� r� r� r� r� r� r� r� r� r� r� r ra re rj r r r r �<module> sp $EA Do(* "c }RO