o ϯi@slddlZddlZddlZddlmZddlmZddlmZe e Z de defdd Z Gd d d eZdS) N) g_pathmgr) DecoderType)VideodecoderreturncCstt|tjkrddlm}|}|St|tjkr"ddlm}|}|St|tjkr3ddlm }|}|St d|)z Select the class for accessing clips based on provided decoder string Args: decoder (str): Defines what type of decoder used to decode a video. r)EncodedVideoPyAV)EncodedVideoTorchVision)EncodedVideoDecordzUnknown decoder type ) rPYAVencoded_video_pyavr TORCHVISIONencoded_video_torchvisionr DECORDencoded_video_decordr NotImplementedError)rr video_clsr r rS/home/ubuntu/.local/lib/python3.10/site-packages/pytorchvideo/data/encoded_video.pyselect_video_classs    rc@s.eZdZdZe d dededefddZd S) EncodedVideoz EncodedVideo is an abstraction for accessing clips from an encoded video. It supports selective decoding when header information is available. Tpyav file_path decode_audiorcCsVt|d}t|}Wdn1swYt|}||t|j|S)z Fetches the given video path using PathManager (allowing remote uris to be fetched) and constructs the EncodedVideo object. Args: file_path (str): a PathManager file-path. rbN) ropenioBytesIOreadrpathlibPathname)clsrrrfh video_filerrrr from_path/s  zEncodedVideo.from_pathN)Tr)__name__ __module__ __qualname____doc__ classmethodstrboolr%rrrrr)sr)rloggingriopath.common.file_iorpytorchvideo.data.decoderrvideor getLoggerr&loggerr+rrrrrrs