Vji$xddlZddlmZddlmZmZddlZddlmZddlm Z m Z ddl m Z mZGddZdS) N)Path)OptionalUnion)Tensor)_core AudioSamples)create_decoderERROR_REPORTING_INSTRUCTIONSc eZdZdZdddddeeeejej e e fde e de e de e fdZd efd Z dd ed e ed efdZdS) AudioDecodera!A single-stream audio decoder. This can be used to decode audio from pure audio files (e.g. mp3, wav, etc.), or from videos that contain audio streams (e.g. mp4 videos). Returned samples are float samples normalized in [-1, 1] Args: source (str, ``Pathlib.path``, bytes, ``torch.Tensor`` or file-like object): The source of the video or audio: - If ``str``: a local path or a URL to a video or audio file. - If ``Pathlib.path``: a path to a local video or audio file. - If ``bytes`` object or ``torch.Tensor``: the raw encoded audio data. - If file-like object: we read video data from the object on demand. The object must expose the methods `read(self, size: int) -> bytes` and `seek(self, offset: int, whence: int) -> int`. Read more in: :ref:`sphx_glr_generated_examples_decoding_file_like.py`. stream_index (int, optional): Specifies which stream in the file to decode samples from. Note that this index is absolute across all media types. If left unspecified, then the :term:`best stream` is used. sample_rate (int, optional): The desired output sample rate of the decoded samples. By default, the sample rate of the source is used. num_channels (int, optional): The desired number of channels of the decoded samples. By default, the number of channels of the source is used. Attributes: metadata (AudioStreamMetadata): Metadata of the audio stream. stream_index (int): The stream index that this decoder is retrieving samples from. If a stream index was provided at initialization, this is the same value. If it was left unspecified, this is the :term:`best stream`. N stream_index sample_rate num_channelssourcerrrc^tjdt|d|_t j|j}||jn||_|jtdtz|jt|j krtd|d|j |j|_ t|j t jjstd|d||n |j j|_t j|j|||dS) Nz torchcodec.decoders.AudioDecoder approximate)r seek_modezCThe best audio stream is unknown and there is no specified stream. zThe stream at index z is not a valid stream.z is not an audio stream. r )torch_C_log_api_usage_oncer _decodercoreget_container_metadatabest_audio_stream_indexr ValueErrorr lenstreamsmetadata isinstance _metadataAudioStreamMetadatar_desired_sample_rateadd_audio_stream)selfrrrrcontainer_metadatas \/root/voice-cloning/.venv/lib/python3.11/site-packages/torchcodec/decoders/_audio_decoder.py__init__zAudioDecoder.__init__7s` $$%GHHH&f NNN !8GG#  6 6    $U./   $6$> ? ? ? ?L|LLL +243DE $-)KLL N|NNN  '2KK 8Q !  M%#%       returnc*|S)aReturns all the audio samples from the source. To decode samples in a specific range, use :meth:`~torchcodec.decoders.AudioDecoder.get_samples_played_in_range`. Returns: AudioSamples: The samples within the file. )get_samples_played_in_range)r%s r'get_all_sampleszAudioDecoder.get_all_samplescs//111r) start_seconds stop_secondsc|||kstd|d|dtj|j||\}}|}|j}|J||krt ||z |z}|}nd}|}|jd}|||z z} ||| kr|t | |z |zz } n|} |dd|| f} t| || jd|z |S) a>Returns audio samples in the given range. Samples are in the half open range [start_seconds, stop_seconds). To decode all the samples from beginning to end, you can call this method while leaving ``start_seconds`` and ``stop_seconds`` to their default values, or use :meth:`~torchcodec.decoders.AudioDecoder.get_all_samples` as a more convenient alias. Args: start_seconds (float): Time, in seconds, of the start of the range. Default: 0. stop_seconds (float or None): Time, in seconds, of the end of the range. As a half open range, the end is excluded. Default: None, which decodes samples until the end. Returns: AudioSamples: The samples within the specified range. NzInvalid start seconds: z1. It must be less than or equal to stop seconds (z).)r/r0r)data pts_secondsduration_secondsr) rr get_frames_by_pts_in_range_audioritemr#roundshaper) r%r/r0frames first_ptsroffset_beginningoutput_pts_seconds num_sampleslast_pts offset_endr3s r'r,z(AudioDecoder.get_samples_played_in_rangens[.  #M\,I,Iz-zzjvzzz !A M'%    NN$$ / &&& } $ $$mi&?;%NOO !.   ! !* l1o {[88  # x(?(?$uh.E-T'U'UUJJ$Jaaa)*445*!Z][8#     r))r.N)__name__ __module__ __qualname____doc__rstrrio RawIOBaseBufferedReaderbytesrrintr(rr-floatr,r)r'r r sJ'+%)&* * * * c4r/@%OP* sm * c] * sm * * * * X 2 2 2 2 2KOK K "K 8@K K K K K K K r)r )rFpathlibrtypingrrrr torchcodecrrr"torchcodec.decoders._decoder_utilsr r r rLr)r'rQs """""""" 22222222 d d d d d d d d d d r)