:j UddlmZmZddlmZeeeeefZdededefdZ dededefdZ dededefd Z eeegefZ e e e d Z eee fed <d Zd edededefdZdZy))CallableUnion) FrameBatchvalues desired_lenreturnc6||dg|t|z zz }|S)Nlenrrs p/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/torchcodec/samplers/_common.py_repeat_last_policyrs'  vbzlkCK788F Mc0||t|zdzzd|S)Nr r s r _wrap_policyrs$ kS[014 5| DDrframes_indicesctd)NzYou set the 'error' policy, and the sampler tried to decode a frame that is beyond the number of frames in the video. Try to leave sampling_range_end to its default value?) ValueError)rrs r _error_policyrs  @ r) repeat_lastwraperror_POLICY_FUNCTIONSct|dkrtdt|d|dkrtd|d|tjvr$td|dtjd y) Nrz,Decoder must have at least one frame, found z frames.rznum_frames_per_clip (z) must be strictly positivezInvalid policy (z). Supported values are .)r rrkeys)decodernum_frames_per_clippolicys r_validate_common_paramsr",s 7|a:3w<. Q  a#$7#88S T  &++--vh&>?P?U?U?W>XXY Z  .rframes num_clipsr c|jjdd}t|jj||g||jj|||j j||S)N)data pts_secondsduration_seconds)r'shaperviewr(r))r#r$r last_3_dimss r_reshape_4d_framebatch_into_5dr-<sq ++##BC(K  V[[  i)< K{ K&&++I7JK0055iATU ra Returns: FrameBatch: The sampled :term:`clips`, as a 5D :class:`~torchcodec.FrameBatch`. The shape of the ``data`` field is (``num_clips``, ``num_frames_per_clips``, ...) where ... is (H, W, C) or (C, H, W) depending on the ``dimension_order`` parameter of :class:`~torchcodec.decoders.VideoDecoder`. The shape of the ``pts_seconds`` and ``duration_seconds`` fields is (``num_clips``, ``num_frames_per_clips``). N)typingrr torchcodecrlistintfloat_LIST_OF_INT_OR_FLOATrrr_POLICY_FUNCTION_TYPErdictstr__annotations__r"r-_FRAMEBATCH_RETURN_DOCSrrr:s""!d3ie45 !03E !E03EE)8;!"7!=?T!TU'  74223         r