o پi!@sdZddlmZddlZddlZddlmZmZddl m Z ddl m Z ddl mZdd lmZdd lmZmZdd lmZejZGd d d eZe ddS)z Video Reader.)absolute_importN)c_arrayc_str) _init_api) DECORDContext) DECORDError)ndarray)cpugpu) bridge_outc@seZdZdZedddddfddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZd ddZdS)! VideoReaderaIndividual video reader with convenient indexing and seeking functions. Parameters ---------- uri : str Path of video file. ctx : decord.Context The context to decode the video file, can be decord.cpu() or decord.gpu(). width : int, default is -1 Desired output width of the video, unchanged if `-1` is specified. height : int, default is -1 Desired output height of the video, unchanged if `-1` is specified. num_threads : int, default is 0 Number of decoding thread, auto if `0` is specified. fault_tol : int, default is -1 The threshold of corupted and recovered frames. This is to prevent silent fault tolerance when for example 50% frames of a video cannot be decoded and duplicate frames are returned. You may find the fault tolerant feature sweet in many cases, but not for training models. Say `N = # recovered frames` If `fault_tol` < 0, nothing will happen. If 0 < `fault_tol` < 1.0, if N > `fault_tol * len(video)`, raise `DECORDLimitReachedError`. If 1 < `fault_tol`, if N > `fault_tol`, raise `DECORDLimitReachedError`. rc Csd|_t|ts Jt|}t|dr/t|}dt|}t ||j |j |||d||_nt ||j |j |||d||_|jdurJt d|dt |j|_|jdks]Jd|jd|_d|_d|_dS)Nreadz{} bytesrzError reading z...zInvalid frame count: {})_handle isinstancerstrhasattr bytearrayrformatlen_CAPI_VideoReaderGetVideoReader device_type device_id RuntimeError_CAPI_VideoReaderGetFrameCount _num_frame _key_indices _frame_pts_avg_fps)selfurictxwidthheight num_threads fault_tolbar)G/home/ubuntu/.local/lib/python3.10/site-packages/decord/video_reader.py__init__,s&     zVideoReader.__init__cCs6z|jdurt|jWdSWdStyYdSw)N)r_CAPI_VideoReaderFree TypeErrorr!r)r)r*__del__@s  zVideoReader.__del__cCs|jS)zGet length of the video. Note that sometimes FFMPEG reports inaccurate number of frames, we always follow what FFMPEG reports. Returns ------- int The number of frames in the video file. )rr.r)r)r*__len__Gs zVideoReader.__len__cCsjt|tr|t|t|S|dkr||j7}||jks#|dkr,td||j| || S)aGet frame at `idx`. Parameters ---------- idx : int or slice The frame index, can be negative which means it will index backwards, or slice of frame indices. Returns ------- ndarray Frame of shape HxWx3 or batch of image frames with shape NxHxWx3, where N is the length of the slice. rzIndex: {} out of bound: {}) rslice get_batchrangeindicesrr IndexErrorr seek_accuratenextr!idxr)r)r* __getitem__Ss   zVideoReader.__getitem__cCs,|jdusJt|j}|jstt|S)zsGrab the next frame. Returns ------- ndarray Frame with shape HxWx3. N)r_CAPI_VideoReaderNextFrameshape StopIterationr )r!arrr)r)r*r7ks  zVideoReader.nextcCs|jdusJtj|tjd}||dk|j7<|dks.td||dk|j||jksAtd|||jk|S)zTValidate int64 integers and convert negative integers to positive by backward searchN)dtyperzInvalid negative indices: {}zOut of bound indices: {})rnparrayint64rallr5r)r!r4r)r)r*_validate_indiceszs zVideoReader._validate_indicescCsf|jdusJt|tr|t|t|}||}|jdur*t |j |_|j|ddfS)aGet frame playback timestamp in unit(second). Parameters ---------- indices: list of integers or slice A list of frame indices. If negative indices detected, the indices will be indexed from backward. Returns ------- numpy.ndarray numpy.ndarray of shape (N, 2), where N is the size of indices. The format is `(start_second, end_second)`. N) rrr1r2r3r4rrDr_CAPI_VideoReaderGetFramePTSasnumpyr8r)r)r*get_frame_timestamps   zVideoReader.get_frame_timestampcCs2|jdusJt||}t|j|}t|S)a!Get entire batch of images. `get_batch` is optimized to handle seeking internally. Duplicate frame indices will be optmized by copying existing frames rather than decode from video again. Parameters ---------- indices : list of integers A list of frame indices. If negative indices detected, the indices will be indexed from backward Returns ------- ndarray An entire batch of image frames with shape NxHxWx3, where N is the length of `indices`. N)r_ndrArD_CAPI_VideoReaderGetBatchr )r!r4r>r)r)r*r2s zVideoReader.get_batchcCs$|jdurt|j|_|jS)z}Get list of key frame indices. Returns ------- list List of key frame indices. N)r_CAPI_VideoReaderGetKeyIndicesrrFtolistr.r)r)r*get_key_indicess zVideoReader.get_key_indicescCs|jdur t|j|_|jS)ztGet average FPS(frame per second). Returns ------- float Average FPS. N)r _CAPI_VideoReaderGetAverageFPSrr.r)r)r* get_avg_fpss zVideoReader.get_avg_fpscCF|jdusJ|dkr||jksJt|j|}|s!td|dS)zFast seek to frame position, this does not guarantee accurate position. To obtain accurate seeking, see `accurate_seek`. Parameters ---------- pos : integer Non negative seeking position. NrzFailed to seek to frame {})rr_CAPI_VideoReaderSeekrrr!possuccessr)r)r*seek   zVideoReader.seekcCrO)zAccurately seek to frame position, this is slower than `seek` but guarantees accurate position. Parameters ---------- pos : integer Non negative seeking position. Nrz#Failed to seek_accurate to frame {})rr_CAPI_VideoReaderSeekAccuraterrrQr)r)r*r6rUzVideoReader.seek_accuratercCs*|jdusJ|dks Jt|j|dS)aSkip reading multiple frames. Skipped frames will still be decoded (required by following frames) but it can save image resize/copy operations. Parameters ---------- num : int, default is 1 The number of frames to be skipped. Nr)r_CAPI_VideoReaderSkipFrames)r!numr)r)r* skip_framess zVideoReader.skip_framesN)r)__name__ __module__ __qualname____doc__r r+r/r0r:r7rDrGr2rLrNrTr6rYr)r)r)r*r s    r zdecord.video_reader)r] __future__rctypesnumpyr@ _ffi.baserr _ffi.functionr _ffi.ndarrayrbaserr rHr r bridger c_void_pVideoReaderHandleobjectr r)r)r)r*s       j