pip4TdZddlZddlmZmZmZmZddlZ ddeeeejfde de de de d eed e d eed eeje ffd Z ddeeejfdejde de d eedeedee d e d eedeeee fd dfdZy)z&TorchCodec integration for TorchAudio.N)BinaryIOOptionalTupleUnionuri frame_offset num_frames normalizechannels_firstformat buffer_sizebackendreturnc" ddlm}|sddl} | j dt d|dk7rddl} | j d t d|ddl} | j d t d|ddl} | j d t d ||} | jj} | td | j} | j}|dkDrh||jdk\rK|r|jddfnd|jdf}tj|tj| fS|dd|df}|dk(rK|r|jddfnd|jdf}tj|tj| fS|dkDr |ddd|f}|s|j!dd}|| fS#t$r} td| d} ~ wwxYw#t $r} td |d | | d} ~ wwxYw#t $r} td| | d} ~ wwxYw)a Load audio data from source using TorchCodec's AudioDecoder. .. note:: This function supports the same API as :func:`~torchaudio.load`, and relies on TorchCodec's decoding capabilities under the hood. It is provided for convenience, but we do recommend that you port your code to natively use ``torchcodec``'s ``AudioDecoder`` class for better performance: https://docs.pytorch.org/torchcodec/stable/generated/torchcodec.decoders.AudioDecoder. As of TorchAudio 2.9, :func:`~torchaudio.load` relies on :func:`~torchaudio.load_with_torchcodec`. Note that some parameters of :func:`~torchaudio.load`, like ``normalize``, ``buffer_size``, and ``backend``, are ignored by :func:`~torchaudio.load_with_torchcodec`. To install torchcodec, follow the instructions at https://github.com/pytorch/torchcodec#installing-torchcodec. Args: uri (path-like object or file-like object): Source of audio data. The following types are accepted: * ``path-like``: File path or URL. * ``file-like``: Object with ``read(size: int) -> bytes`` method. frame_offset (int, optional): Number of samples to skip before start reading data. num_frames (int, optional): Maximum number of samples to read. ``-1`` reads all the remaining samples, starting from ``frame_offset``. normalize (bool, optional): TorchCodec always returns normalized float32 samples. This parameter is ignored and a warning is issued if set to False. Default: ``True``. channels_first (bool, optional): When True, the returned Tensor has dimension `[channel, time]`. Otherwise, the returned Tensor's dimension is `[time, channel]`. format (str or None, optional): Format hint for the decoder. May not be supported by all TorchCodec decoders. (Default: ``None``) buffer_size (int, optional): Not used by TorchCodec AudioDecoder. Provided for API compatibility. backend (str or None, optional): Not used by TorchCodec AudioDecoder. Provided for API compatibility. Returns: (torch.Tensor, int): Resulting Tensor and sample rate. Always returns float32 tensors. If ``channels_first=True``, shape is `[channel, time]`, otherwise `[time, channel]`. Raises: ImportError: If torchcodec is not available. ValueError: If unsupported parameters are used. RuntimeError: If TorchCodec fails to decode the audio. Note: - TorchCodec always returns normalized float32 samples, so the ``normalize`` parameter has no effect. - The ``buffer_size`` and ``backend`` parameters are ignored. - Not all audio formats supported by torchaudio backends may be supported by TorchCodec. r) AudioDecoderz`TorchCodec is required for load_with_torchcodec. Please install torchcodec to use this function.NznTorchCodec AudioDecoder always returns normalized float32 samples. The 'normalize=False' parameter is ignored. stacklevelzCThe 'buffer_size' parameter is not used by TorchCodec AudioDecoder.z?The 'backend' parameter is not used by TorchCodec AudioDecoder.zCThe 'format' parameter is not supported by TorchCodec AudioDecoder.z"Failed to create AudioDecoder for : z3Unable to determine sample rate from audio metadataz Failed to decode audio samples: )dtype)torchcodec.decodersr ImportErrorwarningswarn UserWarning Exception RuntimeErrormetadata sample_rateget_all_samplesdatashapetorchzerosfloat32 transpose)rrr r r r r rrerdecoderr! audio_samplesr# empty_shapes X/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/torchaudio/_torchcodec.pyload_with_torchcodecr. sGP4   :   d []huv w WYdqr s  []huv wSs# ""..KPQQJ//1    Da 4::a= (0>4::a=!,Q ST DVK;;{%--@+M MA|}$%Q,:tzz!}a(DJJqM@R {{;emmboolr8r.r;rAr-rKs, 33  !Q xbkk) *QQQ Q  Q SM QQc]Q 5<< Qp  "%)!/3wI sBKK wI wIwI wI SM wI sm wIc]wIwIc]wI%s +,wI wIrJ