;c]c@`sdZddlmZmZmZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZmZmZejeZidd6d d 6d d 6d d6dd6ZdZdZdZdZdZdZejZdZdZ dZ!dZ"dZ#ejZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,ejZ-ej.Z/ej.Z0ej1Z2dej3fdYZ4dej3fd YZ5d!ej3fd"YZ6d#ej7fd$YZ8d%ej3fd&YZ9d'ej3fd(YZ:ej;e<ej=ejej=Z>d)Z?e<e@d*ZAd+ZBd,efd-YZCeCd.d/d0d1ZDe jEeDdS(2s Plugin for reading videos via AvBin Would be nice if we could capture webcam with this, but unfortunately, avbin does not currently support this. i(tabsolute_importtprint_functiontdivisionNi(tformats(tFormatt get_platformtget_remote_filetInternetNotAllowedErrortNeedDownloadErrorslibavbin-10-osx.dylibtosx64savbin-10-win32.dlltwin32savbin-10-win64.dlltwin64slibavbin-10-linux32.sotlinux32slibavbin-10-linux64.sotlinux64icC`s#|tkrtd|n|S(NsAVBin returned error code %i(tAVBIN_RESULT_OKt RuntimeError(tx((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyt AVbinResult-s iiiiiiii i(i0t AVbinFileInfocB`seZdejfdejfdefdefdejdfdejdfdejdfdejdfd ejdfd ejfd ejfd ejd fg ZRS(tstructure_sizet n_streamst start_timetdurationttitleitauthort copyrighttcommenttalbumtyearttracktgenrei (t__name__t __module__tctypestc_size_ttc_intt Timestamptc_chart_fields_(((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyROs      t_AVbinStreamInfoVideo8cB`sVeZdejfdejfdejfdejfdejfdejfgZRS(twidththeighttsample_aspect_numtsample_aspect_dentframe_rate_numtframe_rate_den(RR R!tc_uintR&(((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyR'`s      t_AVbinStreamInfoAudio8cB`s>eZdejfdejfdejfdejfgZRS(t sample_formatt sample_ratet sample_bitstchannels(RR R!R#R.R&(((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyR/ks   t_AVbinStreamInfoUnion8cB`s eZdefdefgZRS(tvideotaudio(RR R'R/R&(((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyR4tstAVbinStreamInfo8cB`s/eZdejfdejfdefgZRS(Rttypetu(RR R!R"R#R4R&(((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyR7xs  t AVbinPacketcB`sPeZdejfdefdejfdejejfdejfgZRS(Rt timestampt stream_indextdatatsize( RR R!R"R$R#tPOINTERtc_uint8R&(((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyR:s    cC`st|dS(Ni@B(tfloat(R;((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyttimestamp_from_avbinscC`sYt}|o|tks.td|ndt|}td|d|d|dS(s 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). s*AVBIN lib is not available for platform %ssavbin/tfnamet directorytforce_downloadN(RtFNAME_PER_PLATFORMRR(RDREtplatRC((s4lib/python2.7/site-packages/imageio/plugins/avbin.pytdownloads  cC`stjdd}|r|St}yt|}Wn!tk rVtd|nXytd|dtSWnEt k rt dn)t k r}t dt |nXdS(s Get avbin .dll/.dylib/.so tIMAGEIO_AVBIN_LIBs,Avbin plugin is not supported on platform %ssavbin/tautosNeed 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() s Could not download avbin lib: %sN( tostgetenvtNoneRRFtKeyErrorRRtFalseRRtIOErrortstr(tlibtplatformterr((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyt get_avbin_libs    t AvBinFormatcB`sNeZdZdZdZdZddZdej fdYZ RS(s 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. cO`s d|_tj|||dS(N(RMt_avbinRt__init__(tselftargstkwargs((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyRXs cC`s7|jd|jdkr3|j|jkr3tSndS(Nit?(tmodetmodest extensiont extensionstTrue(RYtrequest((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyt _can_readscC`stS(N(RO(RYRb((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyt _can_writescC`sj|jdk r"|dkr"|jStjd|dkrGt}ntjj||_}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_ttjtg|j_ttjtjt g|j!_tj"|j#_ ttjg|j#_t$g|j%_ttjt&g|j'_t|j'_ tj|j(_ t$tj"tj tj"tjtjg|j(_tj|j)_ t$tj"tj tj"g|j)_|j|jt*|jS(NsjThe imageio avbin plugin is deprecated and will be removed in a future version. Please use ffmpeg instead.(+RWRMtloggertwarningRUR!tcdllt LoadLibraryR#tavbin_get_versiontrestypetavbin_get_ffmpeg_revisionR"tavbin_get_audio_buffer_sizetavbin_have_featuretc_char_ptargtypesRt avbin_inittavbin_set_log_levelt AVbinLogLeveltAVbinLogCallbacktavbin_set_log_callbackt AVbinFilePtavbin_open_filenametavbin_open_filename_with_formattavbin_close_fileR$tavbin_seek_fileR?Rtavbin_file_infoR7tavbin_stream_infotc_void_ptavbin_open_streamt AVbinStreamPtavbin_close_streamR:t avbin_readtavbin_decode_audiotavbin_decode_videotAVBIN_LOG_QUIET(RYtlibpathtavbin((s4lib/python2.7/site-packages/imageio/plugins/avbin.pytavbinlibs\         tReadercB`sVeZeddedZdZdZdZdZddZ dZ RS( icC`sot||_t||_||_t||_|jj|_d|_ idd6|_ |j dS(NRtplugin( tboolt _arg_looptintt _arg_streamt_arg_videoformatt_arg_skipemptyRbtget_local_filenamet _filenameRMt_filet_metat _init_video(RYtlooptstreamt videoformatt skipempty((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyt_openGs  cC`s|jj}|jjtj}|jdk rZ|j||jjd|_ n|j ||_ |j st d|jnt |_ tj|j |j _|j|j tj|j t|j|jd<|j jjd|jd<|j jjd|jdt source_sizetfpsg?tnframessStream #%d not found in %r(3tformatRRtencodetsystgetfilesystemencodingRRMRwRRvRPRt_infoR!tsizeofRRztbyrefRQRiRRtdecodeRRBRtrangeRR7R{R8tAVBIN_STREAM_TYPE_VIDEORR}t_streamt _stream_infoR9R5R(t_widthR)t_heightRAR,R-Rt _stream_indexR:t_packett _framecounter(RYRtfilename_bytestvideo_stream_countertitinfo((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyRVsT     %  cC`s>|jdk r:|jj}|j|jd|_ndS(N(RRMRRRx(RYR((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyt_closescC`s |jdS(NR(R(RY((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyt _get_lengthscC`s%tj|j|jdfdtjS(Nitdtype(tnptzerosRRtuint8(RY((s4lib/python2.7/site-packages/imageio/plugins/avbin.pytcreate_empty_imagescC`sC|jj}|jdrU|jdtdkrU|jrU||jd}qUn|dkrptdnm||jdkrtdnK||jkr|dkr|j|j|j dStdn|jd7_|dkr|j }n|j t jkrC|jjrC|j|j|jdfksItxtr)y#|j|jtj|jWntk rtd nX|jj|jkrqLny|jj}Wn"tk r|j d d}nX|j!|j"|jj|jj#|}|j$ s"|d krLPqLqLW|t%d |jj&fS( NRtinfisFrame index must be > 0sReached end of videosAvbin format cannot seekiisReached end of video too soonR=iR;('RRRRARt IndexErrorRRRt _get_dataRMRRRRtflagst c_contiguoustshapeRRtAssertionErrorRaRRR!RRRR<RR=t Exceptiont__array_interface__RRR>RtdictR;(RYtindextoutRtptrtresult((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyRsH&        $ #  !cC`s|jS(N(R(RYR((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyt_get_meta_datasN( RR RORMRRRRRRR(((s4lib/python2.7/site-packages/imageio/plugins/avbin.pyRFs I    sv     (              '