o i_3@sddlmZddlZddlZddlmZddlmZmZm Z ddl m Z m Z ddl mZddl mZdd lmZdd lmZmZdd lmZdd lmZdd lmZddlmZddlm Z eGdddZ!eGdddZ"GdddZ#dS)) annotationsN) dataclass)Any AsyncIteratorOptional) FfiClient FfiHandle)audio_frame_pb2)ffi_pb2) TrackSource) RingQueuetask_done_logger) AudioFrame)logger) Participant)Track)FrameProcessorc@seZdZUdZded<dS)AudioFrameEventzyAn event representing a received audio frame. Attributes: frame (AudioFrame): The received audio frame. rframeN)__name__ __module__ __qualname____doc____annotations__rrZ/var/www/html/livekit_bhavya/venv/lib/python3.10/site-packages/livekit/rtc/audio_stream.pyr"s  rc@seZdZUded<ded<dS)NoiseCancellationOptionsstr module_idzdict[str, Any]optionsN)rrrrrrrrr-s  rc@seZdZdZ      d5d6ddZedddddddd7ddZedddddddd8ddZd9dd Zd:d"d#Z d;d$d%Z d&d'Z d9d(d)Z dd3d4ZdS)? AudioStreamaOAn asynchronous audio stream for receiving audio frames from a participant or track. The `AudioStream` class provides an asynchronous iterator over audio frames received from a specific track or participant. It allows you to receive audio frames in real-time with customizable sample rates and channel configurations. Nr逻rtrackrloop#Optional[asyncio.AbstractEventLoop]capacityint sample_rate num_channels frame_size_ms int | Nonenoise_cancellation?Optional[NoiseCancellationOptions | FrameProcessor[AudioFrame]]returnNonec Ks||_||_||_||_|pt|_tjj j |jddd|_ t ||_ d|_d|_d|_t|tr<|j|_|j|_nt|trD||_|j||_|jtd} d|vre|j|d|dd} n|} t| jj|_ | j!|_"dS)aRInitialize an `AudioStream` instance. Args: track (Optional[Track]): The audio track from which to receive audio. If not provided, you must specify `participant` and `track_source` in `kwargs`. loop (Optional[asyncio.AbstractEventLoop], optional): The event loop to use. Defaults to the current event loop. capacity (int, optional): The capacity of the internal frame queue. Defaults to 0 (unbounded). sample_rate (int, optional): The sample rate for the audio stream in Hz. Defaults to 48000. num_channels (int, optional): The number of audio channels. Defaults to 1. noise_cancellation (Optional[NoiseCancellationOptions | FrameProcessor[AudioFrame]], optional): If noise cancellation is used, pass a `NoiseCancellationOptions` or `FrameProcessor[AudioFrame]` instance created by the noise cancellation module. Example: ```python audio_stream = AudioStream( track=audio_track, sample_rate=44100, num_channels=2, ) audio_stream = AudioStream.from_track( track=audio_track, sample_rate=44100, num_channels=2, ) ``` cSs|ddkS)Nmessageaudio_stream_event) WhichOneof)errrmsz&AudioStream.__init__..) filter_fnN participant track_source)r6r7)#_track _sample_rate _num_channels_frame_size_msasyncioget_event_loop_looprinstancequeue subscribe _ffi_queuer _queue_audio_filter_module_audio_filter_options _processor isinstancerrr r create_task_run_taskadd_done_callbackr%_create_owned_stream_from_participant_create_owned_streamr handleid _ffi_handleinfo_info) selfr#r$r&r(r)r*r,kwargsstreamrrr__init__;s8)       zAudioStream.__init__)r$r&r(r)r*r,r6rr7TrackSource.ValueTypec Cst||||d||||d S)a2Create an `AudioStream` from a participant's audio track. Args: participant (Participant): The participant from whom to receive audio. track_source (TrackSource.ValueType): The source of the audio track (e.g., microphone, screen share). loop (Optional[asyncio.AbstractEventLoop], optional): The event loop to use. Defaults to the current event loop. capacity (int, optional): The capacity of the internal frame queue. Defaults to 0 (unbounded). sample_rate (int, optional): The sample rate for the audio stream in Hz. Defaults to 48000. num_channels (int, optional): The number of audio channels. Defaults to 1. noise_cancellation (Optional[NoiseCancellationOptions], optional): If noise cancellation is used, pass a `NoiseCancellationOptions` instance created by the noise cancellation module. Returns: AudioStream: An instance of `AudioStream` that can be used to receive audio frames. Example: ```python audio_stream = AudioStream.from_participant( participant=participant, track_source=TrackSource.MICROPHONE, sample_rate=24000, num_channels=1, ) ``` N) r6r7r$r&r#r(r)r,r*r!) clsr6r7r$r&r(r)r*r,rrrfrom_participants'zAudioStream.from_participantc Cst|||||||dS)apCreate an `AudioStream` from an existing audio track. Args: track (Track): The audio track from which to receive audio. loop (Optional[asyncio.AbstractEventLoop], optional): The event loop to use. Defaults to the current event loop. capacity (int, optional): The capacity of the internal frame queue. Defaults to 0 (unbounded). sample_rate (int, optional): The sample rate for the audio stream in Hz. Defaults to 48000. num_channels (int, optional): The number of audio channels. Defaults to 1. noise_cancellation (Optional[NoiseCancellationOptions], optional): If noise cancellation is used, pass a `NoiseCancellationOptions` instance created by the noise cancellation module. Returns: AudioStream: An instance of `AudioStream` that can be used to receive audio frames. Example: ```python audio_stream = AudioStream.from_track( track=audio_track, sample_rate=44100, num_channels=2, ) ``` )r#r$r&r(r)r,r*rX)rYr#r$r&r(r)r*r,rrr from_tracks$zAudioStream.from_trackcCstjj|jdSN)rr?r@ unsubscriberBrSrrr__del__szAudioStream.__del__rcCs|jdusJt}|j}|jjj|_|j|_|j |_ |j r#|j |_ t jj|_|jdur1|j|_|jdur=t|j|_tj|}|jjSr\)r8 proto_ffi FfiRequestnew_audio_streamrPrN track_handler9r(r:r)r;r*proto_audio_frameAudioStreamTypeAUDIO_STREAM_NATIVEtyperDaudio_filter_module_idrEjsondumpsaudio_filter_optionsrr?requestrU)rSreqrbresprrrrMs     z AudioStream._create_owned_streamcCst}|j}|jj|_|j|_|j|_ t j j |_ ||_|jr#|j|_|jdur,|j|_|jdur8t|j|_tj|}|jjSr\)r`raaudio_stream_from_participantrPrNparticipant_handler9r(r:r)rdrerfrgr7r;r*rDrhrErirjrkrr?rlrU)rSr6r7rmrornrrrrLs"     z1AudioStream._create_owned_stream_from_participantcs |j|jIdH}|j}|drJ|jj}t|}|j dur?|j j r?z|j |}Wnt y>t jdddYnwt|}|j|n |drV|jdnqtjj|jdS)NTframe_receivedz7Frame processing failed, passing through original frame)exc_infoeos)rBwait_for _is_eventr1HasFieldrqrr_from_owned_inforFenabled_process ExceptionrwarningrrCputrr?r@r])rSevent audio_eventowned_buffer_inforrrrrIs.      zAudioStream._runcs|j|jIdHdS)zAsynchronously close the audio stream. This method cleans up resources associated with the audio stream and waits for any pending operations to complete. N)rPdisposerJr^rrraclose*s zAudioStream.acloser3proto_ffi.FfiEventboolcCs|jj|jjkSr\)r1 stream_handlerPrN)rSr3rrrru3szAudioStream._is_eventAsyncIterator[AudioFrameEvent]cCs|Sr\rr^rrr __aiter__6szAudioStream.__aiter__rcs0|jrt|jIdH}|durt|Sr\)rJdoneStopAsyncIterationrCget)rSitemrrr __anext__9s zAudioStream.__anext__)Nrr"rNN)r#rr$r%r&r'r(r'r)r'r*r+r,r-r.r/)r6rr7rWr$r%r&r'r(r'r)r'r*r+r,r-r.r!)r#rr$r%r&r'r(r'r)r'r*r+r,r-r.r!)r.r/)r.r)r6rr7rWr.r)r3rr.r)r.r)r.r)rrrrrV classmethodrZr[r_rMrLrIrrurrrrrrr!3sB  L2 -    r!)$ __future__rr<ri dataclassesrtypingrrr _ffi_clientrr _protor rdr r`_proto.track_pb2r _utilsr r audio_framerlogrr6rr#rframe_processorrrrr!rrrrs(