B ]t\`@sddlZddlZddlZddlmZddlZddlZddlZddlm Z ddl Z ddl Z ddl Z ddl Z ddlmZddlZddlZddlmZmZmZddlmZmZmZmZeeZe dkrdZ Z!ndZ d d Z"Gd d d e#Z$e$Z%Gd ddej&Z'Gddde'Z(Gddde(Z)e%*dGddde(Z+Gddde#Z,e%*dGddde,e(Z-e%*dGddde,e)Z.Gddde,Z/e%*d Gd!d"d"e/e-Z0e%*d#Gd$d%d%e/e.Z1Gd&d'd'e#Z2e%*d(Gd)d*d*e2e(Z3e%*d+Gd,d-d-e2e)Z4d.d/Z5d0d1Z6e%*d2Gd3d4d4e)Z7Gd5d6d6e#Z8Gd7d8d8e8Z9Gd9d:d:e9Z:Gd;d<d.correct_roundoff)r )whrrrZwnewZhnewrrradjusted_figsize=src@sPeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS)MovieWriterRegistryz.wrapperr)rr"r#r)r"rrregisterls zMovieWriterRegistry.registercCs|jr|dS)z2If dirty, reasks the writers if they are availableN)rreset_available_writers)rrrrensure_not_dirty|sz$MovieWriterRegistry.ensure_not_dirtycCs:i|_x(|jD]\}}|r||j|<qWd|_dS)z3Reset the available state of all registered writersFN)rritemsr r)rr"r!rrrr%s z+MovieWriterRegistry.reset_available_writerscCs|t|jS)z%Get a list of available MovieWriters.)r&listr)rrrrr(szMovieWriterRegistry.listcCs|||jkS)zCheck if given writer is available by name. Parameters ---------- name : str Returns ------- available : bool )r&r)rr"rrr is_availables z MovieWriterRegistry.is_availablecCsH||jstdy |j|Stk rBtd|YnXdS)NzNo MovieWriters available!z(Requested MovieWriter ({}) not available)r&r RuntimeErrorKeyErrorformat)rr"rrr __getitem__s zMovieWriterRegistry.__getitem__N) __name__ __module__ __qualname____doc__rrr$r&r%r(r)r-rrrrrasrc@sJeZdZdZejd ddZejddZejddZe 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()`. NcCsdS)a Perform setup for writing the movie file. Parameters ---------- fig: `matplotlib.figure.Figure` instance 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)rfigoutfilerrrrsetupszAbstractMovieWriter.setupcKsdS)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. Nr)rsavefig_kwargsrrr grab_frameszAbstractMovieWriter.grab_framecCsdS)z,Finish any processing for writing the movie.Nr)rrrrfinishszAbstractMovieWriter.finishcos0|j|||f||z |VWd|XdS)z Context manager to facilitate writing the movie file. ``*args, **kw`` are any parameters that should be passed to `setup`. N)r5r8)rr3r4rargskwargsrrrsavings zAbstractMovieWriter.saving)N) r.r/r0r1abcabstractmethodr5r7r8 contextlibcontextmanagerr;rrrrr2s   r2c@seZdZdZdddZeddZdd Zdd d Zd d Z ddZ ddZ ddZ ddZ ddZeddZeddZdS) MovieWriteraBase class for writing movies. This class is set up to provide for writing movie frame data to a pipe. See examples for how to use these classes. 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. NcCs|||_d|_|dkr td|_n||_|dkr:td|_n||_|dkrZtt|j|_n||_|dkrrt|_ n||_ dS)aMovieWriter 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. rgbaNzanimation.codeczanimation.bitrate) fps frame_formatr codecbitrater(args_key extra_argsrmetadata)rrCrErFrHrIrrrrs   zMovieWriter.__init__cCs*|j\}}t||jt||jfS)z7A tuple ``(width, height)`` in pixels of a movie frame.)r3get_size_inchesr r)rrrrrr frame_size,szMovieWriter.frame_sizecCs|jdkrb|j\}}t|||jd\}}||f||fksp|jj||ddtd||||n|j\}}tjd|j ||fS)Nh264T)Zforwardz>figure size (inches) has been adjusted from %s x %s to %s x %sframe size in pixels is %s x %s)rN) rEr3rJrrset_size_inches_loginfodebugrK)rZwoZhorrrrr_adjust_frame_size2s zMovieWriter._adjust_frame_sizecCs>||_||_|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 : 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. N)r4r3rrS_w_h_run)rr3r4rrrrr5?szMovieWriter.setupcCs8|}tj}td|tj|d||tjtd|_dS)Nz$MovieWriter.run: running command: %sF)shellstdoutstderrstdin creationflags)_args subprocessPIPErPrQPopensubprocess_creation_flags_proc)rZcommandoutputrrrrVXs zMovieWriter._runcCs |dS)z,Finish any processing for writing the movie.N)cleanup)rrrrr8dszMovieWriter.finishc Kstdy8|j|j|j|jj|f|j|j d|WnTt t fk r}z2|j \}}td||t d|||Wdd}~XYnXdS)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. z'MovieWriter.grab_frame: Grabbing frame.)r,rz(MovieWriter -- Error running proc: %s %sz}Error saving animation to file (cause: {0}) Stdout: {1} StdError: {2}. It may help to re-run with logging level set to DEBUG.N)rPrRr3rOrTrUsavefig _frame_sinkrDrr*IOErrorra communicaterQr,)rr6eouterrrrrr7hs zMovieWriter.grab_framecCs|jjS)z4Returns the place to which frames should be written.)rarZ)rrrrreszMovieWriter._frame_sinkcCstdS)z9Assemble list of utility-specific command-line arguments.z)args needs to be implemented by subclass.)NotImplementedError)rrrrr\szMovieWriter._argscCs6|j\}}|td|td|dS)z>Clean-up and collect the process used to write the movie file.z!MovieWriter -- Command stdout: %sz!MovieWriter -- Command stderr: %sN)rargrecloserPrR)rrirjrrrrcs  zMovieWriter.cleanupcCstt|jS)z Returns 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)clsrrrbin_pathszMovieWriter.bin_pathcCst|dk S)zO Check to see if a MovieWriter subclass is actually available. N)shutilZwhichrp)rorrrr szMovieWriter.isAvailable)rANNNN)N)r.r/r0r1rpropertyrKrSr5rVr8r7rer\rc classmethodrpr rrrrr@s 0    r@c@sdeZdZdZddZdddZed d Zejd d Zd d Z ddZ ddZ ddZ ddZ dS)FileMovieWriterzx`MovieWriter` for writing to individual files and stitching at the end. This must be sub-classed to be useful. cOs tj|f||td|_dS)Nzanimation.frame_format)r@rr rD)rr9r:rrrrszFileMovieWriter.__init__N_tmpTcCsN||_||_|dkr|jj}||_|||_||_d|_t|_d|_ dS)a6Perform 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``. Nrz %s%%07d.%s) r3r4rrS clear_temp temp_prefix_frame_counterr( _temp_namesfname_format_str)rr3r4r frame_prefixrvrrrr5szFileMovieWriter.setupcCs|jS)z Format (png, jpeg, etc.) to use for saving the frames, which can be decided by the individual subclasses. ) _frame_format)rrrrrDszFileMovieWriter.frame_formatcCs"||jkr||_n |jd|_dS)Nr)supported_formatsr|)rrDrrrrDs cCs|j|j|jfS)N)rzrwrD)rrrr_base_temp_nameszFileMovieWriter._base_temp_namecCsB||j}|j|td|j||jd7_t|dS)Nz7FileMovieWriter.frame_sink: saving frame %d to fname=%swb)r~rxryappendrPrRopen)rfnamerrrres   zFileMovieWriter._frame_sinkc Ksxtdy6|$}|jj|f|j|jd|WdQRXWn2tk rr|j \}}t d||YnXdS)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. z'MovieWriter.grab_frame: Grabbing frame.)r,rNz(MovieWriter -- Error running proc: %s %s) rPrRrer3rdrDrr*rargrQ)rr6Z myframesinkrirjrrrr7s  zFileMovieWriter.grab_framec Cs|t||jjry@dd|jjD}dd|jjD}td|td|Wn"t k r|}zWdd}~XYnXt d |jjdS)NcSsg|] }|qSr)decode).0srrr sz*FileMovieWriter.finish..cSsg|] }|qSr)r)rrrrrrszMovieWriter.finish: stdout: %szMovieWriter.finish: stderr: %sz%Error creating movie, return code: {}) rVr@r8ra returncodeZ _stdout_buffZ _stderr_buffrPrQ Exceptionr*r,)rrXrYrhrrrr8s  zFileMovieWriter.finishcCs<t||jr8td|jx|jD]}t|q&WdS)Nz)MovieWriter: clearing temporary fnames=%s)r@rcrvrPrRryosremove)rrrrrrcs   zFileMovieWriter.cleanup)NruT)r.r/r0r1rr5rrrDsetterr~rer7r8rcrrrrrts # rtZpillowcsBeZdZeddZfddZd ddZdd Zd d ZZ S) PillowWritercCs&y ddl}Wntk r dSXdS)NrFT)PIL ImportError)rorrrrr %s  zPillowWriter.isAvailablecs(|ddkrd|d<tj||dS)NrHr)getsuperr)rr9r:) __class__rrr-szPillowWriter.__init__NcCsg|_||_||_||_dS)N)_frames_outfileZ_dpi_fig)rr3r4rrrrr52szPillowWriter.setupcKsdddlm}t}|jj|ft|dd|jj}|j | dt |j t |j f|dS)Nr)ImagerB)r,ZRGBA)rrrrrdrcanvasZ get_rendererrrZ frombytesr widthZheightgetvalue)rr6rZbufZrendererrrrr78s   zPillowWriter.grab_framecCs2|jdj|jd|jddtd|jddS)NrTri)Zsave_allZ append_imagesZduration)rsaverr rC)rrrrr8Ds zPillowWriter.finish)N) r.r/r0rsr rr5r7r8 __classcell__rr)rrr#s     rcs8eZdZdZdZdZeddZefddZ Z S) FFMpegBasezMixin class for FFMpeg output. To be useful this must be multiply-inherited from with a `MovieWriterBase` sub-class. zanimation.ffmpeg_pathzanimation.ffmpeg_argscCsd|jg}|jdkr,d|jkr,|ddg|jdkrJ|dd|jg|jr\||jx,|jD]\}}|dd ||fgqhW|d |jgS) Nz-vcodecrLz-pix_fmtZyuv420prz-bz%dkz -metadataz%s=%sz-y)rErHextendrFrIr'r4)rr9kvrrr output_argsVs   zFFMpegBase.output_argscs,to*dtj|gttjtjdjkS)NsLibAv)r[rXrY) rr r]runrpr`ZDEVNULLr^rY)ro)rrrr js  zFFMpegBase.isAvailable) r.r/r0r1rnrGrrrrsr rrr)rrrLs  rZffmpegc@seZdZdZddZdS) FFMpegWriterzwPipe-based ffmpeg writer. Frames are streamed directly to ffmpeg via a pipe and written in a single pass. c Cs\|dddddd|jd|jdt|jg }ttjkrF|dd g7}|d d g|j 7}|S) Nz-fZrawvideoz-vcodecz-sz%dx%dz-pix_fmtz-rz -loglevelquietz-izpipe:) rprKrDrmrCrPZgetEffectiveLevelloggingDEBUGr)rr9rrrr\}s zFFMpegWriter._argsN)r.r/r0r1r\rrrrrvsrZ ffmpeg_filec @s.eZdZdZdddddddd 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. pngjpegppmtiffsgibmppbmrawrBcCs,|dt|jd|dt|jg|jS)Nz-rz-iz-vframes)rprmrCr~rxr)rrrrr\szFFMpegFileWriter._argsN)r.r/r0r1r}r\rrrrrs  rc@s$eZdZdZdZdZeejj ZdS) AVConvBasezMixin class for avconv output. To be useful this must be multiply-inherited from with a `MovieWriterBase` sub-class. zanimation.avconv_pathzanimation.avconv_argsN) r.r/r0r1rnrGrsr@r __func__rrrrrsrZavconvc@seZdZdZdS) AVConvWriterzwPipe-based avconv writer. Frames are streamed directly to avconv via a pipe and written in a single pass. N)r.r/r0r1rrrrrsrZ avconv_filec@seZdZdZdS)AVConvFileWriterz|File-based avconv writer. Frames are written to temporary files on disk and then stitched together at the end. N)r.r/r0r1rrrrrsrcsPeZdZdZdZdZeddZeddZe dd Z e fd d Z Z S) ImageMagickBasezMixin class for ImageMagick output. To be useful this must be multiply-inherited from with a `MovieWriterBase` sub-class. zanimation.convert_pathzanimation.convert_argscCs d|jS)NgY@)rC)rrrrdelayszImageMagickBase.delaycCs|jgS)N)r4)rrrrrszImageMagickBase.output_argsc Cstjdkst|jdkrdSddl}xhd|j|jfD]V}y8||jdd|j |B}| |dd}| |PWq4t k rd}Yq4Xq4W|rx0dD]$}t j||}t j|r|}PqWd}|t|j<t|j<dS)NZwin32convertrzSoftware\Imagemagick\CurrentZBinPath)z convert.exez magick.exe)sysplatformr rnwinregZKEY_WOW64_32KEYZKEY_WOW64_64KEYZ OpenKeyExHKEY_LOCAL_MACHINEZKEY_QUERY_VALUEZ QueryValueExZCloseKeyrrpathjoinexistsr )rorflaghkeyZbinpathZexerrrr_init_from_registrys*   z#ImageMagickBase._init_from_registrycs"|}|dkr|tS)z Check to see if a ImageMagickWriter is actually available. Done by first checking the windows registry (if applicable) and then running the commandline tool. r)rprrr )rorp)rrrr szImageMagickBase.isAvailable) r.r/r0r1rnrGrrrrrsrr rrr)rrrs   rZ imagemagickc@seZdZdZddZdS)ImageMagickWriterz|Pipe-based animated gif. Frames are streamed directly to ImageMagick via a pipe and written in a single pass. c Cs4|dd|jdddt|jddd|jg |jS) Nz-sizez%ix%iz-depth8z-delayz-loop0z%s:-)rprKrmrrDr)rrrrr\ s  zImageMagickWriter._argsN)r.r/r0r1r\rrrrrsrZimagemagick_filec @s.eZdZdZdddddddd d g Zd d Zd S)ImageMagickFileWriterzFile-based animated gif writer. Frames are written to temporary files on disk and then stitched together at the end. rrrrrrrrrBcCs,|dt|jddd|j|jfg|jS)Nz-delayz-looprz%s*.%s)rprmrrwrDr)rrrrr\!szImageMagickFileWriter._argsN)r.r/r0r1r}r\rrrrrs  rcCs tjt|tj|d|dS)z(frame_list should be a list of filenamesr)Nframes frame_dirrD)rr,lenrrdirname) frame_listrDrrr_included_frames)s rcs&dddfddt|DS)z7frame_list should be a list of base64-encoded png filesz, frames[{0}] = "data:image/{1};base64,{2}"  rc 3s(|] \}}||ddVqdS)rz\ N)r,replace)riZ frame_data)rDtemplaterr 4sz#_embedded_frames..)r enumerate)rrDr)rDrr_embedded_frames0s rZhtmlcs\eZdZddddgZdZeddZdfd d Zdfdd ZfddZ ddZ Z S) HTMLWriterrrrZsvgzanimation.html_argscCsdS)NTr)rorrrr =szHTMLWriter.isAvailableNFloopc sj||_||_|dkr$td|_n||_|jd9_|jdkrRd|_tdt|||||dS)Nzanimation.embed_limiti)ronceZreflectrz'unrecognized default_mode: using 'loop') embed_frameslower default_moder _bytes_limitrPwarningrr) rrCrErFrHrIrr embed_limit)rrrrAs    zHTMLWriter.__init__cstj|\}}|dkr tdg|_d|_d|_|jsn|dkrH|d}tj|s^t |tj |d}nd}t j ||||dddS)N)z.htmlz.htmzoutfile must be *.htm or *.htmlrFrframe)rv) rrsplitext ValueError _saved_frames _total_bytes _hit_limitrrmakedirsrrr5)rr3r4rrrootZextr{)rrrr5Vs  zHTMLWriter.setupc s|jr|jrdSt}|jj|f|j|jd|t|  d}|j t |7_ |j |j krtd|j |j d|_q|j|ntjf|SdS)N)r,rasciizAnimation 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)rrrr3rdrDrbase64 encodebytesrrrrrrPrrrrr7)rr6fZ imgdata64)rrrr7js  zHTMLWriter.grab_framec CsGdddt}||_|jr8t|j|j}t|j}nt|j|j}t|j}t dddd}d||j d<d|j }t |j d6}|t|tjftj|||d |WdQRXdS) Nc@seZdZdZddZdS)z'HTMLWriter._run..ProcessStandinrcSsdS)N)rrr)rrrrrgsz3HTMLWriter._run..ProcessStandin.communicateN)r.r/r0rrgrrrrProcessStandinsrr)Z once_checkedZ loop_checkedZreflect_checkedcheckedZ_checkedir)idr fill_framesinterval)objectrarrrrDrrryrrrCrr4writerrr,uuidZuuid4hex)rrrrZ mode_dictrZofrrrrVs*    zHTMLWriter._run)rNNNNFrN)N) r.r/r0r}rGrsr rr5r7rVrrr)rrr8s   rc @seZdZdZd-ddZddZdd Zd.d d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd/d&d'Zd0d)d*Zd+d,ZdS)1 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 NFcCs\||_|o|jj|_||_||_|jd|j|_ |jjd|j |_ |jrX| dS)N draw_eventZ close_event) rrZ supports_blit_blit new_frame_seq frame_seq event_source mpl_connect_start_first_draw_id_stop _close_id _setup_blit)rr3rblitrrrrs   zAnimation.__init__cGs:|jj|jd|_||j|j|jdS)z Starts interactive animation. Adds the draw frame command to the GUI handler, calls show to start the event loop. N) rrmpl_disconnectr _init_drawr add_callback_stepstart)rr9rrrrs zAnimation._startcGs>|jr|jj|j|jj|j|j|jd|_dS)N) rrrr _resize_idrrremove_callbackr)rr9rrrrs zAnimation._stopc  s|dkrtd}n.t|ts@tdd|||||fDr@td| dkrLi} jdk rljjjd} nd} |dkrt drd j }|dkrtd }|d krjj }|dkrtd }|dkrtd }g} | dk r| fdd| Dt|tr~|t jkr"t ||||||d}n\td|t dy"t t d|||||d}Wntk r|tdYnXtdt|d| krtd| dttddkrtddtd<|j||px| D]} | qWxPtdd| DD]:}x&t| |D]\} }| j|ddq"W|jf| qWWdQRXWdQRX| r~jjdj_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. 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. Nzanimation.writercss|]}|dk VqdS)Nr)rargrrrr=sz!Animation.save..zPassing 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 _intervalg@@z savefig.dpifigurezanimation.codeczanimation.bitratec3s|]}|jjkr|VqdS)N)r)ranim)rrrrcs)rHrIz5MovieWriter {} unavailable. Trying to use {} instead.rzZCannot save animation: no writers are available. Please install ffmpeg to save animations.zAnimation.save using %sZ bbox_incheszWarning: 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.cSsg|] }|qSr)new_saved_frame_seq)rarrrrsz"Animation.save..)rr) r isinstancermanyr*rrrrhasattrrrrwritersrrPrr,r( IndexErrorrrQtypepopr r;rzip_draw_next_framer7rr)rfilenamewriterrCrrErFrHrIZ extra_animr6Zreconnect_first_drawZall_animrdatadr)rrrsrD               & zAnimation.savecGs6yt|j}|||jdStk r0dSXdS)z Handler for getting events. By default, gets the next frame in the sequence and hands the data off to be drawn. TFN)nextrrr StopIteration)rr9 framedatarrrrs  zAnimation._stepcCs t|jS)z+Return a new sequence of frame information.)iter _framedata)rrrrrszAnimation.new_frame_seqcCs|S)z8Return a new sequence of saved/cached frame information.)r)rrrrr szAnimation.new_saved_frame_seqcCs&||||||||dS)N) _pre_draw _draw_frame _post_draw)rrrrrrrs  zAnimation._draw_next_framecCsdS)Nr)rrrrrszAnimation._init_drawcCs|r||j|jdS)N) _blit_clear_drawn_artists _blit_cache)rrrrrrrszAnimation._pre_drawcCs tddS)NzDNeeds to be implemented by subclasses to actually make an animation.)rk)rrrrrrszAnimation._draw_framecCs,|r|jr||j|jn |jjdS)N)r! _blit_drawr"rr draw_idle)rrrrrrrs zAnimation._post_drawcCsrg}xF|D]>}|j|kr0|jj|jj||j<|j|||jq Wx t|D]}|jj|jqVWdS)N) axesrrZcopy_from_bboxZbboxZ draw_artistrsetr)rartistsbg_cacheZ updated_axr Zaxrrrr#s   zAnimation._blit_drawcCs:dd|D}x&|D]}||kr|jj||qWdS)NcSsh|] }|jqSr)r%)rr rrr sz(Animation._blit_clear..)rrZrestore_region)rr'r(r%r rrrr s zAnimation._blit_clearcCs4t|_g|_|jjd|j|_|d|j dS)N resize_event) rr"r!rrr_handle_resizerrr)rrrrrs   zAnimation._setup_blitcGsD|jj|j|j|j||jj d|j |_dS)Nr) rrrrrstopr"clearrr _end_redraw)rr9rrrr+s    zAnimation._handle_resizecCs>|dd|j|jj|j|jjd|j|_dS)NFr*) rrrrrrrrr+)rZevtrrrr.s    zAnimation._end_redrawc Csd}t|ds|dkrtd}|d9}tT}t|d}ttd}|dtd d |jd }|jt||d t | }WdQRXt |}||krt d ||n|d|_dj|j|_t|drddg} t|dr|jr| d|j|j|jd| dSdSdS)a Convert the animation to an HTML5 ``