Rc@sddlZddlZddlZddlmZmZddlmZddlm Z ddlm Z de fdYZ e Z de fd YZd efd YZd fd YZe jddeefdYZe jddeefdYZdefdYZe jddeefdYZe jddeefdYZdfdYZe jddeefd YZe jd!d"eefd#YZd$fd%YZe jd&d'eefd(YZe jd)d*eefd+YZd,e fd-YZd.efd/YZd0efd1YZd2efd3YZ dS(4iN(titerabletis_string_like(t subprocess(tverbose(trcParamstMovieWriterRegistrycBs5eZdZdZdZdZdZRS(cCst|_dS(N(tdicttavail(tself((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyt__init__'scsfd}|S(Ncs |jr|jClean-up and collect the process used to write the movie file.s!MovieWriter -- Command stdout: %sR>R,s!MovieWriter -- Command stderr: %sN(R:RCRR6(RRERF((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR=s   cCs t|jS(s 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. (Rtexec_key(tcls((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pytbin_pathscCsLy3tj|jdtdtjdtjtSWntk rGtSXdS(sw Check to see if a MovieWriter subclass is actually available by running the commandline tool. R.R/R0N(RR8RKR9R5tTruetOSError(RJ((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR s   N(RRt__doc__RR tpropertyR%R)t contextlibtcontextmanagerR+R'R*RGRBR2R=t classmethodRKR (((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyREs  /        tFileMovieWritercBskeZdZdZdedZedZejdZdZ dZ dZ d Z RS( s6`MovieWriter` subclass that handles writing to a file.cOs$tj|||td|_dS(Nsanimation.frame_format(RR RR(RR(tkwargs((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR st_tmpcCsO||_||_||_||_||_d|_t|_d|_dS(s Perform setup for writing the movie file. fig: `matplotlib.Figure` instance The figure object that contains the information for frames outfile: string The filename of the resulting movie file dpi: int The DPI (or resolution) for the file. This controls the size in pixels of the resulting movie file. frame_prefix: string, optional The filename prefix to use for the temporary files. Defaults to '_tmp' clear_temp: bool Specifies whether the temporary files should be deleted after the movie is written. (Useful for debugging.) Defaults to True. is %s%%07d.%sN( R R&R"t clear_tempt temp_prefixt_frame_counterRt _temp_namestfname_format_str(RR R&R"t frame_prefixRV((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR)s       cCs|jS(s Format (png, jpeg, etc.) to use for saving the frames, which can be decided by the individual subclasses. (t _frame_format(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCs/||jkr||_n|jd|_dS(Ni(tsupported_formatsR\(RR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR&s cCs|j|j|jfS(N(RZRWR(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyt_base_temp_name-scCs_|j|j}|jj|tjd|j|fdd|jd7_t|dS(Ns7FileMovieWriter.frame_sink: saving frame %d to fname=%sR>R,itwb(R^RXRYtappendRR6topen(Rtfname((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRB2scCsJ|jtj||jjrFtdt|jjdndS(Ns#Error creating movie, return code: s! Try running with --verbose-debug(R'RR*R:t returncodeRtstr(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR*Cs    cCsjtj||jrfddl}tjdt|jddx!|jD]}|j|qLWndS(Nis)MovieWriter: clearing temporary fnames=%sR>R,( RR=RVtosRR6RdRYtremove(RReRb((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR=Ps   ( RRRNR RLR)RORtsetterR^RBR*R=(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRSs    t FFMpegBasecBs#eZdZdZedZRS(sanimation.ffmpeg_pathsanimation.ffmpeg_argscCsd|jg}|jdkr;|jdd|jgn|jrW|j|jnx:|jjD])\}}|jdd||fgqgW|d|jgS(Ns-vcodecis-bs%dks -metadatas%s=%ss-y(RRtextendRRtitemsR&(RR(tktv((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyt output_argsds !(RRRIRRORm(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRh`stffmpegt FFMpegWritercBseZdZRS(c Cs}|jdddddd|jd|jdt|jg }tjdsb|d d g7}n|d d g|j7}|S( Ns-ftrawvideos-vcodecs-ss%dx%ds-pix_fmts-rR,s -logleveltquiets-ispipe:(RKR%RRdRRR3Rm(RR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR2vs(RRR2(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRotst ffmpeg_filetFFMpegFileWriterc Bs2eZdddddddddg Zd ZRS( tpngtjpegtppmttifftsgitbmptpbmtrawRcCs>|jdt|jdt|jd|jg|jS(Ns-vframess-rs-i(RKRdRXRR^Rm(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR2s(RRR]R2(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRsst AVConvBasecBseZdZdZRS(sanimation.avconv_pathsanimation.avconv_args(RRRIR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR|stavconvt AVConvWritercBseZRS((RR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR~st avconv_filetAVConvFileWritercBseZRS((RR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRst MencoderBasecBsGeZdZdZdddddddgZd Zed ZRS( sanimation.mencoder_pathsanimation.mencoder_argsR tartisttgenretsubjectt copyrighttsrcformtcommentcCs*d|jkr&|jd|jds ( RR&RRR`RRiRR7Rj(RR(((RsN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRms   (RRRIRRRRORm(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRs   tmencodertMencoderWritercBseZdZRS(c Cs>|jddddd|jd|j|jfg|jS(Nt-s-demuxerRps -rawvideos w=%i:h=%i:sfps=%i:format=%s(RKR%RRRm(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR2s  (RRR2(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRst mencoder_filetMencoderFileWritercBs#eZddddgZdZRS(RtRuttgaRxcCsL|jd|j|jfdt|jdd|j|jfg|jS(Ns mf://%s*.%ss-framess-mfstype=%s:fps=%d(RKRWRRdRXRRm(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR2s   (RRR]R2(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRstImageMagickBasecBs2eZdZdZedZedZRS(sanimation.convert_pathsanimation.convert_argscCs d|jS(NgY@(R(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pytdelayscCs |jgS(N(R&(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRms(RRRIRRORRm(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRst imagemagicktImageMagickWritercBseZdZRS(c CsF|jdd|jdddt|jddd|jg |jS( Ns-sizes%ix%is-deptht8s-delays-loopt0s%s:-(RKR%RdRRRm(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR2s   (RRR2(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRstimagemagick_filetImageMagickFileWriterc Bs2eZdddddddddg Zd ZRS( RtRuRvRwRxRyRzR{RcCs<|jdt|jddd|j|jfg|jS(Ns-delays-loopRs%s*.%s(RKRdRRWRRm(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR2s(RRR]R2(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRst Animationc BseZdZdedZdZdZdddddddddd ZdZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZRS(s2 This class wraps the creation of an animation using matplotlib. It is only a base class which should be subclassed to provide needed behavior. *fig* is the figure object that is used to get draw, resize, and any other needed events. *event_source* is 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* is a boolean that controls whether blitting is used to optimize drawing. cCs||_|o|jj|_|j|_||_|j|jjd|j |_ |jjjd|j |_ |jr|j ndS(Nt draw_eventt close_event(t_figtcanvast supports_blitt_blitt new_frame_seqt frame_seqt event_sourcet _init_drawt mpl_connectt_startt_first_draw_idt_stopt _close_idt _setup_blit(RR Rtblit((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR s    cGsC|jj|j|jj|jjj|jd|_dS(s Starts interactive animation. Adds the draw frame command to the GUI handler, calls show to start the event loop. N( Rt add_callbackt_steptstartRRtmpl_disconnectRR(RR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR.s cGsX|jr"|jjj|jn|jjj|j|jj|jd|_dS(N( RRRRt _resize_idRRtremove_callbackRR(RR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR;s  c  s| dkri} nd| kr~|dkp?t|ttfs~djt|tr]|n |jjGH| jdq~nj dk rj j j j t } nt} |dkrtdrdj}n|dkrtd}n|dkrtd}n|dkr(td }n|dkrAtd }ng} | dk rv| jfd | Dnt|r |tjkrt||||d |d |}q ddl} | jd|ytjd}Wq tk rtdq Xntjdt|dd|jj ||xxptj g| D]}|j!^qPD]F}x0t"| |D]\}}|j#|dtqW|j$| qiWWdQX| rj j j%dj&_ ndS(s Saves a movie file by drawing every frame. *filename* is the output filename, e.g., :file:`mymovie.mp4` *writer* is either an instance of :class:`MovieWriter` or a string key that identifies a class to use, such as 'ffmpeg' or 'mencoder'. If nothing is passed, the value of the rcparam `animation.writer` is used. *fps* is the frames per second in the movie. Defaults to None, which will use the animation's specified interval to set the frames per second. *dpi* controls the dots per inch for the movie frames. This combined with the figure's size in inches controls the size of the movie. *codec* is the video codec to be used. Not all codecs are supported by a given :class:`MovieWriter`. If none is given, this defaults to the value specified by the rcparam `animation.codec`. *bitrate* specifies the amount 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 no value is given, this defaults to the value given by the rcparam `animation.bitrate`. *extra_args* is a list of extra string arguments to be passed to the underlying movie utiltiy. The default is None, which passes the additional argurments in the 'animation.extra_args' rcParam. *metadata* is 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. *extra_anim* is a list of additional `Animation` objects that should be included in the saved movie file. These need to be from the same `matplotlib.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* is a dictionary containing keyword arguments to be passed on to the 'savefig' command which is called repeatedly to save the individual frames. This can be used to set tight bounding boxes, for example. t bbox_inchesRrRsWarning: discarding the 'bbox_inches' argument in 'savefig_kwargs' as it is only currently supported with the writers 'ffmpeg_file' and 'mencoder_file' (writer used: '{}').t _intervalg@@sanimation.writers savefig.dpisanimation.codecsanimation.bitratec3s'|]}|jjkr|VqdS(N(R(Rtanim(R(sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pys sRRiNsMovieWriter %s unavailableisfCannot save animation: no writers are available. Please install mencoder or ffmpeg to save animations.sAnimation.save using %sR>R@RR(RrR('Rt isinstanceRsRR?Rdt __class__RtpopRRRRRLR9thasattrRRRiRtwritersRtwarningstwarnRt IndexErrort ValueErrorRR6ttypeR+t itertoolstiziptnew_saved_frame_seqtzipt_draw_next_frameRGRR(RtfilenametwriterRR"RRRRt extra_animRDtreconnect_first_drawtall_animRtatdataRtd((RsN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pytsaveCsd1                   #cGsCy*t|j}|j||jtSWntk r>tSXdS(s Handler for getting events. By default, gets the next frame in the sequence and hands the data off to be drawn. N(tnextRRRRLt StopIterationR9(RR(t framedata((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRs  cCs t|jS(s,Creates a new sequence of frame information.(titert _framedata(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCs |jS(s9Creates a new sequence of saved/cached frame information.(R(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCs1|j|||j||j||dS(N(t _pre_drawt _draw_framet _post_draw(RRR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRs cCsdS(N((R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCs#|r|j|j|jndS(N(t _blit_cleart_drawn_artistst _blit_cache(RRR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCstddS(NsDNeeds to be implemented by subclasses to actually make an animation.(RH(RR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCs<|r(|jr(|j|j|jn|jjjdS(N(Rt _blit_drawRRRt draw_idle(RRR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCsg}xb|D]Z}|j|krG|jjj|jj||j$s(RRRtrestore_region(RRRRR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR s cCsJt|_g|_|jjjd|j|_|jd|j dS(Nt resize_event( RRRRRRt_handle_resizeRRRR(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR(s   cGs\|jjj|j|jj|jj|j|jjj d|j |_dS(NR( RRRRRtstopRtclearRRt _end_redraw(RR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR1s    cCsX|jd|j|jj|jjj|j|jjj d|j |_dS(NR( RRRRRRRRRRR(Rtevt((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR=s  N(RRRNRR9R RRRRRRRRRRRRRRRR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRs(             tTimedAnimationcBs>eZdZddeddZdZdZdZRS(sI :class:`Animation` subclass that supports time-based animation, drawing a new frame every *interval* milliseconds. *repeat* controls whether the animation should repeat when the sequence of frames is completed. *repeat_delay* optionally adds a delay in milliseconds before repeating the animation. icOse||_||_||_|dkrE|jj}|j|_ntj||d|||dS(NR( Rt _repeat_delaytrepeatRRt new_timertintervalRR (RR Rt repeat_delayRRR(RT((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR Rs    cGstj||}| r|jr|j|_|jrs|jj|j|jj|j |j|j_ t Stj||Sn|SdS(s- Handler for getting events. N( RRRRRRRRRt _loop_delayRRL(RR(t still_going((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRbs  cGs$|jj|jtj|dS(N(RRRRR(RR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRyscGsF|jj|j|j|j_|jj|jtj|dS(N(RRRRRRRR(RR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRsN( RRRNRRLR RRR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRGs      tArtistAnimationcBs2eZdZdZdZdZdZRS(s  Before calling this function, all plotting should have taken place and the relevant artists saved. frame_info is a list, with each list entry a collection of artists that represent what needs to be enabled on each frame. These will be disabled for other frames. cOs,g|_||_tj||||dS(N(RRRR (RR RR(RT((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR s  cCsg}xT|jD]F}x=|D]5}|jt|j|kr |j|jq q WqWx|D]}|jjjqdWdS(N(Rt set_visibleR9RR`RRtdraw(RRtfRR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRs   cCsD|r|j|j|jn!x|jD]}|jtq)WdS(s5 Clears artists from the last frame. N(RRRRR9(RRRR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCs+||_x|D]}|jtqWdS(N(RRRL(RRR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRs  (RRRNR RRR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRs   t FuncAnimationcBsGeZdZdddddZdZdZdZdZRS(s Makes an animation by repeatedly calling a function *func*, passing in (optional) arguments in *fargs*. *frames* can be a generator, an iterable, or a number of frames. *init_func* is 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. If blit=True, *func* and *init_func* should return an iterable of drawables to clear. c s|r||_n d|_||_||_dkrHtj|_nctr`|_nKtrfd|_t |_nfd|_|_|jdkrd|_n||_ g|_ t j |||g|_ dS(Ncs tS(N(R((tframes(sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pytscsttS(N(Rtrange((R(sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRsid((R2t_funct save_countRRtcountt _iter_gentcallableRtlent _init_funct _save_seqRR (RR tfuncRt init_functfargsRRT((RsN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyR s(            cCs |jS(N(R(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCs3|jrt|jStj|j|jSdS(N(RRRtisliceRR(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRs  cCs>|jdkr+|jt|jn|j|_dS(N(RRRRRR(R((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRscCs@|jj||j|j |_|j||j|_dS(N(RR`RRR2R(RR((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRsN( RRRNRR RRRR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pyRs   +  (!R4RRPtmatplotlib.cbookRRtmatplotlib.compatRt matplotlibRRtobjectRRRRSRhRRoRsR|R~RRRRRRRRRRR(((sN/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/animation.pytsF     c     EA6