U
    ,:%e/                     @   s   d dl Z d dlmZ d dlmZmZmZmZmZm	Z	 d dl
Z
d dlmZmZ ddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZ edeeee f dddZedddZdd Zdd Zdd ZdS )    N)	lru_cache)BinaryIODictOptionalTupleTypeUnion)_FFMPEG_EXT_SOX_INITIALIZED   )soundfile_backend)Backend)AudioMetaData)FFmpegBackend)SoundfileBackend)
SoXBackend)returnc                  C   s2   i } t d k	rt| d< tr t| d< tjr.t| d< | S )Nffmpegsox	soundfile)r	   r   r
   r   r   Z_IS_SOUNDFILE_AVAILABLEr   )Zbackend_specs r   X/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torchaudio/_backend/utils.pyget_available_backends   s    r   c                 C   s8   | |  }r|S td|  ddt|  dd S )NzUnsupported backend 'z' specified; zplease select one of z	 instead.)get
ValueErrorlistkeys)backend_namebackendsbackendr   r   r   get_backend   s    
r    c                     sf   t   ttttjf tt tt td fdddttttjf tt ttt t	dfdd} | S )	Nuriformatr   r   c                    sN   |d k	rt | S   D ]}|| |r|  S qtd|  d| dd S Nz0Couldn't find appropriate backend to handle uri z and format .r    valuesZ
can_decodeRuntimeErrorr"   r#   r   r   r   r   r   
dispatcher+   s    

z!get_info_func.<locals>.dispatcher   )r"   r#   buffer_sizer   r   c                    s    | ||}| | ||S )a  Get signal information of an audio file.

        Note:
            When the input type is file-like object, this function cannot
            get the correct length (``num_samples``) for certain formats,
            such as ``vorbis``.
            In this case, the value of ``num_samples`` is ``0``.

        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,
                  which returns byte string of at most ``size`` length.

            format (str or None, optional):
                If not ``None``, interpreted as hint that may allow backend to override the detected format.
                (Default: ``None``)

            buffer_size (int, optional):
                Size of buffer to use when processing file-like objects, in bytes. (Default: ``4096``)

            backend (str or None, optional):
                I/O backend to use.
                If ``None``, function selects backend given input and available backends.
                Otherwise, must be one of [``"ffmpeg"``, ``"sox"``, ``"soundfile"``],
                with the corresponding backend available.
                (Default: ``None``)

                .. seealso::
                   :ref:`backend`

        Returns:
            AudioMetaData
        info)r"   r#   r-   r   r+   r   r   r/   6   s    *zget_info_func.<locals>.info)Nr,   N)
r   r   r   strosPathLiker   r   intr   r.   r   r   r+   r   get_info_func(   s"         -r6   c                     sx   t   ttttjf tt tt td fdddttttjf ttt	t	tt ttt t
tjtf d	fd	d
} | S )Nr!   c                    sN   |d k	rt | S   D ]}|| |r|  S qtd|  d| dd S r$   r&   r)   r*   r   r   r+   i   s    

z!get_load_func.<locals>.dispatcherr   Tr,   )	r"   frame_offset
num_frames	normalizechannels_firstr#   r-   r   r   c              	      s"    | ||}| | ||||||S )a  Load audio data from source.

        By default (``normalize=True``, ``channels_first=True``), this function returns Tensor with
        ``float32`` dtype, and the shape of `[channel, time]`.

        Note:
            The formats this function can handle depend on the availability of backends.
            Please use the following functions to fetch the supported formats.

            - FFmpeg: :py:func:`torchaudio.utils.ffmpeg_utils.get_audio_decoders`
            - Sox: :py:func:`torchaudio.utils.sox_utils.list_read_formats`
            - SoundFile: Refer to `the official document <https://pysoundfile.readthedocs.io/>`__.

        .. warning::

            ``normalize`` argument does not perform volume normalization.
            It only converts the sample type to `torch.float32` from the native sample
            type.

            When the input format is WAV with integer type, such as 32-bit signed integer, 16-bit
            signed integer, 24-bit signed integer, and 8-bit unsigned integer, by providing ``normalize=False``,
            this function can return integer Tensor, where the samples are expressed within the whole range
            of the corresponding dtype, that is, ``int32`` tensor for 32-bit signed PCM,
            ``int16`` for 16-bit signed PCM and ``uint8`` for 8-bit unsigned PCM. Since torch does not
            support ``int24`` dtype, 24-bit signed PCM are converted to ``int32`` tensors.

            ``normalize`` argument has no effect on 32-bit floating-point WAV and other formats, such as
            ``flac`` and ``mp3``.

            For these formats, this function always returns ``float32`` Tensor with values.


        Args:
            uri (path-like object or file-like object):
                Source of audio data.
            frame_offset (int, optional):
                Number of frames to skip before start reading data.
            num_frames (int, optional):
                Maximum number of frames to read. ``-1`` reads all the remaining samples,
                starting from ``frame_offset``.
                This function may return the less number of frames if there is not enough
                frames in the given file.
            normalize (bool, optional):
                When ``True``, this function converts the native sample type to ``float32``.
                Default: ``True``.

                If input file is integer WAV, giving ``False`` will change the resulting Tensor type to
                integer type.
                This argument has no effect for formats other than integer WAV type.

            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):
                If not ``None``, interpreted as hint that may allow backend to override the detected format.
                (Default: ``None``)

            buffer_size (int, optional):
                Size of buffer to use when processing file-like objects, in bytes. (Default: ``4096``)

            backend (str or None, optional):
                I/O backend to use.
                If ``None``, function selects backend given input and available backends.
                Otherwise, must be one of [``"ffmpeg"``, ``"sox"``, ``"soundfile"``],
                with the corresponding backend being available. (Default: ``None``)

                .. seealso::
                   :ref:`backend`

        Returns:
            (torch.Tensor, int): Resulting Tensor and sample rate.
                If the input file has integer wav format and normalization is off, then it has
                integer type, else ``float32`` type. If ``channels_first=True``, it has
                `[channel, time]` else `[time, channel]`.
        load)r"   r8   r9   r:   r;   r#   r-   r   r0   r   r   r=   t   s    Vzget_load_func.<locals>.load)r   r7   TTNr,   N)r   r   r   r1   r2   r3   r   r   r4   boolr   torchTensorr<   r   r5   r   get_load_funcf   s2             YrA   c                     sx   t   ttttjf tt tt td fddd	ttttjf tj	t
ttt tt tt
 t
tt d	fdd} | S )
Nr!   c                    sN   |d k	rt | S   D ]}|| |r|  S qtd|  d| dd S r$   )r    r'   Z
can_encoder(   r)   r*   r   r   r+      s    

z!get_save_func.<locals>.dispatcherTr,   	r"   srcZsample_rater;   r#   encodingZbits_per_sampler-   r   c	           	   
      s$    | ||}| | |||||||S )a]
  Save audio data to file.

        Note:
            The formats this function can handle depend on the availability of backends.
            Please use the following functions to fetch the supported formats.

            - FFmpeg: :py:func:`torchaudio.utils.ffmpeg_utils.get_audio_encoders`
            - Sox: :py:func:`torchaudio.utils.sox_utils.list_write_formats`
            - SoundFile: Refer to `the official document <https://pysoundfile.readthedocs.io/>`__.

        Args:
            uri (str or pathlib.Path): Path to audio file.
            src (torch.Tensor): Audio data to save. must be 2D tensor.
            sample_rate (int): sampling rate
            channels_first (bool, optional): If ``True``, the given tensor is interpreted as `[channel, time]`,
                otherwise `[time, channel]`.
            format (str or None, optional): Override the audio format.
                When ``uri`` argument is path-like object, audio format is
                inferred from file extension. If the file extension is missing or
                different, you can specify the correct format with this argument.

                When ``uri`` argument is file-like object,
                this argument is required.

                Valid values are ``"wav"``, ``"ogg"``, and ``"flac"``.
            encoding (str or None, optional): Changes the encoding for supported formats.
                This argument is effective only for supported formats, i.e.
                ``"wav"`` and ``""flac"```. Valid values are

                - ``"PCM_S"`` (signed integer Linear PCM)
                - ``"PCM_U"`` (unsigned integer Linear PCM)
                - ``"PCM_F"`` (floating point PCM)
                - ``"ULAW"`` (mu-law)
                - ``"ALAW"`` (a-law)

            bits_per_sample (int or None, optional): Changes the bit depth for the
                supported formats.
                When ``format`` is one of ``"wav"`` and ``"flac"``,
                you can change the bit depth.
                Valid values are ``8``, ``16``, ``24``, ``32`` and ``64``.

            buffer_size (int, optional):
                Size of buffer to use when processing file-like objects, in bytes. (Default: ``4096``)

            backend (str or None, optional):
                I/O backend to use.
                If ``None``, function selects backend given input and available backends.
                Otherwise, must be one of [``"ffmpeg"``, ``"sox"``, ``"soundfile"``],
                with the corresponding backend being available.
                (Default: ``None``)

                .. seealso::
                   :ref:`backend`
        saverB   r0   r   r   rF      s    Azget_save_func.<locals>.save)TNNNr,   N)r   r   r   r1   r2   r3   r   r   r?   r@   r4   r>   rE   r   r5   r   get_save_func   s0            DrG   ) r2   	functoolsr   typingr   r   r   r   r   r   r?   Ztorchaudio._extensionr	   r
    r   r   r   commonr   r   r   r   r   r   r   r1   r   r    r6   rA   rG   r   r   r   r   <module>   s     
>j