U a+d4@sdZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddlmZddlmZedGdd d Zdd d ZdddZdS)z This module implements all the functions to read a video or a picture using ffmpeg. It is quite ugly, as there are many pitfalls to avoid )divisionN)DEVNULLPY3) get_setting)cvsecsTc@sJeZdZdddZdd d Zdd dZddZddZddZddZ dS)FFMPEG_VideoReaderFNrgb24Tbicubictbrc Cs4||_d|_t||||} | d|_| d|_| d|_|r|d|df}d|krd} t|D]\} } | r`| |j| } q`t|jd| t|jd| f|_n||_||_| d|_ | d|_ | d|_ | |_ ||_ |d krd nd |_|dkr|j\} }|j| |d }||_|d|_||_dS) N video_fps video_sizevideo_rotationrvideo_durationduration video_nframesrgbad)filenameprocffmpeg_parse_infosfpssizeZrotation enumerateint resize_algorZffmpeg_durationnframesinfospix_fmtdepthbufsize initializepos read_framelastread)selfr print_infosr"r check_durationZtarget_resolutionr fps_sourcerZratioidxtargetwhr/B/tmp/pip-unpacked-wheel-0yp4gafk/moviepy/video/io/ffmpeg_reader.py__init__s>   (     zFFMPEG_VideoReader.__init__rcCs||dkr:td|}dd||d|jdd|g}n d|jg}tdg|ddd d d d t|jd |jd|jdddg }|jt j t j t d}t j dkrd|d<t j|f||_dS)z"Opens the file, creates the pipe. rrz-ssz%.06f-i FFMPEG_BINARYz -loglevelerror-fZ image2pipez-vfz scale=%d:%dz -sws_flagsz-pix_fmtz-vcodecZrawvideo-r"stdoutstderrstdinnt creationflagsN)closeminrrtuplerrr r"spPIPErosnamePopenr)r'Z starttimeoffsetZi_argcmd popen_paramsr/r/r0r#LsB      zFFMPEG_VideoReader.initializercCsB|j\}}t|D]}|jj|j||q|j|7_dS)zReads and throws away n frames N)rrangerr8readr!r$)r'nr-r.ir/r/r0 skip_framesks  zFFMPEG_VideoReader.skip_framescCs|j\}}|j||}|jj|}t||krtd|jd|t|fd|j |j d|j |j |j fdt t|dstd|j|j}nFttdrtj|d d }ntj|d d }||t|||f|_||_|S) NzWarning: in file %s, z"%d bytes wanted but %d bytes read,z)at frame %d/%d, at time %.02f/%.02f sec. g?z#Using the last valid frame instead.r&a+MoviePy error: failed to read the first frame of video file %s. That might mean that the file is corrupted. That may also mean that you are using a deprecated version of FFMPEG. On Ubuntu/Debian for instance the version in the repos is deprecated. Please update to a recent version from the website. frombufferZuint8)Zdtype)rr!rr8rJlenwarningswarnrr$rrr UserWarninghasattrIOErrorr&nprN fromstringshape)r'r-r.nbytessresultr/r/r0r%ts:     zFFMPEG_VideoReader.read_framecCst|j|dd}|js6||||_||_||jkrF|jS||jks^||jdkrp||||_n|||jd|}||_|S)a6 Read a file video frame at time t. Note for coders: getting an arbitrary frame in the video with ffmpeg can be painfully slow if some decoding has to be done. This function tries to avoid fetching arbitrary frames whenever possible, by moving between adjacent frames. gh㈵>rr)rrrr#r$r%r&rM)r'tr$rZr/r/r0 get_frames    zFFMPEG_VideoReader.get_framecCsJ|jr8|j|jj|jj|jd|_t|drF|`dS)Nr&)r terminater8r>r9waitrSr&r'r/r/r0r>s     zFFMPEG_VideoReader.closecCs |dS)N)r>r_r/r/r0__del__szFFMPEG_VideoReader.__del__)FNrTNr r )r)r) __name__ __module__ __qualname__r1r#rMr%r\r>r`r/r/r/r0rs 0  %# rcCs&|rdnd}t||dd}|j}~|S)a Read an image file (PNG, BMP, JPEG...). Wraps FFMPEG_Videoreader to read just one image. Returns an ImageClip. This function is not meant to be used directly in MoviePy, use ImageClip instead to make clips out of image files. Parameters ----------- filename Name of the image file. Can be of any format supported by ffmpeg. with_mask If the image has a transparency layer, ``with_mask=true`` will save this layer as the mask of the returned ImageClip rrF)r r))rr&)rZ with_maskr readerZimr/r/r0ffmpeg_read_images  reFr cs|d}tdd|g}|r*|dddg7}dtjtjtd}tjd krNd |d <tj|f|}|\}} | d } ~|rt | | } d | dkrt d|t } d| d<|r zN|rdnd|rdnd} fdd| D| tdd}t|| d<Wnt d|| fYnXdd| D}|gk| d<| drzH|dtd}ttt||dd}|| d<Wnt d|| fYnXfd d!}fd"d#}|d$krz|| d%<Wn|| d%<YnXn0|d&kr$z|| d%<Wn|| d%<YnXd'}| d%}d(D]2}||kr4t|||d)kr4||| d%<q4|rt| d| d%d| d*<| d| d+<nd| d*<d| d+<zVd,d| D}t|r|d}td-|}t|||| d.<nd| d.<Wnt d/|| fYnXd0d| D}|gk| d1<| d1r|dz8td2}|d|d3}t|| d4<Wnd5| d4<YnX| S)6aEGet file infos using ffmpeg. Returns a dictionnary with the fields: "video_found", "video_fps", "duration", "video_nframes", "video_duration", "audio_found", "audio_fps" "video_duration" is slightly smaller than "duration" to avoid fetching the uncomplete frames at the end, which raises an error. z.gifr3r2r5nullz /dev/nullir7r;r<r=utf8zNo such file or directoryz^MoviePy error: the file %s could not be found! Please check that you entered the correct path.Nrzframe=z Duration: rcsg|]}|kr|qSr/r/.0l)keywordr/r0 sz&ffmpeg_parse_infos..z-([0-9][0-9]:[0-9][0-9]:[0-9][0-9].[0-9][0-9])zfMoviePy error: failed to read the duration of file %s. Here are the file infos returned by ffmpeg: %scSs$g|]}d|krtd|r|qS)z Video: z\d+x\d+researchrir/r/r0rm&s Z video_foundz [0-9]*x[0-9]*(,| )rxr zjMoviePy error: failed to read video dimensions in file %s. Here are the file infos returned by ffmpeg: %scsTtd}||dd}d|krHt|ddd}nt|}|S)Nz( [0-9]*.| )[0-9]* tbr rki)rorpstartendsplitfloatreplace)matchZs_tbrr liner/r0get_tbr?s  z#ffmpeg_parse_infos..get_tbrcs2td}t||dd}|S)Nz( [0-9]*.| )[0-9]* fpsrrr)rorprxrurvrw)rzrr{r/r0get_fpsJs "z#ffmpeg_parse_infos..get_fpsr r rg .l?)2g{Gz?rrcSs$g|]}d|krtd|r|qS)zrotate :\d+$rnrir/r/r0rmos rr zhMoviePy error: failed to read video rotation in file %s. Here are the file infos returned by ffmpeg: %scSsg|]}d|kr|qS)z Audio: r/rir/r/r0rm|sZ audio_foundz [0-9]* HzrZ audio_fpsunknown)endswithrrArBrrCrDrE communicatedecodeprint splitlinesrTdictrofindallrrplistmaprrurvrwabsrO)rr(r)r*Zis_GIFrGrHroutputr4rlinesrZindexrzZ lines_videorYr}r~ZcoefrrqZrotation_linesZ rotation_lineZ lines_audioZ hz_stringr/)rlr|r0rs           (            r)T)FTr )__doc__ __future__rloggingrCro subprocessrArPZnumpyrUZmoviepy.compatrrZmoviepy.configrZ moviepy.toolsrcaptureWarningsrrerr/r/r/r0s"    1