B %Z\rB@sdZddlmZmZmZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZmZmZeeZddd d d d Zd ZdZddZdZdZdZejZdZdZ dZ!dZ"dZ#ejZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,ejZ-ej.Z/ej.Z0ej1Z2Gdddej3Z4Gdddej3Z5Gdddej3Z6Gd d!d!ej7Z8Gd"d#d#ej3Z9Gd$d%d%ej3Z:e;dejd3d)d*Z?d+d,Z@Gd-d.d.eZAeAd/d0d1d2ZBe CeBdS)4z Plugin for reading videos via AvBin Would be nice if we could capture webcam with this, but unfortunately, avbin does not currently support this. )absolute_importprint_functiondivisionN)formats)Format get_platformget_remote_fileInternetNotAllowedErrorNeedDownloadErrorzlibavbin-10-osx.dylibzavbin-10-win32.dllzavbin-10-win64.dllzlibavbin-10-linux32.sozlibavbin-10-linux64.so)Zosx64Zwin32Zwin64Zlinux32Zlinux64cCs|tkrtd||S)NzAVBin returned error code %i)AVBIN_RESULT_OK RuntimeError)xr4lib/python3.7/site-packages/imageio/plugins/avbin.py AVbinResult-s ri (0c@seZdZdejfdejfdefdefdejdfdejdfdejdfd ejdfd ejdfd ejfd ejfd ejdfg ZdS) AVbinFileInfostructure_size n_streamsZ start_timedurationtitleiauthor copyrightcommentZalbumZyearZtrackZgenrerN) __name__ __module__ __qualname__ctypesc_size_tc_int TimestampZc_char_fields_rrrrrOs     rc@s@eZdZdejfdejfdejfdejfdejfdejfgZdS)_AVbinStreamInfoVideo8widthheightZsample_aspect_numZsample_aspect_denframe_rate_numframe_rate_denN)r$r%r&r'c_uintr+rrrrr,`s r,c@s0eZdZdejfdejfdejfdejfgZdS)_AVbinStreamInfoAudio8Z sample_formatZ sample_rateZ sample_bitsZchannelsN)r$r%r&r'r)r1r+rrrrr2ksr2c@seZdZdefdefgZdS)_AVbinStreamInfoUnion8videoZaudioN)r$r%r&r,r2r+rrrrr3tsr3c@s&eZdZdejfdejfdefgZdS)AVbinStreamInfo8rtypeuN)r$r%r&r'r(r)r3r+rrrrr5xsr5c@s<eZdZdejfdefdejfdeejfdejfgZ dS) AVbinPacketr timestamp stream_indexdatasizeN) r$r%r&r'r(r*r)POINTERZc_uint8r+rrrrr8s r8cCs t|dS)Ni@B)float)r9rrrtimestamp_from_avbinsr?FcCs<t}|r|tkstd|dt|}t|||ddS)a Download the avbin library to your computer. Parameters ---------- directory : str | None The directory where the file will be cached if a download was required to obtain the file. By default, the appdata directory is used. This is also the first directory that is checked for a local version of the file. force_download : bool | str If True, the file will be downloaded even if a local copy exists (and this copy will be overwritten). Can also be a YYYY-MM-DD date to ensure a file is up-to-date (modified date of a file on disk, if present, is checked). z*AVBIN lib is not available for platform %szavbin/)fname directoryforce_downloadN)rFNAME_PER_PLATFORMrr )rArBZplatr@rrrdownloads    rDc Cstdd}|r|St}y t|}Wn tk rFtd|YnXytd|ddStk rttdYn2tk r}zt dt |Wdd}~XYnXdS) z Get avbin .dll/.dylib/.so ZIMAGEIO_AVBIN_LIBNz,Avbin plugin is not supported on platform %szavbin/F)autozNeed avbin library. You can obtain it with either: - download using the command: imageio_download_bin avbin - download by calling (in Python): imageio.plugins.avbin.download() z Could not download avbin lib: %s) osgetenvrrCKeyErrorrr r r IOErrorstr)libplatformerrrrr get_avbin_libs   rNc@sDeZdZdZddZddZddZd d d ZGd d d ej Z dS) AvBinFormata The AvBinFormat uses the AvBin library (based on libav) to read video files. This plugin is more efficient than the ffmpeg plugin, because it uses ctypes (rather than a pipe like the ffmpeg plugin does). Further, it supports reading images into a given numpy array. The limitations of this plugin are that seeking, writing and camera feeds are not supported. See the ffmpeg format for these features. The avbin plugin requires an `avbin` binary. If this binary is not available on the system, it can be downloaded manually from by either - the command line script ``imageio_download_bin avbin`` - the Python method ``imageio.plugins.avbin.download()``. Parameters for reading ---------------------- 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. stream : int Specifies which video stream to read. Default 0. videoformat : str | None Specifies the video format (e.g. 'avi', or 'mp4'). If this is None (default) the format is auto-detected. Parameters for get_data ----------------------- out : np.ndarray destination for the data retrieved. This can be used to save time-consuming memory allocations when reading multiple image sequntially. The shape of out must be (width, height, 3), the dtype must be np.uint8 and it must be C-contiguous. Use the create_empty_image() method of the reader object to create an array that is suitable for get_data. cOsd|_tj|f||dS)N)_avbinr__init__)selfargskwargsrrrrQszAvBinFormat.__init__cCs(|jd|jdkr$|j|jkr$dSdS)Nr?T)modeZmodes extension extensions)rRrequestrrr _can_reads zAvBinFormat._can_readcCsdS)NFr)rRrYrrr _can_writeszAvBinFormat._can_writeNcCs|jdk r|dkr|jStd|dkr0t}tj||_}tj|j_ tj|j _ tj |j _ tj|j _ tjg|j _t|j_ t|j_ tg|j_tg|j_t|j_ tjg|j_t|j_ tjtjg|j_tg|j_ttg|j_tttg|j_ttjttg|j _tj!|j"_ ttjg|j"_t#g|j$_ttt%g|j&_t|j&_ tj|j'_ t#tj!tj tj!ttjg|j'_tj|j(_ t#tj!tj tj!g|j(_||t)|jS)NzjThe imageio avbin plugin is deprecated and will be removed in a future version. Please use ffmpeg instead.)*rPloggerZwarningrNr'ZcdllZ LoadLibraryr)avbin_get_versionZrestypeZavbin_get_ffmpeg_revisionr(Zavbin_get_audio_buffer_sizeZavbin_have_featurec_char_pZargtypesrZ avbin_initZavbin_set_log_level AVbinLogLevelAVbinLogCallbackZavbin_set_log_callback AVbinFilePavbin_open_filenameavbin_open_filename_with_formatavbin_close_filer*Zavbin_seek_filer=ravbin_file_infor5avbin_stream_infoc_void_pavbin_open_stream AVbinStreamPZavbin_close_streamr8 avbin_readZavbin_decode_audioavbin_decode_videoAVBIN_LOG_QUIET)rRZlibpathavbinrrravbinlibs\                 zAvBinFormat.avbinlibc@sHeZdZdddZddZdd Zd d Zd d ZdddZddZ dS)zAvBinFormat.ReaderFrNcCsLt||_t||_||_t||_|j|_d|_ ddi|_ | dS)NZpluginrm) bool _arg_loopint _arg_stream_arg_videoformat_arg_skipemptyrYZget_local_filename _filename_file_meta _init_video)rRZloopstreamZ videoformatZ skipemptyrrr_openGs     zAvBinFormat.Reader._opencCs&|j}|jt}|jdk r<|||jd|_n | ||_|js\t d|jt |_ t |j |j _||jt |j t||jd<|j jd|jd<|j jd|jd<t|j j|jd<d}x&t|j jD]}t}t ||_||j|||jtkr q||j kr6|d 7}q|!|j||_"||_#|j$j%j&|_'|j$j%j(|_)|j'|j)f|jd <|j'|j)f|jd <t*|j$j%j+t*|j$j%j,|jd <|jdd |jd 8<t-|jd|jd |jd<||_.PqWt d|j |jft/|_0t |j0|j0_d|_1dS)NasciizCould not open "%s"Z avbin_versionzutf-8r r!rrrr< source_sizeZfpsg?nframeszStream #%d not found in %r)2formatrnruencodesysgetfilesystemencodingrsrcrvrbrIrZ_infor'ZsizeofrrebyrefrJr]rwr decoder!r?rrangerr5rfr6AVBIN_STREAM_TYPE_VIDEOrrrh_streamZ _stream_infor7r4r-_widthr._heightr>r/r0rq _stream_indexr8_packet _framecounter)rRrmZfilename_bytesZvideo_stream_counteriinforrrrxVsT        zAvBinFormat.Reader._init_videocCs*|jdk r&|j}||jd|_dS)N)rvr~rnrd)rRrmrrr_closes   zAvBinFormat.Reader._closecCs |jdS)Nr})rw)rRrrr _get_lengthszAvBinFormat.Reader._get_lengthcCstj|j|jdftjdS)Nr)dtype)npZzerosrruint8)rRrrrcreate_empty_imagesz%AvBinFormat.Reader.create_empty_imagecCs|j}|jdr:|jdtdkr:|jr:||jd}|dkrLtdnL||jdkrdtdn4||jkr|dkr||| dStd|jd7_|dkr| }|j t j kr|jjr|j|j|jdfkstxy||jt|jWntk rtd YnX|jj|jkr0qy |jj}Wn$tk r`|jd d}YnX||j |jj|jj!|}|j"r|d krPqW|t#|jj$d fS) Nr}infrzFrame index must be > 0zReached end of videozAvbin format cannot seekrrzReached end of video too soonr;r )r9)%r~rnrwr>rp IndexErrorrrrx _get_datarrrrflags c_contiguousshaperrAssertionErrorrjrvr'rrrr:rr; ExceptionZ__array_interface__rkrr<rtdictr9)rRindexoutrmZptrresultrrrrsH       zAvBinFormat.Reader._get_datacCs|jS)N)rw)rRrrrr_get_meta_datasz!AvBinFormat.Reader._get_meta_data)FrNF)N) r$r%r&rzrxrrrrrrrrrReaderFs I sv      (