U a¨+dIiã @sîddlZddlZddlZddlZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZej d¡Zedd „ƒZd ed <d ed <d ed<d ed<ded<ded<ded<dd„Zdd„Zeƒdd„ƒZdd„Zd'dd „Zd(d%d&„ZdS))éN)Ú lru_cache)Ú defaultdicté)Úget_ffmpeg_exeÚ _popen_kwargsÚlogger)Ú LogCatcherÚparse_ffmpeg_headerÚcvsecsÚwincCsdS)Néÿÿÿÿ©r r r ú6/tmp/pip-unpacked-wheel-hya5gyls/imageio_ffmpeg/_io.pyÚórédÚlibx264éZÚ h264_nvencÚ nvenc_h264ÚnvencéPZ h264_vaapiéFZ libopenh264é2Z libx264rgbc Cs>tƒdddddd|dddg }tj|tjtjtjd }|jd kS) Nú -hide_bannerú-fZlavfiú-iznullsrc=s=256x256:d=8ú-vcodecÚnullú-©ÚstdinÚstdoutÚstderrr)rÚ subprocessÚrunÚPIPEÚ returncode)ÚencoderÚcmdÚpr r rÚffmpeg_test_encoder!s&õ ür+cCsètƒddg}tj|tjtjtjd}|j ¡ dd¡}| d¡}|d d¡}g}| d¡D]L}| ¡}| d ¡d}|t krŠ|  |¡q\|d d kr\d |kr\|  |¡q\|j d dd„dd|kràdD]}||krÈ|  |¡qÈt |ƒS)Nrz -encodersr ú Úz------rÚ ú rÚVzH.264TcSst|S)N)Úh264_encoder_preference)Úxr r rrqrz,get_compiled_h264_encoders..)ÚreverseÚkeyr)rr)rr$r%r&r"ÚdecodeÚreplaceÚsplitÚstripr1ÚappendÚsortÚremoveÚtuple)r)r*r"Z header_footerÚfooterÚencodersÚliner(r r rÚget_compiled_h264_encoders:s0 ü    r@cCs,tƒ}|D]}t|ƒr |Sq tdƒ‚dS)Nz=No valid H.264 encoder was found with the ffmpeg installation)r@r+Ú RuntimeError)Zcompiled_encodersr(r r rÚ get_first_available_h264_encoder{s ÿrBc Cs‚t|tjƒrt|ƒ}t|tƒs&tdƒ‚tƒd|ddddddd g }ztj|fd tjit ƒ—Ž}WnDtj k r¢}z$|j j d d }t d  |j|¡ƒ‚W5d}~XYnXd}}t| ¡ƒD]¼}| d¡r¸|j d d }| d¡}|dkr||d… dd¡d ¡ dd¡d ¡}t|ƒ}| d¡}|dkrh||d… dd¡d ¡ dd¡d ¡}t| d¡Ž}||fSq¸t dƒ‚dS)ap Get the number of frames and number of seconds for the given video file. Note that this operation can be quite slow for large files. Disclaimer: I've seen this produce different results from actually reading the frames with older versions of ffmpeg (2.x). Therefore I cannot say with 100% certainty that the returned values are always exact. ú,Video path must be a string or pathlib.Path.rú-mapú0:v:0z-cÚcopyrrrr#Úignore)ÚerrorszFFMEG call failed with {}: {}Nsframe=zframe=rú=rr r/ztime=ú:zCould not get number of frames)Ú isinstanceÚpathlibÚPurePathÚstrÚ TypeErrorrr$Ú check_outputÚSTDOUTrÚCalledProcessErrorÚoutputr5rAÚformatr'ÚreversedÚ splitlinesÚ startswithÚfindr7Úlstripr8Úintr ) Úpathr)ÚoutÚerrÚnframesZnsecsr?ÚiÚsr r rÚcount_frames_and_secs‡sD  ö $    ,  ,raÚrgb24ccsÄt|tjƒrt|ƒ}t|tƒs&tdƒ‚|p,d}|p4d}|p@|d}|pHg}|pPg}t|tƒsdtdƒ‚t|tƒsvtdƒ‚t|tƒsˆtdƒ‚t|tƒsštdƒ‚d |d d d d g}tƒg}||d|g7}|||dg7}t j |ft j t j t j dœt dd—Ž}t |jƒ} d} zÚzŒt ¡d} |  ¡rJ| jsJt ¡| krJt d¡q| jsp|  d¡} d}t| | ¡ƒ‚nd| jkrŠtd |¡ƒ‚t| jƒ}|V|d\}}|||}|d}|  ¡sÌtdƒ‚t|ƒ}d }|d!7}zft!ƒ}t"|ƒ|kr>|j #|t"|ƒ¡}|s2t"|ƒd kr*WWW¢ÀdSt$d"ƒ‚||7}qè|VWnLtk r’} z,t| ƒ}|  d#¡} d$}t$| ||| ¡ƒ‚W5d} ~ XYnXqØWnFt%k r°Yn2tk rÆ‚Ynt&k ràd%} ‚YnXW5|  ¡| ¡dkr¾z|j ¡|j ¡Wn6tk rJ} zt dt| ƒ¡W5d} ~ XYnX| dkr¶z:t ¡d} t ¡| krŽ| ¡dkrŽt d¡qdW5| ¡dkr²t d¡| ¡Xn| ¡XdS)&a: Create a generator to iterate over the frames in a video file. It first yields a small metadata dictionary that contains: * ffmpeg_version: the ffmpeg version in use (as a string). * codec: a hint about the codec used to encode the video, e.g. "h264". * source_size: the width and height of the encoded video frames. * size: the width and height of the frames that will be produced. * fps: the frames per second. Can be zero if it could not be detected. * duration: duration in seconds. Can be zero if it could not be detected. After that, it yields frames until the end of the video is reached. Each frame is a bytes object. This function makes no assumptions about the number of frames in the data. For one because this is hard to predict exactly, but also because it may depend on the provided output_params. If you want to know the number of frames in a video file, use count_frames_and_secs(). It is also possible to estimate the number of frames from the fps and duration, but note that even if both numbers are present, the resulting value is not always correct. Example: gen = read_frames(path) meta = gen.__next__() for frame in gen: print(len(frame)) Parameters: path (str): the filename of the file to read from. pix_fmt (str): the pixel format of the frames to be read. The default is "rgb24" (frames are uint8 RGB images). input_params (list): Additional ffmpeg input command line parameters. output_params (list): Additional ffmpeg output command line parameters. bits_per_pixel (int): The number of bits per pixel in the output frames. This depends on the given pix_fmt. Default is 24 (RGB) bpp (int): DEPRECATED, USE bits_per_pixel INSTEAD. The number of bytes per pixel in the output frames. This depends on the given pix_fmt. Some pixel formats like yuv420p have 12 bits per pixel and cannot be set in bytes as integer. For this reason the bpp argument is deprecated. rCrbéézpix_fmt must be a stringz%bpp and bits_per_pixel must be an intúinput_params must be a listúoutput_params must be a listú-pix_fmtrÚrawvideorZ image2piperrr T©Zprevent_sigintÚtimeoutNz(Error while attempting stop ffmpeg (r): z!We had to kill ffmpeg to stop it.gø?ç{®Gáz„?ç$@gš™™™™™É?z1Could not load meta information === stderr === {}zNo such file or directoryz{} not found! Wrong path?Úsizez;incorrect bits_per_pixel, framesize in bytes must be an intrrz4End of file reached before full frame could be read.gš™™™™™Ù?z-Could not read frame {}: {} === stderr === {}Úkill)'rKrLrMrNrOÚAssertionErrorrZÚlistrr$ÚPopenr&rrr#Zstop_meÚpollr"Úcloser!Ú ExceptionrÚwarningrnÚtimeÚsleepÚis_aliveÚheaderZget_textÚIOErrorrTr Ú is_integerÚbytesÚlenÚreadrAÚ GeneratorExitÚ BaseException)r[Zpix_fmtZbppÚ input_paramsÚ output_paramsZbits_per_pixelZpre_output_paramsr)r*Z log_catcherÚ stop_policyr]ÚetimeZerr2ÚfmtÚmetaÚwÚhZframesize_bitsZframesize_bytesZframenrÚbbZ extra_bytesZerr1r r rÚ read_frames½s´5   ÿüû        ÿþ ÿ   ,  $    rŠÚyuv420pééruc#st|tjƒrt|ƒ}t|tƒs&tdƒ‚|p,d}|p4d}|p—Ž}d?}| s d@}zÄzdd}dV}z|j "|¡Wn8tk rv}zdD  ||¡}t#|ƒ‚W5d}~XYnX|d 7}q(WnZt$k r°|dkr¬t dE¡Yn2tk rÆ‚Ynt%k ràdF}‚YnXW5| ¡dkròz|j ¡Wn6tk r8}zt dAt|ƒ¡W5d}~XYnX|d?kr¤z:t ¡| }t ¡|kr|| ¡dkr|t  dC¡qRW5| ¡dkr t dB¡| ¡XnN|d@krêz | ¡dkrÌt  dC¡q°W5| ¡dkræ| ¡Xn| ¡z|j! ¡Wntk rYnXXdS)GaÔ Create a generator to write frames (bytes objects) into a video file. The frames are written by using the generator's `send()` method. Frames can be anything that can be written to a file. Typically these are bytes objects, but c-contiguous Numpy arrays also work. Example: gen = write_frames(path, size) gen.send(None) # seed the generator for frame in frames: gen.send(frame) gen.close() # don't forget this Parameters: path (str): the filename to write to. size (tuple): the width and height of the frames. pix_fmt_in (str): the pixel format of incoming frames. E.g. "gray", "gray8a", "rgb24", or "rgba". Default "rgb24". pix_fmt_out (str): the pixel format to store frames. Default yuv420p". fps (float): The frames per second. Default 16. quality (float): A measure for quality between 0 and 10. Default 5. Ignored if bitrate is given. bitrate (str): The bitrate, e.g. "192k". The defaults are pretty good. codec (str): The codec. Default "libx264" for .mp4 (if available from the ffmpeg executable) or "msmpeg4" for .wmv. macro_block_size (int): You probably want to align the size of frames to this value to avoid image resizing. Default 16. Can be set to 1 to avoid block alignment, though this is not recommended. ffmpeg_log_level (str): The ffmpeg logging level. Default "warning". ffmpeg_timeout (float): Timeout in seconds to wait for ffmpeg process to finish. Value of 0 or None will wait forever (default). The time that ffmpeg needs depends on CPU speed, compression, and frame size. input_params (list): Additional ffmpeg input command line parameters. output_params (list): Additional ffmpeg output command line parameters. audio_path (str): A input file path for encoding with an audio stream. Default None, no audio. audio_codec (str): The audio codec to use if audio_path is provided. "copy" will try to use audio_path's audio codec without re-encoding. Default None, but some formats must have certain codecs specified. rCrbr‹rŒrurézsize must be a 2-tuplerzsize must be intsz {:d}x{:d}Fzsize must be str or tuplezpix_fmt_in must be strzpix_fmt_out must be strzfps must be floatNzquality must be floaté z*quality must be between 1 and 10 inclusivezmacro_block_size must be intzffmpeg_log_level must be strzffmpeg_timeout must be floatrerfz.wmvZmsmpeg4z-anz.gifrz-acodecrDrEz1:a:0z-yrrhrz-srgz-rz{:.02f}rz-b:vrlré3z-crféz -qscale:vz-vfz scale={}:{}a/IMAGEIO FFMPEG_WRITER WARNING: input image is not divisible by macro_block_size={}, resizing from {} to {} to ensure video compatibility with most codecs and players. To prevent resizing, make your input image divisible by the macro_block_size or set the macro_block_size to 1 (risking incompatibility).z-vr/csg|] }|ˆk‘qSr r )Ú.0Úlevel©Úffmpeg_log_levelr rÚ Esz write_frames..)ÚinfoÚverboseÚdebugÚtracezRUNNING FFMPEG COMMAND: r TrirjÚwaitz(Error while attempting stop ffmpeg (w): ziWe had to kill ffmpeg to stop it. Consider increasing ffmpeg_timeout, or setting it to zero (no timeout).rkz2{0:} FFMPEG COMMAND: {1:} FFMPEG STDERR OUTPUT: z:No frames have been written; the written video is invalid.rn)&rKrLrMrNrOÚfloatrZr<rpr}rorTÚlowerÚendswithrBrrrur9ÚjoinÚanyr—r$rqr&rrrr!rsrtrnrvrwr"Úwriterzrr€)r[rmZ pix_fmt_inZ pix_fmt_outZfpsZqualityZbitrateÚcodecZmacro_block_sizer•Zffmpeg_timeoutrr‚Z audio_pathZ audio_codecZfloatishZsizestrZ audio_paramsr)Zout_wZout_hZcmd_strr*rƒr]r„r^r‰Úmsgr r”rÚ write_frames‡s(>   ÿþ   "  ø       $ úÿ   ÿÿüû ÿÿ  $  ÿ   r¤)rbNNNN) rbr‹rŒrNNrŒruNNNNN)ÚsysrvrLr$Ú functoolsrÚ collectionsrÚ_utilsrrrZ_parsingrr r ÚplatformrWZISWINr1r+r@rBrarŠr¤r r r rÚsT    A 8ú Nñ