;c]c@`sdZddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl mZddlmZmZmZejeZejjdrdZn<ejjd rd Zn!ejjd rd Znd ZeedZdZeadZdefdYZdej fdYZ!dZ"eddddZ#ej$e#dS(s Plugin that uses ffmpeg to read and write series of images to a wide range of video formats. Code inspired/based on code from moviepy: https://github.com/Zulko/moviepy/ by Zulko i(tabsolute_importtprint_functiontdivisionNi(tformats(tFormatt string_typest image_as_uinttwintdshowtlinuxt video4linux2tdarwint avfoundationsunknown-cam-formatcC`stddS(NsYimageio.ffmpeg.download() has been deprecated. Use 'pip install imageio-ffmpeg' instead.'(t RuntimeError(t directorytforce_download((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pytdownload&scC`sddl}|jS(s1 Wrapper for imageio_ffmpeg.get_ffmpeg_exe() iN(timageio_ffmpegtget_ffmpeg_exe(R((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pytget_exe.s cC`sgtdkrctjdkr*tdnyddl}Wntk rYtdnX|antS(Nis-The ffmpeg plugin does not work on Python 2.xisITo use the imageio ffmpeg plugin you need to 'pip install imageio-ffmpeg'(i(t _ffmpeg_apitNonetsyst version_infoR Rt ImportError(R((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt_get_ffmpeg_api9s    t FfmpegFormatcB`sReZdZdZdZdejfdYZdejfdYZRS(s The ffmpeg format provides reading and writing for a wide range of movie formats such as .avi, .mpeg, .mp4, etc. And also to read streams from webcams and USB cameras. To read from camera streams, supply "" as the filename, where the "0" can be replaced with any index of cameras known to the system. To use this plugin, the ``imageio-ffmpeg`` library should be installed (e.g. via pip). For most platforms this includes the ffmpeg executable. One can use the ``IMAGEIO_FFMPEG_EXE`` environment variable to force using a specific ffmpeg executable. When reading from a video, the number of available frames is hard/expensive to calculate, which is why its set to inf by default, indicating "stream mode". To get the number of frames before having read them all, you can use the ``reader.count_frames()`` method (the reader will then use ``imageio_ffmpeg.count_frames_and_secs()`` to get the exact number of frames, note that this operation can take a few seconds on large files). Alternatively, the number of frames can be estimated from the fps and duration in the meta data (though these values themselves are not always present/reliable). Parameters for reading ---------------------- fps : scalar The number of frames per second to read the data at. Default None (i.e. read at the file's own fps). One can use this for files with a variable fps, or in cases where imageio is unable to correctly detect the fps. loop : bool If True, the video will rewind as soon as a frame is requested beyond the last frame. Otherwise, IndexError is raised. Default False. Setting this to True will internally call ``count_frames()``, and set the reader's length to that value instead of inf. size : str | tuple The frame size (i.e. resolution) to read the images, e.g. (100, 100) or "640x480". For camera streams, this allows setting the capture resolution. For normal video data, ffmpeg will rescale the data. dtype : str | type The dtype for the output arrays. Determines the bit-depth that is requested from ffmpeg. Supported dtypes: uint8, uint16. Default: uint8. pixelformat : str The pixel format for the camera to use (e.g. "yuyv422" or "gray"). The camera needs to support the format in order for this to take effect. Note that the images produced by this reader are always RGB. input_params : list List additional arguments to ffmpeg for input file options. (Can also be provided as ``ffmpeg_params`` for backwards compatibility) Example ffmpeg arguments to use aggressive error handling: ['-err_detect', 'aggressive'] output_params : list List additional arguments to ffmpeg for output file options (i.e. the stream being read by imageio). print_info : bool Print information about the video file as reported by ffmpeg. Parameters for saving --------------------- fps : scalar The number of frames per second. Default 10. codec : str the video codec to use. Default 'libx264', which represents the widely available mpeg4. Except when saving .wmv files, then the defaults is 'msmpeg4' which is more commonly supported for windows quality : float | None Video output quality. Default is 5. Uses variable bit rate. Highest quality is 10, lowest is 0. Set to None to prevent variable bitrate flags to FFMPEG so you can manually specify them using output_params instead. Specifying a fixed bitrate using 'bitrate' disables this parameter. bitrate : int | None Set a constant bitrate for the video encoding. Default is None causing 'quality' parameter to be used instead. Better quality videos with smaller file sizes will result from using the 'quality' variable bitrate parameter rather than specifiying a fixed bitrate with this parameter. pixelformat: str The output video pixel format. Default is 'yuv420p' which most widely supported by video players. input_params : list List additional arguments to ffmpeg for input file options (i.e. the stream that imageio provides). output_params : list List additional arguments to ffmpeg for output file options. (Can also be provided as ``ffmpeg_params`` for backwards compatibility) Example ffmpeg arguments to use only intra frames and set aspect ratio: ['-intra', '-aspect', '16:9'] ffmpeg_log_level: str Sets ffmpeg output log level. Default is "warning". Values can be "quiet", "panic", "fatal", "error", "warning", "info" "verbose", or "debug". Also prints the FFMPEG command being used by imageio if "info", "verbose", or "debug". macro_block_size: int Size constraint for video. Width and height, must be divisible by this number. If not divisible by this number imageio will tell ffmpeg to scale the image up to the next closest size divisible by this number. Most codecs are compatible with a macroblock size of 16 (default), some can go smaller (4, 8). To disable this automatic feature set it to None or 1, however be warned many players can't decode videos that are odd in size and some codecs will produce poor results or fail. See https://en.wikipedia.org/wiki/Macroblock. cC`sa|jddkrtS|jgtdD]}d|^q*krGtS|j|jkr]tSdS(NisI?i s (tmodetFalsetfilenametrangetTruet extensiont extensions(tselftrequestti((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt _can_reads ,cC`s7|jd|jdkr3|j|jkr3tSndS(Nit?(RtmodesR R!R(R"R#((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt _can_writestReaderc B`seZd Zd ZdZed d d ed d d d d ZdZdZ dZ dZ dZ ddZ d d Zd ZRS( c C`s2tjjdr'd|jjdd!Stjjdrt}|jdddtd d g}tj |d tj d tj d tj dt }|j j |j|jjjddd}yt||}Wn!tk rtd|nXd|Stjjdr*t|}|SdSdS(NR s/dev/iiRs -list_devicesttrues-fs-itdummytstdintstdouttstderrtshellsutf-8terrorstignoresNo ffdshow camera at index %i.svideo=%sR s??(Rtplatformt startswithR#t_videoRRt CAM_FORMATtsptPopentPIPERR-treadlinet terminateR.treadtdecodetparse_device_namest IndexErrortstr(R"tindext ffmpeg_apitcmdtproctinfostname((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt_get_cam_inputnames2   *    c C`st|_t||_|dkr3d|_nRt|trRd||_n3t|tryd|kry||_n t d|dkrnt|tst dn|dkrt j d|_ nQt j ||_ ddg} |j j | kr$t djdj| n||_|p6g|_|pEg|_|j|p[g7_d|j_|jjgtd D]} d | ^qkr|jj|j_n|jjrt|jjd } |j| |_n*|jj|_|jjd d |_d|_|j j dkrAd|_d|_nd|_d|_d|_idd6|_ d|_!t"d|_#|jr|jj r|j$|_#n|j#|j d<|j%|jjrt&|j'|_(ndS(Ns%ix%itxs"FFMPEG size must be tuple of "NxM"sFFMPEG pixelformat must be strtuint8tuint16sdtype must be one of: {}s, i s it^s^^itrgb24itrgb48leiitffmpegtplugintinftnframes()RRtboolt _arg_loopRt _arg_sizet isinstancettupleRt ValueErrortnptdtypet_dtypeREtformattjoint_arg_pixelformatt_arg_input_paramst_arg_output_paramsR#R4RRtintRFt _filenametget_local_filenametreplacet_deptht_pix_fmtt_bytes_per_channelt_post_metat _lastreadtfloatt_nframest count_framest _initializet FrameCatchert _read_gent_frame_catcher( R"tlooptsizeRXt pixelformatt print_infot ffmpeg_paramst input_paramst output_paramstfpstallowed_dtypesR$R@((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt_opens`          /          cC`sT|jdk r(|jjd|_n|jdk rP|jjd|_ndS(N(RoRtstop_meRntclose(R"((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt_closePs    cC`s|jj}||jdS(s Count the number of frames. Note that this can take a few seconds for large files. Also note that it counts the number of frames in the original video and does not take a given fps into account. i(Rtcount_frames_and_secsR`(R"tcf((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRkZs cC`s|jS(N(Rj(R"((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt _get_lengthhscC`s|jr.|jtdkr.||j;}n||jkrS|jtdtfS|dkrntdn||jkrtdnx||jks||jdkr|j|n|j ||jd|j \}}||_|td|fSdS( s3 Reads a frame at index. 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 fectching arbitrary frames whenever possible, by moving between adjacent frames. ROtnewisFrame index must be >= 0sReached end of videoidiN( RRRjRiRfRhtdictRR>Rlt _skip_framest _read_frame(R"R@tresulttis_new((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt _get_dataks " cC`s|jS(N(Rg(R"R@((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt_get_meta_datasicC`s|jdk r|jjng}g}||j7}|jjr|dtg7}|jrs|d|jg7}n|jr|d|jg7}qna|dkr||j d}t d|}||}|dd|g7}|dd|g7}n|jr|d|jg7}n|jj j ddrZt |jj d}|d d |g7}n||j7}|j}|j|j} |jj} | |j|| d |d ||_|jjrMy|jj} Wnhtk r9} t| } d tjkrd| kr| d7} qntdj|jj| qX|j j| n5|dkru|j j|jjn |jjdS(Ns-fs-pix_fmts-siRwi s-sss%.06fs-rs%.02fRuRvR s$Unknown input format: 'avfoundation'sPTry installing FFMPEG using home brew to get a version with support for cameras.sNo camera at {}. {}(RnRR{R]R#R4R5R\RSRgtmintkwargstgetRiR^RdRcReRt read_framesR`t__next__tIOErrorR?RR2R>RZtupdate(R"R@tiargstoargst starttimet seek_slowt seek_fastRwtpix_fmttbpptrftmetaterrterr_text((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRlsV          !     icC`s7x!t|D]}|jjq W|j|7_dS(s Reads and throws away n frames N(RRnRRf(R"tnR$((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRscC`s|jd\}}|||j|j}|jrL|jj\}}n|jj}t}t||krt dt||fnt j |d|j j }|j|||jf}||_||fS(NRqs#Frame is %i bytes, but expected %i.RX(RgRcReRot get_frameRnRRtlenR RWt frombufferRYtcopytreshapeRh(R"twtht framesizetsRR((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRs  N(t__name__t __module__RRoRnRFRRyR|RkRRRRlRR(((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyR)s( + N     O tWriterc B`sYeZd Zddd dd d d dddd ZdZdZd Zd ZRS( i tlibx264tyuv420ptquietiic C`s=t|_|jj|_d|_d|_d|_dS(N( RRR#RaR`RRdRct_size( R"RwtcodectbitrateRrRtRuRvtffmpeg_log_leveltqualitytmacro_block_size((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRys   cC`s,|jdk r(|jjd|_ndS(N(t _write_genRR{(R"((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyR|s cC`sY|jd \}}||f}|jdkr4dn |jd}t|dd}|jjsqtj|}n|jdkridd6dd6dd6d d 6}|j |d|_ |j dkrt d n||_||_ |j n||jkrt d n||j kr0t d n|jdk sEt|jj|dS(Niitbitdepthitgraytgray8aRKitrgbais%Image must have 1, 2, 3 or 4 channelss+All images in a movie should have same sizes9All images in a movie should have same number of channels(tshapetndimRtflagst c_contiguousRWtascontiguousarrayRRRRdRVRcRlRtAssertionErrortsend(R"timRRRRqtdepthtmap((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt _append_datas* " "    cC`stddS(Ns5The ffmpeg format does not support setting meta data.(R (R"R((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyt set_meta_data>sc C`s|jdk r|jjn|jjjdd}|jjjdd}|jjjdd}|jjjdd}|jjjdpg}|jjjdpg}||jjjdpg7}|jjjd d}|jjjd d }|jjjd d} |p%d }|jj|j|j d|j d|d|d|d|d|d |d | d|d| |_|jj ddS(NRwi RRRRuRvRtRrRiRit pix_fmt_int pix_fmt_out( RRR{R#RRRt write_framesR`RRdR( R"RwRRRRuRvRrRR((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRlCs6     N( RRRRRyR|RRRl(((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRs   ( (RRt__doc__R%R(RR)R(((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRIs j  1RmcB`s2eZdZdZdZdZdZRS(sN Thread to keep reading the frame data from stdout. This is useful when streaming from a webcam. Otherwise, if the user code does not grab frames fast enough, the buffer will fill up, leading to lag, and ffmpeg can also stall (experienced on Linux). The get_frame() method always returns the last available image. cC`s^||_d|_t|_tj|_tjj ||j t t|_ |j dS(N(t_genRt_frameRt _frame_is_newt threadingtRLockt_locktThreadt__init__t setDaemonRt _should_stoptstart(R"tgen((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRts     cC`s-t|_x|jr(tjdq WdS(NgMbP?(RRtis_alivettimetsleep(R"((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRz~s cC`sVx |jdkr"tjdqW|j$|j}t|_|j|fSWdQXdS(NgMbP?(RRRRRRR(R"R((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRs    cC`spyRxK|jsPtjd|jj}|j||_t|_WdQXqWWnt t fk rknXdS(Ni( RRRRRRRRRt StopIterationtEOFError(R"tframe((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pytruns    (RRRRRzRR(((s5lib/python2.7/site-packages/imageio/plugins/ffmpeg.pyRmls   cC`sg}t}x9|jD]+}|jdrtj||jdddj}|r|jdr|dd!}|j|dgqD|r|jjdr|jdddjdd!}t j jd r|j d d }n|j d d }||dd s@           %, &