U a+dh# @sjdZddlZddlZddlZddlmZddlmZm Z ddl m Z GdddZ dd dZ dddZdS)zW On the long term this will implement several methods to make videos out of VideoClips N)proglog)DEVNULLPY3) get_settingc@s:eZdZdZdddZdd Zd d Zd d ZddZdS)FFMPEG_VideoWritera0 A class for FFMPEG-based video writing. A class to write videos using ffmpeg. ffmpeg will write in a large choice of formats. Parameters ----------- filename Any filename like 'video.mp4' etc. but if you want to avoid complications it is recommended to use the generic extension '.avi' for all your videos. size Size (width,height) of the output video in pixels. fps Frames per second in the output video file. codec FFMPEG codec. It seems that in terms of quality the hierarchy is 'rawvideo' = 'png' > 'mpeg4' > 'libx264' 'png' manages the same lossless quality as 'rawvideo' but yields smaller files. Type ``ffmpeg -codecs`` in a terminal to get a list of accepted codecs. Note for default 'libx264': by default the pixel format yuv420p is used. If the video dimensions are not both even (e.g. 720x405) another pixel format is used, and this can cause problem in some video readers. audiofile Optional: The name of an audio file that will be incorporated to the video. preset Sets the time that FFMPEG will take to compress the video. The slower, the better the compression rate. Possibilities are: ultrafast,superfast, veryfast, faster, fast, medium (default), slow, slower, veryslow, placebo. bitrate Only relevant for codecs which accept a bitrate. "5000k" offers nice results in general. withmask Boolean. Set to ``True`` if there is a mask in the video to be encoded. libx264NmediumFc Csl| dkrtj} ||_||_|jdd|_tddd| tjkrDdnddd d d d d |d |dfd|rldnddd|dddg} |dk r| d|ddg| d |d|g| dk r| | |dk r| d|g| dk r| dt| g|dkr*|d dd kr*|ddd kr*| ddg| |gt | tjd} t j d krXd!| d"<tj | f| |_ dS)#N. FFMPEG_BINARY-yz -loglevelerrorinfo-frawvideoz-vcodec-s%dx%dr-pix_fmtrgbargb24z-rz%.02fz-an-i-z-acodeccopyz-presetz-bz-threadsrZyuv420pstdoutstderrstdinnt creationflags)spPIPEfilenamecodecsplitextrextendstrrosnamePopenproc)selfr$sizefpsr% audiofilepresetbitratewithmasklogfilethreads ffmpeg_paramscmd popen_paramsr:B/tmp/pip-unpacked-wheel-0yp4gafk/moviepy/video/io/ffmpeg_writer.py__init__Ds     zFFMPEG_VideoWriter.__init__c Csz.tr|jj|n|jj|Wntk r}z|j\}}t|d|j t|f}d|kr|d|j }n@d|kr|d|j |j f}n"d|kr|d}nd|kr|d }t|W5d }~XYnXd S) z Writes one frame in the file.zS MoviePy error: FFMPEG encountered the following error while writing file %s: %ssUnknown encoderz The video export failed because FFMPEG didn't find the specified codec for video encoding (%s). Please install this codec or change the codec when calling write_videofile. For instance: >>> clip.write_videofile('myvid.webm', codec='libvpx')sincorrect codec parameters ?a The video export failed, possibly because the codec specified for the video (%s) is not compatible with the given extension (%s). Please specify a valid 'codec' argument in write_videofile. This would be 'libx264' or 'mpeg4' for mp4, 'libtheora' for ogv, 'libvpx for webm. Another possible reason is that the audio codec was not compatible with the video codec. For instance the video extensions 'ogv' and 'webm' only allow 'libvorbis' (default) as avideo codec.sencoder setup failedzr The video export failed, possibly because the bitrate you specified was too high or too low for the video codec.sInvalid encoder typezY The video export failed because the codec or file extension you provided is not a videoN) rr-rwritetobytestostringIOError communicater)r$r%r')r.Z img_arrayerr_Z ffmpeg_errorr r:r:r; write_frames,    zFFMPEG_VideoWriter.write_framecCs>|jr4|jj|jjdk r*|jj|jd|_dSN)r-rcloserwaitr.r:r:r;rFs     zFFMPEG_VideoWriter.closecCs|SrEr:rHr:r:r; __enter__szFFMPEG_VideoWriter.__enter__cCs |dSrE)rF)r.exc_type exc_value tracebackr:r:r;__exit__szFFMPEG_VideoWriter.__exit__)rNrNFNNN) __name__ __module__ __qualname____doc__r<rDrFrIrMr:r:r:r;rs3 @2 rrrFTbarc  Cst| } |rt|dd} nd} | d|dt||j||||| || | d p}t|j|}|j| d|dd D]H\}}|rd |j |}|j dkr| d}t ||g}||qpW5QRX|r| | d ddS) ze Write the clip to a videofile. See VideoClip.write_videofile for details on the parameters. .logw+NzMoviepy - Writing video %s )message)r%r2r3r5r1r6r7Tuint8)loggerZ with_timesr0dtypezMoviepy - Done !)rZdefault_bar_loggeropenrr/intdurationZ iter_framesmaskZ get_framerXastypenpZdstackrDrF)Zclipr$r0r%r3r2r4Z write_logfiler1verboser6r7rWr5writerZnframestframer]r:r:r;ffmpeg_write_videos8    rdc Cs|jdkr|d}tdddd|jdddddd d d |jdd krPd nddd|g }|rpt|dd}ntj}t|tjd}tj dkrd|d<tj |f|}| | \}}|j rdd|d|dg}t|~dS)zO Writes an image (HxWx3 or HxWx4 numpy array) to a file, using ffmpeg. rVr r rrNrr rrrrrrrrSrTrrr r! z[MoviePy] Running : %s z(WARNING: this command returned an error:utf8)rXr^rshaperZr"r#rr*r+r,rAr? returncodejoindecoder@) r$imager5r8Zlog_filer9r-outrBr:r:r;ffmpeg_write_images>    rn) rNrFFNTNNrR)F)rQr* subprocessr"Znumpyr_rZmoviepy.compatrrZmoviepy.configrrrdrnr:r:r:r;s&  8 $