gZ*vdZddlZddlZddlmZddlmZddlm Z m Z GddZ d dZ d d Z y) zW On the long term this will implement several methods to make videos out of VideoClips N)proglog) FFMPEG_BINARY)cross_platform_popen_paramsffmpeg_escape_filenamecBeZdZdZ ddZdZdZdZdZy) FFMPEG_VideoWriteraA class for FFMPEG-based video writing. Parameters ---------- filename : str 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 : tuple or list Size of the output video in pixels (width, height). fps : int Frames per second in the output video file. codec : str, optional 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 : str, optional The name of an audio file that will be incorporated to the video. preset : str, optional 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 : str, optional Only relevant for codecs which accept a bitrate. "5000k" offers nice results in general. with_mask : bool, optional Set to ``True`` if there is a mask in the video to be encoded. pixel_format : str, optional Optional: Pixel format for the output video file. If is not specified ``"rgb24"`` will be used as the default format unless ``with_mask`` is set as ``True``, then ``"rgba"`` will be used. logfile : int, optional File descriptor for logging output. If not defined, ``subprocess.PIPE`` will be used. Defined using another value, the log level of the ffmpeg command will be "info", otherwise "error". threads : int, optional Number of threads used to write the output with ffmpeg. ffmpeg_params : list, optional Additional parameters passed to ffmpeg command. Nc v| tj} | |_||_||_|jj dd|_|rdnd} tdd| tjk(rdndd d d d d d |d|dfzd| dd|zdddg} || jd|ddg| jd |d|g| | j| || jd|g| | jdt| g|dk(r)|r'| jddg| jddgn.|dk(r)|dd zdk(r|dd zdk(r| jddg| jt|gttj| tjd!}tj| fi||_y)"N.rgbargb24-yz -loglevelerrorinfo-frawvideoz-vcodec-s%dx%dr-pix_fmtz-rz%.02fz-an-i-z-acodeccopyz-presetz-bz-threadslibvpxyuva420pz -auto-alt-ref0libx264stdoutstderrstdin)spPIPElogfilefilenamecodecsplitextrextendstrrrDEVNULLPopenproc)selfr&sizefpsr' audiofilepresetbitrate with_maskr%threads ffmpeg_params pixel_formatcmd popen_paramss g/var/www/it7/html/youtubeDownloader/venv/lib/python3.12/site-packages/moviepy/video/io/ffmpeg_writer.py__init__zFFMPEG_VideoWriter.__init__Ms ?ggG    ==&&s+B/!*v    "'')Gv      tAwQ( (    cM   # &  JJiF; < Iui89  $ JJ} %   JJg '   JJ CL1 2 H  JJ J/ 0 JJ- .y tAw{a'7d1gkQ>N JJ J/ 0 *84562zzWrww G HHS1L1 cr |jjj|jy#t$r}|jj \}}||j }n5|jjd|jj}|d|jd|}d|vr|d|jdz }t |d|vr+|d |jd |jd z }t |d |vr|d z }t |d|vr|dz }t |d}~wwxYw)zWrites one frame in the file.NrK MoviePy error: FFMPEG encountered the following error while writing file : zUnknown encoderz\ The video export failed because FFMPEG didn't find the specified codec for video encoding z. Please install this codec or change the codec when calling write_videofile. For instance: >>> clip.write_videofile('myvid.webm', codec='libvpx')zincorrect codec parameters ?zN The video export failed, possibly because the codec specified for the video z, is not compatible with the given extension aO. 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.zbitrate not specifiedzn The video export failed, possibly because the bitrate specified was too high or too low for the video codec.zInvalid encoder typezd The video export failed because the codec or file extension you provided is not suitable for video) r.r"writetobytesIOError communicatedecoder%seekreadr&r'r))r/ img_arrayerr_ ffmpeg_errorrs r; write_framezFFMPEG_VideoWriter.write_frames{4 ! IIOO ! !)"3"3"5 62 !"ii335OA|'+224  !!!$#||002 % $ f\ND  !L0::>**FOOD%. 50<?//3zzl;++/88*5# # 2%. )L8M%. (<7O %. e2 !s36 D6C2D11D6c|jr|jjj|jj$|jjj|jj d|_yy)z@Closes the writer, terminating the subprocess if is still alive.N)r.r"closer!waitr/s r;rNzFFMPEG_VideoWriter.closes] 99 IIOO ! ! #yy+   &&( IINN DI r=c|SNrPs r; __enter__zFFMPEG_VideoWriter.__enter__s r=c$|jyrR)rN)r/exc_type exc_value tracebacks r;__exit__zFFMPEG_VideoWriter.__exit__s  r=) rNmediumNFNNNN) __name__ __module__ __qualname____doc__r<rLrNrTrYrSr=r;rrsA;DG2R6!pr=rc Ltj| } |rt|dzd} nd} | d|z|jdu} t ||j ||||| | ||| |  5}|j | d|d D]w\}}|jUd |jj|z}|jdk7r|jd}tj||g}|j|y ddd|r| j| d y#1swY%xYw) zdWrite the clip to a videofile. See VideoClip.write_videofile for details on the parameters. .logw+NzMoviePy - Writing video %s )message) r'r3r4r5r%r2r6r7r8Tuint8)logger with_timesr1dtypezMoviePy - Done !)rdefault_bar_loggeropenmaskrr0 iter_frames get_framerfastypenpdstackrLrN)clipr&r1r'r4r3 write_logfiler2r6r7rdr8r%has_maskwritertframerjs r;ffmpeg_write_videorvs5" ' ' /Fx&($/ 1H<=yy$H   #! & ((d7) &HAuyy$TYY0033::(;;w/D 5$-0   u % &&2  %&7&&s %BDD#c D|jdk7r|jd}|s|jddk(rdnd}tddd|jd dd d d zd d d |ddt |g }|rt |dzd}nt j}tt j|t jd}t j|fi|}|j|j\}} |jr&| d|d|d| j} t| ~y )a,Writes an image (HxWx3 or HxWx4 numpy array) to a file, using ffmpeg. Parameters ---------- filename : str Path to the output file. image : np.ndarray Numpy array with the image data. logfile : bool, optional Writes the ffmpeg output inside a logging file (``True``) or not (``False``). pixel_format : str, optional Pixel format for ffmpeg. If not defined, it will be discovered checking if the image data contains an alpha channel (``"rgba"``) or not (``"rgb24"``). rcrr r rrrNr rrrrrr`rarr?z with command r@)rfrmshaperrrir#r$rr,r-rDrB returncoderErC) r&imager%r8r9log_filer:r.outrIrs r;ffmpeg_write_imager~s** {{g W% "'++a.A"5vG    5;;r?4R4()   x( C6)4077.::BGGDL 88C (< (D 0HC e$:^C5szz|n N  en r=) rNrZFNNNbarN)FN)r^ subprocessr#numpyrnrmoviepy.configr moviepy.toolsrrrrvr~rSr=r;rsP (MOOl     7't= r=