Vjiv9hddlmZmZddlZddlmZddlmZmZm Z m Z m Z dZ dZ dejded ed ed ed eef d Zdddeddeedeeded eedeedeededd efdZddddddddeded eedeedeededd efdZdddddddeded eedeedeededd efdZdZdZd ed!ee"d#e_d$Zd%Zd&eeee"d#e_dS)')LiteralOptionalN) FrameBatch)_FRAMEBATCH_RETURN_DOCS_POLICY_FUNCTION_TYPE_POLICY_FUNCTIONS_reshape_4d_framebatch_into_5d_validate_common_paramscf||||td||dkrtd|d||dkrtd|d|jjtd|jjtdd|jjz }||}n|dkrtd|d |S) Nz.This is internal only and should never happen.rzseconds_between_clip_starts (z ) must be > 0z num_clips (zZCould not infer average fps from video metadata. Try using an index-based sampler instead.zYCould not infer stream end from video metadata. Try using an index-based sampler instead.z) must be > 0, got) ValueErrormetadata average_fpsend_stream_seconds)decoder num_clipsseconds_between_clip_startsseconds_between_framesaverage_frame_duration_secondss Y/root/voice-cloning/.venv/lib/python3.11/site-packages/torchcodec/samplers/_time_based.py_validate_params_time_basedrs& 9A"="IIJJJ".3NRS3S3S V,G V V V   a?y???@@@#+ 8   *2 8   &')9)E%E"%!? 1 $ $ [,G [ [ [    "!c*||}n6||krtd|d|||krtd|d|| ||dz |zz }n||krtd|d|||krtd|d|dt||}||fS)Nzsampling_range_start (z) must be at least z) must be smaller than r zsampling_range_end (z+) must be smaller than sampling_range_end ())r minnum_frames_per_cliprsampling_range_startsampling_range_endbegin_stream_secondsrs r#_validate_sampling_range_time_basedr!>s.#3 "6 6 6h)=hhRfhh  #5 5 5j)=jjVhjj !0 "5"9=S!S S  3 3 3 `#5 ` `J^ ` `   111 {%9 { {fx { { {   /1CDD !3 33rclip_start_secondsrrr policy_funreturncg}|D]Dfdt|D}t||kr |||}||z }E|S)Nc0g|]}|zzxkSr').0irr start_seconds timestamps r z/_build_all_clips_timestamps..sI   *Q1G-GGG  !! !!!r)rangelen) r"rrrr#all_clips_timestampsclip_timestampsr*r+s `` @@r_build_all_clips_timestampsr1}s)++ 0 0        .//      "5 5 5(j:MNNO/ r repeat_last)policykind)randomregularrrrrr3)r2wraperrorct|||t||||}t|||||jj|jj\}}|dkr$|J||z } t j|| z|z} nA|Jt j|||} | d|kr | | |k} t| }t| |||jjt|} | | } t| ||S) N)rrr3)rrrrrr5)r"rrrr#)seconds)framesrr)r rr!rr rtorchrandaranger.r1rget_frames_played_atr ) r4rrrrrrrr3sampling_range_widthr"r/r<s r_generic_time_based_samplerrBs"/ 9$?5 0S/51-$-B"+> 000,, x$$$14HH Jy ! !$8 8;O O +666"\  '   b !%7 7 7!3"%77" *++ 6-/5"+>$V,  ) )2F ) G GF )/   rr )rrrrrr3c ptjdtd||d||||| S)Nz.torchcodec.samplers.clips_at_random_timestampsr5 r4rrrrrrrr3r=_C_log_api_usage_oncerB)rrrrrrr3s rclips_at_random_timestampsrHsM H  !QRRR & $(/51-    r)rrrrr3c ptjdtd|d|||||| S)Nz/torchcodec.samplers.clips_at_regular_timestampsr6rDrE)rrrrrrr3s rclips_at_regular_timestampsrJsM H  !RSSS & $?/51-    ra {maybe_note} Args: decoder (VideoDecoder): The :class:`~torchcodec.decoders.VideoDecoder` instance to sample clips from. {num_clips_or_seconds_between_clip_starts} num_frames_per_clip (int, optional): The number of frames per clips. Default: 1. seconds_between_frames (float or None, optional): The time (in seconds) between each frame within a clip. More accurately, this defines the time between the *frame sampling point*, i.e. the timestamps at which we sample the frames. Because frames span intervals in time , the resulting start of frames within a clip may not be exactly spaced by ``seconds_between_frames`` - but on average, they will be. Default is None, which is set to the average frame duration (``1/average_fps``). sampling_range_start (float or None, optional): The start of the sampling range, which defines the first timestamp (in seconds) that a clip may *start* at. Default: None, which corresponds to the start of the video. (Note: some videos start at negative values, which is why the default is not 0). sampling_range_end (float or None, optional): The end of the sampling range, which defines the last timestamp (in seconds) 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, i.e. it is set to ``end_video_seconds - (num_frames_per_clip - 1) * seconds_between_frames``. 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 (seekable) timestamp in a video is 10.9, and a clip was sampled to start at timestamp 10.5, with ``num_frames_per_clip=5`` and ``seconds_between_frames=0.2``, the sampling timestamps of the frames in the clip are supposed to be [10.5, 10.7, 10.9, 11.1, 11.2]. But 11.1 and 11.2 are invalid timestamps, so the ``policy`` parameter defines how to replace those frames, with valid sampling timestamps: - "repeat_last": repeats the last valid frame of the clip. We would get frames sampled at timestamps [10.5, 10.7, 10.9, 10.9, 10.9]. - "wrap": wraps around to the beginning of the clip. We would get frames sampled at timestamps [10.5, 10.7, 10.9, 10.5, 10.7]. - "error": raises an error. Default is "repeat_last". Note that when ``sampling_range_end=None`` (default), this policy parameter is unlikely to be relevant. {return_docs} zO num_clips (int, optional): The number of clips to return. Default: 1. z+Sample :term:`clips` at random timestamps. ) maybe_note(num_clips_or_seconds_between_clip_starts return_docs zj seconds_between_clip_starts (float): The space (in seconds) between each clip start. a .. note:: For consistency with existing sampling APIs (such as torchvision), this sampler takes a ``seconds_between_clip_starts`` parameter instead of ``num_clips``. If you find that supporting ``num_clips`` would be useful, please let us know by `opening a feature request `_. z=Sample :term:`clips` at regular (equally-spaced) timestamps. )typingrrr= torchcodecrtorchcodec.samplers._commonrrrr r rr!Tensorintfloatlistr1rBrHrJ _COMMON_DOCS_NUM_CLIPS_DOCSformat__doc___SECONDS_BETWEEN_CLIP_STARTS _NOTE_DOCSr'rrr]sO$$$$$$$$ !!!!!!,",","^<4<4<4~    "    &   %[    H7DMMM % &M} M "*% M  M%UOM#5/M!M 2 3MMMMMf .2,0*.6C  %UO #5/! 2 3< !.2,0*.6C"'  %UO #5/! 2 342 j& _kBCC&&&"    '  Uq@WXX'''###r