Vji,ddlmZmZddlZddlmZddlmZddlm Z m Z m Z m Z m Z dZdZdZd ejd ed ed ed e deef dZdeddeded ed ededeededdefdZddddddddeded ed ededeededdefdZddddddddeded ed ededeededdefdZde dZd ede_d!ede_dS)")LiteralOptionalN) FrameBatch) VideoDecoder)_FRAMEBATCH_RETURN_DOCS_POLICY_FUNCTION_TYPE_POLICY_FUNCTIONS_reshape_4d_framebatch_into_5d_validate_common_paramscj|dkrtd|d|dkrtd|ddS)Nrz num_clips (z ) must be > 0znum_indices_between_frames (z) must be strictly positive) ValueError num_clipsnum_indices_between_framess Z/root/voice-cloning/.venv/lib/python3.11/site-packages/torchcodec/samplers/_index_based.py_validate_params_index_basedrsYA~~?y???@@@!Q&& b+E b b b   '&cR|dkr||z}||krtd|d|dt||}|3t||z dzd}||krtd|d|dn7|dkr||z}t||}||krtd|d |d||fS) Nrzsampling_range_start (z-) must be smaller than the number of frames (z).rnum_frames_per_clipz0We determined that sampling_range_end should be z;, but it is smaller than or equal to sampling_range_start (z+) must be smaller than sampling_range_end ()r _get_clip_spanmaxmin)rrsampling_range_startsampling_range_endnum_frames_in_video clip_spans r$_validate_sampling_range_index_basedrs^a25II222 =%9 = =%8 = = =   #=/I ! !4y!@1!DaHH #5 5 5-CU--(---  6  ! !!47I!I  !35HII #5 5 5>)=>>'9>>>  !3 33rc||dz zdzS)aReturn the span of a clip, i.e. the number of frames (or indices) between the first and last frame in the clip, both included. This isn't the same as the number of frames in a clip! Example: f means a frame in the clip, x means a frame excluded from the clip num_frames_per_clip = 4 num_indices_between_frames = 1, clip = ffff , span = 4 num_indices_between_frames = 2, clip = fxfxfxf , span = 7 num_indices_between_frames = 3, clip = fxxfxxfxxf, span = 10 rrs rrrFs &)_C_log_api_usage_oncerCr1rrrrrr2s rclips_at_random_indicesrJsJ H  !NOOO ' /#=1-    rc ntjdtd|||||||S)Nz,torchcodec.samplers.clips_at_regular_indicesr0rErFrIs rclips_at_regular_indicesrLsJ H  !OPPP ' /#=1-    ra Args: decoder (VideoDecoder): The :class:`~torchcodec.decoders.VideoDecoder` instance to sample clips from. num_clips (int, optional): The number of clips to return. Default: 1. num_frames_per_clip (int, optional): The number of frames per clips. Default: 1. num_indices_between_frames(int, optional): The number of indices between the frames *within* a clip. Default: 1, which means frames are consecutive. This is sometimes refered-to as "dilation". sampling_range_start (int, optional): The start of the sampling range, which defines the first index that a clip may *start* at. Default: 0, i.e. the start of the video. sampling_range_end (int or None, optional): The end of the sampling range, which defines the last index that a clip may *start* at. This value is exclusive, i.e. a clip may only start within [``sampling_range_start``, ``sampling_range_end``). If None (default), the value is set automatically such that the clips never span beyond the end of the video. For example if the last valid index in a video is 99 and the clips span 10 frames, this value is set to 99 - 10 + 1 = 90. Negative values are accepted and are equivalent to ``len(video) - val``. When a clip spans beyond the end of the video, the ``policy`` parameter defines how to construct such clip. policy (str, optional): Defines how to construct clips that span beyond the end of the video. This is best described with an example: assuming the last valid index in a video is 99, and a clip was sampled to start at index 95, with ``num_frames_per_clip=5`` and ``num_indices_between_frames=2``, the indices of the frames in the clip are supposed to be [95, 97, 99, 101, 103]. But 101 and 103 are invalid indices, so the ``policy`` parameter defines how to replace those frames, with valid indices: - "repeat_last": repeats the last valid frame of the clip. We would get [95, 97, 99, 99, 99]. - "wrap": wraps around to the beginning of the clip. We would get [95, 97, 99, 95, 97]. - "error": raises an error. Default is "repeat_last". Note that when ``sampling_range_end=None`` (default), this policy parameter is unlikely to be relevant.  z(Sample :term:`clips` at random indices. z:Sample :term:`clips` at regular (equally-spaced) indices. )typingrrr> torchcodecrtorchcodec.decodersrtorchcodec.samplers._commonrrr r r rrrTensorrAr&r-rCrJrL _COMMON_DOCS__doc__r!rrrUs$$$$$$$$ !!!!!!,,,,,,   )4)4)4X F F F! !!!$ !  ! & ! #Y!!!!HA % &A A A  A !$ AA! A 2 3AAAAAN &' !(,6C   !$   !  2 36 &' !(,6C   !$   !  2 30*RS*** X# ### $ $$$ r