o §¼–i®ã@sRddlmZddlmZmZddlmZddlm Z ddl m Z Gdd„dƒZ d S) é)Ú annotationsé)Ú FfiClientÚ FfiHandle)Úffi_pb2)Ú get_address)Ú AudioFramec@sDeZdZdZdddddœdd d „Zddd„Zddd„Zddd„ZdS)ÚAudioProcessingModulez’ Provides WebRTC audio processing capabilities including echo cancellation, noise suppression, high-pass filtering, and gain control. F)Úecho_cancellationÚnoise_suppressionÚhigh_pass_filterÚauto_gain_controlr Úboolr r r ÚreturnÚNonecCsJt ¡}||j_||j_||j_||j_tj  |¡}t |jj j j ƒ|_dS)aÏ Initialize an AudioProcessingModule instance with the specified audio processing features. Args: echo_cancellation (bool, optional): Whether to enable echo cancellation. noise_suppression (bool, optional): Whether to enable noise suppression. high_pass_filter (bool, optional): Whether to enable a high-pass filter. auto_gain_control (bool, optional): Whether to enable auto gain control. N)Ú proto_ffiÚ FfiRequestÚnew_apmÚecho_canceller_enabledÚnoise_suppression_enabledÚhigh_pass_filter_enabledÚgain_controller_enabledrÚinstanceÚrequestrÚapmÚhandleÚidÚ _ffi_handle)Úselfr r r r ÚreqÚresp©r!úQ/var/www/html/livekit_bhavya/venv/lib/python3.10/site-packages/livekit/rtc/apm.pyÚ__init__s zAudioProcessingModule.__init__ÚdatarcCs”t|jtƒst|jtƒr|jjrt|jƒ|_t ¡}|jj |j _ t |jƒ|j _ t|jƒ|j _|j|j _|j|j _tj |¡}|j jrHt|j jƒ‚dS)a\ Process the provided audio frame using the configured audio processing features. The input audio frame is modified in-place (if applicable) by the underlying audio processing module (e.g., echo cancellation, noise suppression, etc.). Important: Audio frames must be exactly 10 ms in duration. N)Ú isinstanceÚ_dataÚbytesÚ memoryviewÚreadonlyÚ bytearrayrrrrÚapm_process_streamÚ apm_handlerÚdata_ptrÚlenÚsizeÚ sample_rateÚ num_channelsrrrÚerrorÚ RuntimeError©rr$rr r!r!r"Úprocess_stream)s ÿÿ      ÿz$AudioProcessingModule.process_streamcCs”t|jtƒst|jtƒr|jjrt|jƒ|_t ¡}|jj |j _ t |jƒ|j _ t|jƒ|j _|j|j _|j|j _tj |¡}|jjrHt|jjƒ‚dS)aÌ Process the reverse audio frame (typically used for echo cancellation in a full-duplex setup). In an echo cancellation scenario, this method is used to process the "far-end" audio prior to mixing or feeding it into the echo canceller. Like `process_stream`, the input audio frame is modified in-place by the underlying processing module. Important: Audio frames must be exactly 10 ms in duration. N)r%r&r'r(r)r*rrrrÚapm_process_reverse_streamr,rr-r.r/r0r1rrrr+r2r3r4r!r!r"Úprocess_reverse_streamDs ÿÿ      ÿz,AudioProcessingModule.process_reverse_streamÚdelay_msÚintcCs@t ¡}|jj|j_||j_tj  |¡}|jj rt |jj ƒ‚dS)a! This must be called if and only if echo processing is enabled. Sets the `delay` in ms between `process_reverse_stream()` receiving a far-end frame and `process_stream()` receiving a near-end frame containing the corresponding echo. On the client-side this can be expressed as delay = (t_render - t_analyze) + (t_process - t_capture) where, - t_analyze is the time a frame is passed to `process_reverse_stream()` and t_render is the time the first sample of the same frame is rendered by the audio hardware. - t_capture is the time the first sample of a frame is captured by the audio hardware and t_process is the time the same frame is passed to `process_stream()`. N) rrrrÚapm_set_stream_delayr,r8rrrr2r3)rr8rr r!r!r"Úset_stream_delay_ms`s   ÿz)AudioProcessingModule.set_stream_delay_msN) r rr rr rr rrr)r$rrr)r8r9rr)Ú__name__Ú __module__Ú __qualname__Ú__doc__r#r5r7r;r!r!r!r"r sú  r N) Ú __future__rÚ _ffi_clientrrÚ_protorrÚ_utilsrÚ audio_framerr r!r!r!r"Ús