Vji\ddlmZddlmZmZddlZddlmZddlmZGddZ dS))Path)OptionalUnionN)Tensor)_corec eZdZdZdedefdZdddddeee fde ed e ede ed df d Z ddddd ede ed e ede ed ef d Z ddddd ede ed e ede ed df dZ dS) AudioEncoderaAn audio encoder. Args: samples (``torch.Tensor``): The samples to encode. This must be a 2D tensor of shape ``(num_channels, num_samples)``, or a 1D tensor in which case ``num_channels = 1`` is assumed. Values must be float values in ``[-1, 1]``. sample_rate (int): The sample rate of the **input** ``samples``. The sample rate of the encoded output can be specified using the encoding methods (``to_file``, etc.). samples sample_ratectjdt|ts t dt |d|jdkrtj|d}|jdkrt d|j d|j tj krt d|j d|dkrt d |d ||_ ||_ dS) Nz torchcodec.encoders.AudioEncoderz5Expected samples to be a Tensor, got type(samples) = .rz/Expected 1D or 2D samples, got samples.shape = z.Expected float32 samples, got samples.dtype = zsample_rate = z must be > 0.)torch_C_log_api_usage_once isinstancer ValueErrortypendim unsqueezeshapedtypefloat32_samples _sample_rate)selfr r s \/root/voice-cloning/.venv/lib/python3.11/site-packages/torchcodec/encoders/_audio_encoder.py__init__zAudioEncoder.__init__s $$%GHHH'6** KW KKK  <1  ogq11G <1  Rw}RRRSS S =EM ) )QgmQQQRR R !  = ===>> > 'N)bit_rate num_channelsr destr!r"returnchtj|j|jt ||||dS)abEncode samples into a file. Args: dest (str or ``pathlib.Path``): The path to the output file, e.g. ``audio.mp3``. The extension of the file determines the audio format and container. bit_rate (int, optional): The output bit rate. Encoders typically support a finite set of bit rate values, so ``bit_rate`` will be matched to one of those supported values. The default is chosen by FFmpeg. num_channels (int, optional): The number of channels of the encoded output samples. By default, the number of channels of the input ``samples`` is used. sample_rate (int, optional): The sample rate of the encoded output. By default, the sample rate of the input ``samples`` is used. )r r filenamer!r"desired_sample_rateN)rencode_audio_to_filerrstr)rr#r!r"r s rto_filezAudioEncoder.to_file-sE0 "M)YY% +       r formatcJtj|j|j||||S)asEncode samples into raw bytes, as a 1D uint8 Tensor. Args: format (str): The format of the encoded samples, e.g. "mp3", "wav" or "flac". bit_rate (int, optional): The output bit rate. Encoders typically support a finite set of bit rate values, so ``bit_rate`` will be matched to one of those supported values. The default is chosen by FFmpeg. num_channels (int, optional): The number of channels of the encoded output samples. By default, the number of channels of the input ``samples`` is used. sample_rate (int, optional): The sample rate of the encoded output. By default, the sample rate of the input ``samples`` is used. Returns: Tensor: The raw encoded bytes as 1D uint8 Tensor. )r r r+r!r"r')rencode_audio_to_tensorrr)rr+r!r"r s r to_tensorzAudioEncoder.to_tensorNs54+M)% +     r c Ptj|j|j|||||dS)alEncode samples into a file-like object. Args: file_like: A file-like object that supports ``write()`` and ``seek()`` methods, such as io.BytesIO(), an open file in binary write mode, etc. Methods must have the following signature: ``write(data: bytes) -> int`` and ``seek(offset: int, whence: int = 0) -> int``. format (str): The format of the encoded samples, e.g. "mp3", "wav" or "flac". bit_rate (int, optional): The output bit rate. Encoders typically support a finite set of bit rate values, so ``bit_rate`` will be matched to one of those supported values. The default is chosen by FFmpeg. num_channels (int, optional): The number of channels of the encoded output samples. By default, the number of channels of the input ``samples`` is used. sample_rate (int, optional): The sample rate of the encoded output. By default, the sample rate of the input ``samples`` is used. )r r r+ file_liker!r"r'N)rencode_audio_to_file_likerr)rr0r+r!r"r s r to_file_likezAudioEncoder.to_file_likeqsB: 'M)% +      r )__name__ __module__ __qualname____doc__rintrrr)rrr*r.r2r rr r s|  ((((((4#'&*%)    CI 3-  sm  c]       J#'&*%) ! ! ! ! 3- ! sm ! c] !  ! ! ! ! P#'&*%)% % % % 3- % sm % c]%  % % % % % % r r ) pathlibrtypingrrrr torchcodecrr r8r rr<s"""""""" L L L L L L L L L L r