U ,:%e.@sddlZddlmZmZmZddlZddlmZddlmZm Z ddl m Z m Z GdddZ ejd d d ZGd d d Zeeeeeeee dddZejd ddZGdddZdS)N)IteratorListOptional)Tensor)_get_afilter_desc StreamReader) CodecConfig StreamWriterc@s.eZdZdZddZedddZddZd S) _StreamingIOBufferz6Streaming Bytes IO buffer. Data are dropped when read.cCs g|_dSN)_buffer)selfrV/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torchaudio/io/_effector.py__init__sz_StreamingIOBuffer.__init__)bcCs|r|j|t|Sr )r appendlen)rrrrrwrites z_StreamingIOBuffer.writecCsV|js dSt|jd|kr(|jdS|jdd|}|jd|d|jd<|S)zMPop the oldest byte string. It does not necessary return the requested amountrN)r rpop)rnretrrrrs z_StreamingIOBuffer.popN)__name__ __module__ __qualname____doc__rbytesrrrrrrr sr )dtypec CsJtjdtjdtjdtjdtjdi}||krBtd|d|||S)Nu8Zs16Zs32ZfltZdblUnsupported dtype is provided . Supported dtypes are: torchZuint8Zint16Zint32Zfloat32Zfloat64 ValueErrorkeysrtypesrrr_get_sample_fmt!sr)c@s:eZdZdZeeeeeeeeedddZ ddZ dS)_AudioStreamingEncoderz3Given a waveform, encode on-demand and return bytes)src sample_rateeffectmuxerencoder codec_configframes_per_chunkcCs\||_t|_t|j|d|_|jj|d|t|j|||d|j ||_ d|_ dSN)formatr) num_channelsr,r3r/ filter_descr0r) r+r bufferr writeradd_audio_streamsizer)ropenfpci_iter)rr+r,r-r.r/r0r1rrrr1s  z_AudioStreamingEncoder.__init__cCs~|jjsr|jdkrr|jd|j|j|j|j|j|j7_|j|jdkr|j|j d|_q|j |S)Nr) r6r r<r7write_audio_chunkr+r;r9flushcloser)rrrrrreadNs"  z_AudioStreamingEncoder.readN) rrrrrintstrrr rrArrrrr*.s r*)r+r,r-r.r/r0c Csdt}t||d}|j|d|t|j|||d||d|W5QRX| d|Sr2) ioBytesIOr r8r9r)rr:r>seek)r+r,r-r.r/r0r6r7rrr_encodeYs   rGc CsJtjdtjdtjdtjdtjdi}||krBtd|d|||S)Nr Zs16leZs32leZf32leZf64ler!r"r#r'rrr _get_muxerqsrHc@seZdZdZdddddeeeeeeeeedddZddd Z de e ee e d d d Z de e e ee e e d ddZdS) AudioEffectoraApply various filters and/or codecs to waveforms. .. versionadded:: 2.1 Args: effect (str or None, optional): Filter expressions or ``None`` to apply no filter. See https://ffmpeg.org/ffmpeg-filters.html#Audio-Filters for the details of filter syntax. format (str or None, optional): When provided, encode the audio into the corresponding format. Default: ``None``. encoder (str or None, optional): When provided, override the encoder used by the ``format``. Default: ``None``. codec_config (CodecConfig or None, optional): When provided, configure the encoding codec. Should be provided in conjunction with ``format`` option. pad_end (bool, optional): When enabled, and if the waveform becomes shorter after applying effects/codec, then pad the end with silence. Example - Basic usage To use ``AudioEffector``, first instantiate it with a set of ``effect`` and ``format``. >>> # instantiate the effector >>> effector = AudioEffector(effect=..., format=...) Then, use :py:meth:`~AudioEffector.apply` or :py:meth:`~AudioEffector.stream` method to apply them. >>> # Apply the effect to the whole waveform >>> applied = effector.apply(waveform, sample_rate) >>> # Apply the effect chunk-by-chunk >>> for chunk in effector.stream(waveform, sample_rate): >>> ... Example - Applying effects Please refer to https://ffmpeg.org/ffmpeg-filters.html#Filtergraph-description for the overview of filter description, and https://ffmpeg.org/ffmpeg-filters.html#toc-Audio-Filters for the list of available filters. Tempo - https://ffmpeg.org/ffmpeg-filters.html#atempo >>> AudioEffector(effect="atempo=1.5") Echo - https://ffmpeg.org/ffmpeg-filters.html#aecho >>> AudioEffector(effect="aecho=0.8:0.88:60:0.4") Flanger - https://ffmpeg.org/ffmpeg-filters.html#flanger >>> AudioEffector(effect="aflanger") Vibrato - https://ffmpeg.org/ffmpeg-filters.html#vibrato >>> AudioEffector(effect="vibrato") Tremolo - https://ffmpeg.org/ffmpeg-filters.html#tremolo >>> AudioEffector(effect="vibrato") You can also apply multiple effects at once. >>> AudioEffector(effect="") Example - Applying codec One can apply codec using ``format`` argument. ``format`` can be audio format or container format. If the container format supports multiple encoders, you can specify it with ``encoder`` argument. Wav format (no compression is applied but samples are converted to 16-bit signed integer) >>> AudioEffector(format="wav") Ogg format with default encoder >>> AudioEffector(format="ogg") Ogg format with vorbis >>> AudioEffector(format="ogg", encoder="vorbis") Ogg format with opus >>> AudioEffector(format="ogg", encoder="opus") Webm format with opus >>> AudioEffector(format="webm", encoder="opus") Example - Applying codec with configuration Reference: https://trac.ffmpeg.org/wiki/Encode/MP3 MP3 with default config >>> AudioEffector(format="mp3") MP3 with variable bitrate >>> AudioEffector(format="mp3", codec_config=CodecConfig(qscale=5)) MP3 with constant bitrate >>> AudioEffector(format="mp3", codec_config=CodecConfig(bit_rate=32_000)) NT)r/r0pad_end)r-r3r/r0rJcCsB|dkr |dk s|dk r td||_||_||_||_||_dS)NzM`encoder` and/or `condec_config` opions are provided without `format` option.)r%r-r3r/r0rJ)rr-r3r/r0rJrrrrs zAudioEffector.__init__cCs|j\}}|jdk r>|j}|j}i} |jdkrZ||d} nt|j}d}||d} |dkrzt|||j|||j} nt|||j|||j|} |dkr|n|} t | t |j|} |j r| d|} t | || d} | j |pdd| d| S)NZmulaw)r,Zchannelsz,apad=whole_len=)r3optionr=)r5)shaper3r/rHrrGr-r0r*rr)rJrr8)rwaveformr,output_sample_rater1Z num_framesr4r.r/rKr+Z output_srr5readerrrr _get_readers8    zAudioEffector._get_reader)rMr,rNreturncCsR|jdkrtd|j|dkr*|S||||}||\}t|S)aApply the effect and/or codecs to the whole tensor. Args: waveform (Tensor): The input waveform. Shape: ``(time, channel)`` sample_rate (int): Sample rate of the input waveform. output_sample_rate (int or None, optional): Output sample rate. If provided, override the output sample rate. Otherwise, the resulting tensor is resampled to have the same sample rate as the input. Default: ``None``. Returns: Tensor: Resulting Tensor. Shape: ``(time, channel)``. The number of frames could be different from that of the input. -Expected the input waveform to be 2D. Found: r)ndimr%numelrPZprocess_all_packetsZ pop_chunksr)rrMr,rNrOappliedrrrapply"s   zAudioEffector.apply)rMr,r1rNrQccsX|jdkrtd|j|dkr*|S|||||}|D]\}t|VqBdS)aZApply the effect and/or codecs to the given tensor chunk by chunk. Args: waveform (Tensor): The input waveform. Shape: ``(time, channel)`` sample_rate (int): Sample rate of the waveform. frames_per_chunk (int): The number of frames to return at a time. output_sample_rate (int or None, optional): Output sample rate. If provided, override the output sample rate. Otherwise, the resulting tensor is resampled to have the same sample rate as the input. Default: ``None``. Returns: Iterator[Tensor]: Series of processed chunks. Shape: ``(time, channel)``, where the the number of frames matches ``frames_per_chunk`` except the last chunk, which could be shorter. rRrSrN)rTr%rUrPstreamr)rrMr,r1rNrOrVrrrrX>s  zAudioEffector.stream)NN)N)N)N)rrrrrrCr boolrrPrrBrWrrXrrrrrIs0r  rI)rDtypingrrrr$r_stream_readerrr_stream_writerr r r rr)r*rBrCrGrHrIrrrrs"  ,