o ϯi @s\ddlmZmZddlmZmZmZddlZddlm Z Gddde Z GdddeZ dS) )ABCabstractmethod)BinaryIODictOptionalN) g_pathmgrc@s$eZdZdZd ddZd d d ZdS) VideoPathHandlerzq Utility class that handles all deciphering and caching of video paths for encoded and frame videos. returnNcCs i|_dSNpath_order_cacheselfrK/home/ubuntu/.local/lib/python3.10/site-packages/pytorchvideo/data/video.py__init__s zVideoPathHandler.__init__Fpyavc Csz t|}t|}Wn tyd}d}Ynw|r)ddlm}||||S|r@ddlm}|r7Jd|j |||j dSt |d) NTFr) EncodedVideo) FrameVideoz0decode_audio must be False when using FrameVideor z not found.) risfileisdirNotImplementedErrorpytorchvideo.data.encoded_videor from_pathpytorchvideo.data.frame_videorfrom_directoryr FileNotFoundError) rfilepath decode_audiodecoderfpsis_fileis_dirrrrrrvideo_from_paths"     z VideoPathHandler.video_from_path)r N)Frr)__name__ __module__ __qualname____doc__rr$rrrrr s rc @s~eZdZdZe  ddedeededdfdd Z e ede fd d Z ed e d e de eeejffddZddZdS)VideozM Video provides an interface to access clips from a video container. NTfile video_namerr cCdS)z Args: file (BinaryIO): a file-like object (e.g. io.BytesIO or io.StringIO) that contains the encoded video. Nr)rr*r+rrrrr4s zVideo.__init__cCr,)zG Returns: duration of the video in seconds Nrr rrrdurationBszVideo.duration start_secend_seccCr,)a Retrieves frames from the internal video at the specified start and end times in seconds (the video always starts at 0 seconds). Args: start_sec (float): the clip start time in seconds end_sec (float): the clip end time in seconds Returns: video_data_dictonary: A dictionary mapping strings to tensor of the clip's underlying data. Nr)rr.r/rrrget_clipKszVideo.get_clipcCsdSr rr rrrclose]sz Video.close)NT)r%r&r'r(rrrstrboolrpropertyfloatr-rtorchTensorr0r1rrrrr)/s4   r)) abcrrtypingrrrr6iopath.common.file_iorobjectrr)rrrrs  %